graphql: Fix

dolthub · Jun 5, 2024 · c0686f7 · c0686f7
1 parent 4015d5f
commit c0686f7
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/graphql-server/src/queryFactory/doltgres/index.ts b/graphql-server/src/queryFactory/doltgres/index.ts
@@ -17,15 +17,11 @@ export class DoltgresQueryFactory
   isDolt = true;
 
   async getTablePKColumns(args: t.TableArgs): Promise<string[]> {
-    console.log(args, this.ds);
-    // const res: t.RawRows = await this.query(
-    //   qh.tableColsQuery,
-    //   [args.tableName],
-    //   args.databaseName,
-    //   args.refName,
-    // );
-    // return res.filter(c => c.Key === "PRI").map(c => c.Field);
-    return [];
+    return this.queryQR(async qr => {
+      const table = await qr.getTable(`${args.databaseName}.${args.tableName}`);
+      if (!table) return [];
+      return table.columns.filter(c => c.isPrimary).map(c => c.name);
+    }, args.databaseName);
   }
 
   async getSchemas(args: t.RefArgs, type?: SchemaType): Promise<SchemaItem[]> {