Support for QTEMP/session

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

Author: worksofliam

package.json

@@ -2,7 +2,7 @@
   "name": "vscode-db2i",
   "displayName": "Db2 for IBM i",
   "description": "Db2 for IBM i tools in VS Code",
-  "version": "1.6.3-scott11",
+  "version": "1.6.3-scott12",
   "engines": {
     "vscode": "^1.95.0"
   },

src/database/table.ts

@@ -30,11 +30,38 @@ export default class Table {
       `    column.table_schema = key.table_schema and`,
       `    column.table_name = key.table_name and`,
       `    column.column_name = key.column_name`,
-      `WHERE column.TABLE_SCHEMA = '${schema}' AND column.TABLE_NAME = '${name}'`,
+      `WHERE column.TABLE_SCHEMA = ? AND column.TABLE_NAME = ?`,
       `ORDER BY column.ORDINAL_POSITION`,
     ].join(` `);
 
-    return JobManager.runSQL(sql);
+    return JobManager.runSQL(sql, {parameters: [schema, name]});
+  }
+
+  /**
+   * This is to be used instead of getItems when the table is in session/QTEMP
+   */
+  static async getSessionItems(name: string): Promise<TableColumn[]> {
+    const sql = [
+      `SELECT `,
+      `  column.TABLE_SCHEMA,`,
+      `  column.TABLE_NAME,`,
+      `  column.COLUMN_NAME,`,
+      `  '' as CONSTRAINT_NAME,`,
+      `  column.DATA_TYPE, `,
+      `  column.CHARACTER_MAXIMUM_LENGTH,`,
+      `  column.NUMERIC_SCALE, `,
+      `  column.NUMERIC_PRECISION,`,
+      `  column.IS_NULLABLE, `,
+      `  column.HAS_DEFAULT, `,
+      `  column.COLUMN_DEFAULT, `,
+      `  column.COLUMN_TEXT, `,
+      `  column.IS_IDENTITY`,
+      `FROM QSYS2.SYSCOLUMNS2_SESSION as column`,
+      `WHERE column.TABLE_NAME = ?`,
+      `ORDER BY column.ORDINAL_POSITION`,
+    ].join(` `);
+
+    return JobManager.runSQL(sql, {parameters: [name]});
   }
 
   static async isPartitioned(schema: string, name: string): Promise<boolean> {

src/views/results/resultSetPanelProvider.ts

@@ -194,10 +194,16 @@ export class ResultSetPanelProvider implements WebviewViewProvider {
 
         const isPartitioned = await Table.isPartitioned(goodSchema, goodName);
         if (!isPartitioned) {
-          const tableInfo = await Table.getItems(
-            goodSchema,
-            goodName
-          );
+          let tableInfo: TableColumn[] = [];
+
+          if ([`SESSION`, `QTEMP`].includes(goodSchema)) {
+            tableInfo = await Table.getSessionItems(goodName);
+          } else {
+            tableInfo = await Table.getItems(
+              goodSchema,
+              goodName
+            );
+          }
 
           const uneditableTypes = [`VARBIN`, `BINARY`, `ROWID`, `DATALINK`, `DBCLOB`, `BLOB`, `GRAPHIC`]