@dremio/js-sdk 0.3.0 → 0.4.0

package/dist/cloud/ai/AiResource.d.ts

@@ -0,0 +1,24 @@
+import { Err, Ok } from "ts-results-es";
+import type { ResourceConfig, V3Config } from "../../common/Config.ts";
+import type { SignalParam } from "../../common/Params.ts";
+export declare class AiResource {
+    #private;
+    constructor(config: ResourceConfig & V3Config);
+    autoProduceQuery: (project_id: string) => ({ datasetPaths, text }: {
+        datasetPaths: Set<string>;
+        text: string;
+    }, { signal }?: SignalParam) => Promise<Ok<string> | Err<{
+        readonly detail: "Please clarify your request and try again.";
+        readonly title: "SQL could not be generated from the input text.";
+        readonly type: "https://api.dremio.dev/problems/auto-produce-sql/clarify-request";
+    }> | Err<{
+        readonly detail: string;
+        readonly title: "An unexpected error occurred while processing this request.";
+        readonly type: "https://api.dremio.dev/problems/unexpected-error";
+    }>>;
+}
+export declare const clarifyRequestErr: {
+    readonly detail: "Please clarify your request and try again.";
+    readonly title: "SQL could not be generated from the input text.";
+    readonly type: "https://api.dremio.dev/problems/auto-produce-sql/clarify-request";
+};

package/dist/cloud/ai/AiResource.js

@@ -0,0 +1,55 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { Err, Ok } from "ts-results-es";
+import { HttpError } from "../../common/HttpError.js";
+import { isProblem } from "../../common/Problem.js";
+export class AiResource {
+    #config;
+    constructor(config) {
+        this.#config = config;
+    }
+    autoProduceQuery = (project_id) => ({ datasetPaths, text }, { signal } = {}) => this.#config
+        .v3Request("gen-ai/auto-produce-query", {
+        body: JSON.stringify({
+            datasetIdentifiers: Array.from(datasetPaths).map((datasetPath) => ({
+                datasetPath,
+            })),
+            projectId: project_id,
+            userInputtedText: text,
+        }),
+        headers: {
+            Accept: "application/json",
+            "Content-Type": "application/json",
+        },
+        method: "POST",
+        signal,
+    })
+        .then((res) => res.json())
+        .then((response) => Ok(response.sql))
+        .catch((e) => {
+        if (e instanceof HttpError && isProblem(e.body)) {
+            if (e.body.detail?.includes("INVALID_ARGUMENT: SQL could not be generated. Please clarify your request."))
+                return Err(clarifyRequestErr);
+        }
+        return Err(e);
+    });
+}
+export const clarifyRequestErr = {
+    detail: "Please clarify your request and try again.",
+    title: "SQL could not be generated from the input text.",
+    type: "https://api.dremio.dev/problems/auto-produce-sql/clarify-request",
+};
+//# sourceMappingURL=AiResource.js.map

package/dist/cloud/ai/AiResource.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AiResource.js","sourceRoot":"","sources":["../../../src/cloud/ai/AiResource.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,eAAe,CAAC;AAGxC,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AACtD,OAAO,EAAE,SAAS,EAAgB,MAAM,yBAAyB,CAAC;AAGlE,MAAM,OAAO,UAAU;IACZ,OAAO,CAA4B;IAE5C,YAAY,MAAiC;QAC3C,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAED,gBAAgB,GACd,CAAC,UAAkB,EAAE,EAAE,CACvB,CACE,EAAE,YAAY,EAAE,IAAI,EAA+C,EACnE,EAAE,MAAM,KAAkB,EAAE,EAC5B,EAAE,CACF,IAAI,CAAC,OAAO;SACT,SAAS,CAAC,2BAA2B,EAAE;QACtC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;YACnB,kBAAkB,EAAE,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;gBACjE,WAAW;aACZ,CAAC,CAAC;YACH,SAAS,EAAE,UAAU;YACrB,gBAAgB,EAAE,IAAI;SACvB,CAAC;QACF,OAAO,EAAE;YACP,MAAM,EAAE,kBAAkB;YAC1B,cAAc,EAAE,kBAAkB;SACnC;QACD,MAAM,EAAE,MAAM;QACd,MAAM;KACP,CAAC;SACD,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAA8B,CAAC;SACrD,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;SACpC,KAAK,CAAC,CAAC,CAAU,EAAE,EAAE;QACpB,IAAI,CAAC,YAAY,SAAS,IAAI,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;YAChD,IACE,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,CACrB,4EAA4E,CAC7E;gBAED,OAAO,GAAG,CAAC,iBAAiB,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,GAAG,CAAC,CAAuC,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;CACV;AAED,MAAM,CAAC,MAAM,iBAAiB,GAAG;IAC/B,MAAM,EAAE,4CAA4C;IACpD,KAAK,EAAE,iDAAiD;IACxD,IAAI,EAAE,kEAAkE;CAC9C,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { Err, Ok } from \"ts-results-es\";\nimport type { ResourceConfig, V3Config } from \"../../common/Config.ts\";\nimport type { SignalParam } from \"../../common/Params.ts\";\nimport { HttpError } from \"../../common/HttpError.ts\";\nimport { isProblem, type Problem } from \"../../common/Problem.ts\";\nimport type { unexpectedError } from \"../../common/problems.ts\";\n\nexport class AiResource {\n  readonly #config: ResourceConfig & V3Config;\n\n  constructor(config: ResourceConfig & V3Config) {\n    this.#config = config;\n  }\n\n  autoProduceQuery =\n    (project_id: string) =>\n    (\n      { datasetPaths, text }: { datasetPaths: Set<string>; text: string },\n      { signal }: SignalParam = {},\n    ) =>\n      this.#config\n        .v3Request(\"gen-ai/auto-produce-query\", {\n          body: JSON.stringify({\n            datasetIdentifiers: Array.from(datasetPaths).map((datasetPath) => ({\n              datasetPath,\n            })),\n            projectId: project_id,\n            userInputtedText: text,\n          }),\n          headers: {\n            Accept: \"application/json\",\n            \"Content-Type\": \"application/json\",\n          },\n          method: \"POST\",\n          signal,\n        })\n        .then((res) => res.json() as Promise<{ sql: string }>)\n        .then((response) => Ok(response.sql))\n        .catch((e: unknown) => {\n          if (e instanceof HttpError && isProblem(e.body)) {\n            if (\n              e.body.detail?.includes(\n                \"INVALID_ARGUMENT: SQL could not be generated. Please clarify your request.\",\n              )\n            )\n              return Err(clarifyRequestErr);\n          }\n          return Err(e as ReturnType<typeof unexpectedError>);\n        });\n}\n\nexport const clarifyRequestErr = {\n  detail: \"Please clarify your request and try again.\",\n  title: \"SQL could not be generated from the input text.\",\n  type: \"https://api.dremio.dev/problems/auto-produce-sql/clarify-request\",\n} as const satisfies Problem;\n"]}

package/dist/cloud/index.d.ts

@@ -16,6 +16,7 @@ export declare const Dremio: (config: Config) => {
         v3Request: (path: string, init?: RequestInit) => Promise<Response>;
     };
     _sonarV3Request: (projectId: string) => (path: string, init?: RequestInit) => Promise<Response>;
