Function loadCompletionCache

src/language/server.ts:177–317

(connectionOptions: IDBConnection)

}

async function loadCompletionCache(connectionOptions: IDBConnection) {
  if (!connectionOptions || !dbConnection) return;
  // setup database caches for schemas, functions, tables, and fields
  let vQueries = SqlQueryManager.getVersionQueries(dbConnection.pg_version);
  try {
    if (connectionOptions.database) {
      let schemas = await dbConnection.query(`
        SELECT nspname as name
        FROM pg_namespace
        WHERE
          nspname not in ('information_schema', 'pg_catalog', 'pg_toast')
          AND nspname not like 'pg_temp_%'
          AND nspname not like 'pg_toast_temp_%'
          AND has_schema_privilege(oid, 'CREATE, USAGE')
        ORDER BY nspname;
        `);
      schemaCache = schemas.rows;
    }
  }
  catch (err) {
    console.log(err.message);
  }

  try {
    if (connectionOptions.database) {
      /*
      SELECT tablename as name, true as is_table FROM pg_tables WHERE schemaname not in ('information_schema', 'pg_catalog')
      union all
      SELECT viewname as name, false as is_table FROM pg_views WHERE schemaname not in ('information_schema', 'pg_catalog') order by name;
      */
      let tablesAndColumns = await dbConnection.query(`
        SELECT
          tbl.schemaname,
          tbl.tablename,
          tbl.quoted_name,
          tbl.is_table,
          json_agg(a) as columns
        FROM
          (
            SELECT
              n.nspname as schemaname,
              c.relname as tablename,
              (quote_ident(n.nspname) || '.' || quote_ident(c.relname)) as quoted_name,
              true as is_table
            FROM
              pg_catalog.pg_class c
              JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace
            WHERE
              c.relkind = 'r'
              AND n.nspname not in ('information_schema', 'pg_catalog', 'pg_toast')
              AND n.nspname not like 'pg_temp_%'
              AND n.nspname not like 'pg_toast_temp_%'
              AND has_schema_privilege(n.oid, 'USAGE') = true
              AND has_table_privilege(quote_ident(n.nspname) || '.' || quote_ident(c.relname), 'SELECT, INSERT, UPDATE, DELETE, TRUNCATE, REFERENCES, TRIGGER') = true
            union all
            SELECT
              n.nspname as schemaname,
              c.relname as tablename,