@query-farm/vgi 0.42.0 → 0.44.0

package/src/protocol/handlers/catalog/table.ts

@@ -2,337 +2,272 @@
 // Catalog table handlers: table_get/create/drop, scan_function_get,
 // column_statistics_get, comment_set, rename, plus all column_* mutations.
 
-import { type VgiSchema, schema, type VgiField, field, type VgiDataType, binary, utf8, bool, int32, list } from "../../../arrow/index.js";
-import { Protocol } from "@query-farm/vgi-rpc";
+import type { VgiService } from "../../../generated/vgi-service.js";
 import { encodeTableInfo } from "../../../generated/vgi-client.js";
 import {
-  CatalogTableColumnAddParamsSchema,
-  CatalogTableColumnDefaultDropParamsSchema,
-  CatalogTableColumnDefaultSetParamsSchema,
-  CatalogTableColumnDropParamsSchema,
-  CatalogTableColumnRenameParamsSchema,
-  CatalogTableColumnStatisticsGetParamsSchema,
-  CatalogTableColumnTypeChangeParamsSchema,
-  CatalogTableCommentSetParamsSchema,
-  CatalogTableCreateParamsSchema,
-  CatalogTableDropParamsSchema,
-  CatalogTableGetParamsSchema,
   CatalogTableGetResultSchema,
-  CatalogTableNotNullDropParamsSchema,
-  CatalogTableNotNullSetParamsSchema,
-  CatalogTableRenameParamsSchema,
-  CatalogTableScanBranchesGetParamsSchema,
-  CatalogTableScanFunctionGetParamsSchema,
   ScanBranchesResultSchema,
   ScanFunctionResultSchema,
 } from "../../../generated/vgi-protocol-schemas.js";
 import { toUint8Array } from "../../../util/bytes.js";
 import {
-  RESULT_BINARY_SCHEMA,
-  RESULT_BINARY_NULLABLE_SCHEMA,
   wrapResult,
 } from "../shared.js";
 import {
   type GetCatalog,
-  emptyResultSchema,
-  catalogUnary,
+  catalogHandler,
 } from "./shared.js";
 
