azure-kusto-data 5.0.4 → 5.1.0

package/dist-esm/src/client.js

@@ -80,27 +80,30 @@ class KustoClient {
         return this.executeQuery(db, query, properties);
     }
     async executeQuery(db, query, properties) {
-        return this._execute(this.endpoints[ExecutionType.Query], ExecutionType.Query, db, query, null, properties);
+        return this._execute(this.endpoints[ExecutionType.Query], ExecutionType.Query, db, { query }, properties);
     }
     async executeQueryV1(db, query, properties) {
-        return this._execute(this.endpoints[ExecutionType.QueryV1], ExecutionType.QueryV1, db, query, null, properties);
+        return this._execute(this.endpoints[ExecutionType.QueryV1], ExecutionType.QueryV1, db, { query }, properties);
     }
     async executeMgmt(db, query, properties) {
-        return this._execute(this.endpoints[ExecutionType.Mgmt], ExecutionType.Mgmt, db, query, null, properties);
+        return this._execute(this.endpoints[ExecutionType.Mgmt], ExecutionType.Mgmt, db, { query }, properties);
     }
-    async executeStreamingIngest(db, table, stream, streamFormat, mappingName, clientRequestId) {
+    async executeStreamingIngest(db, table, stream, streamFormat, mappingName, blob, clientRequestId) {
         let endpoint = `${this.endpoints[ExecutionType.Ingest]}/${this.getDb(db)}/${table}?streamFormat=${streamFormat}`;
         if (mappingName != null) {
             endpoint += `&mappingName=${mappingName}`;
         }
+        if (blob) {
+            endpoint += `&sourceKind=uri`;
+        }
         let properties = null;
         if (clientRequestId) {
             properties = new clientRequestProperties_1.default();
             properties.clientRequestId = clientRequestId;
         }
-        return this._execute(endpoint, ExecutionType.Ingest, db, null, stream, properties);
+        return this._execute(endpoint, ExecutionType.Ingest, db, blob ? { blob } : { stream }, properties);
     }
-    async _execute(endpoint, executionType, db, query, stream, properties) {
+    async _execute(endpoint, executionType, db, entity, properties) {
         this.ensureOpen();
         kustoTrustedEndpoints_1.kustoTrustedEndpoints.validateTrustedEndpoint(endpoint, (await cloudSettings_1.default.getCloudInfoForCluster(this.cluster)).LoginEndpoint);
         db = this.getDb(db);
@@ -109,10 +112,10 @@ class KustoClient {
         let clientRequestPrefix = "";
         const timeout = this._getClientTimeout(executionType, properties);
         let payloadContent = "";
-        if (query != null) {
+        if ("query" in entity) {
             payload = {
                 db,
-                csl: query,
+                csl: entity.query,
             };
             if (properties != null) {
                 payload.properties = properties.toJSON();
@@ -121,8 +124,8 @@ class KustoClient {
             headers["Content-Type"] = "application/json; charset=utf-8";
             clientRequestPrefix = "KNC.execute;";
         }
-        else if (stream != null) {
-            payloadContent = stream;
+        else if ("stream" in entity) {
+            payloadContent = entity.stream;
             clientRequestPrefix = "KNC.executeStreamingIngest;";
             if (core_util_1.isNode) {
                 headers["Content-Encoding"] = "gzip";
@@ -132,6 +135,13 @@ class KustoClient {
                 headers["Content-Type"] = "application/json";
             }
         }
+        else if ("blob" in entity) {
+            payloadContent = {
+                sourceUri: entity.blob,
+            };
+            clientRequestPrefix = "KNC.executeStreamingIngestFromBlob;";
+            headers["Content-Type"] = "application/json";
+        }
         else {
             throw new Error("Invalid parameters - expected query or streaming ingest");
         }

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "azure-kusto-data",
-    "version": "5.0.4",
+    "version": "5.1.0",
     "description": "Azure Data Explorer Query SDK",
     "module": "dist-esm/src/index.js",
     "types": "./types/src/index.d.ts",
@@ -68,5 +68,5 @@
     "overrides": {
         "jsonwebtoken": "^9.0.0"
     },
-    "gitHead": "c50520216e9db02516a3ae5a85e2f088f788595e"
+    "gitHead": "120a8d91fde4b0e520225e09895a28b085b85a7e"
 }

package/types/src/client.d.ts

@@ -9,6 +9,13 @@ declare enum ExecutionType {
     Ingest = "ingest",
     QueryV1 = "queryv1"
 }
+export type RequestEntity = {
+    query: string;
+} | {
+    stream: any;
+} | {
+    blob: string;
+};
 export declare class KustoClient {
     connectionString: ConnectionStringBuilder;
     cluster: string;
@@ -25,8 +32,8 @@ export declare class KustoClient {
     executeQuery(db: string | null, query: string, properties?: ClientRequestProperties): Promise<KustoResponseDataSet>;
     executeQueryV1(db: string | null, query: string, properties?: ClientRequestProperties): Promise<KustoResponseDataSet>;
     executeMgmt(db: string | null, query: string, properties?: ClientRequestProperties): Promise<KustoResponseDataSet>;
-    executeStreamingIngest(db: string | null, table: string, stream: any, streamFormat: any, mappingName: string | null, clientRequestId?: string): Promise<KustoResponseDataSet>;
-    _execute(endpoint: string, executionType: ExecutionType, db: string | null, query: string | null, stream: any, properties?: ClientRequestProperties | null): Promise<KustoResponseDataSet>;
+    executeStreamingIngest(db: string | null, table: string, stream: any, streamFormat: any, mappingName: string | null, blob?: string, clientRequestId?: string): Promise<KustoResponseDataSet>;
+    _execute(endpoint: string, executionType: ExecutionType, db: string | null, entity: RequestEntity, properties?: ClientRequestProperties | null): Promise<KustoResponseDataSet>;
     private getDb;
     _doRequest(endpoint: string, executionType: ExecutionType, headers: {
         [header: string]: string;