get tables by schema names

packages/cubejs-base-driver/src/BaseDriver.ts

@@ -35,7 +35,8 @@ import {
   TableQueryResult,
   TableStructure,
   DriverCapabilities,
-  SchemaQueryResult
+  QuerySchemasResult,
+  QueryTablesResult
 } from './driver.interface';
 
 const sortByKeys = (unordered: any) => {
@@ -172,6 +173,17 @@ export abstract class BaseDriver implements DriverInterface {
     `;
   }
 
+  protected informationSchemaGetTablesQuery(schemaNames: string[]) {
+    const schemas = schemaNames.map(s => this.singleQuoteIdentifier(s)).join(', ');
+    return `
+      SELECT table_schema as ${this.quoteIdentifier('schema_name')},
+             table_name as ${this.quoteIdentifier('table_name')}
+      FROM information_schema.tables
+      WHERE table_schema IN (${schemas})
+      ORDER BY table_schema, table_name
+    `;
+  }
+
   protected getSslOptions(dataSource: string): TLSConnectionOptions | undefined {
     if (
       getEnv('dbSsl', { dataSource }) ||
@@ -320,9 +332,18 @@ export abstract class BaseDriver implements DriverInterface {
     }
   }
 
-  public getSchemasQuery() {
+  public getSchemas() {
     const query = this.informationSchemaGetSchemasQuery();
-    return this.query<SchemaQueryResult>(query);
+    return this.query<QuerySchemasResult>(query);
+  }
+
+  public getTables(schemaNames?: string[]) {
+    if (!schemaNames) {
+      throw new Error('getTables without schemaNames is not supported');
+    }
+
+    const query = this.informationSchemaGetTablesQuery(schemaNames);
+    return this.query<QueryTablesResult>(query);
   }
 
   public getTablesQuery(schemaName: string) {

packages/cubejs-base-driver/src/driver.interface.ts

@@ -166,7 +166,9 @@ export type DownloadQueryResultsResult = DownloadQueryResultsBase & (DownloadTab
 // eslint-disable-next-line camelcase
 export type TableQueryResult = { table_name?: string, TABLE_NAME?: string };
 
-export type SchemaQueryResult = { schemaName: string };
+export type QuerySchemasResult = { schemaName: string };
+
+export type QueryTablesResult = { schemaName: string, tableName: string };
 
 export interface DriverInterface {
   createSchemaIfNotExists(schemaName: string): Promise<void>;
@@ -180,7 +182,9 @@ export interface DriverInterface {
   tableColumnTypes: (table: string) => Promise<TableStructure>;
   queryColumnTypes: (sql: string, params?: unknown[]) => Promise<{ name: any; type: string; }[]>;
   // Returns all schemas for current database
-  getSchemasQuery: () => Promise<SchemaQueryResult[]>;
+  getSchemas: () => Promise<QuerySchemasResult[]>;
+  // Returns all tables for defined schemas
+  getTables: (schemaNames: string[]) => Promise<QueryTablesResult[]>;
   // eslint-disable-next-line camelcase
   getTablesQuery: (schemaName: string) => Promise<TableQueryResult[]>;
   // Remove table from database

packages/cubejs-server-core/src/core/DevServer.ts

@@ -113,10 +113,28 @@ export class DevServer {
         requestId: getRequestIdFromRequest(req),
       });
 
-      const schemas = await driver.getSchemasQuery();
+      const schemas = await driver.getSchemas();
       res.json({ schemas });
     }));
 
+    /** WIP */
+    app.post('/playground/list-tables', catchErrors(async (req, res) => {
+      if (!req.body.schemaNames) {
+        throw new Error('schemaNames parameter is required');
+      }
+
+      // this.cubejsServer.event('Dev Server List Tables');
+      const driver = await this.cubejsServer.getDriver({
+        dataSource: req.body.dataSource || 'default',
+        authInfo: null,
+        securityContext: null,
+        requestId: getRequestIdFromRequest(req),
+      });
+
+      const tables = await driver.getTables(req.body.schemaNames);
+      res.json({ tables });
+    }));
+
     app.get('/playground/db-schema', catchErrors(async (req, res) => {
       this.cubejsServer.event('Dev Server DB Schema Load');
       const driver = await this.cubejsServer.getDriver({