-export function registerCatalogTableMethods(protocol: Protocol, getCatalog: GetCatalog, signingKey?: Uint8Array): void {
+export function catalogTableHandlers(getCatalog: GetCatalog, signingKey?: Uint8Array): Partial<VgiService> {
   // catalog_table_get
-  catalogUnary(protocol, signingKey, "catalog_table_get", {
-    params: CatalogTableGetParamsSchema,
-    result: RESULT_BINARY_SCHEMA,
-    handler: async (params) => {
-      const cat = getCatalog();
-      const info = await cat.tableGet(
-        toUint8Array(params.attach_opaque_data),
-        params.schema_path,
-        params.name,
-        params.at_unit,
-        params.at_value,
-        params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
-      );
-      return wrapResult({
-        items: info ? [encodeTableInfo(info)] : [],
-      }, CatalogTableGetResultSchema);
-    },
+  const catalogTableGet = catalogHandler(signingKey, async (params) => {
+    const cat = getCatalog();
+    const info = await cat.tableGet(
+      toUint8Array(params.attach_opaque_data),
+      params.schema_path,
+      params.name,
+      params.at_unit,
+      params.at_value,
+      params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
+    );
+    return wrapResult({
+      items: info ? [encodeTableInfo(info)] : [],
+    }, CatalogTableGetResultSchema);
   });
 
   // catalog_table_create
-  catalogUnary(protocol, signingKey, "catalog_table_create", {
-    params: CatalogTableCreateParamsSchema,
-    result: emptyResultSchema,
-    handler: async (params) => {
-      const cat = getCatalog();
-      await cat.tableCreate(
-        toUint8Array(params.attach_opaque_data),
-        params.schema_path,
-        params.name,
-        toUint8Array(params.columns),
-        params.on_conflict,
-        params.not_null_constraints ?? [],
-        params.unique_constraints ?? [],
-        params.check_constraints ?? [],
-        params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
-      );
-      return {};
-    },
+  const catalogTableCreate = catalogHandler(signingKey, async (params) => {
+    const cat = getCatalog();
+    await cat.tableCreate(
+      toUint8Array(params.attach_opaque_data),
+      params.schema_path,
+      params.name,
+      toUint8Array(params.columns),
+      params.on_conflict,
+      params.not_null_constraints ?? [],
+      params.unique_constraints ?? [],
+      params.check_constraints ?? [],
+      params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
+    );
+    return {};
   });
 
   // catalog_table_drop
-  catalogUnary(protocol, signingKey, "catalog_table_drop", {
-    params: CatalogTableDropParamsSchema,
-    result: emptyResultSchema,
-    handler: async (params) => {
-      const cat = getCatalog();
-      await cat.tableDrop(
-        toUint8Array(params.attach_opaque_data),
-        params.schema_path,
-        params.name,
-        params.ignore_not_found,
-        params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
-      );
-      return {};
-    },
+  const catalogTableDrop = catalogHandler(signingKey, async (params) => {
+    const cat = getCatalog();
+    await cat.tableDrop(
+      toUint8Array(params.attach_opaque_data),
+      params.schema_path,
+      params.name,
+      params.ignore_not_found,
+      params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
+    );
+    return {};
   });
 
   // catalog_table_column_statistics_get
-  catalogUnary(protocol, signingKey, "catalog_table_column_statistics_get", {
-    params: CatalogTableColumnStatisticsGetParamsSchema,
-    result: RESULT_BINARY_NULLABLE_SCHEMA,
-    handler: async (params) => {
-      const cat = getCatalog();
-      const stats = await cat.tableColumnStatisticsGet(
-        toUint8Array(params.attach_opaque_data),
-        params.schema_path,
-        params.name,
-        params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined,
-      );
-      return { result: stats?.bytes ?? null };
-    },
+  const catalogTableColumnStatisticsGet = catalogHandler(signingKey, async (params) => {
+    const cat = getCatalog();
+    const stats = await cat.tableColumnStatisticsGet(
+      toUint8Array(params.attach_opaque_data),
+      params.schema_path,
+      params.name,
+      params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined,
+    );
+    return { result: stats?.bytes ?? null };
   });
 
   // catalog_table_scan_function_get
-  catalogUnary(protocol, signingKey, "catalog_table_scan_function_get", {
-    params: CatalogTableScanFunctionGetParamsSchema,
-    result: RESULT_BINARY_SCHEMA,
-    handler: async (params) => {
-      const cat = getCatalog();
-      const scanResult = await cat.tableScanFunctionGet(
-        toUint8Array(params.attach_opaque_data),
-        params.schema_path,
-        params.name,
-        params.at_unit,
-        params.at_value,
-        params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
-      );
-      return wrapResult(scanResult, ScanFunctionResultSchema);
-    },
+  const catalogTableScanFunctionGet = catalogHandler(signingKey, async (params) => {
+    const cat = getCatalog();
+    const scanResult = await cat.tableScanFunctionGet(
+      toUint8Array(params.attach_opaque_data),
+      params.schema_path,
+      params.name,
+      params.at_unit,
+      params.at_value,
+      params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
+    );
+    return wrapResult(scanResult, ScanFunctionResultSchema);
   });
 
   // catalog_table_scan_branches_get — branches-aware variant of
   // scan_function_get. The branches-aware C++ extension calls this for every
   // table scan; single-source catalogs synthesise a one-branch result.
-  catalogUnary(protocol, signingKey, "catalog_table_scan_branches_get", {
-    params: CatalogTableScanBranchesGetParamsSchema,
-    result: RESULT_BINARY_SCHEMA,
-    handler: async (params) => {
-      const cat = getCatalog();
-      const branchesResult = await cat.tableScanBranchesGet(
-        toUint8Array(params.attach_opaque_data),
-        params.schema_path,
-        params.name,
-        params.at_unit,
-        params.at_value,
-        params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
-      );
-      return wrapResult(branchesResult, ScanBranchesResultSchema);
-    },
+  const catalogTableScanBranchesGet = catalogHandler(signingKey, async (params) => {
+    const cat = getCatalog();
+    const branchesResult = await cat.tableScanBranchesGet(
+      toUint8Array(params.attach_opaque_data),
+      params.schema_path,
+      params.name,
+      params.at_unit,
+      params.at_value,
+      params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
+    );
+    return wrapResult(branchesResult, ScanBranchesResultSchema);
   });
 
   // catalog_table_comment_set
-  catalogUnary(protocol, signingKey, "catalog_table_comment_set", {
-    params: CatalogTableCommentSetParamsSchema,
-    result: emptyResultSchema,
-    handler: async (params) => {
-      const cat = getCatalog();
-      await cat.tableCommentSet(
-        toUint8Array(params.attach_opaque_data),
-        params.schema_path,
-        params.name,
-        params.comment,
-        params.ignore_not_found,
-        params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
-      );
-      return {};
-    },
+  const catalogTableCommentSet = catalogHandler(signingKey, async (params) => {
+    const cat = getCatalog();
+    await cat.tableCommentSet(
+      toUint8Array(params.attach_opaque_data),
+      params.schema_path,
+      params.name,
+      params.comment,
+      params.ignore_not_found,
+      params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
+    );
+    return {};
   });
 
   // catalog_table_rename
-  catalogUnary(protocol, signingKey, "catalog_table_rename", {
-    params: CatalogTableRenameParamsSchema,
-    result: emptyResultSchema,
-    handler: async (params) => {
-      const cat = getCatalog();
-      await cat.tableRename(
-        toUint8Array(params.attach_opaque_data),
-        params.schema_path,
-        params.name,
-        params.new_name,
-        params.ignore_not_found,
-        params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
-      );
-      return {};
-    },
+  const catalogTableRename = catalogHandler(signingKey, async (params) => {
+    const cat = getCatalog();
+    await cat.tableRename(
+      toUint8Array(params.attach_opaque_data),
+      params.schema_path,
+      params.name,
+      params.new_name,
+      params.ignore_not_found,
+      params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
+    );
+    return {};
   });
 
   // catalog_table_column_add
-  catalogUnary(protocol, signingKey, "catalog_table_column_add", {
-    params: CatalogTableColumnAddParamsSchema,
-    result: emptyResultSchema,
-    handler: async (params) => {
-      const cat = getCatalog();
-      await cat.tableColumnAdd(
-        toUint8Array(params.attach_opaque_data),
-        params.schema_path,
-        params.name,
-        params.column_name,
-        params.column_type,
-        params.default_value,
-        params.ignore_not_found,
-        params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
-      );
-      return {};
-    },
+  const catalogTableColumnAdd = catalogHandler(signingKey, async (params) => {
+    const cat = getCatalog();
+    await cat.tableColumnAdd(
+      toUint8Array(params.attach_opaque_data),
+      params.schema_path,
+      params.name,
+      params.column_name,
+      params.column_type,
+      params.default_value,
+      params.ignore_not_found,
+      params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
+    );
+    return {};
   });
 
   // catalog_table_column_drop
-  catalogUnary(protocol, signingKey, "catalog_table_column_drop", {
-    params: CatalogTableColumnDropParamsSchema,
-    result: emptyResultSchema,
-    handler: async (params) => {
-      const cat = getCatalog();
-      await cat.tableColumnDrop(
-        toUint8Array(params.attach_opaque_data),
-        params.schema_path,
-        params.name,
-        params.column_name,
-        params.ignore_not_found,
-        params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
-      );
-      return {};
-    },
+  const catalogTableColumnDrop = catalogHandler(signingKey, async (params) => {
+    const cat = getCatalog();
+    await cat.tableColumnDrop(
+      toUint8Array(params.attach_opaque_data),
+      params.schema_path,
+      params.name,
+      params.column_name,
+      params.ignore_not_found,
+      params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
+    );
+    return {};
   });
 
   // catalog_table_column_rename
-  catalogUnary(protocol, signingKey, "catalog_table_column_rename", {
-    params: CatalogTableColumnRenameParamsSchema,
-    result: emptyResultSchema,
-    handler: async (params) => {
-      const cat = getCatalog();
-      await cat.tableColumnRename(
-        toUint8Array(params.attach_opaque_data),
-        params.schema_path,
-        params.name,
-        params.column_name,
-        params.new_name,
-        params.ignore_not_found,
-        params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
-      );
-      return {};
-    },
+  const catalogTableColumnRename = catalogHandler(signingKey, async (params) => {
+    const cat = getCatalog();
+    await cat.tableColumnRename(
+      toUint8Array(params.attach_opaque_data),
+      params.schema_path,
+      params.name,
+      params.column_name,
+      params.new_name,
+      params.ignore_not_found,
+      params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
+    );
+    return {};
   });
 
   // catalog_table_column_default_set
-  catalogUnary(protocol, signingKey, "catalog_table_column_default_set", {
-    params: CatalogTableColumnDefaultSetParamsSchema,
-    result: emptyResultSchema,
-    handler: async (params) => {
-      const cat = getCatalog();
-      await cat.tableColumnDefaultSet(
-        toUint8Array(params.attach_opaque_data),
-        params.schema_path,
-        params.name,
-        params.column_name,
-        params.default_value,
-        params.ignore_not_found,
-        params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
-      );
-      return {};
-    },
+  const catalogTableColumnDefaultSet = catalogHandler(signingKey, async (params) => {
+    const cat = getCatalog();
+    await cat.tableColumnDefaultSet(
+      toUint8Array(params.attach_opaque_data),
+      params.schema_path,
+      params.name,
+      params.column_name,
+      params.default_value,
+      params.ignore_not_found,
+      params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
+    );
+    return {};
   });
 
   // catalog_table_column_default_drop
-  catalogUnary(protocol, signingKey, "catalog_table_column_default_drop", {
-    params: CatalogTableColumnDefaultDropParamsSchema,
-    result: emptyResultSchema,
-    handler: async (params) => {
-      const cat = getCatalog();
-      await cat.tableColumnDefaultDrop(
-        toUint8Array(params.attach_opaque_data),
-        params.schema_path,
-        params.name,
-        params.column_name,
-        params.ignore_not_found,
-        params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
-      );
-      return {};
-    },
+  const catalogTableColumnDefaultDrop = catalogHandler(signingKey, async (params) => {
+    const cat = getCatalog();
+    await cat.tableColumnDefaultDrop(
+      toUint8Array(params.attach_opaque_data),
+      params.schema_path,
+      params.name,
+      params.column_name,
+      params.ignore_not_found,
+      params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
+    );
+    return {};
   });
 
   // catalog_table_column_type_change
-  catalogUnary(protocol, signingKey, "catalog_table_column_type_change", {
-    params: CatalogTableColumnTypeChangeParamsSchema,
-    result: emptyResultSchema,
-    handler: async (params) => {
-      const cat = getCatalog();
-      await cat.tableColumnTypeChange(
-        toUint8Array(params.attach_opaque_data),
-        params.schema_path,
-        params.name,
-        params.column_name,
-        params.new_type,
-        params.ignore_not_found,
-        params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
-      );
-      return {};
-    },
+  const catalogTableColumnTypeChange = catalogHandler(signingKey, async (params) => {
+    const cat = getCatalog();
+    await cat.tableColumnTypeChange(
+      toUint8Array(params.attach_opaque_data),
+      params.schema_path,
+      params.name,
+      params.column_name,
+      params.new_type,
+      params.ignore_not_found,
+      params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
+    );
+    return {};
   });
 
   // catalog_table_not_null_set
-  catalogUnary(protocol, signingKey, "catalog_table_not_null_set", {
-    params: CatalogTableNotNullSetParamsSchema,
-    result: emptyResultSchema,
-    handler: async (params) => {
-      const cat = getCatalog();
-      await cat.tableNotNullSet(
-        toUint8Array(params.attach_opaque_data),
-        params.schema_path,
-        params.name,
-        params.column_name,
-        params.ignore_not_found,
-        params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
-      );
-      return {};
-    },
+  const catalogTableNotNullSet = catalogHandler(signingKey, async (params) => {
+    const cat = getCatalog();
+    await cat.tableNotNullSet(
+      toUint8Array(params.attach_opaque_data),
+      params.schema_path,
+      params.name,
+      params.column_name,
+      params.ignore_not_found,
+      params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
+    );
+    return {};
   });
 
   // catalog_table_not_null_drop
-  catalogUnary(protocol, signingKey, "catalog_table_not_null_drop", {
-    params: CatalogTableNotNullDropParamsSchema,
-    result: emptyResultSchema,
-    handler: async (params) => {
-      const cat = getCatalog();
-      await cat.tableNotNullDrop(
-        toUint8Array(params.attach_opaque_data),
-        params.schema_path,
-        params.name,
-        params.column_name,
-        params.ignore_not_found,
-        params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
-      );
-      return {};
-    },
+  const catalogTableNotNullDrop = catalogHandler(signingKey, async (params) => {
+    const cat = getCatalog();
+    await cat.tableNotNullDrop(
+      toUint8Array(params.attach_opaque_data),
+      params.schema_path,
+      params.name,
+      params.column_name,
+      params.ignore_not_found,
+      params.transaction_opaque_data ? toUint8Array(params.transaction_opaque_data) : undefined
+    );
+    return {};
   });
+
+  return {
+    catalogTableGet,
+    catalogTableCreate,
+    catalogTableDrop,
+    catalogTableColumnStatisticsGet,
+    catalogTableScanFunctionGet,
+    catalogTableScanBranchesGet,
+    catalogTableCommentSet,
+    catalogTableRename,
+    catalogTableColumnAdd,
+    catalogTableColumnDrop,
+    catalogTableColumnRename,
+    catalogTableColumnDefaultSet,
+    catalogTableColumnDefaultDrop,
+    catalogTableColumnTypeChange,
+    catalogTableNotNullSet,
+    catalogTableNotNullDrop,
+  };
 }