If table is partitioned, no updates available

Signed-off-by: worksofliam <[email protected]>

Author: worksofliam

src/database/table.ts

@@ -37,6 +37,14 @@ export default class Table {
     return JobManager.runSQL(sql);
   }
 
+  static async isPartitioned(schema: string, name: string): Promise<boolean> {
+    const sql = `select table_name, partitioned_table from qsys2.sysfiles where table_schema = ? and table_name = ? and partitioned_table is not null and partitioned_table = 'YES'`;
+    const parameters = [schema, name];
+
+    const result = await JobManager.runSQL(sql, {parameters});
+    return result.length > 0;
+  }
+
   static async clearFile(library: string, objectName: string): Promise<void> {
     const command = `CLRPFM ${library}/${objectName}`;
 

src/views/results/html.ts

@@ -511,7 +511,7 @@ export function generateScroller(basicSelect: string, isCL: boolean, withCancel?
           <tbody></tbody>
           <tfoot id="resultfooter"></tfoot>
           </table>
-        <p id="messageSpan"></p>
+        <p style="padding-left: 20px;" id="messageSpan"></p>
         <div id="spinnerContent" class="center-screen">
           <p id="loadingText">Running statement</p>
           <span class="loader"></span>

src/views/results/resultSetPanelProvider.ts

@@ -63,7 +63,7 @@ export class ResultSetPanelProvider implements WebviewViewProvider {
           if (message.id && message.update && message.bindings) {
             console.log(message);
             try {
-              const result = await JobManager.runSQL(message.update, {parameters: message.bindings});
+              const result = await JobManager.runSQL(message.update, { parameters: message.bindings });
               postCellResponse(message.id, true);
             } catch (e) {
               // this.setError(e.message);
@@ -98,7 +98,7 @@ export class ResultSetPanelProvider implements WebviewViewProvider {
 
               if (this.currentQuery.getState() !== "RUN_DONE") {
                 setCancelButtonVisibility(true);
-                
+
                 let queryResults = this.currentQuery.getState() == "RUN_MORE_DATA_AVAILABLE" ? await this.currentQuery.fetchMore() : await this.currentQuery.execute();
 
                 const jobId = this.currentQuery.getHostJob().id;
@@ -184,63 +184,73 @@ export class ResultSetPanelProvider implements WebviewViewProvider {
     this.loadingState = false;
     await this.focus();
 
-    let updatable: html.UpdatableInfo|undefined;
+    let updatable: html.UpdatableInfo | undefined;
 
     if (ref) {
       const schema = ref.object.schema || ref.object.system;
       if (schema) {
-        const tableInfo = await Table.getItems(
-          Statement.delimName(schema, true), 
-          Statement.delimName(ref.object.name, true)
-        );
-
-        if (tableInfo.length > 0) {
-          let currentColumns: html.BasicColumn[]|undefined;
-      
-          currentColumns = tableInfo.map((column) => ({
-            name: column.COLUMN_NAME, 
-            jsType: column.NUMERIC_PRECISION ? `number` : `asString`, 
-            useInWhere: column.IS_IDENTITY === `YES`,
-            maxInputLength: column.CHARACTER_MAXIMUM_LENGTH
-          }));
-
-          if (!currentColumns.some(c => c.useInWhere)) {
-            const cName = ref.alias || `t`;
-
-            // Support for using a custom column list
-            const selectClauseStart = basicSelect.toLowerCase().indexOf(`select `);
-            const fromClauseStart = basicSelect.toLowerCase().indexOf(`from`);
-            let possibleColumnList: string|undefined;
-
-            possibleColumnList = `${cName}.*`;
-            if (fromClauseStart > 0) {
-              possibleColumnList = basicSelect.substring(0, fromClauseStart);
-              if (selectClauseStart >= 0) {
-                possibleColumnList = possibleColumnList.substring(selectClauseStart + 7);
-
-                if (possibleColumnList.trim() === `*`) {
-                  possibleColumnList = `${cName}.*`;
+        const goodSchema = Statement.delimName(schema, true);
+        const goodName = Statement.delimName(ref.object.name, true);
+
+        const isPartitioned = await Table.isPartitioned(goodSchema, goodName);
+        if (!isPartitioned) {
+          const tableInfo = await Table.getItems(
+            goodSchema,
+            goodName
+          );
+
+          const uneditableTypes = [`VARBIN`, `BINARY`, `ROWID`, `DATALINK`, `DBCLOB`, `BLOB`, `GRAPHIC`]
+
+          if (tableInfo.length > 0) {
+            let currentColumns: html.BasicColumn[] | undefined;
+
+            currentColumns = tableInfo
+              .filter((column) => !uneditableTypes.includes(column.DATA_TYPE))
+              .map((column) => ({
+                name: column.COLUMN_NAME,
+                jsType: column.NUMERIC_PRECISION ? `number` : `asString`,
+                useInWhere: column.IS_IDENTITY === `YES`,
+                maxInputLength: column.CHARACTER_MAXIMUM_LENGTH
+              }));
+
+            if (!currentColumns.some(c => c.useInWhere)) {
+              const cName = ref.alias || `t`;
+
+              // Support for using a custom column list
+              const selectClauseStart = basicSelect.toLowerCase().indexOf(`select `);
+              const fromClauseStart = basicSelect.toLowerCase().indexOf(`from`);
+              let possibleColumnList: string | undefined;
+
+              possibleColumnList = `${cName}.*`;
+              if (fromClauseStart > 0) {
+                possibleColumnList = basicSelect.substring(0, fromClauseStart);
+                if (selectClauseStart >= 0) {
+                  possibleColumnList = possibleColumnList.substring(selectClauseStart + 7);
+
+                  if (possibleColumnList.trim() === `*`) {
+                    possibleColumnList = `${cName}.*`;
+                  }
                 }
               }
-            }
 
-            // We need to override the input statement if they want to do updatable
-            const whereClauseStart = basicSelect.toLowerCase().indexOf(`where`);
-            let fromWhereClause: string|undefined;
+              // We need to override the input statement if they want to do updatable
+              const whereClauseStart = basicSelect.toLowerCase().indexOf(`where`);
+              let fromWhereClause: string | undefined;
 
-            if (whereClauseStart > 0) {
-              fromWhereClause = basicSelect.substring(whereClauseStart);
-            }
+              if (whereClauseStart > 0) {
+                fromWhereClause = basicSelect.substring(whereClauseStart);
+              }
 
 
-            basicSelect = `select rrn(${cName}) as RRN, ${possibleColumnList} from ${schema}.${ref.object.name} as ${cName} ${fromWhereClause || ``}`;
-            currentColumns = [{name: `RRN`, jsType: `number`, useInWhere: true}, ...currentColumns];
-          }
+              basicSelect = `select rrn(${cName}) as RRN, ${possibleColumnList} from ${schema}.${ref.object.name} as ${cName} ${fromWhereClause || ``}`;
+              currentColumns = [{ name: `RRN`, jsType: `number`, useInWhere: true }, ...currentColumns];
+            }
 
-          updatable = {
-            table: schema + `.` + ref.object.name,
-            columns: currentColumns
-          };
+            updatable = {
+              table: schema + `.` + ref.object.name,
+              columns: currentColumns
+            };
+          }
         }
       }
     }