@databricks/sdk-jobs 0.1.0-dev.1 → 0.1.0-dev.3

package/src/v2/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> {