@databricks/sdk-postgres 0.1.0-dev.2 → 0.1.0-dev.4

package/src/v1/index.ts

@@ -41,6 +41,7 @@ export {
 } from './model';
 
 export type {
+  ApiError,
   Branch,
   BranchOperationMetadata,
   BranchSpec,
@@ -61,7 +62,6 @@ export type {
   Database_DatabaseStatus,
   DatabaseCredential,
   DatabaseOperationMetadata,
-  DatabricksServiceExceptionWithDetailsProto,
   DeleteBranchRequest,
   DeleteCatalogRequest,
   DeleteDatabaseRequest,

package/src/v1/model.ts

@@ -685,6 +685,14 @@ export enum SyncedTable_SyncedTableSpec_SyncedTableSchedulingPolicy {
   SNAPSHOT = 'SNAPSHOT',
 }
 
+/** Databricks Error that is returned by all Databricks APIs. */
+export interface ApiError {
+  errorCode?: ErrorCode | undefined;
+  message?: string | undefined;
+  stackTrace?: string | undefined;
+  details?: Record<string, unknown>[] | undefined;
+}
+
 export interface Branch {
   /**
    * Output only. The full resource path of the branch.
@@ -1061,14 +1069,6 @@ export interface DatabaseCredential {
 // eslint-disable-next-line @typescript-eslint/no-empty-object-type
 export interface DatabaseOperationMetadata {}
 
-/** Databricks Error that is returned by all Databricks APIs. */
-export interface DatabricksServiceExceptionWithDetailsProto {
-  errorCode?: ErrorCode | undefined;
-  message?: string | undefined;
-  stackTrace?: string | undefined;
-  details?: Record<string, unknown>[] | undefined;
-}
-
 export interface DeleteBranchRequest {
   /**
    * The full resource path of the branch to delete.
@@ -1555,7 +1555,7 @@ export interface Operation {
     | {
         $case: 'error';
         /** The error result of the operation in case of failure or cancellation. */
-        error: DatabricksServiceExceptionWithDetailsProto;
+        error: ApiError;
       }
     | {
         $case: 'response';
@@ -2058,6 +2058,20 @@ export interface UpdateRoleRequest {
   updateMask?: FieldMask<Role> | undefined;
 }
 
+export const unmarshalApiErrorSchema: z.ZodType<ApiError> = z
+  .object({
+    error_code: z.enum(ErrorCode).optional(),
+    message: z.string().optional(),
+    stack_trace: z.string().optional(),
+    details: z.array(z.record(z.string(), z.unknown())).optional(),
+  })
+  .transform(d => ({
+    errorCode: d.error_code,
+    message: d.message,
+    stackTrace: d.stack_trace,
+    details: d.details,
+  }));
+
 export const unmarshalBranchSchema: z.ZodType<Branch> = z
   .object({
     name: z.string().optional(),
@@ -2293,21 +2307,6 @@ export const unmarshalDatabaseCredentialSchema: z.ZodType<DatabaseCredential> =
 export const unmarshalDatabaseOperationMetadataSchema: z.ZodType<DatabaseOperationMetadata> =
   z.object({});
 
-export const unmarshalDatabricksServiceExceptionWithDetailsProtoSchema: z.ZodType<DatabricksServiceExceptionWithDetailsProto> =
-  z
-    .object({
-      error_code: z.enum(ErrorCode).optional(),
-      message: z.string().optional(),
-      stack_trace: z.string().optional(),
-      details: z.array(z.record(z.string(), z.unknown())).optional(),
-    })
-    .transform(d => ({
-      errorCode: d.error_code,
-      message: d.message,
-      stackTrace: d.stack_trace,
-      details: d.details,
-    }));
-
 export const unmarshalDeltaTableSyncInfoSchema: z.ZodType<DeltaTableSyncInfo> =
   z
     .object({
@@ -2540,9 +2539,7 @@ export const unmarshalOperationSchema: z.ZodType<Operation> = z
     name: z.string().optional(),
     metadata: z.record(z.string(), z.unknown()).optional(),
     done: z.boolean().optional(),
-    error: z
-      .lazy(() => unmarshalDatabricksServiceExceptionWithDetailsProtoSchema)
-      .optional(),
+    error: z.lazy(() => unmarshalApiErrorSchema).optional(),
     response: z.record(z.string(), z.unknown()).optional(),
   })
   .transform(d => ({

package/src/v1/utils.ts

@@ -1,7 +1,7 @@
 // Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
 
-import type {Call, Options} from '@databricks/sdk-core/api';
-import {execute} from '@databricks/sdk-core/api';
+import type {Options} from '@databricks/sdk-core/ops';
+import {execute, retryOn} from '@databricks/sdk-core/ops';
 import {ApiError} from '@databricks/sdk-core/apierror';
 import type {
   HttpClient,
@@ -10,6 +10,7 @@ import type {
 } from '@databricks/sdk-core/http';
 import type {Logger} from '@databricks/sdk-core/logger';
 import type {CallOptions} from '@databricks/sdk-options/call';
+import type {LroOptions} from '@databricks/sdk-options/lro';
 import JSONBig from 'json-bigint';
 import type {z} from 'zod';
 
@@ -30,7 +31,7 @@ export interface HttpCallOptions {
  * API from the executor's internal type so they can diverge.
  */
 export async function executeCall(
-  call: Call,
+  call: (signal?: AbortSignal) => Promise<void>,
   options?: CallOptions
 ): Promise<void> {
   const opts: Options = {
@@ -43,6 +44,29 @@ export async function executeCall(
   return execute(options?.signal, call, opts);
 }
 
+/**
+ * Sentinel thrown by a polling call to signal that the operation has not
+ * yet reached a terminal state. {@link executeWait} treats this error as
+ * retriable; any other error aborts the wait.
+ */
+export class StillRunningError extends Error {}
+
+/**
+ * Polls until the call returns without throwing {@link StillRunningError}.
+ * Abort and overall-deadline behavior come from the supplied LroOptions.
+ */
+export async function executeWait(
+  call: (signal?: AbortSignal) => Promise<void>,
+  options?: LroOptions
+): Promise<void> {
+  const opts: Options = {
+    ...(options?.timeout !== undefined && {timeout: options.timeout}),
+    retrier: () =>
+      retryOn({}, (err: Error) => err instanceof StillRunningError),
+  };
+  return execute(options?.signal, call, opts);
+}
+
 async function readAll(
   body: ReadableStream<Uint8Array> | null
 ): Promise<Uint8Array> {