+    ai: import("./ai/AiResource.ts").AiResource;
     arctic: import("moize").Moized<(projectId: string) => {
         _createFromEntity: (entity: import("./arctic/ArcticCatalog.ts").ArcticCatalogEntity) => import("./interfaces.ts").ArcticCatalog;
         list: () => {
@@ -217,6 +218,7 @@ export declare const Dremio: (config: Config) => {
     }>>;
     jobs: import("moize").Moized<(projectId: string) => {
         create: (query: import("../common/Query.ts").Query) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<string>>;
+        createAndRetrieve: (query: import("../common/Query.ts").Query) => Promise<import("ts-results-es").Result<import("./interfaces.ts").Job, unknown>>;
         observe: (id: string) => import("rxjs").Observable<import("../oss/jobs/Job.ts").JobResult>;
         retrieve: (id: string, { signal }?: import("../common/Params.ts").SignalParam) => Promise<import("../oss/jobs/Job.ts").JobResult>;
     }, Partial<{
@@ -232,16 +234,19 @@ export declare const Dremio: (config: Config) => {
         maxSize: number;
         onCacheAdd: import("moize").OnCacheOperation<(projectId: string) => {
             create: (query: import("../common/Query.ts").Query) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<string>>;
+            createAndRetrieve: (query: import("../common/Query.ts").Query) => Promise<import("ts-results-es").Result<import("./interfaces.ts").Job, unknown>>;
             observe: (id: string) => import("rxjs").Observable<import("../oss/jobs/Job.ts").JobResult>;
             retrieve: (id: string, { signal }?: import("../common/Params.ts").SignalParam) => Promise<import("../oss/jobs/Job.ts").JobResult>;
         }>;
         onCacheChange: import("moize").OnCacheOperation<(projectId: string) => {
             create: (query: import("../common/Query.ts").Query) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<string>>;
+            createAndRetrieve: (query: import("../common/Query.ts").Query) => Promise<import("ts-results-es").Result<import("./interfaces.ts").Job, unknown>>;
             observe: (id: string) => import("rxjs").Observable<import("../oss/jobs/Job.ts").JobResult>;
             retrieve: (id: string, { signal }?: import("../common/Params.ts").SignalParam) => Promise<import("../oss/jobs/Job.ts").JobResult>;
         }>;
         onCacheHit: import("moize").OnCacheOperation<(projectId: string) => {
             create: (query: import("../common/Query.ts").Query) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<string>>;
+            createAndRetrieve: (query: import("../common/Query.ts").Query) => Promise<import("ts-results-es").Result<import("./interfaces.ts").Job, unknown>>;
             observe: (id: string) => import("rxjs").Observable<import("../oss/jobs/Job.ts").JobResult>;
             retrieve: (id: string, { signal }?: import("../common/Params.ts").SignalParam) => Promise<import("../oss/jobs/Job.ts").JobResult>;
         }>;

package/dist/cloud/resources.d.ts

@@ -1,10 +1,12 @@
 import type { ResourceConfig, SonarV3Config, V3Config } from "../common/Config.ts";
 import moize from "moize";
+import { AiResource } from "./ai/AiResource.ts";
 /**
  * @internal
  * @hidden
  */
 export declare const Resources: (config: (projectId?: string) => ResourceConfig & SonarV3Config & V3Config) => {
+    ai: AiResource;
     arctic: moize.Moized<(projectId: string) => {
         _createFromEntity: (entity: import("./arctic/ArcticCatalog.ts").ArcticCatalogEntity) => import("./interfaces.ts").ArcticCatalog;
         list: () => {
@@ -206,6 +208,7 @@ export declare const Resources: (config: (projectId?: string) => ResourceConfig
     }>>;
     jobs: moize.Moized<(projectId: string) => {
         create: (query: import("./index.ts").Query) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<string>>;
+        createAndRetrieve: (query: import("./index.ts").Query) => Promise<import("ts-results-es").Result<import("./interfaces.ts").Job, unknown>>;
         observe: (id: string) => import("rxjs").Observable<import("../oss/jobs/Job.ts").JobResult>;
         retrieve: (id: string, { signal }?: import("../common/Params.ts").SignalParam) => Promise<import("../oss/jobs/Job.ts").JobResult>;
     }, Partial<{
@@ -221,16 +224,19 @@ export declare const Resources: (config: (projectId?: string) => ResourceConfig
         maxSize: number;
         onCacheAdd: moize.OnCacheOperation<(projectId: string) => {
             create: (query: import("./index.ts").Query) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<string>>;
+            createAndRetrieve: (query: import("./index.ts").Query) => Promise<import("ts-results-es").Result<import("./interfaces.ts").Job, unknown>>;
             observe: (id: string) => import("rxjs").Observable<import("../oss/jobs/Job.ts").JobResult>;
             retrieve: (id: string, { signal }?: import("../common/Params.ts").SignalParam) => Promise<import("../oss/jobs/Job.ts").JobResult>;
         }>;
         onCacheChange: moize.OnCacheOperation<(projectId: string) => {
             create: (query: import("./index.ts").Query) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<string>>;
+            createAndRetrieve: (query: import("./index.ts").Query) => Promise<import("ts-results-es").Result<import("./interfaces.ts").Job, unknown>>;
             observe: (id: string) => import("rxjs").Observable<import("../oss/jobs/Job.ts").JobResult>;
             retrieve: (id: string, { signal }?: import("../common/Params.ts").SignalParam) => Promise<import("../oss/jobs/Job.ts").JobResult>;
         }>;
         onCacheHit: moize.OnCacheOperation<(projectId: string) => {
             create: (query: import("./index.ts").Query) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<string>>;
+            createAndRetrieve: (query: import("./index.ts").Query) => Promise<import("ts-results-es").Result<import("./interfaces.ts").Job, unknown>>;
             observe: (id: string) => import("rxjs").Observable<import("../oss/jobs/Job.ts").JobResult>;
             retrieve: (id: string, { signal }?: import("../common/Params.ts").SignalParam) => Promise<import("../oss/jobs/Job.ts").JobResult>;
         }>;

package/dist/cloud/resources.js

@@ -23,11 +23,13 @@ import { CloudUsersResource } from "./users/CloudUsersResource.js";
 import { JobsResource } from "../oss/jobs/JobsResource.js";
 import moize from "moize";
 import { OAuthApplicationsResource } from "./oauth/OAuthApplicationsResource.js";
+import { AiResource } from "./ai/AiResource.js";
 /**
  * @internal
  * @hidden
  */
 export const Resources = (config) => ({
+    ai: new AiResource(config()),
     arctic: moize.default((projectId) => ArcticResource(config(projectId))),
     catalog: moize.default((projectId) => EnterpriseCatalogResource(config(projectId))),
     engines: moize.default((projectId) => EnginesResource(config(projectId))),

package/dist/cloud/resources.js.map

@@ -1 +1 @@
-{"version":3,"file":"resources.js","sourceRoot":"","sources":["../../src/cloud/resources.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,yBAAyB,EAAE,MAAM,oDAAoD,CAAC;AAC/F,OAAO,EAAE,aAAa,EAAE,MAAM,sCAAsC,CAAC;AACrE,OAAO,EAAE,yBAAyB,EAAE,MAAM,oDAAoD,CAAC;AAC/F,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAClE,OAAO,EAAE,kBAAkB,EAAE,MAAM,+BAA+B,CAAC;AACnE,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,yBAAyB,EAAE,MAAM,sCAAsC,CAAC;AAEjF;;;GAGG;AACH,MAAM,CAAC,MAAM,SAAS,GAAG,CACvB,MAAyE,EACzE,EAAE,CAAC,CAAC;IACJ,MAAM,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,SAAiB,EAAE,EAAE,CAAC,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;IAC/E,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,SAAiB,EAAE,EAAE,CAAC,yBAAyB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;IAC3F,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,SAAiB,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;IACjF,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,SAAiB,EAAE,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;IAC3E,iBAAiB,EAAE,yBAAyB,CAAC,MAAM,EAAE,CAAC;IACtD,QAAQ,EAAE,gBAAgB,CAAC,MAAM,EAAE,CAAC;IACpC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,SAAiB,EAAE,EAAE,CAAC,aAAa,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;IAC7E,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,SAAiB,EAAE,EAAE,CAAC,yBAAyB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;IAC3F,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,SAAiB,EAAE,EAAE,CAAC,kBAAkB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;CACnF,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { ResourceConfig, SonarV3Config, V3Config } from \"../common/Config.ts\";\nimport { ArcticResource } from \"./arctic/ArcticResource.ts\";\nimport { EnterpriseCatalogResource } from \"../enterprise/catalog/EnterpriseCatalogResource.ts\";\nimport { RolesResource } from \"../enterprise/roles/RolesResource.ts\";\nimport { EnterpriseScriptsResource } from \"../enterprise/scripts/EnterpriseScriptsResource.ts\";\nimport { EnginesResource } from \"./engines/EnginesResource.ts\";\nimport { ProjectsResource } from \"./projects/ProjectsResource.ts\";\nimport { CloudUsersResource } from \"./users/CloudUsersResource.ts\";\nimport { JobsResource } from \"../oss/jobs/JobsResource.ts\";\nimport moize from \"moize\";\nimport { OAuthApplicationsResource } from \"./oauth/OAuthApplicationsResource.ts\";\n\n/**\n * @internal\n * @hidden\n */\nexport const Resources = (\n  config: (projectId?: string) => ResourceConfig & SonarV3Config & V3Config,\n) => ({\n  arctic: moize.default((projectId: string) => ArcticResource(config(projectId))),\n  catalog: moize.default((projectId: string) => EnterpriseCatalogResource(config(projectId))),\n  engines: moize.default((projectId: string) => EnginesResource(config(projectId))),\n  jobs: moize.default((projectId: string) => JobsResource(config(projectId))),\n  oauthApplications: OAuthApplicationsResource(config()),\n  projects: ProjectsResource(config()),\n  roles: moize.default((projectId: string) => RolesResource(config(projectId))),\n  scripts: moize.default((projectId: string) => EnterpriseScriptsResource(config(projectId))),\n  users: moize.default((projectId: string) => CloudUsersResource(config(projectId))),\n});\n"]}
+{"version":3,"file":"resources.js","sourceRoot":"","sources":["../../src/cloud/resources.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,yBAAyB,EAAE,MAAM,oDAAoD,CAAC;AAC/F,OAAO,EAAE,aAAa,EAAE,MAAM,sCAAsC,CAAC;AACrE,OAAO,EAAE,yBAAyB,EAAE,MAAM,oDAAoD,CAAC;AAC/F,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAClE,OAAO,EAAE,kBAAkB,EAAE,MAAM,+BAA+B,CAAC;AACnE,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,yBAAyB,EAAE,MAAM,sCAAsC,CAAC;AACjF,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAEhD;;;GAGG;AACH,MAAM,CAAC,MAAM,SAAS,GAAG,CACvB,MAAyE,EACzE,EAAE,CAAC,CAAC;IACJ,EAAE,EAAE,IAAI,UAAU,CAAC,MAAM,EAAE,CAAC;IAC5B,MAAM,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,SAAiB,EAAE,EAAE,CAAC,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;IAC/E,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,SAAiB,EAAE,EAAE,CAAC,yBAAyB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;IAC3F,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,SAAiB,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;IACjF,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,SAAiB,EAAE,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;IAC3E,iBAAiB,EAAE,yBAAyB,CAAC,MAAM,EAAE,CAAC;IACtD,QAAQ,EAAE,gBAAgB,CAAC,MAAM,EAAE,CAAC;IACpC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,SAAiB,EAAE,EAAE,CAAC,aAAa,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;IAC7E,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,SAAiB,EAAE,EAAE,CAAC,yBAAyB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;IAC3F,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,SAAiB,EAAE,EAAE,CAAC,kBAAkB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;CACnF,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { ResourceConfig, SonarV3Config, V3Config } from \"../common/Config.ts\";\nimport { ArcticResource } from \"./arctic/ArcticResource.ts\";\nimport { EnterpriseCatalogResource } from \"../enterprise/catalog/EnterpriseCatalogResource.ts\";\nimport { RolesResource } from \"../enterprise/roles/RolesResource.ts\";\nimport { EnterpriseScriptsResource } from \"../enterprise/scripts/EnterpriseScriptsResource.ts\";\nimport { EnginesResource } from \"./engines/EnginesResource.ts\";\nimport { ProjectsResource } from \"./projects/ProjectsResource.ts\";\nimport { CloudUsersResource } from \"./users/CloudUsersResource.ts\";\nimport { JobsResource } from \"../oss/jobs/JobsResource.ts\";\nimport moize from \"moize\";\nimport { OAuthApplicationsResource } from \"./oauth/OAuthApplicationsResource.ts\";\nimport { AiResource } from \"./ai/AiResource.ts\";\n\n/**\n * @internal\n * @hidden\n */\nexport const Resources = (\n  config: (projectId?: string) => ResourceConfig & SonarV3Config & V3Config,\n) => ({\n  ai: new AiResource(config()),\n  arctic: moize.default((projectId: string) => ArcticResource(config(projectId))),\n  catalog: moize.default((projectId: string) => EnterpriseCatalogResource(config(projectId))),\n  engines: moize.default((projectId: string) => EnginesResource(config(projectId))),\n  jobs: moize.default((projectId: string) => JobsResource(config(projectId))),\n  oauthApplications: OAuthApplicationsResource(config()),\n  projects: ProjectsResource(config()),\n  roles: moize.default((projectId: string) => RolesResource(config(projectId))),\n  scripts: moize.default((projectId: string) => EnterpriseScriptsResource(config(projectId))),\n  users: moize.default((projectId: string) => CloudUsersResource(config(projectId))),\n});\n"]}

package/dist/enterprise/index.d.ts

@@ -29,6 +29,7 @@ export declare const Dremio: (config: Config) => {
     engines: import("./engines/EnginesResource.ts").EnginesResource;
     jobs: {
         create: (query: import("../common/Query.ts").Query) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<string>>;
+        createAndRetrieve: (query: import("../common/Query.ts").Query) => Promise<import("ts-results-es").Result<import("./interfaces.ts").Job, unknown>>;
         observe: (id: string) => import("rxjs").Observable<import("../oss/jobs/Job.ts").JobResult>;
         retrieve: (id: string, { signal }?: import("../common/Params.ts").SignalParam) => Promise<import("../oss/jobs/Job.ts").JobResult>;
     };

package/dist/enterprise/resources.d.ts

@@ -16,6 +16,7 @@ export declare const Resources: (config: ResourceConfig & SonarV2Config & SonarV
     engines: EnginesResource;
     jobs: {
         create: (query: import("./index.ts").Query) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<string>>;
+        createAndRetrieve: (query: import("./index.ts").Query) => Promise<import("ts-results-es").Result<import("./interfaces.ts").Job, unknown>>;
         observe: (id: string) => import("rxjs").Observable<import("../oss/jobs/Job.ts").JobResult>;
         retrieve: (id: string, { signal }?: import("../common/Params.ts").SignalParam) => Promise<import("../oss/jobs/Job.ts").JobResult>;
     };

package/dist/oss/catalog/CatalogReferences/BaseCatalogReference.d.ts

@@ -8,7 +8,7 @@ export declare abstract class BaseCatalogReference {
     abstract catalogObject(): Promise<Result<CatalogObject, unknown>>;
     constructor(properties: BaseCatalogReferenceProperties);
     get name(): string;
-    pathString: (SEPARATOR?: string) => string;
+    pathString(SEPARATOR?: string): string;
 }
 export type BaseCatalogReferenceProperties = {
     id: string;

package/dist/oss/catalog/CatalogReferences/BaseCatalogReference.js

@@ -13,6 +13,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
+import { catalogPathToString } from "./utils/catalogPathToString.js";
 export class BaseCatalogReference {
     id;
     path;
@@ -23,12 +24,8 @@ export class BaseCatalogReference {
     get name() {
         return this.path.at(-1);
     }
-    pathString = pathString(() => this.path);
+    pathString(SEPARATOR = ".") {
+        return catalogPathToString(this.path, SEPARATOR);
+    }
 }
-const requiresQuotes = /\W/;
-const pathString = (getPath) => (SEPARATOR = ".") => {
-    return getPath()
-        .map((part) => (requiresQuotes.test(part) ? `"${part}"` : part))
-        .join(SEPARATOR);
-};
 //# sourceMappingURL=BaseCatalogReference.js.map

package/dist/oss/catalog/CatalogReferences/BaseCatalogReference.js.map

@@ -1 +1 @@
-{"version":3,"file":"BaseCatalogReference.js","sourceRoot":"","sources":["../../../../src/oss/catalog/CatalogReferences/BaseCatalogReference.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAOH,MAAM,OAAgB,oBAAoB;IAC/B,EAAE,CAAuC;IACzC,IAAI,CAAyC;IAItD,YAAY,UAA0C;QACpD,IAAI,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC;QACxB,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC;IAC9B,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC;IAC3B,CAAC;IAED,UAAU,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;CAC1C;AAED,MAAM,cAAc,GAAG,IAAI,CAAC;AAE5B,MAAM,UAAU,GACd,CAAC,OAAuB,EAAE,EAAE,CAC5B,CAAC,YAAoB,GAAG,EAAU,EAAE;IAClC,OAAO,OAAO,EAAE;SACb,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;SAC/D,IAAI,CAAC,SAAS,CAAC,CAAC;AACrB,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { Result } from \"ts-results-es\";\nimport type { CatalogObject } from \"../CatalogObjects/index.ts\";\n\nexport type RetrieveByPath = (path: string[]) => Promise<Result<CatalogObject, unknown>>;\n\nexport abstract class BaseCatalogReference {\n  readonly id: BaseCatalogReferenceProperties[\"id\"];\n  readonly path: BaseCatalogReferenceProperties[\"path\"];\n  abstract readonly type: string;\n  abstract catalogObject(): Promise<Result<CatalogObject, unknown>>;\n\n  constructor(properties: BaseCatalogReferenceProperties) {\n    this.id = properties.id;\n    this.path = properties.path;\n  }\n\n  get name() {\n    return this.path.at(-1)!;\n  }\n\n  pathString = pathString(() => this.path);\n}\n\nconst requiresQuotes = /\\W/;\n\nconst pathString =\n  (getPath: () => string[]) =>\n  (SEPARATOR: string = \".\"): string => {\n    return getPath()\n      .map((part) => (requiresQuotes.test(part) ? `\"${part}\"` : part))\n      .join(SEPARATOR);\n  };\n\nexport type BaseCatalogReferenceProperties = { id: string; path: string[] };\n"]}
+{"version":3,"file":"BaseCatalogReference.js","sourceRoot":"","sources":["../../../../src/oss/catalog/CatalogReferences/BaseCatalogReference.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,OAAO,EAAE,mBAAmB,EAAE,MAAM,gCAAgC,CAAC;AAIrE,MAAM,OAAgB,oBAAoB;IAC/B,EAAE,CAAuC;IACzC,IAAI,CAAyC;IAItD,YAAY,UAA0C;QACpD,IAAI,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC;QACxB,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC;IAC9B,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC;IAC3B,CAAC;IAED,UAAU,CAAC,YAAoB,GAAG;QAChC,OAAO,mBAAmB,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IACnD,CAAC;CACF","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { Result } from \"ts-results-es\";\nimport type { CatalogObject } from \"../CatalogObjects/index.ts\";\nimport { catalogPathToString } from \"./utils/catalogPathToString.ts\";\n\nexport type RetrieveByPath = (path: string[]) => Promise<Result<CatalogObject, unknown>>;\n\nexport abstract class BaseCatalogReference {\n  readonly id: BaseCatalogReferenceProperties[\"id\"];\n  readonly path: BaseCatalogReferenceProperties[\"path\"];\n  abstract readonly type: string;\n  abstract catalogObject(): Promise<Result<CatalogObject, unknown>>;\n\n  constructor(properties: BaseCatalogReferenceProperties) {\n    this.id = properties.id;\n    this.path = properties.path;\n  }\n\n  get name() {\n    return this.path.at(-1)!;\n  }\n\n  pathString(SEPARATOR: string = \".\") {\n    return catalogPathToString(this.path, SEPARATOR);\n  }\n}\n\nexport type BaseCatalogReferenceProperties = { id: string; path: string[] };\n"]}

package/dist/oss/catalog/CatalogReferences/utils/catalogPathToString.d.ts

@@ -0,0 +1 @@
+export declare const catalogPathToString: (path: string[], SEPARATOR?: string) => string;

package/dist/oss/catalog/CatalogReferences/utils/catalogPathToString.js

@@ -0,0 +1,18 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+export const catalogPathToString = (path, SEPARATOR = ".") => path.map((part) => (requiresQuotes.test(part) ? `"${part}"` : part)).join(SEPARATOR);
+const requiresQuotes = /\W/;
+//# sourceMappingURL=catalogPathToString.js.map

package/dist/oss/catalog/CatalogReferences/utils/catalogPathToString.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"catalogPathToString.js","sourceRoot":"","sources":["../../../../../src/oss/catalog/CatalogReferences/utils/catalogPathToString.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,IAAc,EAAE,YAAoB,GAAG,EAAE,EAAE,CAC7E,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AAEvF,MAAM,cAAc,GAAG,IAAI,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nexport const catalogPathToString = (path: string[], SEPARATOR: string = \".\") =>\n  path.map((part) => (requiresQuotes.test(part) ? `\"${part}\"` : part)).join(SEPARATOR);\n\nconst requiresQuotes = /\\W/;\n"]}

package/dist/oss/index.d.ts

@@ -28,6 +28,7 @@ export declare const Dremio: (config: Config) => {
     };
     jobs: {
         create: (query: import("../common/Query.ts").Query) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<string>>;
+        createAndRetrieve: (query: import("../common/Query.ts").Query) => Promise<import("ts-results-es").Result<import("./interfaces.ts").Job, unknown>>;
         observe: (id: string) => import("rxjs").Observable<import("./jobs/Job.ts").JobResult>;
         retrieve: (id: string, { signal }?: import("../common/Params.ts").SignalParam) => Promise<import("./jobs/Job.ts").JobResult>;
     };

package/dist/oss/jobs/Job.d.ts

@@ -30,7 +30,7 @@ export declare class Job {
             };
             readonly totalRows: number;
         }, void, unknown>;
-        slice: <T extends Record<string, unknown> = Record<string, unknown>>(start?: number, end?: number) => Promise<T[]>;
+        slice: <T extends Record<string, unknown> = Record<string, unknown>>(start?: number, end?: number) => Promise<Err<unknown> | Ok<T[]>>;
     };
     cancel(): Promise<Err<any> | Ok<undefined>>;
 }

package/dist/oss/jobs/Job.js

@@ -112,11 +112,16 @@ export class Job {
         }
     }
     async #slice(start = 0, end = Infinity) {
-        const rows = [];
-        for await (const batch of this.#jsonBatches({ limit: end - start, offset: start })) {
-            rows.push(...batch.rows);
+        try {
+            const rows = [];
+            for await (const batch of this.#jsonBatches({ limit: end - start, offset: start })) {
+                rows.push(...batch.rows);
+            }
+            return Ok(rows);
+        }
+        catch (e) {
+            return Err(e);
         }
-        return rows;
     }
     async cancel() {
         return this.#config

package/dist/oss/jobs/Job.js.map

@@ -1 +1 @@
-{"version":3,"file":"Job.js","sourceRoot":"","sources":["../../../src/oss/jobs/Job.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAChD,OAAO,EAAE,aAAa,EAAE,GAAG,EAAE,EAAE,EAAE,SAAS,EAAmB,MAAM,MAAM,CAAC;AAE1E,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnD,MAAM,OAAO,GAAG;IACL,YAAY,CAAgC;IAC5C,kBAAkB,CAAsC;IACxD,OAAO,CAA2B;IAClC,EAAE,CAAsB;IACxB,OAAO,CAA2B;IAClC,SAAS,CAA6B;IACtC,yBAAyB,CAA6C;IACtE,2BAA2B,CAA+C;IAC1E,QAAQ,CAA4B;IACpC,SAAS,CAA6B;IACtC,KAAK,CAAyB;IAEvB,QAAQ,CAAkB;IAC1C,OAAO,CAAgB;IAEvB,YACE,UAAyB,EACzB,MAAqB,EACrB,OAA8C;QAE9C,IAAI,CAAC,YAAY,GAAG,UAAU,CAAC,YAAY,CAAC;QAC5C,IAAI,CAAC,kBAAkB,GAAG,UAAU,CAAC,kBAAkB,CAAC;QACxD,IAAI,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;QAClC,IAAI,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC;QACxB,IAAI,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;QAClC,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,yBAAyB,GAAG,UAAU,CAAC,yBAAyB,CAAC;QACtE,IAAI,CAAC,2BAA2B,GAAG,UAAU,CAAC,2BAA2B,CAAC;QAC1E,IAAI,CAAC,QAAQ,GAAG,UAAU,CAAC,QAAQ,CAAC;QACpC,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC;QAC9B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAC3B,SAAS,CAAC,CAAC,GAAG,EAAE,EAAE;YAChB,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;gBACjB,OAAO,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAC1B,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;oBACb,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;wBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;oBACrB,CAAC;oBACD,OAAO,MAAM,CAAC,KAAK,CAAC;gBACtB,CAAC,CAAC,CACH,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,OAAO,EAAE,CAAC,GAAG,CAAC,CAAC;YACjB,CAAC;QACH,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAED,IAAI,OAAO;QACT,OAAO,CACL,IAAI,CAAC,KAAK,KAAK,WAAW;YAC1B,IAAI,CAAC,KAAK,KAAK,QAAQ;YACvB,IAAI,CAAC,KAAK,KAAK,eAAe;YAC9B,IAAI,CAAC,KAAK,KAAK,UAAU,CAC1B,CAAC;IACJ,CAAC;IAED,IAAI,OAAO;QACT,OAAO;YACL,WAAW,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;YACzC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;SAC9B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,YAAY,CACjB,OAA4C,EAAE;QAE9C,IAAI,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,IAAI,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC;YACrD,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvD,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC/C,CAAC;QAED,2EAA2E;QAC3E,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,MAAM,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrC,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,IAAI,QAAQ,CAAC;QACxC,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,IAAI,CAAC,CAAC;QAEnC,6EAA6E;QAC7E,IAAI,OAAO,GAAG,IAAI,CAAC;QAEnB,uGAAuG;QACvG,IAAI,MAAM,GAAG,SAAS,CAAC;QAEvB,iEAAiE;QACjE,MAAM,eAAe,GAAG,QAAQ,GAAG,SAAS,CAAC;QAE7C,OAAO,OAAO,IAAI,MAAM,GAAG,eAAe,EAAE,CAAC;YAC3C,uEAAuE;YACvE,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,eAAe,GAAG,MAAM,CAAC,CAAC;YAEtE,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,OAAO;iBAC7B,cAAc,CAAC,OAAO,IAAI,CAAC,EAAE,mBAAmB,MAAM,UAAU,UAAU,EAAE,CAAC;iBAC7E,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAoC,CAAC,CAAC;YAE/D,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC;YAC5B,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,KAAK,UAAU,CAAC;YAE3C,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACtB,MAAM,MAAM,GAAG,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;gBACxC,UAAU,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;gBACrC,MAAM;oBACJ,IAAI,OAAO;wBACT,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC;oBACjC,CAAC;oBACD,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,MAAM;oBACN,SAAS,EAAE,KAAK,CAAC,QAAQ;iBACjB,CAAC;YACb,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CACV,QAAgB,CAAC,EACjB,MAAc,QAAQ;QAEtB,MAAM,IAAI,GAAQ,EAAE,CAAC;QACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,YAAY,CAAI,EAAE,KAAK,EAAE,GAAG,GAAG,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;YACtF,IAAI,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,MAAM;QACV,OAAO,IAAI,CAAC,OAAO;aAChB,cAAc,CAAC,OAAO,IAAI,CAAC,EAAE,SAAS,EAAE;YACvC,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,MAAM;SACf,CAAC;aACD,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC;aACzB,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1B,CAAC;CACF;AAED,MAAM,CAAC,MAAM,qBAAqB,GAAG,CACnC,EAAU,EACV,MA6CC,EACD,EAAE,CACF,CAAC;IACC,YAAY,EAAE,MAAM,CAAC,YAAY;IACjC,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;IAC7C,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI;IACzD,EAAE;IACF,OAAO,EAAE,IAAsB;IAC/B,SAAS,EAAE,MAAM,CAAC,SAAS;IAC3B,yBAAyB,EAAE,MAAM,CAAC,yBAAyB;QACzD,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,yBAAyB,CAAC;QAC5C,CAAC,CAAC,IAAI;IACR,2BAA2B,EAAE,MAAM,CAAC,2BAA2B;QAC7D,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,2BAA2B,CAAC;QAC9C,CAAC,CAAC,IAAI;IACR,QAAQ,EAAE,MAAM,CAAC,QAAQ;IACzB,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI;IAC/D,KAAK,EAAE,MAAM,CAAC,QAAQ;CACvB,CAAU,CAAC;AAMd,MAAM,cAAc,GAAG,GAAG,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { SonarV3Config } from \"../../common/Config.ts\";\nimport type { Problem } from \"../../common/Problem.ts\";\nimport { Err, Ok, Result } from \"ts-results-es\";\nimport { lastValueFrom, map, of, switchMap, type Observable } from \"rxjs\";\nimport type { JobResultsResponse } from \"./utils/JobResultsResponse.ts\";\nimport { mapRowsToColumns } from \"./utils/mapRowsToColumns.ts\";\nimport { mapRowData } from \"./utils/mapRowData.ts\";\n\nexport class Job {\n  readonly acceleration: JobProperties[\"acceleration\"];\n  readonly cancellationReason: JobProperties[\"cancellationReason\"];\n  readonly endedAt: JobProperties[\"endedAt\"];\n  readonly id: JobProperties[\"id\"];\n  readonly problem: JobProperties[\"problem\"];\n  readonly queryType: JobProperties[\"queryType\"];\n  readonly resourceSchedulingEndedAt: JobProperties[\"resourceSchedulingEndedAt\"];\n  readonly resourceSchedulingStartedAt: JobProperties[\"resourceSchedulingStartedAt\"];\n  readonly rowCount: JobProperties[\"rowCount\"];\n  readonly startedAt: JobProperties[\"startedAt\"];\n  readonly state: JobProperties[\"state\"];\n\n  public readonly observer: Observable<Job>;\n  #config: SonarV3Config;\n\n  constructor(\n    properties: JobProperties,\n    config: SonarV3Config,\n    observe: (id: string) => Observable<JobResult>,\n  ) {\n    this.acceleration = properties.acceleration;\n    this.cancellationReason = properties.cancellationReason;\n    this.endedAt = properties.endedAt;\n    this.id = properties.id;\n    this.problem = properties.problem;\n    this.queryType = properties.queryType;\n    this.resourceSchedulingEndedAt = properties.resourceSchedulingEndedAt;\n    this.resourceSchedulingStartedAt = properties.resourceSchedulingStartedAt;\n    this.rowCount = properties.rowCount;\n    this.startedAt = properties.startedAt;\n    this.state = properties.state;\n    this.#config = config;\n    this.observer = of(this).pipe(\n      switchMap((job) => {\n        if (!job.settled) {\n          return observe(this.id).pipe(\n            map((result) => {\n              if (result.isErr()) {\n                throw result.error;\n              }\n              return result.value;\n            }),\n          );\n        } else {\n          return of(job);\n        }\n      }),\n    );\n  }\n\n  get settled() {\n    return (\n      this.state === \"COMPLETED\" ||\n      this.state === \"FAILED\" ||\n      this.state === \"INVALID_STATE\" ||\n      this.state === \"CANCELED\"\n    );\n  }\n\n  get results() {\n    return {\n      jsonBatches: this.#jsonBatches.bind(this),\n      slice: this.#slice.bind(this),\n    };\n  }\n\n  async *#jsonBatches<T extends Record<string, unknown> = Record<string, unknown>>(\n    args: { limit?: number; offset?: number } = {},\n  ) {\n    if (typeof args.limit === \"number\" && args.limit < 0) {\n      throw new Error(\"Limit cannot be negative\");\n    }\n\n    if (typeof args.offset === \"number\" && args.offset < 0) {\n      throw new Error(\"Offset cannot be negative\");\n    }\n\n    // Wait for job to enter a settled state before attempting to fetch batches\n    if (!this.settled) {\n      await lastValueFrom(this.observer);\n    }\n\n    const limitArg = args.limit ?? Infinity;\n    const offsetArg = args.offset || 0;\n\n    // Tracks whether there are more rows available from the job results endpoint\n    let hasMore = true;\n\n    // Tracks the currently requested offset. If the offset arg is provided, start from there instead of 0.\n    let offset = offsetArg;\n\n    // Keeps track of the total number of rows that need to be loaded\n    const stopAfterOffset = limitArg + offsetArg;\n\n    while (hasMore && offset < stopAfterOffset) {\n      // Make batch_size dynamic to allow for requesting a smaller final page\n      const batch_size = Math.min(MAX_BATCH_SIZE, stopAfterOffset - offset);\n\n      const batch = await this.#config\n        .sonarV3Request(`job/${this.id}/results?offset=${offset}&limit=${batch_size}`)\n        .then((res) => res.json() as Promise<JobResultsResponse<T>>);\n\n      offset += batch.rows.length;\n      hasMore = batch.rows.length === batch_size;\n\n      if (batch.rows.length) {\n        const schema = { fields: batch.schema };\n        mapRowData(batch.rows, batch.schema);\n        yield {\n          get columns() {\n            return mapRowsToColumns(batch);\n          },\n          rows: batch.rows,\n          schema,\n          totalRows: batch.rowCount,\n        } as const;\n      }\n    }\n  }\n\n  async #slice<T extends Record<string, unknown> = Record<string, unknown>>(\n    start: number = 0,\n    end: number = Infinity,\n  ) {\n    const rows: T[] = [];\n    for await (const batch of this.#jsonBatches<T>({ limit: end - start, offset: start })) {\n      rows.push(...batch.rows);\n    }\n    return rows;\n  }\n\n  async cancel() {\n    return this.#config\n      .sonarV3Request(`job/${this.id}/cancel`, {\n        keepalive: true,\n        method: \"POST\",\n      })\n      .then(() => Ok(undefined))\n      .catch((e) => Err(e));\n  }\n}\n\nexport const jobEntityToProperties = (\n  id: string,\n  entity: {\n    jobState:\n      | \"NOT_SUBMITTED\"\n      | \"STARTING\"\n      | \"RUNNING\"\n      | \"COMPLETED\"\n      | \"CANCELED\"\n      | \"FAILED\"\n      | \"CANCELLATION_REQUESTED\"\n      | \"PLANNING\"\n      | \"PENDING\"\n      | \"METADATA_RETRIEVAL\"\n      | \"QUEUED\"\n      | \"ENGINE_START\"\n      | \"EXECUTION_PLANNING\"\n      | \"INVALID_STATE\";\n    rowCount: number | null;\n    errorMessage: string;\n    startedAt: string | null;\n    endedAt: string | null;\n    acceleration: {\n      reflectionRelationships: {\n        datasetId: string;\n        reflectionId: string;\n        relationship: string;\n      }[];\n    };\n    queryType:\n      | \"UI_RUN\"\n      | \"UI_PREVIEW\"\n      | \"UI_INTERNAL_PREVIEW\"\n      | \"UI_INTERNAL_RUN\"\n      | \"UI_EXPORT\"\n      | \"ODBC\"\n      | \"JDBC\"\n      | \"REST\"\n      | \"ACCELERATOR_CREATE\"\n      | \"ACCELERATOR_DROP\"\n      | \"UNKNOWN\"\n      | \"PREPARE_INTERNAL\"\n      | \"ACCELERATOR_EXPLAIN\"\n      | \"UI_INITIAL_PREVIEW\";\n    resourceSchedulingStartedAt: string | null;\n    resourceSchedulingEndedAt: string | null;\n    cancellationReason: string | null;\n  },\n) =>\n  ({\n    acceleration: entity.acceleration,\n    cancellationReason: entity.cancellationReason,\n    endedAt: entity.endedAt ? new Date(entity.endedAt) : null,\n    id,\n    problem: null as Problem | null,\n    queryType: entity.queryType,\n    resourceSchedulingEndedAt: entity.resourceSchedulingEndedAt\n      ? new Date(entity.resourceSchedulingEndedAt)\n      : null,\n    resourceSchedulingStartedAt: entity.resourceSchedulingStartedAt\n      ? new Date(entity.resourceSchedulingStartedAt)\n      : null,\n    rowCount: entity.rowCount,\n    startedAt: entity.startedAt ? new Date(entity.startedAt) : null,\n    state: entity.jobState,\n  }) as const;\n\ntype JobProperties = ReturnType<typeof jobEntityToProperties>;\n\nexport type JobResult = Result<Job, unknown>;\n\nconst MAX_BATCH_SIZE = 500;\n"]}
+{"version":3,"file":"Job.js","sourceRoot":"","sources":["../../../src/oss/jobs/Job.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAChD,OAAO,EAAE,aAAa,EAAE,GAAG,EAAE,EAAE,EAAE,SAAS,EAAmB,MAAM,MAAM,CAAC;AAE1E,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnD,MAAM,OAAO,GAAG;IACL,YAAY,CAAgC;IAC5C,kBAAkB,CAAsC;IACxD,OAAO,CAA2B;IAClC,EAAE,CAAsB;IACxB,OAAO,CAA2B;IAClC,SAAS,CAA6B;IACtC,yBAAyB,CAA6C;IACtE,2BAA2B,CAA+C;IAC1E,QAAQ,CAA4B;IACpC,SAAS,CAA6B;IACtC,KAAK,CAAyB;IAEvB,QAAQ,CAAkB;IAC1C,OAAO,CAAgB;IAEvB,YACE,UAAyB,EACzB,MAAqB,EACrB,OAA8C;QAE9C,IAAI,CAAC,YAAY,GAAG,UAAU,CAAC,YAAY,CAAC;QAC5C,IAAI,CAAC,kBAAkB,GAAG,UAAU,CAAC,kBAAkB,CAAC;QACxD,IAAI,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;QAClC,IAAI,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC;QACxB,IAAI,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;QAClC,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,yBAAyB,GAAG,UAAU,CAAC,yBAAyB,CAAC;QACtE,IAAI,CAAC,2BAA2B,GAAG,UAAU,CAAC,2BAA2B,CAAC;QAC1E,IAAI,CAAC,QAAQ,GAAG,UAAU,CAAC,QAAQ,CAAC;QACpC,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACtC,IAAI,CAAC,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC;QAC9B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAC3B,SAAS,CAAC,CAAC,GAAG,EAAE,EAAE;YAChB,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;gBACjB,OAAO,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAC1B,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;oBACb,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;wBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;oBACrB,CAAC;oBACD,OAAO,MAAM,CAAC,KAAK,CAAC;gBACtB,CAAC,CAAC,CACH,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,OAAO,EAAE,CAAC,GAAG,CAAC,CAAC;YACjB,CAAC;QACH,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAED,IAAI,OAAO;QACT,OAAO,CACL,IAAI,CAAC,KAAK,KAAK,WAAW;YAC1B,IAAI,CAAC,KAAK,KAAK,QAAQ;YACvB,IAAI,CAAC,KAAK,KAAK,eAAe;YAC9B,IAAI,CAAC,KAAK,KAAK,UAAU,CAC1B,CAAC;IACJ,CAAC;IAED,IAAI,OAAO;QACT,OAAO;YACL,WAAW,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;YACzC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;SAC9B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,YAAY,CACjB,OAA4C,EAAE;QAE9C,IAAI,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,IAAI,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC;YACrD,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvD,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC/C,CAAC;QAED,2EAA2E;QAC3E,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,MAAM,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrC,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,IAAI,QAAQ,CAAC;QACxC,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,IAAI,CAAC,CAAC;QAEnC,6EAA6E;QAC7E,IAAI,OAAO,GAAG,IAAI,CAAC;QAEnB,uGAAuG;QACvG,IAAI,MAAM,GAAG,SAAS,CAAC;QAEvB,iEAAiE;QACjE,MAAM,eAAe,GAAG,QAAQ,GAAG,SAAS,CAAC;QAE7C,OAAO,OAAO,IAAI,MAAM,GAAG,eAAe,EAAE,CAAC;YAC3C,uEAAuE;YACvE,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,eAAe,GAAG,MAAM,CAAC,CAAC;YAEtE,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,OAAO;iBAC7B,cAAc,CAAC,OAAO,IAAI,CAAC,EAAE,mBAAmB,MAAM,UAAU,UAAU,EAAE,CAAC;iBAC7E,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAoC,CAAC,CAAC;YAE/D,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC;YAC5B,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,KAAK,UAAU,CAAC;YAE3C,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACtB,MAAM,MAAM,GAAG,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;gBACxC,UAAU,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;gBACrC,MAAM;oBACJ,IAAI,OAAO;wBACT,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC;oBACjC,CAAC;oBACD,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,MAAM;oBACN,SAAS,EAAE,KAAK,CAAC,QAAQ;iBACjB,CAAC;YACb,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CACV,QAAgB,CAAC,EACjB,MAAc,QAAQ;QAEtB,IAAI,CAAC;YACH,MAAM,IAAI,GAAQ,EAAE,CAAC;YACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,YAAY,CAAI,EAAE,KAAK,EAAE,GAAG,GAAG,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;gBACtF,IAAI,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;YAC3B,CAAC;YACD,OAAO,EAAE,CAAC,IAAI,CAAC,CAAC;QAClB,CAAC;QAAC,OAAO,CAAU,EAAE,CAAC;YACpB,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC;QAChB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM;QACV,OAAO,IAAI,CAAC,OAAO;aAChB,cAAc,CAAC,OAAO,IAAI,CAAC,EAAE,SAAS,EAAE;YACvC,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,MAAM;SACf,CAAC;aACD,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC;aACzB,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1B,CAAC;CACF;AAED,MAAM,CAAC,MAAM,qBAAqB,GAAG,CACnC,EAAU,EACV,MA6CC,EACD,EAAE,CACF,CAAC;IACC,YAAY,EAAE,MAAM,CAAC,YAAY;IACjC,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;IAC7C,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI;IACzD,EAAE;IACF,OAAO,EAAE,IAAsB;IAC/B,SAAS,EAAE,MAAM,CAAC,SAAS;IAC3B,yBAAyB,EAAE,MAAM,CAAC,yBAAyB;QACzD,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,yBAAyB,CAAC;QAC5C,CAAC,CAAC,IAAI;IACR,2BAA2B,EAAE,MAAM,CAAC,2BAA2B;QAC7D,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,2BAA2B,CAAC;QAC9C,CAAC,CAAC,IAAI;IACR,QAAQ,EAAE,MAAM,CAAC,QAAQ;IACzB,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI;IAC/D,KAAK,EAAE,MAAM,CAAC,QAAQ;CACvB,CAAU,CAAC;AAMd,MAAM,cAAc,GAAG,GAAG,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { SonarV3Config } from \"../../common/Config.ts\";\nimport type { Problem } from \"../../common/Problem.ts\";\nimport { Err, Ok, Result } from \"ts-results-es\";\nimport { lastValueFrom, map, of, switchMap, type Observable } from \"rxjs\";\nimport type { JobResultsResponse } from \"./utils/JobResultsResponse.ts\";\nimport { mapRowsToColumns } from \"./utils/mapRowsToColumns.ts\";\nimport { mapRowData } from \"./utils/mapRowData.ts\";\n\nexport class Job {\n  readonly acceleration: JobProperties[\"acceleration\"];\n  readonly cancellationReason: JobProperties[\"cancellationReason\"];\n  readonly endedAt: JobProperties[\"endedAt\"];\n  readonly id: JobProperties[\"id\"];\n  readonly problem: JobProperties[\"problem\"];\n  readonly queryType: JobProperties[\"queryType\"];\n  readonly resourceSchedulingEndedAt: JobProperties[\"resourceSchedulingEndedAt\"];\n  readonly resourceSchedulingStartedAt: JobProperties[\"resourceSchedulingStartedAt\"];\n  readonly rowCount: JobProperties[\"rowCount\"];\n  readonly startedAt: JobProperties[\"startedAt\"];\n  readonly state: JobProperties[\"state\"];\n\n  public readonly observer: Observable<Job>;\n  #config: SonarV3Config;\n\n  constructor(\n    properties: JobProperties,\n    config: SonarV3Config,\n    observe: (id: string) => Observable<JobResult>,\n  ) {\n    this.acceleration = properties.acceleration;\n    this.cancellationReason = properties.cancellationReason;\n    this.endedAt = properties.endedAt;\n    this.id = properties.id;\n    this.problem = properties.problem;\n    this.queryType = properties.queryType;\n    this.resourceSchedulingEndedAt = properties.resourceSchedulingEndedAt;\n    this.resourceSchedulingStartedAt = properties.resourceSchedulingStartedAt;\n    this.rowCount = properties.rowCount;\n    this.startedAt = properties.startedAt;\n    this.state = properties.state;\n    this.#config = config;\n    this.observer = of(this).pipe(\n      switchMap((job) => {\n        if (!job.settled) {\n          return observe(this.id).pipe(\n            map((result) => {\n              if (result.isErr()) {\n                throw result.error;\n              }\n              return result.value;\n            }),\n          );\n        } else {\n          return of(job);\n        }\n      }),\n    );\n  }\n\n  get settled() {\n    return (\n      this.state === \"COMPLETED\" ||\n      this.state === \"FAILED\" ||\n      this.state === \"INVALID_STATE\" ||\n      this.state === \"CANCELED\"\n    );\n  }\n\n  get results() {\n    return {\n      jsonBatches: this.#jsonBatches.bind(this),\n      slice: this.#slice.bind(this),\n    };\n  }\n\n  async *#jsonBatches<T extends Record<string, unknown> = Record<string, unknown>>(\n    args: { limit?: number; offset?: number } = {},\n  ) {\n    if (typeof args.limit === \"number\" && args.limit < 0) {\n      throw new Error(\"Limit cannot be negative\");\n    }\n\n    if (typeof args.offset === \"number\" && args.offset < 0) {\n      throw new Error(\"Offset cannot be negative\");\n    }\n\n    // Wait for job to enter a settled state before attempting to fetch batches\n    if (!this.settled) {\n      await lastValueFrom(this.observer);\n    }\n\n    const limitArg = args.limit ?? Infinity;\n    const offsetArg = args.offset || 0;\n\n    // Tracks whether there are more rows available from the job results endpoint\n    let hasMore = true;\n\n    // Tracks the currently requested offset. If the offset arg is provided, start from there instead of 0.\n    let offset = offsetArg;\n\n    // Keeps track of the total number of rows that need to be loaded\n    const stopAfterOffset = limitArg + offsetArg;\n\n    while (hasMore && offset < stopAfterOffset) {\n      // Make batch_size dynamic to allow for requesting a smaller final page\n      const batch_size = Math.min(MAX_BATCH_SIZE, stopAfterOffset - offset);\n\n      const batch = await this.#config\n        .sonarV3Request(`job/${this.id}/results?offset=${offset}&limit=${batch_size}`)\n        .then((res) => res.json() as Promise<JobResultsResponse<T>>);\n\n      offset += batch.rows.length;\n      hasMore = batch.rows.length === batch_size;\n\n      if (batch.rows.length) {\n        const schema = { fields: batch.schema };\n        mapRowData(batch.rows, batch.schema);\n        yield {\n          get columns() {\n            return mapRowsToColumns(batch);\n          },\n          rows: batch.rows,\n          schema,\n          totalRows: batch.rowCount,\n        } as const;\n      }\n    }\n  }\n\n  async #slice<T extends Record<string, unknown> = Record<string, unknown>>(\n    start: number = 0,\n    end: number = Infinity,\n  ) {\n    try {\n      const rows: T[] = [];\n      for await (const batch of this.#jsonBatches<T>({ limit: end - start, offset: start })) {\n        rows.push(...batch.rows);\n      }\n      return Ok(rows);\n    } catch (e: unknown) {\n      return Err(e);\n    }\n  }\n\n  async cancel() {\n    return this.#config\n      .sonarV3Request(`job/${this.id}/cancel`, {\n        keepalive: true,\n        method: \"POST\",\n      })\n      .then(() => Ok(undefined))\n      .catch((e) => Err(e));\n  }\n}\n\nexport const jobEntityToProperties = (\n  id: string,\n  entity: {\n    jobState:\n      | \"NOT_SUBMITTED\"\n      | \"STARTING\"\n      | \"RUNNING\"\n      | \"COMPLETED\"\n      | \"CANCELED\"\n      | \"FAILED\"\n      | \"CANCELLATION_REQUESTED\"\n      | \"PLANNING\"\n      | \"PENDING\"\n      | \"METADATA_RETRIEVAL\"\n      | \"QUEUED\"\n      | \"ENGINE_START\"\n      | \"EXECUTION_PLANNING\"\n      | \"INVALID_STATE\";\n    rowCount: number | null;\n    errorMessage: string;\n    startedAt: string | null;\n    endedAt: string | null;\n    acceleration: {\n      reflectionRelationships: {\n        datasetId: string;\n        reflectionId: string;\n        relationship: string;\n      }[];\n    };\n    queryType:\n      | \"UI_RUN\"\n      | \"UI_PREVIEW\"\n      | \"UI_INTERNAL_PREVIEW\"\n      | \"UI_INTERNAL_RUN\"\n      | \"UI_EXPORT\"\n      | \"ODBC\"\n      | \"JDBC\"\n      | \"REST\"\n      | \"ACCELERATOR_CREATE\"\n      | \"ACCELERATOR_DROP\"\n      | \"UNKNOWN\"\n      | \"PREPARE_INTERNAL\"\n      | \"ACCELERATOR_EXPLAIN\"\n      | \"UI_INITIAL_PREVIEW\";\n    resourceSchedulingStartedAt: string | null;\n    resourceSchedulingEndedAt: string | null;\n    cancellationReason: string | null;\n  },\n) =>\n  ({\n    acceleration: entity.acceleration,\n    cancellationReason: entity.cancellationReason,\n    endedAt: entity.endedAt ? new Date(entity.endedAt) : null,\n    id,\n    problem: null as Problem | null,\n    queryType: entity.queryType,\n    resourceSchedulingEndedAt: entity.resourceSchedulingEndedAt\n      ? new Date(entity.resourceSchedulingEndedAt)\n      : null,\n    resourceSchedulingStartedAt: entity.resourceSchedulingStartedAt\n      ? new Date(entity.resourceSchedulingStartedAt)\n      : null,\n    rowCount: entity.rowCount,\n    startedAt: entity.startedAt ? new Date(entity.startedAt) : null,\n    state: entity.jobState,\n  }) as const;\n\ntype JobProperties = ReturnType<typeof jobEntityToProperties>;\n\nexport type JobResult = Result<Job, unknown>;\n\nconst MAX_BATCH_SIZE = 500;\n"]}

package/dist/oss/jobs/JobsResource.d.ts

@@ -1,10 +1,14 @@
 import type { SonarV3Config } from "../../common/Config.ts";
-import { type JobResult } from "./Job.ts";
+import { Job, type JobResult } from "./Job.ts";
 import { Query } from "../../common/Query.ts";
 import { Err, Ok } from "ts-results-es";
 import type { SignalParam } from "../../common/Params.ts";
 export declare const JobsResource: (config: SonarV3Config) => {
     create: (query: Query) => Promise<Err<unknown> | Ok<string>>;
+    /**
+     * A convenience method to combine job creation and retrieval
+     */
+    createAndRetrieve: (query: Query) => Promise<import("ts-results-es").Result<Job, unknown>>;
     observe: (id: string) => import("rxjs").Observable<JobResult>;
     retrieve: (id: string, { signal }?: SignalParam) => Promise<JobResult>;
 };

package/dist/oss/jobs/JobsResource.js

@@ -15,7 +15,7 @@
  */
 import { Job, jobEntityToProperties } from "./Job.js";
 import { Query } from "../../common/Query.js";
-import { Err, Ok } from "ts-results-es";
+import { Err, Ok, AsyncResult } from "ts-results-es";
 import { concatMap, shareReplay, takeWhile, timer } from "rxjs";
 import { fromAbortable } from "../../common/fromAbortable.js";
 export const JobsResource = (config) => {
@@ -43,30 +43,35 @@ export const JobsResource = (config) => {
         bufferSize: 1,
         refCount: true,
     }));
+    const create = (query) => {
+        return config
+            .sonarV3Request(`sql`, {
+            body: JSON.stringify({
+                context: query.context,
+                sql: query.sql,
+            }),
+            headers: {
+                Accept: "application/json",
+                "Content-Type": "application/json",
+            },
+            keepalive: true,
+            method: "POST",
+        })
+            .then((res) => res.json())
+            .then((response) => Ok(response.id))
+            .catch((e) => Err(e));
+    };
     const retrieve = (id, { signal } = {}) => config
         .sonarV3Request(`job/${id}`, { signal })
         .then((res) => res.json())
         .then((properties) => Ok(new Job(jobEntityToProperties(id, properties), config, observe)))
         .catch((e) => Err(e));
     return {
-        create: (query) => {
-            return config
-                .sonarV3Request(`sql`, {
-                body: JSON.stringify({
-                    context: query.context,
-                    sql: query.sql,
-                }),
-                headers: {
-                    Accept: "application/json",
-                    "Content-Type": "application/json",
-                },
-                keepalive: true,
-                method: "POST",
-            })
-                .then((res) => res.json())
-                .then((response) => Ok(response.id))
-                .catch((e) => Err(e));
-        },
+        create,
+        /**
+         * A convenience method to combine job creation and retrieval
+         */
+        createAndRetrieve: (query) => new AsyncResult(create(query)).andThen((jobId) => retrieve(jobId)).promise,
         observe,
         retrieve,
     };

package/dist/oss/jobs/JobsResource.js.map

@@ -1 +1 @@
-{"version":3,"file":"JobsResource.js","sourceRoot":"","sources":["../../../src/oss/jobs/JobsResource.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EAAE,GAAG,EAAE,qBAAqB,EAAkB,MAAM,UAAU,CAAC;AACtE,OAAO,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAC9C,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,eAAe,CAAC;AAExC,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,MAAM,CAAC;AAChE,OAAO,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AAE9D,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,MAAqB,EAAE,EAAE;IACpD,MAAM,OAAO,GAAG,CAAC,EAAU,EAAE,EAAE;IAC7B;;;OAGG;IACH,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI;IACjB;;;;OAIG;IACH,SAAS,CAAC,GAAG,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;IACxE;;OAEG;IACH,SAAS,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,EAAE,IAAI,CAAC;IAC5E;;;OAGG;IACH,WAAW,CAAC;QACV,UAAU,EAAE,CAAC;QACb,QAAQ,EAAE,IAAI;KACf,CAAC,CACH,CAAC;IACJ,MAAM,QAAQ,GAAG,CAAC,EAAU,EAAE,EAAE,MAAM,KAAkB,EAAE,EAAsB,EAAE,CAChF,MAAM;SACH,cAAc,CAAC,OAAO,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC;SACvC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;SACzB,IAAI,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,GAAG,CAAC,qBAAqB,CAAC,EAAE,EAAE,UAAU,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;SACzF,KAAK,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACnC,OAAO;QACL,MAAM,EAAE,CAAC,KAAY,EAAE,EAAE;YACvB,OAAO,MAAM;iBACV,cAAc,CAAC,KAAK,EAAE;gBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,OAAO,EAAE,KAAK,CAAC,OAAO;oBACtB,GAAG,EAAE,KAAK,CAAC,GAAG;iBACf,CAAC;gBACF,OAAO,EAAE;oBACP,MAAM,EAAE,kBAAkB;oBAC1B,cAAc,EAAE,kBAAkB;iBACnC;gBACD,SAAS,EAAE,IAAI;gBACf,MAAM,EAAE,MAAM;aACf,CAAC;iBACD,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAA6B,CAAC;iBACpD,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;iBACnC,KAAK,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACnC,CAAC;QACD,OAAO;QACP,QAAQ;KACT,CAAC;AACJ,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { SonarV3Config } from \"../../common/Config.ts\";\nimport { Job, jobEntityToProperties, type JobResult } from \"./Job.ts\";\nimport { Query } from \"../../common/Query.ts\";\nimport { Err, Ok } from \"ts-results-es\";\nimport type { SignalParam } from \"../../common/Params.ts\";\nimport { concatMap, shareReplay, takeWhile, timer } from \"rxjs\";\nimport { fromAbortable } from \"../../common/fromAbortable.ts\";\n\nexport const JobsResource = (config: SonarV3Config) => {\n  const observe = (id: string) =>\n    /**\n     * Check every 1000ms for job status changes. 1000ms may seem too frequent,\n     * but it helps keep the client feeling snappy, especially for jobs that complete quickly.\n     */\n    timer(0, 1000).pipe(\n      /**\n       * Use `concatMap` instead of `switchMap` to avoid a situation where the call takes longer than\n       * 1000ms (possibly due to network conditions or API slowdowns) which would cause switchMap to\n       * constantly cancel unfinished requests (in which case, no requests would ever complete)\n       */\n      concatMap(() => fromAbortable(({ signal }) => retrieve(id, { signal }))),\n      /**\n       * Continue polling until the job has reached a settled state\n       */\n      takeWhile((jobResult) => jobResult.isOk() && !jobResult.value.settled, true),\n      /**\n       * Reuse the same polling for multiple subscribers, buffering only the latest result.\n       * Make sure to unsubscribe from the source to stop polling when there are no subscribers (refCount: true)\n       */\n      shareReplay({\n        bufferSize: 1,\n        refCount: true,\n      }),\n    );\n  const retrieve = (id: string, { signal }: SignalParam = {}): Promise<JobResult> =>\n    config\n      .sonarV3Request(`job/${id}`, { signal })\n      .then((res) => res.json())\n      .then((properties) => Ok(new Job(jobEntityToProperties(id, properties), config, observe)))\n      .catch((e: unknown) => Err(e));\n  return {\n    create: (query: Query) => {\n      return config\n        .sonarV3Request(`sql`, {\n          body: JSON.stringify({\n            context: query.context,\n            sql: query.sql,\n          }),\n          headers: {\n            Accept: \"application/json\",\n            \"Content-Type\": \"application/json\",\n          },\n          keepalive: true,\n          method: \"POST\",\n        })\n        .then((res) => res.json() as Promise<{ id: string }>)\n        .then((response) => Ok(response.id))\n        .catch((e: unknown) => Err(e));\n    },\n    observe,\n    retrieve,\n  };\n};\n"]}
+{"version":3,"file":"JobsResource.js","sourceRoot":"","sources":["../../../src/oss/jobs/JobsResource.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EAAE,GAAG,EAAE,qBAAqB,EAAkB,MAAM,UAAU,CAAC;AACtE,OAAO,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAC9C,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAErD,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,MAAM,CAAC;AAChE,OAAO,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AAE9D,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,MAAqB,EAAE,EAAE;IACpD,MAAM,OAAO,GAAG,CAAC,EAAU,EAAE,EAAE;IAC7B;;;OAGG;IACH,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI;IACjB;;;;OAIG;IACH,SAAS,CAAC,GAAG,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;IACxE;;OAEG;IACH,SAAS,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,EAAE,IAAI,CAAC;IAC5E;;;OAGG;IACH,WAAW,CAAC;QACV,UAAU,EAAE,CAAC;QACb,QAAQ,EAAE,IAAI;KACf,CAAC,CACH,CAAC;IACJ,MAAM,MAAM,GAAG,CAAC,KAAY,EAAE,EAAE;QAC9B,OAAO,MAAM;aACV,cAAc,CAAC,KAAK,EAAE;YACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,GAAG,EAAE,KAAK,CAAC,GAAG;aACf,CAAC;YACF,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;gBAC1B,cAAc,EAAE,kBAAkB;aACnC;YACD,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,MAAM;SACf,CAAC;aACD,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAA6B,CAAC;aACpD,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;aACnC,KAAK,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACnC,CAAC,CAAC;IACF,MAAM,QAAQ,GAAG,CAAC,EAAU,EAAE,EAAE,MAAM,KAAkB,EAAE,EAAsB,EAAE,CAChF,MAAM;SACH,cAAc,CAAC,OAAO,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC;SACvC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;SACzB,IAAI,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,GAAG,CAAC,qBAAqB,CAAC,EAAE,EAAE,UAAU,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;SACzF,KAAK,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACnC,OAAO;QACL,MAAM;QACN;;WAEG;QACH,iBAAiB,EAAE,CAAC,KAAY,EAAE,EAAE,CAClC,IAAI,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO;QAC5E,OAAO;QACP,QAAQ;KACT,CAAC;AACJ,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { SonarV3Config } from \"../../common/Config.ts\";\nimport { Job, jobEntityToProperties, type JobResult } from \"./Job.ts\";\nimport { Query } from \"../../common/Query.ts\";\nimport { Err, Ok, AsyncResult } from \"ts-results-es\";\nimport type { SignalParam } from \"../../common/Params.ts\";\nimport { concatMap, shareReplay, takeWhile, timer } from \"rxjs\";\nimport { fromAbortable } from \"../../common/fromAbortable.ts\";\n\nexport const JobsResource = (config: SonarV3Config) => {\n  const observe = (id: string) =>\n    /**\n     * Check every 1000ms for job status changes. 1000ms may seem too frequent,\n     * but it helps keep the client feeling snappy, especially for jobs that complete quickly.\n     */\n    timer(0, 1000).pipe(\n      /**\n       * Use `concatMap` instead of `switchMap` to avoid a situation where the call takes longer than\n       * 1000ms (possibly due to network conditions or API slowdowns) which would cause switchMap to\n       * constantly cancel unfinished requests (in which case, no requests would ever complete)\n       */\n      concatMap(() => fromAbortable(({ signal }) => retrieve(id, { signal }))),\n      /**\n       * Continue polling until the job has reached a settled state\n       */\n      takeWhile((jobResult) => jobResult.isOk() && !jobResult.value.settled, true),\n      /**\n       * Reuse the same polling for multiple subscribers, buffering only the latest result.\n       * Make sure to unsubscribe from the source to stop polling when there are no subscribers (refCount: true)\n       */\n      shareReplay({\n        bufferSize: 1,\n        refCount: true,\n      }),\n    );\n  const create = (query: Query) => {\n    return config\n      .sonarV3Request(`sql`, {\n        body: JSON.stringify({\n          context: query.context,\n          sql: query.sql,\n        }),\n        headers: {\n          Accept: \"application/json\",\n          \"Content-Type\": \"application/json\",\n        },\n        keepalive: true,\n        method: \"POST\",\n      })\n      .then((res) => res.json() as Promise<{ id: string }>)\n      .then((response) => Ok(response.id))\n      .catch((e: unknown) => Err(e));\n  };\n  const retrieve = (id: string, { signal }: SignalParam = {}): Promise<JobResult> =>\n    config\n      .sonarV3Request(`job/${id}`, { signal })\n      .then((res) => res.json())\n      .then((properties) => Ok(new Job(jobEntityToProperties(id, properties), config, observe)))\n      .catch((e: unknown) => Err(e));\n  return {\n    create,\n    /**\n     * A convenience method to combine job creation and retrieval\n     */\n    createAndRetrieve: (query: Query) =>\n      new AsyncResult(create(query)).andThen((jobId) => retrieve(jobId)).promise,\n    observe,\n    retrieve,\n  };\n};\n"]}

package/dist/oss/resources.d.ts

@@ -14,6 +14,7 @@ export declare const Resources: (config: ResourceConfig & SonarV2Config & SonarV
     };
     jobs: {
         create: (query: import("./index.ts").Query) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<string>>;
+        createAndRetrieve: (query: import("./index.ts").Query) => Promise<import("ts-results-es").Result<import("./interfaces.ts").Job, unknown>>;
         observe: (id: string) => import("rxjs").Observable<import("./jobs/Job.ts").JobResult>;
         retrieve: (id: string, { signal }?: import("../common/Params.ts").SignalParam) => Promise<import("./jobs/Job.ts").JobResult>;
     };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dremio/js-sdk",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "description": "JavaScript library for the Dremio API",
   "keywords": [
     "dremio",