@deephaven-enterprise/query-utils 1.20250801.264-beta → 1.20250801.266-beta

package/dist/DraftQuery.d.ts

@@ -1,4 +1,4 @@
-import type { EditableQueryInfo, TypeSpecificFields } from '@deephaven-enterprise/jsapi-types';
+import type { EditableQueryInfo, PlainEditableQueryInfo, TypeSpecificFields } from '@deephaven-enterprise/jsapi-types';
 import QueryScheduler from './QueryScheduler';
 export type ScriptDraftQuery = IDraftQuery & {
     scriptLanguage: string;
@@ -10,7 +10,7 @@ export type ScriptDraftQuery = IDraftQuery & {
  * it causes issues upstream where `Record<string, unknown>` generics are expected.
  * e.g. `WorkspaceStorageItem<IDraftQuery>`
  */
-export type IDraftQuery = EditableQueryInfo & {
+export type IDraftQuery = PlainEditableQueryInfo & {
     isClientSide: boolean;
     draftOwner: string;
     initializationThreads?: string;
@@ -96,23 +96,23 @@ export declare class DraftQuery implements IDraftQuery {
     dbServerName: string;
     scriptLanguage: string | null;
     scriptPath: string | null;
-    scriptCode: string;
+    scriptCode: string | null;
     scheduling: readonly string[];
     adminGroups: readonly string[];
     viewerGroups: readonly string[];
     restartUsers: number;
     timeout: number;
     scheduler: QueryScheduler;
-    kubernetesControl?: string | null;
-    pythonControl?: string | null;
-    genericWorkerControl?: string | null;
+    kubernetesControl: string | null;
+    pythonControl: string | null;
+    genericWorkerControl: string | null;
     /**
      * Should be the name (not title) of the workerKind
      * The server gives null if it should be the default
      * Not sure if the value will be set if the PQ is from pre-vermilion and the customer upgrades
      * Treat both null and undefined as server default (not necessarily 1st option in the UI)
      */
-    workerKind: string | null | undefined;
+    workerKind: string | null;
     typeSpecificFields: TypeSpecificFields | null;
     replicaCount: number;
     spareCount: number;
@@ -139,7 +139,7 @@ export declare class DraftQuery implements IDraftQuery {
      *
      * @param editableQuery the Editable Query from the Server
      */
-    saveDraftToQuery(editableQuery: Partial<EditableQueryInfo>): void;
+    saveDraftToQuery<T extends EditableQueryInfo>(editableQuery: Partial<T> & Pick<T, 'serial'>): T;
     /**
      * Updates the JVM args to include the thread count for initialization and update threads.
      * It is safe to call from saveDraftToQuery because the the thead counts will be set to
@@ -148,9 +148,10 @@ export declare class DraftQuery implements IDraftQuery {
      * manually in that case.
      */
     addThreadCountToJVMArgs(): void;
-    toEditableQueryInfo(): {
-        serial: string | null;
-    };
+    /**
+     * Get a PlainEditableQueryInfo object populated from this DraftQuery
+     */
+    toEditableQueryInfo(): PlainEditableQueryInfo;
     toRowMap(queryTypes: Map<string, string>): Map<string, string>;
     formatIdForSerialColumn(): string;
 }

package/dist/DraftQuery.js

@@ -26,7 +26,7 @@ export class DraftQuery {
         // Remove the 'Draft ID: ' marking
         return serial === null || serial === void 0 ? void 0 : serial.substring(DRAFT_SERIAL_PREFIX.length);
     }
-    constructor({ isClientSide = false, draftOwner, serial = DraftQuery.NO_SERIAL, name, owner, type = QueryType.SCRIPT, enabled = true, enableGcLogs = true, envVars = '', heapSize = 4.0, additionalMemory = 0.0, dataMemoryRatio = 0.25, jvmArgs = '', extraClasspaths = '', jvmProfile = 'Default', dbServerName, scriptLanguage = 'Groovy', scriptPath = null, scriptCode = '', scheduling, adminGroups = [], viewerGroups = [], restartUsers = DraftQuery.RESTART_MODE.ADMIN, timeout = 0, workerKind, kubernetesControl, pythonControl, genericWorkerControl, typeSpecificFields = null, replicaCount = 1, spareCount = 0, assignmentPolicy = null, assignmentPolicyParams = null, initializationThreads = undefined, updateThreads = undefined, }) {
+    constructor({ isClientSide = false, draftOwner, serial = DraftQuery.NO_SERIAL, name, owner, type = QueryType.SCRIPT, enabled = true, enableGcLogs = true, envVars = '', heapSize = 4.0, additionalMemory = 0.0, dataMemoryRatio = 0.25, jvmArgs = '', extraClasspaths = '', jvmProfile = 'Default', dbServerName, scriptLanguage = 'Groovy', scriptPath = null, scriptCode = '', scheduling, adminGroups = [], viewerGroups = [], restartUsers = DraftQuery.RESTART_MODE.ADMIN, timeout = 0, workerKind = null, kubernetesControl = null, pythonControl = null, genericWorkerControl = null, typeSpecificFields = null, replicaCount = 1, spareCount = 0, assignmentPolicy = null, assignmentPolicyParams = null, initializationThreads = undefined, updateThreads = undefined, }) {
         this.scheduling = [];
         // Meta Data
         this.id = shortid.generate();
@@ -105,38 +105,43 @@ export class DraftQuery {
      * @param editableQuery the Editable Query from the Server
      */
     saveDraftToQuery(editableQuery) {
-        var _a, _b;
         this.addThreadCountToJVMArgs();
-        editableQuery.name = this.name;
-        editableQuery.owner = this.owner;
-        editableQuery.type = this.type;
-        editableQuery.enabled = this.enabled;
-        editableQuery.enableGcLogs = this.enableGcLogs;
-        editableQuery.envVars = this.envVars;
-        editableQuery.heapSize = this.heapSize;
-        editableQuery.dataMemoryRatio = this.dataMemoryRatio;
-        editableQuery.additionalMemory = this.additionalMemory;
-        editableQuery.jvmArgs = this.jvmArgs;
-        editableQuery.extraClasspaths = this.extraClasspaths;
-        editableQuery.jvmProfile = this.jvmProfile;
-        editableQuery.dbServerName = this.dbServerName;
-        editableQuery.scriptLanguage = this.scriptLanguage;
-        editableQuery.scriptPath = (_a = this.scriptPath) !== null && _a !== void 0 ? _a : undefined;
-        editableQuery.scriptCode = this.scriptCode;
-        editableQuery.scheduling = this.scheduler.toStringArray();
-        editableQuery.adminGroups = this.adminGroups;
-        editableQuery.viewerGroups = this.viewerGroups;
-        editableQuery.restartUsers = this.restartUsers;
-        editableQuery.timeout = this.timeout;
-        editableQuery.workerKind = (_b = this.workerKind) !== null && _b !== void 0 ? _b : undefined;
-        editableQuery.kubernetesControl = this.kubernetesControl;
-        editableQuery.pythonControl = this.pythonControl;
-        editableQuery.genericWorkerControl = this.genericWorkerControl;
-        editableQuery.typeSpecificFields = this.typeSpecificFields;
-        editableQuery.replicaCount = this.replicaCount;
-        editableQuery.spareCount = this.spareCount;
-        editableQuery.assignmentPolicy = this.assignmentPolicy;
-        editableQuery.assignmentPolicyParams = this.assignmentPolicyParams;
+        // Type check to make sure none of the required fields are missing
+        const completeQuery = {
+            serial: editableQuery.serial,
+            name: this.name,
+            owner: this.owner,
+            type: this.type,
+            enabled: this.enabled,
+            enableGcLogs: this.enableGcLogs,
+            envVars: this.envVars,
+            heapSize: this.heapSize,
+            dataMemoryRatio: this.dataMemoryRatio,
+            additionalMemory: this.additionalMemory,
+            jvmArgs: this.jvmArgs,
+            extraClasspaths: this.extraClasspaths,
+            jvmProfile: this.jvmProfile,
+            dbServerName: this.dbServerName,
+            scriptLanguage: this.scriptLanguage,
+            scriptPath: this.scriptPath,
+            scriptCode: this.scriptCode,
+            scheduling: this.scheduler.toStringArray(),
+            adminGroups: [...this.adminGroups],
+            viewerGroups: this.viewerGroups,
+            restartUsers: this.restartUsers,
+            timeout: this.timeout,
+            workerKind: this.workerKind,
+            kubernetesControl: this.kubernetesControl,
+            pythonControl: this.pythonControl,
+            genericWorkerControl: this.genericWorkerControl,
+            typeSpecificFields: this.typeSpecificFields,
+            replicaCount: this.replicaCount,
+            spareCount: this.spareCount,
+            assignmentPolicy: this.assignmentPolicy,
+            assignmentPolicyParams: this.assignmentPolicyParams,
+        };
+        Object.assign(editableQuery, completeQuery);
+        return editableQuery;
     }
     /**
      * Updates the JVM args to include the thread count for initialization and update threads.
@@ -149,10 +154,11 @@ export class DraftQuery {
         const { initializationThreads, updateThreads } = this;
         this.jvmArgs = WorkerThreadUtils.addArgs(this.jvmArgs, initializationThreads, updateThreads);
     }
+    /**
+     * Get a PlainEditableQueryInfo object populated from this DraftQuery
+     */
     toEditableQueryInfo() {
-        const editableQuery = { serial: this.serial };
-        this.saveDraftToQuery(editableQuery);
-        return editableQuery;
+        return this.saveDraftToQuery({ serial: this.serial });
     }
     toRowMap(queryTypes) {
         var _a, _b, _c, _d, _e, _f;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@deephaven-enterprise/query-utils",
-  "version": "1.20250801.264-beta",
+  "version": "1.20250801.266-beta",
   "description": "Deephaven Enterprise Query Utils",
   "author": "Deephaven Data Labs LLC",
   "license": "SEE LICENSE IN LICENSE.md",
@@ -29,5 +29,5 @@
   "publishConfig": {
     "access": "public"
   },
-  "gitHead": "93d0b6f170ed43555d7d7b99ff25d1c1d9964517"
+  "gitHead": "a0b6108c8cfdda6363d2cdefc6933f1181cc1ff9"
 }