azure-kusto-ingest 5.2.2 → 6.0.1

package/dist-esm/src/abstractKustoClient.js

@@ -1,9 +1,15 @@
 "use strict";
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AbstractKustoClient = void 0;
 const ingestionProperties_1 = require("./ingestionProperties");
+const is_ip_1 = __importDefault(require("is-ip"));
+const INGEST_PREFIX = "ingest-";
+const PROTOCOL_SUFFIX = "://";
 class AbstractKustoClient {
     constructor(defaultProps) {
         this._isClosed = false;
@@ -34,6 +40,33 @@ class AbstractKustoClient {
             throw new Error("Client is closed");
         }
     }
+    getIngestionEndpoint(clusterUrl) {
+        if (!clusterUrl || clusterUrl.includes(INGEST_PREFIX) || this.isReservedHostname(clusterUrl)) {
+            return clusterUrl;
+        }
+        return clusterUrl.replace(PROTOCOL_SUFFIX, PROTOCOL_SUFFIX + INGEST_PREFIX);
+    }
+    getQueryEndpoint(clusterUrl) {
+        if (clusterUrl && clusterUrl.includes(INGEST_PREFIX)) {
+            return clusterUrl.replace(INGEST_PREFIX, "");
+        }
+        return clusterUrl;
+    }
+    isReservedHostname(clusterUrl) {
+        try {
+            const parsedUrl = new URL(clusterUrl);
+            const authority = parsedUrl.hostname;
+            if (!authority) {
+                return true;
+            }
+            const is_ip = (0, is_ip_1.default)(authority);
+            const is_localhost = authority.includes("localhost");
+            return is_localhost || is_ip || authority.toLowerCase() === "onebox.dev.kusto.windows.net";
+        }
+        catch (_a) {
+            return false;
+        }
+    }
 }
 exports.AbstractKustoClient = AbstractKustoClient;
 //# sourceMappingURL=abstractKustoClient.js.map

package/dist-esm/src/abstractKustoClient.js.map

@@ -1 +1 @@
-{"version":3,"file":"abstractKustoClient.js","sourceRoot":"","sources":["../../src/abstractKustoClient.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAElC,+DAAsF;AAGtF,MAAsB,mBAAmB;IAKrC,YAAsB,YAAsC;QAFlD,cAAS,GAAY,KAAK,CAAC;QAGjC,IAAI,CAAC,YAAY,EAAE;YACf,IAAI,CAAC,YAAY,GAAG,IAAI,yCAAmB,CAAC,EAAE,CAAC,CAAC;SACnD;aAAM,IAAI,CAAC,CAAC,YAAY,YAAY,yCAAmB,CAAC,EAAE;YACvD,IAAI,CAAC,YAAY,GAAG,IAAI,yCAAmB,CAAC,YAAY,CAAC,CAAC;SAC7D;aAAM;YACH,IAAI,CAAC,YAAY,GAAG,IAAI,yCAAmB,CAAC,YAAY,CAAC,CAAC;SAC7D;IACL,CAAC;IAED,eAAe,CAAC,aAAwC;QACpD,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;QACrD,KAAK,CAAC,WAAW,EAAE,CAAC;QACpB,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE;YACjB,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC;SACzC;QAED,KAAK,CAAC,QAAQ,EAAE,CAAC;QACjB,OAAO,KAAK,CAAC;IACjB,CAAC;IAQM,KAAK;QACR,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;IAC1B,CAAC;IAES,UAAU;QAChB,IAAI,IAAI,CAAC,SAAS,EAAE;YAChB,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;SACvC;IACL,CAAC;CACJ;AAzCD,kDAyCC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { IngestionProperties, IngestionPropertiesInput } from \"./ingestionProperties\";\nimport { StreamDescriptor, FileDescriptorBase, BlobDescriptor } from \"./descriptors\";\n\nexport abstract class AbstractKustoClient {\n    public defaultProps: IngestionProperties;\n    public defaultDatabase?: string;\n    protected _isClosed: boolean = false;\n\n    protected constructor(defaultProps: IngestionPropertiesInput) {\n        if (!defaultProps) {\n            this.defaultProps = new IngestionProperties({});\n        } else if (!(defaultProps instanceof IngestionProperties)) {\n            this.defaultProps = new IngestionProperties(defaultProps);\n        } else {\n            this.defaultProps = new IngestionProperties(defaultProps);\n        }\n    }\n\n    _getMergedProps(newProperties?: IngestionPropertiesInput): IngestionProperties {\n        const props = this.defaultProps.merge(newProperties);\n        props.setDefaults();\n        if (!props.database) {\n            props.database = this.defaultDatabase;\n        }\n\n        props.validate();\n        return props;\n    }\n\n    abstract ingestFromStream(stream: StreamDescriptor, ingestionProperties: IngestionPropertiesInput): Promise<any>;\n\n    abstract ingestFromFile(file: FileDescriptorBase | string | Blob, ingestionProperties: IngestionPropertiesInput): Promise<any>;\n\n    abstract ingestFromBlob(blob: string | BlobDescriptor, ingestionProperties?: IngestionPropertiesInput): Promise<any>;\n\n    public close(): void {\n        this._isClosed = true;\n    }\n\n    protected ensureOpen() {\n        if (this._isClosed) {\n            throw new Error(\"Client is closed\");\n        }\n    }\n}\n"]}
+{"version":3,"file":"abstractKustoClient.js","sourceRoot":"","sources":["../../src/abstractKustoClient.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;;;;AAElC,+DAAsF;AAEtF,kDAAyB;AAEzB,MAAM,aAAa,GAAG,SAAS,CAAC;AAChC,MAAM,eAAe,GAAG,KAAK,CAAC;AAE9B,MAAsB,mBAAmB;IAKrC,YAAsB,YAAsC;QAFlD,cAAS,GAAY,KAAK,CAAC;QAGjC,IAAI,CAAC,YAAY,EAAE;YACf,IAAI,CAAC,YAAY,GAAG,IAAI,yCAAmB,CAAC,EAAE,CAAC,CAAC;SACnD;aAAM,IAAI,CAAC,CAAC,YAAY,YAAY,yCAAmB,CAAC,EAAE;YACvD,IAAI,CAAC,YAAY,GAAG,IAAI,yCAAmB,CAAC,YAAY,CAAC,CAAC;SAC7D;aAAM;YACH,IAAI,CAAC,YAAY,GAAG,IAAI,yCAAmB,CAAC,YAAY,CAAC,CAAC;SAC7D;IACL,CAAC;IAED,eAAe,CAAC,aAAwC;QACpD,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;QACrD,KAAK,CAAC,WAAW,EAAE,CAAC;QACpB,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE;YACjB,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC;SACzC;QAED,KAAK,CAAC,QAAQ,EAAE,CAAC;QACjB,OAAO,KAAK,CAAC;IACjB,CAAC;IAQM,KAAK;QACR,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;IAC1B,CAAC;IAES,UAAU;QAChB,IAAI,IAAI,CAAC,SAAS,EAAE;YAChB,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;SACvC;IACL,CAAC;IAED,oBAAoB,CAAC,UAAmB;QACpC,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAAC,EAAE;YAC1F,OAAO,UAAU,CAAC;SACrB;QACD,OAAO,UAAU,CAAC,OAAO,CAAC,eAAe,EAAE,eAAe,GAAG,aAAa,CAAC,CAAC;IAChF,CAAC;IAED,gBAAgB,CAAC,UAAmB;QAChC,IAAI,UAAU,IAAI,UAAU,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE;YAClD,OAAO,UAAU,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;SAChD;QACD,OAAO,UAAU,CAAC;IACtB,CAAC;IAED,kBAAkB,CAAC,UAAkB;QACjC,IAAI;YACA,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC;YACtC,MAAM,SAAS,GAAG,SAAS,CAAC,QAAQ,CAAC;YACrC,IAAI,CAAC,SAAS,EAAE;gBACZ,OAAO,IAAI,CAAC;aACf;YACD,MAAM,KAAK,GAAG,IAAA,eAAI,EAAC,SAAS,CAAC,CAAC;YAC9B,MAAM,YAAY,GAAG,SAAS,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;YACrD,OAAO,YAAY,IAAI,KAAK,IAAI,SAAS,CAAC,WAAW,EAAE,KAAK,8BAA8B,CAAC;SAC9F;QAAC,WAAM;YACJ,OAAO,KAAK,CAAC;SAChB;IACL,CAAC;CACJ;AAtED,kDAsEC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { IngestionProperties, IngestionPropertiesInput } from \"./ingestionProperties\";\nimport { StreamDescriptor, FileDescriptorBase, BlobDescriptor } from \"./descriptors\";\nimport isIP from \"is-ip\";\n\nconst INGEST_PREFIX = \"ingest-\";\nconst PROTOCOL_SUFFIX = \"://\";\n\nexport abstract class AbstractKustoClient {\n    public defaultProps: IngestionProperties;\n    public defaultDatabase?: string;\n    protected _isClosed: boolean = false;\n\n    protected constructor(defaultProps: IngestionPropertiesInput) {\n        if (!defaultProps) {\n            this.defaultProps = new IngestionProperties({});\n        } else if (!(defaultProps instanceof IngestionProperties)) {\n            this.defaultProps = new IngestionProperties(defaultProps);\n        } else {\n            this.defaultProps = new IngestionProperties(defaultProps);\n        }\n    }\n\n    _getMergedProps(newProperties?: IngestionPropertiesInput): IngestionProperties {\n        const props = this.defaultProps.merge(newProperties);\n        props.setDefaults();\n        if (!props.database) {\n            props.database = this.defaultDatabase;\n        }\n\n        props.validate();\n        return props;\n    }\n\n    abstract ingestFromStream(stream: StreamDescriptor, ingestionProperties: IngestionPropertiesInput): Promise<any>;\n\n    abstract ingestFromFile(file: FileDescriptorBase | string | Blob, ingestionProperties: IngestionPropertiesInput): Promise<any>;\n\n    abstract ingestFromBlob(blob: string | BlobDescriptor, ingestionProperties?: IngestionPropertiesInput): Promise<any>;\n\n    public close(): void {\n        this._isClosed = true;\n    }\n\n    protected ensureOpen() {\n        if (this._isClosed) {\n            throw new Error(\"Client is closed\");\n        }\n    }\n\n    getIngestionEndpoint(clusterUrl?: string): string | undefined {\n        if (!clusterUrl || clusterUrl.includes(INGEST_PREFIX) || this.isReservedHostname(clusterUrl)) {\n            return clusterUrl;\n        }\n        return clusterUrl.replace(PROTOCOL_SUFFIX, PROTOCOL_SUFFIX + INGEST_PREFIX);\n    }\n\n    getQueryEndpoint(clusterUrl?: string): string | undefined {\n        if (clusterUrl && clusterUrl.includes(INGEST_PREFIX)) {\n            return clusterUrl.replace(INGEST_PREFIX, \"\");\n        }\n        return clusterUrl;\n    }\n\n    isReservedHostname(clusterUrl: string): boolean {\n        try {\n            const parsedUrl = new URL(clusterUrl);\n            const authority = parsedUrl.hostname;\n            if (!authority) {\n                return true;\n            }\n            const is_ip = isIP(authority);\n            const is_localhost = authority.includes(\"localhost\");\n            return is_localhost || is_ip || authority.toLowerCase() === \"onebox.dev.kusto.windows.net\";\n        } catch {\n            return false;\n        }\n    }\n}\n"]}

package/dist-esm/src/descriptors.js

@@ -5,7 +5,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.generateBlobName = exports.BlobDescriptor = exports.StreamDescriptor = exports.AbstractDescriptor = exports.getSourceId = exports.CompressionType = void 0;
+exports.generateBlobName = exports.BlobDescriptor = exports.StreamDescriptor = exports.AbstractDescriptor = exports.shouldCompressFileByExtension = exports.getSourceId = exports.CompressionType = void 0;
 const uuid_1 = require("uuid");
 const uuid_validate_1 = __importDefault(require("uuid-validate"));
 const storage_blob_1 = require("@azure/storage-blob");
@@ -25,6 +25,10 @@ const getSourceId = (sourceId) => {
     return (0, uuid_1.v4)();
 };
 exports.getSourceId = getSourceId;
+function shouldCompressFileByExtension(extension) {
+    return !(extension === ".avro" || extension === ".apacheavro" || extension === ".parquet" || extension === ".sstream" || extension === ".orc");
+}
+exports.shouldCompressFileByExtension = shouldCompressFileByExtension;
 class AbstractDescriptor {
     constructor(sourceId = null, size = null) {
         this.sourceId = sourceId;

package/dist-esm/src/descriptors.js.map

@@ -1 +1 @@
-{"version":3,"file":"descriptors.js","sourceRoot":"","sources":["../../src/descriptors.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;;;;AAElC,+BAAoC;AACpC,kEAAyC;AAGzC,sDAAiD;AAEjD,IAAY,eAIX;AAJD,WAAY,eAAe;IACvB,+BAAY,CAAA;IACZ,+BAAY,CAAA;IACZ,4BAAS,CAAA;AACb,CAAC,EAJW,eAAe,+BAAf,eAAe,QAI1B;AAEM,MAAM,WAAW,GAAG,CAAC,QAAuB,EAAU,EAAE;IAC3D,IAAI,QAAQ,EAAE;QACV,IAAI,CAAC,IAAA,uBAAY,EAAC,QAAQ,EAAE,CAAC,CAAC,EAAE;YAC5B,MAAM,KAAK,CAAC,iCAAiC,CAAC,CAAC;SAClD;QACD,OAAO,QAAQ,CAAC;KACnB;IACD,OAAO,IAAA,SAAM,GAAE,CAAC;AACpB,CAAC,CAAC;AARW,QAAA,WAAW,eAQtB;AAEF,MAAsB,kBAAkB;IACpC,YAAmB,WAA0B,IAAI,EAAS,OAAsB,IAAI;QAAjE,aAAQ,GAAR,QAAQ,CAAsB;QAAS,SAAI,GAAJ,IAAI,CAAsB;QAChF,IAAI,CAAC,QAAQ,GAAG,IAAA,mBAAW,EAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;CACJ;AAJD,gDAIC;AAED,MAAa,gBAAiB,SAAQ,kBAAkB;IACpD;;OAEG;IACH,YACa,MAA8B,EACvC,WAA0B,IAAI,EACvB,kBAAmC,eAAe,CAAC,IAAI,EAC9D,OAAsB,IAAI;QAE1B,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QALb,WAAM,GAAN,MAAM,CAAwB;QAEhC,oBAAe,GAAf,eAAe,CAAwC;IAIlE,CAAC;IAED,KAAK,CAAC,KAAuB;QACzB,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;QACvB,IAAI,CAAC,eAAe,GAAG,KAAK,CAAC,eAAe,CAAC;QAC7C,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;QAC/B,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,6CAA6C;IAC7C,oBAAoB;QAChB,OAAO,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAClE,CAAC;CACJ;AAxBD,4CAwBC;AAED,MAAa,cAAe,SAAQ,kBAAkB;IAClD,YAAqB,IAAY,EAAE,OAAsB,IAAI,EAAE,WAA0B,IAAI;QACzF,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QADL,SAAI,GAAJ,IAAI,CAAQ;IAEjC,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;YACZ,MAAM,UAAU,GAAG,IAAI,yBAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7C,MAAM,SAAS,GAAG,MAAM,UAAU,CAAC,aAAa,EAAE,CAAC;YACnD,MAAM,MAAM,GAAG,SAAS,CAAC,aAAa,CAAC;YACvC,IAAI,MAAM,KAAK,SAAS,EAAE;gBACtB,IAAI,MAAM,KAAK,CAAC,EAAE;oBACd,MAAM,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC;iBAClC;gBACD,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC;aACtB;SACJ;IACL,CAAC;CACJ;AAlBD,wCAkBC;AAaM,MAAM,gBAAgB,GAAG,CAAC,IAA2C,EAAE,KAA0B,EAAU,EAAE;IAChH,MAAM,SAAS,GAAG,IAAI,YAAY,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC;IAEhJ,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;IAChE,MAAM,iBAAiB,GAAG,IAAI,CAAC,oBAAoB,EAAE,CAAC;IACtD,OAAO,GAAG,KAAK,CAAC,QAAQ,KAAK,KAAK,CAAC,KAAK,KAAK,IAAI,CAAC,QAAQ,GAAG,SAAS,IAAI,YAAY,GAAG,iBAAiB,EAAE,CAAC;AACjH,CAAC,CAAC;AANW,QAAA,gBAAgB,oBAM3B","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { v4 as uuidv4 } from \"uuid\";\nimport uuidValidate from \"uuid-validate\";\nimport { Readable } from \"stream\";\nimport IngestionProperties from \"./ingestionProperties\";\nimport { BlobClient } from \"@azure/storage-blob\";\n\nexport enum CompressionType {\n    ZIP = \".zip\",\n    GZIP = \".gz\",\n    None = \"\",\n}\n\nexport const getSourceId = (sourceId: string | null): string => {\n    if (sourceId) {\n        if (!uuidValidate(sourceId, 4)) {\n            throw Error(\"sourceId is not a valid uuid/v4\");\n        }\n        return sourceId;\n    }\n    return uuidv4();\n};\n\nexport abstract class AbstractDescriptor {\n    constructor(public sourceId: string | null = null, public size: number | null = null) {\n        this.sourceId = getSourceId(sourceId);\n    }\n}\n\nexport class StreamDescriptor extends AbstractDescriptor {\n    /**\n     * Use Readable for Node.js and ArrayBuffer in browser\n     */\n    constructor(\n        readonly stream: Readable | ArrayBuffer,\n        sourceId: string | null = null,\n        public compressionType: CompressionType = CompressionType.None,\n        size: number | null = null\n    ) {\n        super(sourceId, size);\n    }\n\n    merge(other: StreamDescriptor) {\n        this.size = other.size;\n        this.compressionType = other.compressionType;\n        this.sourceId = other.sourceId;\n        return this;\n    }\n\n    // Currently streams are not compressed by us\n    getCompressionSuffix() {\n        return this.compressionType ? `.${this.compressionType}` : \"\";\n    }\n}\n\nexport class BlobDescriptor extends AbstractDescriptor {\n    constructor(readonly path: string, size: number | null = null, sourceId: string | null = null) {\n        super(sourceId, size);\n    }\n\n    async fillSize(): Promise<void> {\n        if (!this.size) {\n            const blobClient = new BlobClient(this.path);\n            const blobProps = await blobClient.getProperties();\n            const length = blobProps.contentLength;\n            if (length !== undefined) {\n                if (length === 0) {\n                    throw new Error(\"Empty blob.\");\n                }\n                this.size = length;\n            }\n        }\n    }\n}\n\nexport interface FileDescriptorBase {\n    size: number | null;\n    zipped: boolean;\n    compressionType: CompressionType;\n    cleanupTmp?: () => Promise<void>;\n    extension?: string;\n    name?: string;\n    sourceId: string | null;\n    getCompressionSuffix: () => string;\n}\n\nexport const generateBlobName = (desc: StreamDescriptor | FileDescriptorBase, props: IngestionProperties): string => {\n    const extension = desc instanceof StreamDescriptor ? null : `${desc.name ? \"__\" + desc.name : `${desc.extension ? \".\" + desc.extension : \"\"}`}`;\n\n    const formatSuffix = props.format ? `.${props.format}` : \".csv\";\n    const compressionString = desc.getCompressionSuffix();\n    return `${props.database}__${props.table}__${desc.sourceId}${extension || formatSuffix}${compressionString}`;\n};\n"]}
+{"version":3,"file":"descriptors.js","sourceRoot":"","sources":["../../src/descriptors.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;;;;AAElC,+BAAoC;AACpC,kEAAyC;AAGzC,sDAAiD;AAEjD,IAAY,eAIX;AAJD,WAAY,eAAe;IACvB,+BAAY,CAAA;IACZ,+BAAY,CAAA;IACZ,4BAAS,CAAA;AACb,CAAC,EAJW,eAAe,+BAAf,eAAe,QAI1B;AAEM,MAAM,WAAW,GAAG,CAAC,QAAuB,EAAU,EAAE;IAC3D,IAAI,QAAQ,EAAE;QACV,IAAI,CAAC,IAAA,uBAAY,EAAC,QAAQ,EAAE,CAAC,CAAC,EAAE;YAC5B,MAAM,KAAK,CAAC,iCAAiC,CAAC,CAAC;SAClD;QACD,OAAO,QAAQ,CAAC;KACnB;IACD,OAAO,IAAA,SAAM,GAAE,CAAC;AACpB,CAAC,CAAC;AARW,QAAA,WAAW,eAQtB;AAEF,SAAgB,6BAA6B,CAAC,SAAkB;IAC5D,OAAO,CAAC,CAAC,SAAS,KAAK,OAAO,IAAI,SAAS,KAAK,aAAa,IAAI,SAAS,KAAK,UAAU,IAAI,SAAS,KAAK,UAAU,IAAI,SAAS,KAAK,MAAM,CAAC,CAAC;AACnJ,CAAC;AAFD,sEAEC;AAED,MAAsB,kBAAkB;IACpC,YAAmB,WAA0B,IAAI,EAAS,OAAsB,IAAI;QAAjE,aAAQ,GAAR,QAAQ,CAAsB;QAAS,SAAI,GAAJ,IAAI,CAAsB;QAChF,IAAI,CAAC,QAAQ,GAAG,IAAA,mBAAW,EAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;CACJ;AAJD,gDAIC;AAED,MAAa,gBAAiB,SAAQ,kBAAkB;IACpD;;OAEG;IACH,YACa,MAA8B,EACvC,WAA0B,IAAI,EACvB,kBAAmC,eAAe,CAAC,IAAI,EAC9D,OAAsB,IAAI;QAE1B,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QALb,WAAM,GAAN,MAAM,CAAwB;QAEhC,oBAAe,GAAf,eAAe,CAAwC;IAIlE,CAAC;IAED,KAAK,CAAC,KAAuB;QACzB,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;QACvB,IAAI,CAAC,eAAe,GAAG,KAAK,CAAC,eAAe,CAAC;QAC7C,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;QAC/B,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,6CAA6C;IAC7C,oBAAoB;QAChB,OAAO,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAClE,CAAC;CACJ;AAxBD,4CAwBC;AAED,MAAa,cAAe,SAAQ,kBAAkB;IAClD,YAAqB,IAAY,EAAE,OAAsB,IAAI,EAAE,WAA0B,IAAI;QACzF,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QADL,SAAI,GAAJ,IAAI,CAAQ;IAEjC,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;YACZ,MAAM,UAAU,GAAG,IAAI,yBAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7C,MAAM,SAAS,GAAG,MAAM,UAAU,CAAC,aAAa,EAAE,CAAC;YACnD,MAAM,MAAM,GAAG,SAAS,CAAC,aAAa,CAAC;YACvC,IAAI,MAAM,KAAK,SAAS,EAAE;gBACtB,IAAI,MAAM,KAAK,CAAC,EAAE;oBACd,MAAM,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC;iBAClC;gBACD,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC;aACtB;SACJ;IACL,CAAC;CACJ;AAlBD,wCAkBC;AAaM,MAAM,gBAAgB,GAAG,CAAC,IAA2C,EAAE,KAA0B,EAAU,EAAE;IAChH,MAAM,SAAS,GAAG,IAAI,YAAY,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC;IAEhJ,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;IAChE,MAAM,iBAAiB,GAAG,IAAI,CAAC,oBAAoB,EAAE,CAAC;IACtD,OAAO,GAAG,KAAK,CAAC,QAAQ,KAAK,KAAK,CAAC,KAAK,KAAK,IAAI,CAAC,QAAQ,GAAG,SAAS,IAAI,YAAY,GAAG,iBAAiB,EAAE,CAAC;AACjH,CAAC,CAAC;AANW,QAAA,gBAAgB,oBAM3B","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { v4 as uuidv4 } from \"uuid\";\nimport uuidValidate from \"uuid-validate\";\nimport { Readable } from \"stream\";\nimport IngestionProperties from \"./ingestionProperties\";\nimport { BlobClient } from \"@azure/storage-blob\";\n\nexport enum CompressionType {\n    ZIP = \".zip\",\n    GZIP = \".gz\",\n    None = \"\",\n}\n\nexport const getSourceId = (sourceId: string | null): string => {\n    if (sourceId) {\n        if (!uuidValidate(sourceId, 4)) {\n            throw Error(\"sourceId is not a valid uuid/v4\");\n        }\n        return sourceId;\n    }\n    return uuidv4();\n};\n\nexport function shouldCompressFileByExtension(extension?: string): boolean {\n    return !(extension === \".avro\" || extension === \".apacheavro\" || extension === \".parquet\" || extension === \".sstream\" || extension === \".orc\");\n}\n\nexport abstract class AbstractDescriptor {\n    constructor(public sourceId: string | null = null, public size: number | null = null) {\n        this.sourceId = getSourceId(sourceId);\n    }\n}\n\nexport class StreamDescriptor extends AbstractDescriptor {\n    /**\n     * Use Readable for Node.js and ArrayBuffer in browser\n     */\n    constructor(\n        readonly stream: Readable | ArrayBuffer,\n        sourceId: string | null = null,\n        public compressionType: CompressionType = CompressionType.None,\n        size: number | null = null\n    ) {\n        super(sourceId, size);\n    }\n\n    merge(other: StreamDescriptor) {\n        this.size = other.size;\n        this.compressionType = other.compressionType;\n        this.sourceId = other.sourceId;\n        return this;\n    }\n\n    // Currently streams are not compressed by us\n    getCompressionSuffix() {\n        return this.compressionType ? `.${this.compressionType}` : \"\";\n    }\n}\n\nexport class BlobDescriptor extends AbstractDescriptor {\n    constructor(readonly path: string, size: number | null = null, sourceId: string | null = null) {\n        super(sourceId, size);\n    }\n\n    async fillSize(): Promise<void> {\n        if (!this.size) {\n            const blobClient = new BlobClient(this.path);\n            const blobProps = await blobClient.getProperties();\n            const length = blobProps.contentLength;\n            if (length !== undefined) {\n                if (length === 0) {\n                    throw new Error(\"Empty blob.\");\n                }\n                this.size = length;\n            }\n        }\n    }\n}\n\nexport interface FileDescriptorBase {\n    size: number | null;\n    zipped: boolean;\n    compressionType: CompressionType;\n    cleanupTmp?: () => Promise<void>;\n    extension?: string;\n    name?: string;\n    sourceId: string | null;\n    getCompressionSuffix: () => string;\n}\n\nexport const generateBlobName = (desc: StreamDescriptor | FileDescriptorBase, props: IngestionProperties): string => {\n    const extension = desc instanceof StreamDescriptor ? null : `${desc.name ? \"__\" + desc.name : `${desc.extension ? \".\" + desc.extension : \"\"}`}`;\n\n    const formatSuffix = props.format ? `.${props.format}` : \".csv\";\n    const compressionString = desc.getCompressionSuffix();\n    return `${props.database}__${props.table}__${desc.sourceId}${extension || formatSuffix}${compressionString}`;\n};\n"]}

package/dist-esm/src/fileDescriptor.browser.js

@@ -8,6 +8,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.FileDescriptor = void 0;
 const pako_1 = __importDefault(require("pako"));
 const descriptors_1 = require("./descriptors");
+const ingestionProperties_1 = require("./ingestionProperties");
 class FileDescriptor extends descriptors_1.AbstractDescriptor {
     constructor(file, sourceId = null, size = null, compressionType = descriptors_1.CompressionType.None, extension, name) {
         super(sourceId);
@@ -17,9 +18,11 @@ class FileDescriptor extends descriptors_1.AbstractDescriptor {
         this.compressionType = compressionType;
         this.size = size || file.size;
         this.zipped = compressionType !== descriptors_1.CompressionType.None || this.extension === ".gz" || this.extension === ".zip";
+        this.shouldNotCompress = !(0, descriptors_1.shouldCompressFileByExtension)(this.extension);
     }
-    async prepare() {
-        if (!this.zipped) {
+    async prepare(ingestionProperties) {
+        const shouldNotCompressByFormat = !(0, ingestionProperties_1.shouldCompressFileByFormat)(ingestionProperties);
+        if (!this.zipped && !this.shouldNotCompress && !shouldNotCompressByFormat) {
             try {
                 const gzipped = pako_1.default.gzip(await this.file.arrayBuffer());
                 return new Blob([gzipped]);

package/dist-esm/src/fileDescriptor.browser.js.map

@@ -1 +1 @@
-{"version":3,"file":"fileDescriptor.browser.js","sourceRoot":"","sources":["../../src/fileDescriptor.browser.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;;;;AAElC,gDAAwB;AACxB,+CAAwF;AAExF,MAAa,cAAe,SAAQ,gCAAkB;IAMlD,YACa,IAAU,EACnB,WAA0B,IAAI,EAC9B,OAAsB,IAAI,EAC1B,kBAAmC,6BAAe,CAAC,IAAI,EAC9C,SAAkB,EAClB,IAAa;QAEtB,KAAK,CAAC,QAAQ,CAAC,CAAC;QAPP,SAAI,GAAJ,IAAI,CAAM;QAIV,cAAS,GAAT,SAAS,CAAS;QAClB,SAAI,GAAJ,IAAI,CAAS;QAGtB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,IAAI,GAAG,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QAC9B,IAAI,CAAC,MAAM,GAAG,eAAe,KAAK,6BAAe,CAAC,IAAI,IAAI,IAAI,CAAC,SAAS,KAAK,KAAK,IAAI,IAAI,CAAC,SAAS,KAAK,MAAM,CAAC;IACpH,CAAC;IAED,KAAK,CAAC,OAAO;QACT,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;YACd,IAAI;gBACA,MAAM,OAAO,GAAG,cAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;gBACzD,OAAO,IAAI,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;aAC9B;YAAC,OAAO,CAAC,EAAE;gBACR,kCAAkC;aACrC;SACJ;QAED,OAAO,IAAI,CAAC,IAAI,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,OAAO;QACT,IAAI,IAAI,CAAC,UAAU,EAAE;YACjB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;SAC3B;IACL,CAAC;IAED,oBAAoB;QAChB,OAAO,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;IACrE,CAAC;CACJ;AA1CD,wCA0CC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport pako from \"pako\";\nimport { AbstractDescriptor, CompressionType, FileDescriptorBase } from \"./descriptors\";\n\nexport class FileDescriptor extends AbstractDescriptor implements FileDescriptorBase {\n    size: number | null;\n    zipped: boolean;\n    compressionType: CompressionType;\n    cleanupTmp?: () => Promise<void>;\n\n    constructor(\n        readonly file: Blob,\n        sourceId: string | null = null,\n        size: number | null = null,\n        compressionType: CompressionType = CompressionType.None,\n        readonly extension?: string,\n        readonly name?: string\n    ) {\n        super(sourceId);\n        this.compressionType = compressionType;\n        this.size = size || file.size;\n        this.zipped = compressionType !== CompressionType.None || this.extension === \".gz\" || this.extension === \".zip\";\n    }\n\n    async prepare(): Promise<Blob> {\n        if (!this.zipped) {\n            try {\n                const gzipped = pako.gzip(await this.file.arrayBuffer());\n                return new Blob([gzipped]);\n            } catch (e) {\n                // Ignore - return the file itself\n            }\n        }\n\n        return this.file;\n    }\n\n    async cleanup(): Promise<void> {\n        if (this.cleanupTmp) {\n            await this.cleanupTmp();\n        }\n    }\n\n    getCompressionSuffix() {\n        return this.compressionType ? `.${this.compressionType}` : \".gz\";\n    }\n}\n"]}
+{"version":3,"file":"fileDescriptor.browser.js","sourceRoot":"","sources":["../../src/fileDescriptor.browser.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;;;;AAElC,gDAAwB;AACxB,+CAAuH;AACvH,+DAA6F;AAE7F,MAAa,cAAe,SAAQ,gCAAkB;IAOlD,YACa,IAAU,EACnB,WAA0B,IAAI,EAC9B,OAAsB,IAAI,EAC1B,kBAAmC,6BAAe,CAAC,IAAI,EAC9C,SAAkB,EAClB,IAAa;QAEtB,KAAK,CAAC,QAAQ,CAAC,CAAC;QAPP,SAAI,GAAJ,IAAI,CAAM;QAIV,cAAS,GAAT,SAAS,CAAS;QAClB,SAAI,GAAJ,IAAI,CAAS;QAGtB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,IAAI,GAAG,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QAC9B,IAAI,CAAC,MAAM,GAAG,eAAe,KAAK,6BAAe,CAAC,IAAI,IAAI,IAAI,CAAC,SAAS,KAAK,KAAK,IAAI,IAAI,CAAC,SAAS,KAAK,MAAM,CAAC;QAChH,IAAI,CAAC,iBAAiB,GAAG,CAAC,IAAA,2CAA6B,EAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC5E,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,mBAA8C;QACxD,MAAM,yBAAyB,GAAG,CAAC,IAAA,gDAA0B,EAAC,mBAAmB,CAAC,CAAC;QACnF,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,iBAAiB,IAAI,CAAC,yBAAyB,EAAE;YACvE,IAAI;gBACA,MAAM,OAAO,GAAG,cAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;gBACzD,OAAO,IAAI,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;aAC9B;YAAC,OAAO,CAAC,EAAE;gBACR,kCAAkC;aACrC;SACJ;QAED,OAAO,IAAI,CAAC,IAAI,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,OAAO;QACT,IAAI,IAAI,CAAC,UAAU,EAAE;YACjB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;SAC3B;IACL,CAAC;IAED,oBAAoB;QAChB,OAAO,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;IACrE,CAAC;CACJ;AA7CD,wCA6CC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport pako from \"pako\";\nimport { AbstractDescriptor, CompressionType, FileDescriptorBase, shouldCompressFileByExtension } from \"./descriptors\";\nimport { IngestionPropertiesInput, shouldCompressFileByFormat } from \"./ingestionProperties\";\n\nexport class FileDescriptor extends AbstractDescriptor implements FileDescriptorBase {\n    size: number | null;\n    zipped: boolean;\n    compressionType: CompressionType;\n    shouldNotCompress: boolean;\n    cleanupTmp?: () => Promise<void>;\n\n    constructor(\n        readonly file: Blob,\n        sourceId: string | null = null,\n        size: number | null = null,\n        compressionType: CompressionType = CompressionType.None,\n        readonly extension?: string,\n        readonly name?: string\n    ) {\n        super(sourceId);\n        this.compressionType = compressionType;\n        this.size = size || file.size;\n        this.zipped = compressionType !== CompressionType.None || this.extension === \".gz\" || this.extension === \".zip\";\n        this.shouldNotCompress = !shouldCompressFileByExtension(this.extension);\n    }\n\n    async prepare(ingestionProperties?: IngestionPropertiesInput): Promise<Blob> {\n        const shouldNotCompressByFormat = !shouldCompressFileByFormat(ingestionProperties);\n        if (!this.zipped && !this.shouldNotCompress && !shouldNotCompressByFormat) {\n            try {\n                const gzipped = pako.gzip(await this.file.arrayBuffer());\n                return new Blob([gzipped]);\n            } catch (e) {\n                // Ignore - return the file itself\n            }\n        }\n\n        return this.file;\n    }\n\n    async cleanup(): Promise<void> {\n        if (this.cleanupTmp) {\n            await this.cleanupTmp();\n        }\n    }\n\n    getCompressionSuffix() {\n        return this.compressionType ? `.${this.compressionType}` : \".gz\";\n    }\n}\n"]}

package/dist-esm/src/fileDescriptor.js

@@ -12,6 +12,7 @@ const fs_1 = __importDefault(require("fs"));
 const tmp_promise_1 = require("tmp-promise");
 const util_1 = require("util");
 const descriptors_1 = require("./descriptors");
+const ingestionProperties_1 = require("./ingestionProperties");
 /**
  * Describes a file to be ingested. Use string to describe a local path in Node.JS and Blob object in browsers
  */
@@ -31,6 +32,7 @@ class FileDescriptor extends descriptors_1.AbstractDescriptor {
         this.name = name ? name : path_1.default.basename(this.file);
         this.extension = extension ? extension : path_1.default.extname(this.file).toLowerCase();
         this.zipped = compressionType !== descriptors_1.CompressionType.None || this.extension === ".gz" || this.extension === ".zip";
+        this.shouldNotCompress = !(0, descriptors_1.shouldCompressFileByExtension)(this.extension);
     }
     async _gzip() {
         const { path, cleanup } = await (0, tmp_promise_1.file)({ postfix: ".gz", keep: false });
@@ -51,8 +53,9 @@ class FileDescriptor extends descriptors_1.AbstractDescriptor {
         });
         return path;
     }
-    async prepare() {
-        if (this.zipped) {
+    async prepare(ingestionProperties) {
+        const shouldNotCompressByFormat = !(0, ingestionProperties_1.shouldCompressFileByFormat)(ingestionProperties);
+        if (this.zipped || this.shouldNotCompress || shouldNotCompressByFormat) {
             const estimatedCompressionModifier = 11;
             await this._calculateSize(estimatedCompressionModifier);
             return this.file;

package/dist-esm/src/fileDescriptor.js.map

@@ -1 +1 @@
-{"version":3,"file":"fileDescriptor.js","sourceRoot":"","sources":["../../src/fileDescriptor.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;;;;AAElC,gDAAwB;AACxB,gDAA2B;AAC3B,4CAAoB;AACpB,6CAA8C;AAC9C,+BAAiC;AACjC,+CAAwF;AAExF;;GAEG;AACH,MAAa,cAAe,SAAQ,gCAAkB;IAKlD;IACI;;OAEG;IACM,IAAmB,EAC5B,WAA0B,IAAI,EAC9B,OAAsB,IAAI,EAC1B,kBAAmC,6BAAe,CAAC,IAAI,EAC9C,SAAkB,EAAE,sCAAsC;IAC1D,IAAa,CAAC,sCAAsC;;QAE7D,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAPb,SAAI,GAAJ,IAAI,CAAe;QAInB,cAAS,GAAT,SAAS,CAAS;QAClB,SAAI,GAAJ,IAAI,CAAS;QAGtB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,cAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAc,CAAC,CAAC;QAChE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,cAAO,CAAC,OAAO,CAAC,IAAI,CAAC,IAAc,CAAC,CAAC,WAAW,EAAE,CAAC;QAE5F,IAAI,CAAC,MAAM,GAAG,eAAe,KAAK,6BAAe,CAAC,IAAI,IAAI,IAAI,CAAC,SAAS,KAAK,KAAK,IAAI,IAAI,CAAC,SAAS,KAAK,MAAM,CAAC;IACpH,CAAC;IAED,KAAK,CAAC,KAAK;QACP,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,MAAM,IAAA,kBAAO,EAAC,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QACzE,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC;QAE1B,MAAM,MAAM,GAAG,cAAI,CAAC,UAAU,EAAE,CAAC;QACjC,MAAM,KAAK,GAAG,YAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAc,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5E,MAAM,MAAM,GAAG,YAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAE1C,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAClC,KAAK;iBACA,IAAI,CAAC,MAAM,CAAC;iBACZ,IAAI,CAAC,MAAM,CAAC;iBACZ,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBACjB,MAAM,CAAC,GAAG,CAAC,CAAC;YAChB,CAAC,CAAC,CAAC;YACP,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE;gBACtB,OAAO,CAAC,IAAI,CAAC,CAAC;YAClB,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,OAAO;QACT,IAAI,IAAI,CAAC,MAAM,EAAE;YACb,MAAM,4BAA4B,GAAG,EAAE,CAAC;YACxC,MAAM,IAAI,CAAC,cAAc,CAAC,4BAA4B,CAAC,CAAC;YACxD,OAAO,IAAI,CAAC,IAAc,CAAC;SAC9B;QAED,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QAChC,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC5B,OAAO,IAAI,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,WAAmB,CAAC;QAC7C,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,IAAI,CAAC,EAAE;YACrC,MAAM,SAAS,GAAG,IAAA,gBAAS,EAAC,YAAE,CAAC,IAAI,CAAC,CAAC;YACrC,IAAI,CAAC,IAAI,GAAG,CAAC,MAAM,SAAS,CAAC,IAAI,CAAC,IAAc,CAAC,CAAC,CAAC,IAAI,GAAG,QAAQ,CAAC;SACtE;IACL,CAAC;IAED,KAAK,CAAC,OAAO;QACT,IAAI,IAAI,CAAC,UAAU,EAAE;YACjB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;SAC3B;IACL,CAAC;IAED,oBAAoB;QAChB,OAAO,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;IACrE,CAAC;CACJ;AA3ED,wCA2EC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport zlib from \"zlib\";\nimport pathlib from \"path\";\nimport fs from \"fs\";\nimport { file as tmpFile } from \"tmp-promise\";\nimport { promisify } from \"util\";\nimport { AbstractDescriptor, CompressionType, FileDescriptorBase } from \"./descriptors\";\n\n/**\n * Describes a file to be ingested. Use string to describe a local path in Node.JS and Blob object in browsers\n */\nexport class FileDescriptor extends AbstractDescriptor implements FileDescriptorBase {\n    zipped: boolean;\n    compressionType: CompressionType;\n    cleanupTmp?: () => Promise<void>;\n\n    constructor(\n        /**\n         * Use string in Node.JS and Blob in browser\n         */\n        readonly file: string | Blob,\n        sourceId: string | null = null,\n        size: number | null = null,\n        compressionType: CompressionType = CompressionType.None,\n        readonly extension?: string, // Extracted from file name by default\n        readonly name?: string // Extracted from file name by default\n    ) {\n        super(sourceId, size);\n        this.compressionType = compressionType;\n        this.name = name ? name : pathlib.basename(this.file as string);\n        this.extension = extension ? extension : pathlib.extname(this.file as string).toLowerCase();\n\n        this.zipped = compressionType !== CompressionType.None || this.extension === \".gz\" || this.extension === \".zip\";\n    }\n\n    async _gzip(): Promise<string> {\n        const { path, cleanup } = await tmpFile({ postfix: \".gz\", keep: false });\n        this.cleanupTmp = cleanup;\n\n        const zipper = zlib.createGzip();\n        const input = fs.createReadStream(this.file as string, { autoClose: true });\n        const output = fs.createWriteStream(path);\n\n        await new Promise((resolve, reject) => {\n            input\n                .pipe(zipper)\n                .pipe(output)\n                .on(\"error\", (err) => {\n                    reject(err);\n                });\n            output.once(\"close\", () => {\n                resolve(null);\n            });\n        });\n\n        return path;\n    }\n\n    async prepare(): Promise<string> {\n        if (this.zipped) {\n            const estimatedCompressionModifier = 11;\n            await this._calculateSize(estimatedCompressionModifier);\n            return this.file as string;\n        }\n\n        const path = await this._gzip();\n        await this._calculateSize();\n        return path;\n    }\n\n    private async _calculateSize(modifier: number = 1): Promise<void> {\n        if (this.size == null || this.size <= 0) {\n            const asyncStat = promisify(fs.stat);\n            this.size = (await asyncStat(this.file as string)).size * modifier;\n        }\n    }\n\n    async cleanup(): Promise<void> {\n        if (this.cleanupTmp) {\n            await this.cleanupTmp();\n        }\n    }\n\n    getCompressionSuffix() {\n        return this.compressionType ? `.${this.compressionType}` : \".gz\";\n    }\n}\n"]}
+{"version":3,"file":"fileDescriptor.js","sourceRoot":"","sources":["../../src/fileDescriptor.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;;;;AAElC,gDAAwB;AACxB,gDAA2B;AAC3B,4CAAoB;AACpB,6CAA8C;AAC9C,+BAAiC;AACjC,+CAAuH;AACvH,+DAA6F;AAE7F;;GAEG;AACH,MAAa,cAAe,SAAQ,gCAAkB;IAMlD;IACI;;OAEG;IACM,IAAmB,EAC5B,WAA0B,IAAI,EAC9B,OAAsB,IAAI,EAC1B,kBAAmC,6BAAe,CAAC,IAAI,EAC9C,SAAkB,EAAE,sCAAsC;IAC1D,IAAa,CAAC,sCAAsC;;QAE7D,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAPb,SAAI,GAAJ,IAAI,CAAe;QAInB,cAAS,GAAT,SAAS,CAAS;QAClB,SAAI,GAAJ,IAAI,CAAS;QAGtB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,cAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAc,CAAC,CAAC;QAChE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,cAAO,CAAC,OAAO,CAAC,IAAI,CAAC,IAAc,CAAC,CAAC,WAAW,EAAE,CAAC;QAE5F,IAAI,CAAC,MAAM,GAAG,eAAe,KAAK,6BAAe,CAAC,IAAI,IAAI,IAAI,CAAC,SAAS,KAAK,KAAK,IAAI,IAAI,CAAC,SAAS,KAAK,MAAM,CAAC;QAChH,IAAI,CAAC,iBAAiB,GAAG,CAAC,IAAA,2CAA6B,EAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC5E,CAAC;IAED,KAAK,CAAC,KAAK;QACP,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,MAAM,IAAA,kBAAO,EAAC,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QACzE,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC;QAE1B,MAAM,MAAM,GAAG,cAAI,CAAC,UAAU,EAAE,CAAC;QACjC,MAAM,KAAK,GAAG,YAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAc,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5E,MAAM,MAAM,GAAG,YAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAE1C,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAClC,KAAK;iBACA,IAAI,CAAC,MAAM,CAAC;iBACZ,IAAI,CAAC,MAAM,CAAC;iBACZ,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBACjB,MAAM,CAAC,GAAG,CAAC,CAAC;YAChB,CAAC,CAAC,CAAC;YACP,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE;gBACtB,OAAO,CAAC,IAAI,CAAC,CAAC;YAClB,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,mBAA8C;QACxD,MAAM,yBAAyB,GAAG,CAAC,IAAA,gDAA0B,EAAC,mBAAmB,CAAC,CAAC;QACnF,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,iBAAiB,IAAI,yBAAyB,EAAE;YACpE,MAAM,4BAA4B,GAAG,EAAE,CAAC;YACxC,MAAM,IAAI,CAAC,cAAc,CAAC,4BAA4B,CAAC,CAAC;YACxD,OAAO,IAAI,CAAC,IAAc,CAAC;SAC9B;QAED,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QAChC,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC5B,OAAO,IAAI,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,WAAmB,CAAC;QAC7C,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,IAAI,CAAC,EAAE;YACrC,MAAM,SAAS,GAAG,IAAA,gBAAS,EAAC,YAAE,CAAC,IAAI,CAAC,CAAC;YACrC,IAAI,CAAC,IAAI,GAAG,CAAC,MAAM,SAAS,CAAC,IAAI,CAAC,IAAc,CAAC,CAAC,CAAC,IAAI,GAAG,QAAQ,CAAC;SACtE;IACL,CAAC;IAED,KAAK,CAAC,OAAO;QACT,IAAI,IAAI,CAAC,UAAU,EAAE;YACjB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;SAC3B;IACL,CAAC;IAED,oBAAoB;QAChB,OAAO,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;IACrE,CAAC;CACJ;AA9ED,wCA8EC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport zlib from \"zlib\";\nimport pathlib from \"path\";\nimport fs from \"fs\";\nimport { file as tmpFile } from \"tmp-promise\";\nimport { promisify } from \"util\";\nimport { AbstractDescriptor, CompressionType, FileDescriptorBase, shouldCompressFileByExtension } from \"./descriptors\";\nimport { IngestionPropertiesInput, shouldCompressFileByFormat } from \"./ingestionProperties\";\n\n/**\n * Describes a file to be ingested. Use string to describe a local path in Node.JS and Blob object in browsers\n */\nexport class FileDescriptor extends AbstractDescriptor implements FileDescriptorBase {\n    zipped: boolean;\n    compressionType: CompressionType;\n    shouldNotCompress: boolean;\n    cleanupTmp?: () => Promise<void>;\n\n    constructor(\n        /**\n         * Use string in Node.JS and Blob in browser\n         */\n        readonly file: string | Blob,\n        sourceId: string | null = null,\n        size: number | null = null,\n        compressionType: CompressionType = CompressionType.None,\n        readonly extension?: string, // Extracted from file name by default\n        readonly name?: string // Extracted from file name by default\n    ) {\n        super(sourceId, size);\n        this.compressionType = compressionType;\n        this.name = name ? name : pathlib.basename(this.file as string);\n        this.extension = extension ? extension : pathlib.extname(this.file as string).toLowerCase();\n\n        this.zipped = compressionType !== CompressionType.None || this.extension === \".gz\" || this.extension === \".zip\";\n        this.shouldNotCompress = !shouldCompressFileByExtension(this.extension);\n    }\n\n    async _gzip(): Promise<string> {\n        const { path, cleanup } = await tmpFile({ postfix: \".gz\", keep: false });\n        this.cleanupTmp = cleanup;\n\n        const zipper = zlib.createGzip();\n        const input = fs.createReadStream(this.file as string, { autoClose: true });\n        const output = fs.createWriteStream(path);\n\n        await new Promise((resolve, reject) => {\n            input\n                .pipe(zipper)\n                .pipe(output)\n                .on(\"error\", (err) => {\n                    reject(err);\n                });\n            output.once(\"close\", () => {\n                resolve(null);\n            });\n        });\n\n        return path;\n    }\n\n    async prepare(ingestionProperties?: IngestionPropertiesInput): Promise<string> {\n        const shouldNotCompressByFormat = !shouldCompressFileByFormat(ingestionProperties);\n        if (this.zipped || this.shouldNotCompress || shouldNotCompressByFormat) {\n            const estimatedCompressionModifier = 11;\n            await this._calculateSize(estimatedCompressionModifier);\n            return this.file as string;\n        }\n\n        const path = await this._gzip();\n        await this._calculateSize();\n        return path;\n    }\n\n    private async _calculateSize(modifier: number = 1): Promise<void> {\n        if (this.size == null || this.size <= 0) {\n            const asyncStat = promisify(fs.stat);\n            this.size = (await asyncStat(this.file as string)).size * modifier;\n        }\n    }\n\n    async cleanup(): Promise<void> {\n        if (this.cleanupTmp) {\n            await this.cleanupTmp();\n        }\n    }\n\n    getCompressionSuffix() {\n        return this.compressionType ? `.${this.compressionType}` : \".gz\";\n    }\n}\n"]}

package/dist-esm/src/index.js

@@ -5,7 +5,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.IngestionPropertiesValidationError = exports.IngestionPropertiesEnums = exports.dataFormatMappingKind = exports.W3CLogFileMapping = exports.ValidationPolicy = exports.ValidationOptions = exports.ValidationImplications = exports.StreamDescriptor = exports.SStreamColumnMapping = exports.ReportMethod = exports.ReportLevel = exports.ParquetColumnMapping = exports.OrcColumnMapping = exports.JsonColumnMapping = exports.IngestionMappingKind = exports.FileDescriptor = exports.FieldTransformation = exports.DataFormat = exports.CsvColumnMapping = exports.ConstantTransformation = exports.CompressionType = exports.ColumnMapping = exports.BlobDescriptor = exports.AvroColumnMapping = exports.ApacheAvroColumnMapping = exports.IngestionDescriptors = exports.StreamingIngestClient = exports.ManagedStreamingIngestClient = exports.IngestStatusQueues = exports.IngestClient = exports.IngestionProperties = void 0;
+exports.IngestionPropertiesValidationError = exports.IngestionPropertiesEnums = exports.IngestionStatusInTableDescription = exports.OperationStatus = exports.dataFormatMappingKind = exports.W3CLogFileMapping = exports.ValidationPolicy = exports.ValidationOptions = exports.ValidationImplications = exports.StreamDescriptor = exports.SStreamColumnMapping = exports.ReportMethod = exports.ReportLevel = exports.ParquetColumnMapping = exports.OrcColumnMapping = exports.JsonColumnMapping = exports.IngestionMappingKind = exports.FileDescriptor = exports.FieldTransformation = exports.DataFormat = exports.CsvColumnMapping = exports.ConstantTransformation = exports.CompressionType = exports.ColumnMapping = exports.BlobDescriptor = exports.AvroColumnMapping = exports.ApacheAvroColumnMapping = exports.IngestionDescriptors = exports.StreamingIngestClient = exports.ManagedStreamingIngestClient = exports.IngestStatusQueues = exports.IngestClient = exports.IngestionProperties = void 0;
 const ingestClient_1 = __importDefault(require("./ingestClient"));
 exports.IngestClient = ingestClient_1.default;
 const streamingIngestClient_1 = __importDefault(require("./streamingIngestClient"));
@@ -14,6 +14,9 @@ const managedStreamingIngestClient_1 = __importDefault(require("./managedStreami
 exports.ManagedStreamingIngestClient = managedStreamingIngestClient_1.default;
 const status_1 = __importDefault(require("./status"));
 exports.IngestStatusQueues = status_1.default;
+const ingestionResult_1 = require("./ingestionResult");
+Object.defineProperty(exports, "OperationStatus", { enumerable: true, get: function () { return ingestionResult_1.OperationStatus; } });
+Object.defineProperty(exports, "IngestionStatusInTableDescription", { enumerable: true, get: function () { return ingestionResult_1.IngestionStatusInTableDescription; } });
 const ingestionProperties_1 = require("./ingestionProperties");
 Object.defineProperty(exports, "DataFormat", { enumerable: true, get: function () { return ingestionProperties_1.DataFormat; } });
 Object.defineProperty(exports, "IngestionMappingKind", { enumerable: true, get: function () { return ingestionProperties_1.IngestionMappingKind; } });

package/dist-esm/src/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;;;;AAElC,kEAA+C;AAsCtB,uBAtClB,sBAAiB,CAsCa;AApCrC,oFAA4D;AAuC/B,gCAvCtB,+BAAqB,CAuCsB;AArClD,kGAA0E;AAoCtC,uCApC7B,sCAA4B,CAoC6B;AAlChE,sDAA+C;AAiChB,6BAjCxB,gBAAuB,CAiCmB;AA/BjD,+DAS+B;AA4C3B,2FApDA,gCAAU,OAoDA;AAGV,qGAtDA,0CAAoB,OAsDA;AAIpB,4FAzDA,iCAAW,OAyDA;AACX,6FAzDA,kCAAY,OAyDA;AAGZ,uGA3DA,4CAAsB,OA2DA;AACtB,kGA3DA,uCAAiB,OA2DA;AACjB,iGA3DA,sCAAgB,OA2DA;AAEhB,sGA5DA,2CAAqB,OA4DA;AAzDzB,qDAY0B;AAuBtB,wGAlCA,wCAAuB,OAkCA;AACvB,kGAlCA,kCAAiB,OAkCA;AAEjB,8FAnCA,8BAAa,OAmCA;AAEb,uGApCA,uCAAsB,OAoCA;AACtB,iGApCA,iCAAgB,OAoCA;AAEhB,oGArCA,oCAAmB,OAqCA;AAGnB,kGAvCA,kCAAiB,OAuCA;AACjB,iGAvCA,iCAAgB,OAuCA;AAChB,qGAvCA,qCAAoB,OAuCA;AAGpB,qGAzCA,qCAAoB,OAyCA;AAKpB,kGA7CA,kCAAiB,OA6CA;AA3CrB,+CAAkF;AAwB9E,+FAxBK,4BAAc,OAwBL;AAEd,gGA1BqB,6BAAe,OA0BrB;AAaf,iGAvCsC,8BAAgB,OAuCtC;AAtCpB,qDAAkD;AA8B9C,+FA9BK,+BAAc,OA8BL;AA3BlB,6DAA4D;AAAnD,0HAAA,mBAAmB,OAAA;AAQ5B;;GAEG;AACU,QAAA,oBAAoB,GAAG;IAChC,cAAc,EAAd,4BAAc;IACd,cAAc,EAAd,+BAAc;IACd,gBAAgB,EAAhB,8BAAgB;CACnB,CAAC;AA4BF;;GAEG;AACU,QAAA,wBAAwB,GAAG;IACpC,iBAAiB,EAAjB,kCAAiB;IACjB,gBAAgB,EAAhB,iCAAgB;IAChB,iBAAiB,EAAjB,kCAAiB;IACjB,oBAAoB,EAApB,qCAAoB;IACpB,gBAAgB,EAAhB,iCAAgB;IAChB,iBAAiB,EAAjB,kCAAiB;IACjB,gBAAgB,EAAhB,sCAAgB;IAChB,WAAW,EAAX,iCAAW;IACX,YAAY,EAAZ,kCAAY;IACZ,sBAAsB,EAAtB,4CAAsB;IACtB,iBAAiB,EAAjB,uCAAiB;IACjB,UAAU,EAAV,gCAAU;IACV;;OAEG;IACH,oBAAoB,EAAE,0CAAoB;IAC1C,oBAAoB,EAApB,0CAAoB;IACpB,eAAe,EAAf,6BAAe;IACf,uBAAuB,EAAvB,wCAAuB;IACvB,oBAAoB,EAApB,qCAAoB;IACpB,sBAAsB,EAAtB,uCAAsB;IACtB,mBAAmB,EAAnB,oCAAmB;IACnB,aAAa,EAAb,8BAAa;CAChB,CAAC;AAEF,mCAA8D;AAArD,4HAAA,kCAAkC,OAAA;AAC3C,sCAAsC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport KustoIngestClient from \"./ingestClient\";\n\nimport streamingIngestClient from \"./streamingIngestClient\";\n\nimport managedStreamingIngestClient from \"./managedStreamingIngestClient\";\n\nimport KustoIngestStatusQueues from \"./status\";\n\nimport {\n    DataFormat,\n    IngestionMappingKind,\n    ReportLevel,\n    ReportMethod,\n    ValidationImplications,\n    ValidationOptions,\n    ValidationPolicy,\n    dataFormatMappingKind,\n} from \"./ingestionProperties\";\n\nimport {\n    ApacheAvroColumnMapping,\n    AvroColumnMapping,\n    ColumnMapping,\n    ConstantTransformation,\n    CsvColumnMapping,\n    FieldTransformation,\n    JsonColumnMapping,\n    OrcColumnMapping,\n    ParquetColumnMapping,\n    SStreamColumnMapping,\n    W3CLogFileMapping,\n} from \"./columnMappings\";\nimport { BlobDescriptor, CompressionType, StreamDescriptor } from \"./descriptors\";\nimport { FileDescriptor } from \"./fileDescriptor\";\n\nexport { Transformation as ColumnMappingTransformation } from \"./columnMappings\";\nexport { IngestionProperties } from \"./ingestionProperties\";\nexport {\n    KustoIngestClient as IngestClient,\n    KustoIngestStatusQueues as IngestStatusQueues,\n    managedStreamingIngestClient as ManagedStreamingIngestClient,\n    streamingIngestClient as StreamingIngestClient,\n};\n\n/**\n * @deprecated - import directly instead. Export const is not exporting type.\n */\nexport const IngestionDescriptors = {\n    BlobDescriptor,\n    FileDescriptor,\n    StreamDescriptor,\n};\n\nexport {\n    ApacheAvroColumnMapping,\n    AvroColumnMapping,\n    BlobDescriptor,\n    ColumnMapping,\n    CompressionType,\n    ConstantTransformation,\n    CsvColumnMapping,\n    DataFormat,\n    FieldTransformation,\n    FileDescriptor,\n    IngestionMappingKind,\n    JsonColumnMapping,\n    OrcColumnMapping,\n    ParquetColumnMapping,\n    ReportLevel,\n    ReportMethod,\n    SStreamColumnMapping,\n    StreamDescriptor,\n    ValidationImplications,\n    ValidationOptions,\n    ValidationPolicy,\n    W3CLogFileMapping,\n    dataFormatMappingKind,\n};\n\n/**\n * @deprecated - import directly instead\n */\nexport const IngestionPropertiesEnums = {\n    JsonColumnMapping,\n    CsvColumnMapping,\n    AvroColumnMapping,\n    ParquetColumnMapping,\n    OrcColumnMapping,\n    W3CLogFileMapping,\n    ValidationPolicy,\n    ReportLevel,\n    ReportMethod,\n    ValidationImplications,\n    ValidationOptions,\n    DataFormat,\n    /**\n     * @deprecated - use IngestionMappingKind instead\n     */\n    IngestionMappingType: IngestionMappingKind,\n    IngestionMappingKind,\n    CompressionType,\n    ApacheAvroColumnMapping,\n    SStreamColumnMapping,\n    ConstantTransformation,\n    FieldTransformation,\n    ColumnMapping,\n};\n\nexport { IngestionPropertiesValidationError } from \"./errors\";\n// eslint-disable-next-line no-console\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;;;;AAElC,kEAA+C;AAuCtB,uBAvClB,sBAAiB,CAuCa;AArCrC,oFAA4D;AAwC/B,gCAxCtB,+BAAqB,CAwCsB;AAtClD,kGAA0E;AAqCtC,uCArC7B,sCAA4B,CAqC6B;AAnChE,sDAA+C;AAkChB,6BAlCxB,gBAAuB,CAkCmB;AAhCjD,uDAAyH;AAuErH,gGAvEsB,iCAAe,OAuEtB;AAEf,kHAzEwD,mDAAiC,OAyExD;AAxErC,+DAS+B;AA4C3B,2FApDA,gCAAU,OAoDA;AAGV,qGAtDA,0CAAoB,OAsDA;AAIpB,4FAzDA,iCAAW,OAyDA;AACX,6FAzDA,kCAAY,OAyDA;AAGZ,uGA3DA,4CAAsB,OA2DA;AACtB,kGA3DA,uCAAiB,OA2DA;AACjB,iGA3DA,sCAAgB,OA2DA;AAEhB,sGA5DA,2CAAqB,OA4DA;AAzDzB,qDAY0B;AAuBtB,wGAlCA,wCAAuB,OAkCA;AACvB,kGAlCA,kCAAiB,OAkCA;AAEjB,8FAnCA,8BAAa,OAmCA;AAEb,uGApCA,uCAAsB,OAoCA;AACtB,iGApCA,iCAAgB,OAoCA;AAEhB,oGArCA,oCAAmB,OAqCA;AAGnB,kGAvCA,kCAAiB,OAuCA;AACjB,iGAvCA,iCAAgB,OAuCA;AAChB,qGAvCA,qCAAoB,OAuCA;AAGpB,qGAzCA,qCAAoB,OAyCA;AAKpB,kGA7CA,kCAAiB,OA6CA;AA3CrB,+CAAkF;AAwB9E,+FAxBK,4BAAc,OAwBL;AAEd,gGA1BqB,6BAAe,OA0BrB;AAaf,iGAvCsC,8BAAgB,OAuCtC;AAtCpB,qDAAkD;AA8B9C,+FA9BK,+BAAc,OA8BL;AA3BlB,6DAA4D;AAAnD,0HAAA,mBAAmB,OAAA;AAQ5B;;GAEG;AACU,QAAA,oBAAoB,GAAG;IAChC,cAAc,EAAd,4BAAc;IACd,cAAc,EAAd,+BAAc;IACd,gBAAgB,EAAhB,8BAAgB;CACnB,CAAC;AAgCF;;GAEG;AACU,QAAA,wBAAwB,GAAG;IACpC,iBAAiB,EAAjB,kCAAiB;IACjB,gBAAgB,EAAhB,iCAAgB;IAChB,iBAAiB,EAAjB,kCAAiB;IACjB,oBAAoB,EAApB,qCAAoB;IACpB,gBAAgB,EAAhB,iCAAgB;IAChB,iBAAiB,EAAjB,kCAAiB;IACjB,gBAAgB,EAAhB,sCAAgB;IAChB,WAAW,EAAX,iCAAW;IACX,YAAY,EAAZ,kCAAY;IACZ,sBAAsB,EAAtB,4CAAsB;IACtB,iBAAiB,EAAjB,uCAAiB;IACjB,UAAU,EAAV,gCAAU;IACV;;OAEG;IACH,oBAAoB,EAAE,0CAAoB;IAC1C,oBAAoB,EAApB,0CAAoB;IACpB,eAAe,EAAf,6BAAe;IACf,uBAAuB,EAAvB,wCAAuB;IACvB,oBAAoB,EAApB,qCAAoB;IACpB,sBAAsB,EAAtB,uCAAsB;IACtB,mBAAmB,EAAnB,oCAAmB;IACnB,aAAa,EAAb,8BAAa;CAChB,CAAC;AAEF,mCAA8D;AAArD,4HAAA,kCAAkC,OAAA;AAC3C,sCAAsC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport KustoIngestClient from \"./ingestClient\";\n\nimport streamingIngestClient from \"./streamingIngestClient\";\n\nimport managedStreamingIngestClient from \"./managedStreamingIngestClient\";\n\nimport KustoIngestStatusQueues from \"./status\";\n\nimport { IngestionResult, OperationStatus, IngestionStatus, IngestionStatusInTableDescription } from \"./ingestionResult\";\nimport {\n    DataFormat,\n    IngestionMappingKind,\n    ReportLevel,\n    ReportMethod,\n    ValidationImplications,\n    ValidationOptions,\n    ValidationPolicy,\n    dataFormatMappingKind,\n} from \"./ingestionProperties\";\n\nimport {\n    ApacheAvroColumnMapping,\n    AvroColumnMapping,\n    ColumnMapping,\n    ConstantTransformation,\n    CsvColumnMapping,\n    FieldTransformation,\n    JsonColumnMapping,\n    OrcColumnMapping,\n    ParquetColumnMapping,\n    SStreamColumnMapping,\n    W3CLogFileMapping,\n} from \"./columnMappings\";\nimport { BlobDescriptor, CompressionType, StreamDescriptor } from \"./descriptors\";\nimport { FileDescriptor } from \"./fileDescriptor\";\n\nexport { Transformation as ColumnMappingTransformation } from \"./columnMappings\";\nexport { IngestionProperties } from \"./ingestionProperties\";\nexport {\n    KustoIngestClient as IngestClient,\n    KustoIngestStatusQueues as IngestStatusQueues,\n    managedStreamingIngestClient as ManagedStreamingIngestClient,\n    streamingIngestClient as StreamingIngestClient,\n};\n\n/**\n * @deprecated - import directly instead. Export const is not exporting type.\n */\nexport const IngestionDescriptors = {\n    BlobDescriptor,\n    FileDescriptor,\n    StreamDescriptor,\n};\n\nexport {\n    ApacheAvroColumnMapping,\n    AvroColumnMapping,\n    BlobDescriptor,\n    ColumnMapping,\n    CompressionType,\n    ConstantTransformation,\n    CsvColumnMapping,\n    DataFormat,\n    FieldTransformation,\n    FileDescriptor,\n    IngestionMappingKind,\n    JsonColumnMapping,\n    OrcColumnMapping,\n    ParquetColumnMapping,\n    ReportLevel,\n    ReportMethod,\n    SStreamColumnMapping,\n    StreamDescriptor,\n    ValidationImplications,\n    ValidationOptions,\n    ValidationPolicy,\n    W3CLogFileMapping,\n    dataFormatMappingKind,\n    IngestionResult,\n    OperationStatus,\n    IngestionStatus,\n    IngestionStatusInTableDescription,\n};\n\n/**\n * @deprecated - import directly instead\n */\nexport const IngestionPropertiesEnums = {\n    JsonColumnMapping,\n    CsvColumnMapping,\n    AvroColumnMapping,\n    ParquetColumnMapping,\n    OrcColumnMapping,\n    W3CLogFileMapping,\n    ValidationPolicy,\n    ReportLevel,\n    ReportMethod,\n    ValidationImplications,\n    ValidationOptions,\n    DataFormat,\n    /**\n     * @deprecated - use IngestionMappingKind instead\n     */\n    IngestionMappingType: IngestionMappingKind,\n    IngestionMappingKind,\n    CompressionType,\n    ApacheAvroColumnMapping,\n    SStreamColumnMapping,\n    ConstantTransformation,\n    FieldTransformation,\n    ColumnMapping,\n};\n\nexport { IngestionPropertiesValidationError } from \"./errors\";\n// eslint-disable-next-line no-console\n"]}

package/dist-esm/src/ingestClient.browser.js

@@ -7,8 +7,8 @@ const descriptors_1 = require("./descriptors");
 const fileDescriptor_browser_1 = require("./fileDescriptor.browser");
 const ingestClientBase_1 = require("./ingestClientBase");
 class KustoIngestClient extends ingestClientBase_1.KustoIngestClientBase {
-    constructor(kcsb, defaultProps) {
-        super(kcsb, defaultProps, true);
+    constructor(kcsb, defaultProps, autoCorrectEndpoint) {
+        super(kcsb, defaultProps, autoCorrectEndpoint, true);
     }
     /**
      * Use string for Node.js and Blob in browser
@@ -18,12 +18,10 @@ class KustoIngestClient extends ingestClientBase_1.KustoIngestClientBase {
         const descriptor = file instanceof fileDescriptor_browser_1.FileDescriptor ? file : new fileDescriptor_browser_1.FileDescriptor(file);
         const blob = descriptor.file;
         const props = this._getMergedProps(ingestionProperties);
-        const [fileToUpload, blockBlobClient] = await Promise.all([
-            descriptor.prepare(),
-            this.resourceManager.getBlockBlobClient((0, descriptors_1.generateBlobName)(descriptor, props)),
-        ]);
-        await blockBlobClient.uploadData(fileToUpload);
-        return this.ingestFromBlob(new descriptors_1.BlobDescriptor(blockBlobClient.url, blob.size, descriptor.sourceId), props);
+        const fileToUpload = await descriptor.prepare(ingestionProperties);
+        const blobName = (0, descriptors_1.generateBlobName)(descriptor, props);
+        const blobUri = await this.uploadToBlobWithRetry(fileToUpload, blobName);
+        return this.ingestFromBlob(new descriptors_1.BlobDescriptor(blobUri, blob.size, descriptor.sourceId), props);
     }
     /**
      * Use Readable for Node.js and ArrayBuffer in browser
@@ -33,9 +31,8 @@ class KustoIngestClient extends ingestClientBase_1.KustoIngestClientBase {
         const props = this._getMergedProps(ingestionProperties);
         const descriptor = stream instanceof descriptors_1.StreamDescriptor ? stream : new descriptors_1.StreamDescriptor(stream);
         const blobName = (0, descriptors_1.generateBlobName)(descriptor, props);
-        const blockBlobClient = await this.resourceManager.getBlockBlobClient(blobName);
-        await blockBlobClient.uploadData(descriptor.stream);
-        return this.ingestFromBlob(new descriptors_1.BlobDescriptor(blockBlobClient.url), props); // descriptor.size?
+        const blobUri = await this.uploadToBlobWithRetry(descriptor.stream, blobName);
+        return this.ingestFromBlob(new descriptors_1.BlobDescriptor(blobUri, descriptor.size, descriptor.sourceId), props);
     }
 }
 exports.KustoIngestClient = KustoIngestClient;

package/dist-esm/src/ingestClient.browser.js.map

@@ -1 +1 @@
-{"version":3,"file":"ingestClient.browser.js","sourceRoot":"","sources":["../../src/ingestClient.browser.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAIlC,+CAAmF;AACnF,qEAA0D;AAI1D,yDAA2D;AAE3D,MAAa,iBAAkB,SAAQ,wCAAqB;IACxD,YAAY,IAA2C,EAAE,YAAuC;QAC5F,KAAK,CAAC,IAAI,EAAE,YAAY,EAAE,IAAI,CAAC,CAAC;IACpC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,IAA2B,EAAE,mBAA8C;QAC5F,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,UAAU,GAAG,IAAI,YAAY,uCAAc,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,uCAAc,CAAC,IAAI,CAAC,CAAC;QAEpF,MAAM,IAAI,GAAG,UAAU,CAAC,IAAY,CAAC;QACrC,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QAExD,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACtD,UAAU,CAAC,OAAO,EAAE;YACpB,IAAI,CAAC,eAAe,CAAC,kBAAkB,CAAC,IAAA,8BAAgB,EAAC,UAAU,EAAE,KAAK,CAAC,CAAC;SAC/E,CAAC,CAAC;QAEH,MAAM,eAAe,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,4BAAc,CAAC,eAAe,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,EAAE,UAAU,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC;IAC/G,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB,CAAC,MAAsC,EAAE,mBAA8C;QACzG,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QACxD,MAAM,UAAU,GAAqB,MAAM,YAAY,8BAAgB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,8BAAgB,CAAC,MAAM,CAAC,CAAC;QAChH,MAAM,QAAQ,GAAG,IAAA,8BAAgB,EAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAErD,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAChF,MAAM,eAAe,CAAC,UAAU,CAAC,UAAU,CAAC,MAAqB,CAAC,CAAC;QACnE,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,4BAAc,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,mBAAmB;IACnG,CAAC;CACJ;AArCD,8CAqCC;AAED,kBAAe,iBAAiB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { KustoConnectionStringBuilder } from \"azure-kusto-data\";\n\nimport { BlobDescriptor, generateBlobName, StreamDescriptor } from \"./descriptors\";\nimport { FileDescriptor } from \"./fileDescriptor.browser\";\nimport { QueueSendMessageResponse } from \"@azure/storage-queue\";\n\nimport { IngestionPropertiesInput } from \"./ingestionProperties\";\nimport { KustoIngestClientBase } from \"./ingestClientBase\";\n\nexport class KustoIngestClient extends KustoIngestClientBase {\n    constructor(kcsb: string | KustoConnectionStringBuilder, defaultProps?: IngestionPropertiesInput) {\n        super(kcsb, defaultProps, true);\n    }\n\n    /**\n     * Use string for Node.js and Blob in browser\n     */\n    async ingestFromFile(file: Blob | FileDescriptor, ingestionProperties?: IngestionPropertiesInput): Promise<QueueSendMessageResponse> {\n        this.ensureOpen();\n        const descriptor = file instanceof FileDescriptor ? file : new FileDescriptor(file);\n\n        const blob = descriptor.file as Blob;\n        const props = this._getMergedProps(ingestionProperties);\n\n        const [fileToUpload, blockBlobClient] = await Promise.all([\n            descriptor.prepare(),\n            this.resourceManager.getBlockBlobClient(generateBlobName(descriptor, props)),\n        ]);\n\n        await blockBlobClient.uploadData(fileToUpload);\n        return this.ingestFromBlob(new BlobDescriptor(blockBlobClient.url, blob.size, descriptor.sourceId), props);\n    }\n\n    /**\n     * Use Readable for Node.js and ArrayBuffer in browser\n     */\n    async ingestFromStream(stream: ArrayBuffer | StreamDescriptor, ingestionProperties?: IngestionPropertiesInput): Promise<QueueSendMessageResponse> {\n        this.ensureOpen();\n        const props = this._getMergedProps(ingestionProperties);\n        const descriptor: StreamDescriptor = stream instanceof StreamDescriptor ? stream : new StreamDescriptor(stream);\n        const blobName = generateBlobName(descriptor, props);\n\n        const blockBlobClient = await this.resourceManager.getBlockBlobClient(blobName);\n        await blockBlobClient.uploadData(descriptor.stream as ArrayBuffer);\n        return this.ingestFromBlob(new BlobDescriptor(blockBlobClient.url), props); // descriptor.size?\n    }\n}\n\nexport default KustoIngestClient;\n"]}
+{"version":3,"file":"ingestClient.browser.js","sourceRoot":"","sources":["../../src/ingestClient.browser.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAIlC,+CAAmF;AACnF,qEAA0D;AAG1D,yDAA2D;AAG3D,MAAa,iBAAkB,SAAQ,wCAAqB;IACxD,YAAY,IAA2C,EAAE,YAAuC,EAAE,mBAA6B;QAC3H,KAAK,CAAC,IAAI,EAAE,YAAY,EAAE,mBAAmB,EAAE,IAAI,CAAC,CAAC;IACzD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,IAA2B,EAAE,mBAA8C;QAC5F,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,UAAU,GAAG,IAAI,YAAY,uCAAc,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,uCAAc,CAAC,IAAI,CAAC,CAAC;QAEpF,MAAM,IAAI,GAAG,UAAU,CAAC,IAAY,CAAC;QACrC,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QAExD,MAAM,YAAY,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;QACnE,MAAM,QAAQ,GAAG,IAAA,8BAAgB,EAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAErD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAEzE,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,4BAAc,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,EAAE,UAAU,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC;IACnG,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB,CAAC,MAAsC,EAAE,mBAA8C;QACzG,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QACxD,MAAM,UAAU,GAAqB,MAAM,YAAY,8BAAgB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,8BAAgB,CAAC,MAAM,CAAC,CAAC;QAChH,MAAM,QAAQ,GAAG,IAAA,8BAAgB,EAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QACrD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,UAAU,CAAC,MAAqB,EAAE,QAAQ,CAAC,CAAC;QAC7F,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,4BAAc,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC;IACzG,CAAC;CACJ;AAlCD,8CAkCC;AAED,kBAAe,iBAAiB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { KustoConnectionStringBuilder } from \"azure-kusto-data\";\n\nimport { BlobDescriptor, generateBlobName, StreamDescriptor } from \"./descriptors\";\nimport { FileDescriptor } from \"./fileDescriptor.browser\";\n\nimport { IngestionPropertiesInput } from \"./ingestionProperties\";\nimport { KustoIngestClientBase } from \"./ingestClientBase\";\nimport { IngestionResult } from \"./ingestionResult\";\n\nexport class KustoIngestClient extends KustoIngestClientBase {\n    constructor(kcsb: string | KustoConnectionStringBuilder, defaultProps?: IngestionPropertiesInput, autoCorrectEndpoint?: boolean) {\n        super(kcsb, defaultProps, autoCorrectEndpoint, true);\n    }\n\n    /**\n     * Use string for Node.js and Blob in browser\n     */\n    async ingestFromFile(file: Blob | FileDescriptor, ingestionProperties?: IngestionPropertiesInput): Promise<IngestionResult> {\n        this.ensureOpen();\n        const descriptor = file instanceof FileDescriptor ? file : new FileDescriptor(file);\n\n        const blob = descriptor.file as Blob;\n        const props = this._getMergedProps(ingestionProperties);\n\n        const fileToUpload = await descriptor.prepare(ingestionProperties);\n        const blobName = generateBlobName(descriptor, props);\n\n        const blobUri = await this.uploadToBlobWithRetry(fileToUpload, blobName);\n\n        return this.ingestFromBlob(new BlobDescriptor(blobUri, blob.size, descriptor.sourceId), props);\n    }\n\n    /**\n     * Use Readable for Node.js and ArrayBuffer in browser\n     */\n    async ingestFromStream(stream: ArrayBuffer | StreamDescriptor, ingestionProperties?: IngestionPropertiesInput): Promise<IngestionResult> {\n        this.ensureOpen();\n        const props = this._getMergedProps(ingestionProperties);\n        const descriptor: StreamDescriptor = stream instanceof StreamDescriptor ? stream : new StreamDescriptor(stream);\n        const blobName = generateBlobName(descriptor, props);\n        const blobUri = await this.uploadToBlobWithRetry(descriptor.stream as ArrayBuffer, blobName);\n        return this.ingestFromBlob(new BlobDescriptor(blobUri, descriptor.size, descriptor.sourceId), props);\n    }\n}\n\nexport default KustoIngestClient;\n"]}

package/dist-esm/src/ingestClient.js

@@ -7,8 +7,8 @@ const descriptors_1 = require("./descriptors");
 const fileDescriptor_1 = require("./fileDescriptor");
 const ingestClientBase_1 = require("./ingestClientBase");
 class KustoIngestClient extends ingestClientBase_1.KustoIngestClientBase {
-    constructor(kcsb, defaultProps) {
-        super(kcsb, defaultProps);
+    constructor(kcsb, defaultProps, autoCorrectEndpoint) {
+        super(kcsb, defaultProps, autoCorrectEndpoint);
     }
     /**
      * Use string in Node.JS and Blob in browser
@@ -19,9 +19,9 @@ class KustoIngestClient extends ingestClientBase_1.KustoIngestClientBase {
         const descriptor = file instanceof fileDescriptor_1.FileDescriptor ? file : new fileDescriptor_1.FileDescriptor(file);
         try {
             const blobName = (0, descriptors_1.generateBlobName)(descriptor, props);
-            const [fileToUpload, blockBlobClient] = await Promise.all([descriptor.prepare(), this.resourceManager.getBlockBlobClient(blobName)]);
-            await blockBlobClient.uploadFile(fileToUpload);
-            return this.ingestFromBlob(new descriptors_1.BlobDescriptor(blockBlobClient.url, descriptor.size, descriptor.sourceId), props);
+            const fileToUpload = await descriptor.prepare(ingestionProperties);
+            const blobUri = await this.uploadToBlobWithRetry(fileToUpload, blobName);
+            return this.ingestFromBlob(new descriptors_1.BlobDescriptor(blobUri, descriptor.size, descriptor.sourceId), props);
         }
         finally {
             await descriptor.cleanup();
@@ -34,15 +34,10 @@ class KustoIngestClient extends ingestClientBase_1.KustoIngestClientBase {
         this.ensureOpen();
         const props = this._getMergedProps(ingestionProperties);
         const descriptor = stream instanceof descriptors_1.StreamDescriptor ? stream : new descriptors_1.StreamDescriptor(stream);
+        // TODO we don't gzip on stream here, just ont streamingIngestClient
         const blobName = (0, descriptors_1.generateBlobName)(descriptor, props);
-        const blockBlobClient = await this.resourceManager.getBlockBlobClient(blobName);
-        if (descriptor.stream instanceof Buffer) {
-            await blockBlobClient.uploadData(descriptor.stream);
-        }
-        else {
-            await blockBlobClient.uploadStream(descriptor.stream);
-        }
-        return this.ingestFromBlob(new descriptors_1.BlobDescriptor(blockBlobClient.url), props); // descriptor.size?
+        const blobUri = await this.uploadToBlobWithRetry(descriptor, blobName);
+        return this.ingestFromBlob(new descriptors_1.BlobDescriptor(blobUri), props); // descriptor.size?
     }
 }
 exports.KustoIngestClient = KustoIngestClient;

package/dist-esm/src/ingestClient.js.map

@@ -1 +1 @@
-{"version":3,"file":"ingestClient.js","sourceRoot":"","sources":["../../src/ingestClient.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAIlC,+CAAmF;AACnF,qDAAkD;AAKlD,yDAA2D;AAG3D,MAAa,iBAAkB,SAAQ,wCAAqB;IACxD,YAAY,IAA2C,EAAE,YAAuC;QAC5F,KAAK,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;IAC9B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,IAAoC,EAAE,mBAA8C;QACrG,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QAExD,MAAM,UAAU,GAAG,IAAI,YAAY,+BAAc,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,+BAAc,CAAC,IAAI,CAAC,CAAC;QAEpF,IAAI;YACA,MAAM,QAAQ,GAAG,IAAA,8BAAgB,EAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YACrD,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,IAAI,CAAC,eAAe,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;YACrI,MAAM,eAAe,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;YAC/C,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,4BAAc,CAAC,eAAe,CAAC,GAAG,EAAE,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC;SACpH;gBAAS;YACN,MAAM,UAAU,CAAC,OAAO,EAAE,CAAC;SAC9B;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB,CAClB,MAAiD,EACjD,mBAA8C;QAE9C,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QACxD,MAAM,UAAU,GAAqB,MAAM,YAAY,8BAAgB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,8BAAgB,CAAC,MAAM,CAAC,CAAC;QAEhH,MAAM,QAAQ,GAAG,IAAA,8BAAgB,EAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAErD,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAChF,IAAI,UAAU,CAAC,MAAM,YAAY,MAAM,EAAE;YACrC,MAAM,eAAe,CAAC,UAAU,CAAC,UAAU,CAAC,MAAgB,CAAC,CAAC;SACjE;aAAM;YACH,MAAM,eAAe,CAAC,YAAY,CAAC,UAAU,CAAC,MAAkB,CAAC,CAAC;SACrE;QAED,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,4BAAc,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,mBAAmB;IACnG,CAAC;CACJ;AA9CD,8CA8CC;AAED,kBAAe,iBAAiB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { KustoConnectionStringBuilder } from \"azure-kusto-data\";\n\nimport { BlobDescriptor, generateBlobName, StreamDescriptor } from \"./descriptors\";\nimport { FileDescriptor } from \"./fileDescriptor\";\n\nimport { QueueSendMessageResponse } from \"@azure/storage-queue\";\n\nimport { IngestionPropertiesInput } from \"./ingestionProperties\";\nimport { KustoIngestClientBase } from \"./ingestClientBase\";\nimport { Readable } from \"stream\";\n\nexport class KustoIngestClient extends KustoIngestClientBase {\n    constructor(kcsb: string | KustoConnectionStringBuilder, defaultProps?: IngestionPropertiesInput) {\n        super(kcsb, defaultProps);\n    }\n\n    /**\n     * Use string in Node.JS and Blob in browser\n     */\n    async ingestFromFile(file: FileDescriptor | string | Blob, ingestionProperties?: IngestionPropertiesInput): Promise<QueueSendMessageResponse> {\n        this.ensureOpen();\n        const props = this._getMergedProps(ingestionProperties);\n\n        const descriptor = file instanceof FileDescriptor ? file : new FileDescriptor(file);\n\n        try {\n            const blobName = generateBlobName(descriptor, props);\n            const [fileToUpload, blockBlobClient] = await Promise.all([descriptor.prepare(), this.resourceManager.getBlockBlobClient(blobName)]);\n            await blockBlobClient.uploadFile(fileToUpload);\n            return this.ingestFromBlob(new BlobDescriptor(blockBlobClient.url, descriptor.size, descriptor.sourceId), props);\n        } finally {\n            await descriptor.cleanup();\n        }\n    }\n\n    /**\n     * Use Readable in Node.JS and ArrayBuffer in browser\n     */\n    async ingestFromStream(\n        stream: StreamDescriptor | Readable | ArrayBuffer,\n        ingestionProperties?: IngestionPropertiesInput\n    ): Promise<QueueSendMessageResponse> {\n        this.ensureOpen();\n        const props = this._getMergedProps(ingestionProperties);\n        const descriptor: StreamDescriptor = stream instanceof StreamDescriptor ? stream : new StreamDescriptor(stream);\n\n        const blobName = generateBlobName(descriptor, props);\n\n        const blockBlobClient = await this.resourceManager.getBlockBlobClient(blobName);\n        if (descriptor.stream instanceof Buffer) {\n            await blockBlobClient.uploadData(descriptor.stream as Buffer);\n        } else {\n            await blockBlobClient.uploadStream(descriptor.stream as Readable);\n        }\n\n        return this.ingestFromBlob(new BlobDescriptor(blockBlobClient.url), props); // descriptor.size?\n    }\n}\n\nexport default KustoIngestClient;\n"]}
+{"version":3,"file":"ingestClient.js","sourceRoot":"","sources":["../../src/ingestClient.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAIlC,+CAAmF;AACnF,qDAAkD;AAGlD,yDAA2D;AAI3D,MAAa,iBAAkB,SAAQ,wCAAqB;IACxD,YAAY,IAA2C,EAAE,YAAuC,EAAE,mBAA6B;QAC3H,KAAK,CAAC,IAAI,EAAE,YAAY,EAAE,mBAAmB,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,IAAoC,EAAE,mBAA8C;QACrG,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QAExD,MAAM,UAAU,GAAG,IAAI,YAAY,+BAAc,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,+BAAc,CAAC,IAAI,CAAC,CAAC;QAEpF,IAAI;YACA,MAAM,QAAQ,GAAG,IAAA,8BAAgB,EAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YACrD,MAAM,YAAY,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;YACnE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;YACzE,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,4BAAc,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC;SACxG;gBAAS;YACN,MAAM,UAAU,CAAC,OAAO,EAAE,CAAC;SAC9B;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB,CAAC,MAAiD,EAAE,mBAA8C;QACpH,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QACxD,MAAM,UAAU,GAAqB,MAAM,YAAY,8BAAgB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,8BAAgB,CAAC,MAAM,CAAC,CAAC;QAChH,oEAAoE;QACpE,MAAM,QAAQ,GAAG,IAAA,8BAAgB,EAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAErD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAEvE,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,4BAAc,CAAC,OAAO,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,mBAAmB;IACvF,CAAC;CACJ;AAtCD,8CAsCC;AAED,kBAAe,iBAAiB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { KustoConnectionStringBuilder } from \"azure-kusto-data\";\n\nimport { BlobDescriptor, generateBlobName, StreamDescriptor } from \"./descriptors\";\nimport { FileDescriptor } from \"./fileDescriptor\";\n\nimport { IngestionPropertiesInput } from \"./ingestionProperties\";\nimport { KustoIngestClientBase } from \"./ingestClientBase\";\nimport { Readable } from \"stream\";\nimport { IngestionResult } from \"./ingestionResult\";\n\nexport class KustoIngestClient extends KustoIngestClientBase {\n    constructor(kcsb: string | KustoConnectionStringBuilder, defaultProps?: IngestionPropertiesInput, autoCorrectEndpoint?: boolean) {\n        super(kcsb, defaultProps, autoCorrectEndpoint);\n    }\n\n    /**\n     * Use string in Node.JS and Blob in browser\n     */\n    async ingestFromFile(file: FileDescriptor | string | Blob, ingestionProperties?: IngestionPropertiesInput): Promise<IngestionResult> {\n        this.ensureOpen();\n        const props = this._getMergedProps(ingestionProperties);\n\n        const descriptor = file instanceof FileDescriptor ? file : new FileDescriptor(file);\n\n        try {\n            const blobName = generateBlobName(descriptor, props);\n            const fileToUpload = await descriptor.prepare(ingestionProperties);\n            const blobUri = await this.uploadToBlobWithRetry(fileToUpload, blobName);\n            return this.ingestFromBlob(new BlobDescriptor(blobUri, descriptor.size, descriptor.sourceId), props);\n        } finally {\n            await descriptor.cleanup();\n        }\n    }\n\n    /**\n     * Use Readable in Node.JS and ArrayBuffer in browser\n     */\n    async ingestFromStream(stream: StreamDescriptor | Readable | ArrayBuffer, ingestionProperties?: IngestionPropertiesInput): Promise<IngestionResult> {\n        this.ensureOpen();\n        const props = this._getMergedProps(ingestionProperties);\n        const descriptor: StreamDescriptor = stream instanceof StreamDescriptor ? stream : new StreamDescriptor(stream);\n        // TODO we don't gzip on stream here, just ont streamingIngestClient\n        const blobName = generateBlobName(descriptor, props);\n\n        const blobUri = await this.uploadToBlobWithRetry(descriptor, blobName);\n\n        return this.ingestFromBlob(new BlobDescriptor(blobUri), props); // descriptor.size?\n    }\n}\n\nexport default KustoIngestClient;\n"]}

package/dist-esm/src/ingestClientBase.js

@@ -7,33 +7,116 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.KustoIngestClientBase = void 0;
 const azure_kusto_data_1 = require("azure-kusto-data");
-const descriptors_1 = require("./descriptors");
 const resourceManager_1 = __importDefault(require("./resourceManager"));
 const ingestionBlobInfo_1 = __importDefault(require("./ingestionBlobInfo"));
+const storage_blob_1 = require("@azure/storage-blob");
 const storage_queue_1 = require("@azure/storage-queue");
+const ingestionProperties_1 = require("./ingestionProperties");
 const abstractKustoClient_1 = require("./abstractKustoClient");
+const ingestionResult_1 = require("./ingestionResult");
+const descriptors_1 = require("./descriptors");
 class KustoIngestClientBase extends abstractKustoClient_1.AbstractKustoClient {
-    constructor(kcsb, defaultProps, isBrowser) {
+    constructor(kcsb, defaultProps, autoCorrectEndpoint = true, isBrowser) {
         super(defaultProps);
+        if (typeof kcsb === "string") {
+            kcsb = new azure_kusto_data_1.KustoConnectionStringBuilder(kcsb);
+        }
+        if (autoCorrectEndpoint) {
+            kcsb.dataSource = this.getIngestionEndpoint(kcsb.dataSource);
+        }
         const kustoClient = new azure_kusto_data_1.Client(kcsb);
         this.resourceManager = new resourceManager_1.default(kustoClient, isBrowser);
         this.defaultDatabase = kustoClient.defaultDatabase;
+        const clientDetails = kcsb.clientDetails();
+        this.applicationForTracing = clientDetails.applicationNameForTracing;
+        this.clientVersionForTracing = clientDetails.versionForTracing;
     }
-    async ingestFromBlob(blob, ingestionProperties) {
+    async ingestFromBlob(blob, ingestionProperties, maxRetries = KustoIngestClientBase.MaxNumberOfRetryAttempts) {
         this.ensureOpen();
         const props = this._getMergedProps(ingestionProperties);
         const descriptor = blob instanceof descriptors_1.BlobDescriptor ? blob : new descriptors_1.BlobDescriptor(blob);
+        const authorizationContext = await this.resourceManager.getAuthorizationContext();
+        const ingestionBlobInfo = new ingestionBlobInfo_1.default(descriptor, props, authorizationContext, this.applicationForTracing, this.clientVersionForTracing);
+        const reportToTable = props.reportLevel !== ingestionProperties_1.ReportLevel.DoNotReport && props.reportMethod !== ingestionProperties_1.ReportMethod.Queue;
+        if (reportToTable) {
+            const statusTableClient = await this.resourceManager.createStatusTable();
+            const status = this.createStatusObject(props, ingestionResult_1.OperationStatus.Pending, ingestionBlobInfo);
+            await (0, ingestionResult_1.putRecordInTable)(statusTableClient, Object.assign(Object.assign({}, status), { partitionKey: ingestionBlobInfo.Id, rowKey: ingestionBlobInfo.Id }));
+            const desc = new ingestionResult_1.IngestionStatusInTableDescription(statusTableClient.url, ingestionBlobInfo.Id, ingestionBlobInfo.Id);
+            ingestionBlobInfo.IngestionStatusInTable = desc;
+            await this.sendQueueMessage(maxRetries, ingestionBlobInfo);
+            return new ingestionResult_1.TableReportIngestionResult(desc, statusTableClient);
+        }
+        await this.sendQueueMessage(maxRetries, ingestionBlobInfo);
+        return new ingestionResult_1.IngestionStatusResult(this.createStatusObject(props, ingestionResult_1.OperationStatus.Queued, ingestionBlobInfo));
+    }
+    async sendQueueMessage(maxRetries, blobInfo) {
         const queues = await this.resourceManager.getIngestionQueues();
         if (queues == null) {
             throw new Error("Failed to get queues");
         }
-        const authorizationContext = await this.resourceManager.getAuthorizationContext();
-        const queueDetails = queues[Math.floor(Math.random() * queues.length)];
-        const queueClient = new storage_queue_1.QueueClient(queueDetails.uri);
-        const ingestionBlobInfo = new ingestionBlobInfo_1.default(descriptor, props, authorizationContext);
-        const ingestionBlobInfoJson = JSON.stringify(ingestionBlobInfo);
+        const ingestionBlobInfoJson = JSON.stringify(blobInfo);
         const encoded = Buffer.from(ingestionBlobInfoJson).toString("base64");
-        return queueClient.sendMessage(encoded);
+        const retryCount = Math.min(maxRetries, queues.length);
+        for (let i = 0; i < retryCount; i++) {
+            const queueClient = new storage_queue_1.QueueClient(queues[i].uri);
+            try {
+                const queueResponse = await queueClient.sendMessage(encoded);
+                this.resourceManager.reportResourceUsageResult(queueClient.accountName, true);
+                return queueResponse;
+            }
+            catch (_) {
+                this.resourceManager.reportResourceUsageResult(queueClient.accountName, false);
+            }
+        }
+        throw new Error("Failed to send message to queue.");
+    }
+    createStatusObject(props, status, ingestionBlobInfo) {
+        const time = Date.now().toString();
+        return {
+            Status: status,
+            Timestamp: time,
+            IngestionSourceId: ingestionBlobInfo.Id,
+            IngestionSourcePath: ingestionBlobInfo.BlobPath.split(/[?;]/)[0],
+            Database: props.database,
+            Table: props.table,
+            UpdatedOn: time,
+            Details: "",
+        };
+    }
+    async uploadToBlobWithRetry(descriptor, blobName, maxRetries = KustoIngestClientBase.MaxNumberOfRetryAttempts) {
+        const containers = await this.resourceManager.getContainers();
+        if (containers == null || containers.length === 0) {
+            throw new Error("Failed to get containers");
+        }
+        const retryCount = Math.min(maxRetries, containers.length);
+        // Go over all containers and try to upload the file to the first one that succeeds
+        for (let i = 0; i < retryCount; i++) {
+            const containerClient = new storage_blob_1.ContainerClient(containers[i].uri);
+            const blockBlobClient = containerClient.getBlockBlobClient(blobName);
+            try {
+                if (typeof descriptor == "string") {
+                    await blockBlobClient.uploadFile(descriptor);
+                }
+                else if (descriptor instanceof descriptors_1.StreamDescriptor) {
+                    if (descriptor.stream instanceof Buffer) {
+                        await blockBlobClient.uploadData(descriptor.stream);
+                    }
+                    else {
+                        await blockBlobClient.uploadStream(descriptor.stream);
+                    }
+                }
+                else {
+                    await blockBlobClient.uploadData(descriptor);
+                }
+                this.resourceManager.reportResourceUsageResult(containerClient.accountName, true);
+                return blockBlobClient.url;
+            }
+            catch (ex) {
+                this.resourceManager.reportResourceUsageResult(containerClient.accountName, false);
+            }
+        }
+        throw new Error("Failed to upload to blob.");
     }
     close() {
         if (!this._isClosed) {
@@ -43,5 +126,6 @@ class KustoIngestClientBase extends abstractKustoClient_1.AbstractKustoClient {
     }
 }
 exports.KustoIngestClientBase = KustoIngestClientBase;
+KustoIngestClientBase.MaxNumberOfRetryAttempts = 3;
 exports.default = KustoIngestClientBase;
 //# sourceMappingURL=ingestClientBase.js.map

package/dist-esm/src/ingestClientBase.js.map

@@ -1 +1 @@
-{"version":3,"file":"ingestClientBase.js","sourceRoot":"","sources":["../../src/ingestClientBase.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;;;;AAElC,uDAAuF;AAEvF,+CAA+C;AAE/C,wEAAgD;AAEhD,4EAAoD;AAEpD,wDAA6E;AAG7E,+DAA4D;AAE5D,MAAsB,qBAAsB,SAAQ,yCAAmB;IAGnE,YAAY,IAA2C,EAAE,YAAuC,EAAE,SAAmB;QACjH,KAAK,CAAC,YAAY,CAAC,CAAC;QACpB,MAAM,WAAW,GAAG,IAAI,yBAAW,CAAC,IAAI,CAAC,CAAC;QAC1C,IAAI,CAAC,eAAe,GAAG,IAAI,yBAAe,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;QACnE,IAAI,CAAC,eAAe,GAAG,WAAW,CAAC,eAAe,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,IAA6B,EAAE,mBAA8C;QAC9F,IAAI,CAAC,UAAU,EAAE,CAAC;QAElB,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QAExD,MAAM,UAAU,GAAG,IAAI,YAAY,4BAAc,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,4BAAc,CAAC,IAAI,CAAC,CAAC;QACpF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,kBAAkB,EAAE,CAAC;QAC/D,IAAI,MAAM,IAAI,IAAI,EAAE;YAChB,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;SAC3C;QAED,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,uBAAuB,EAAE,CAAC;QAElF,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QAEvE,MAAM,WAAW,GAAG,IAAI,2BAAW,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;QAEtD,MAAM,iBAAiB,GAAG,IAAI,2BAAiB,CAAC,UAAU,EAAE,KAAK,EAAE,oBAAoB,CAAC,CAAC;QACzF,MAAM,qBAAqB,GAAG,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;QAChE,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAEtE,OAAO,WAAW,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK;QACD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;YACjB,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;SAChC;QACD,KAAK,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC;CACJ;AAxCD,sDAwCC;AAED,kBAAe,qBAAqB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { Client as KustoClient, KustoConnectionStringBuilder } from \"azure-kusto-data\";\n\nimport { BlobDescriptor } from \"./descriptors\";\n\nimport ResourceManager from \"./resourceManager\";\n\nimport IngestionBlobInfo from \"./ingestionBlobInfo\";\n\nimport { QueueClient, QueueSendMessageResponse } from \"@azure/storage-queue\";\n\nimport { IngestionPropertiesInput } from \"./ingestionProperties\";\nimport { AbstractKustoClient } from \"./abstractKustoClient\";\n\nexport abstract class KustoIngestClientBase extends AbstractKustoClient {\n    resourceManager: ResourceManager;\n\n    constructor(kcsb: string | KustoConnectionStringBuilder, defaultProps?: IngestionPropertiesInput, isBrowser?: boolean) {\n        super(defaultProps);\n        const kustoClient = new KustoClient(kcsb);\n        this.resourceManager = new ResourceManager(kustoClient, isBrowser);\n        this.defaultDatabase = kustoClient.defaultDatabase;\n    }\n\n    async ingestFromBlob(blob: string | BlobDescriptor, ingestionProperties?: IngestionPropertiesInput): Promise<QueueSendMessageResponse> {\n        this.ensureOpen();\n\n        const props = this._getMergedProps(ingestionProperties);\n\n        const descriptor = blob instanceof BlobDescriptor ? blob : new BlobDescriptor(blob);\n        const queues = await this.resourceManager.getIngestionQueues();\n        if (queues == null) {\n            throw new Error(\"Failed to get queues\");\n        }\n\n        const authorizationContext = await this.resourceManager.getAuthorizationContext();\n\n        const queueDetails = queues[Math.floor(Math.random() * queues.length)];\n\n        const queueClient = new QueueClient(queueDetails.uri);\n\n        const ingestionBlobInfo = new IngestionBlobInfo(descriptor, props, authorizationContext);\n        const ingestionBlobInfoJson = JSON.stringify(ingestionBlobInfo);\n        const encoded = Buffer.from(ingestionBlobInfoJson).toString(\"base64\");\n\n        return queueClient.sendMessage(encoded);\n    }\n\n    close() {\n        if (!this._isClosed) {\n            this.resourceManager.close();\n        }\n        super.close();\n    }\n}\n\nexport default KustoIngestClientBase;\n"]}
+{"version":3,"file":"ingestClientBase.js","sourceRoot":"","sources":["../../src/ingestClientBase.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;;;;AAElC,uDAAuF;AAEvF,wEAAgD;AAEhD,4EAAoD;AACpD,sDAAsD;AAEtD,wDAAmD;AAEnD,+DAAiH;AACjH,+DAA4D;AAC5D,uDAQ2B;AAG3B,+CAAiE;AAEjE,MAAsB,qBAAsB,SAAQ,yCAAmB;IAMnE,YACI,IAA2C,EAC3C,YAAuC,EACvC,sBAA+B,IAAI,EACnC,SAAmB;QAEnB,KAAK,CAAC,YAAY,CAAC,CAAC;QACpB,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE;YAC1B,IAAI,GAAG,IAAI,+CAA4B,CAAC,IAAI,CAAC,CAAC;SACjD;QACD,IAAI,mBAAmB,EAAE;YACrB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;SAChE;QACD,MAAM,WAAW,GAAG,IAAI,yBAAW,CAAC,IAAI,CAAC,CAAC;QAC1C,IAAI,CAAC,eAAe,GAAG,IAAI,yBAAe,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;QACnE,IAAI,CAAC,eAAe,GAAG,WAAW,CAAC,eAAe,CAAC;QACnD,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;QAC3C,IAAI,CAAC,qBAAqB,GAAG,aAAa,CAAC,yBAAyB,CAAC;QACrE,IAAI,CAAC,uBAAuB,GAAG,aAAa,CAAC,iBAAiB,CAAC;IACnE,CAAC;IAED,KAAK,CAAC,cAAc,CAChB,IAA6B,EAC7B,mBAA8C,EAC9C,aAAqB,qBAAqB,CAAC,wBAAwB;QAEnE,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QAExD,MAAM,UAAU,GAAG,IAAI,YAAY,4BAAc,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,4BAAc,CAAC,IAAI,CAAC,CAAC;QAEpF,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,uBAAuB,EAAE,CAAC;QAClF,MAAM,iBAAiB,GAAG,IAAI,2BAAiB,CAAC,UAAU,EAAE,KAAK,EAAE,oBAAoB,EAAE,IAAI,CAAC,qBAAqB,EAAE,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAEnJ,MAAM,aAAa,GAAG,KAAK,CAAC,WAAW,KAAK,iCAAW,CAAC,WAAW,IAAI,KAAK,CAAC,YAAY,KAAK,kCAAY,CAAC,KAAK,CAAC;QAEjH,IAAI,aAAa,EAAE;YACf,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,iBAAiB,EAAE,CAAC;YACzE,MAAM,MAAM,GAAG,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,iCAAe,CAAC,OAAO,EAAE,iBAAiB,CAAC,CAAC;YAC1F,MAAM,IAAA,kCAAgB,EAAC,iBAAiB,kCAAO,MAAM,KAAE,YAAY,EAAE,iBAAiB,CAAC,EAAE,EAAE,MAAM,EAAE,iBAAiB,CAAC,EAAE,IAAG,CAAC;YAE3H,MAAM,IAAI,GAAG,IAAI,mDAAiC,CAAC,iBAAiB,CAAC,GAAG,EAAE,iBAAiB,CAAC,EAAE,EAAE,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACtH,iBAAiB,CAAC,sBAAsB,GAAG,IAAI,CAAC;YAChD,MAAM,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,iBAAiB,CAAC,CAAC;YAC3D,OAAO,IAAI,4CAA0B,CAAC,IAAI,EAAE,iBAAiB,CAAC,CAAC;SAClE;QAED,MAAM,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,iBAAiB,CAAC,CAAC;QAC3D,OAAO,IAAI,uCAAqB,CAAC,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,iCAAe,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC,CAAC;IAChH,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,UAAkB,EAAE,QAA2B;QAC1E,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,kBAAkB,EAAE,CAAC;QAC/D,IAAI,MAAM,IAAI,IAAI,EAAE;YAChB,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;SAC3C;QACD,MAAM,qBAAqB,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACvD,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACtE,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QAEvD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE;YACjC,MAAM,WAAW,GAAG,IAAI,2BAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YACnD,IAAI;gBACA,MAAM,aAAa,GAAG,MAAM,WAAW,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;gBAC7D,IAAI,CAAC,eAAe,CAAC,yBAAyB,CAAC,WAAW,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;gBAC9E,OAAO,aAAa,CAAC;aACxB;YAAC,OAAO,CAAC,EAAE;gBACR,IAAI,CAAC,eAAe,CAAC,yBAAyB,CAAC,WAAW,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;aAClF;SACJ;QAED,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;IACxD,CAAC;IAEO,kBAAkB,CAAC,KAA0B,EAAE,MAAuB,EAAE,iBAAoC;QAChH,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;QACnC,OAAO;YACH,MAAM,EAAE,MAAM;YACd,SAAS,EAAE,IAAI;YACf,iBAAiB,EAAE,iBAAiB,CAAC,EAAE;YACvC,mBAAmB,EAAE,iBAAiB,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAChE,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,SAAS,EAAE,IAAI;YACf,OAAO,EAAE,EAAE;SACK,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,qBAAqB,CACvB,UAA0D,EAC1D,QAAgB,EAChB,aAAqB,qBAAqB,CAAC,wBAAwB;QAEnE,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE,CAAC;QAE9D,IAAI,UAAU,IAAI,IAAI,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE;YAC/C,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;SAC/C;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC;QAE3D,mFAAmF;QACnF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE;YACjC,MAAM,eAAe,GAAG,IAAI,8BAAe,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YAC/D,MAAM,eAAe,GAAG,eAAe,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YACrE,IAAI;gBACA,IAAI,OAAO,UAAU,IAAI,QAAQ,EAAE;oBAC/B,MAAM,eAAe,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;iBAChD;qBAAM,IAAI,UAAU,YAAY,8BAAgB,EAAE;oBAC/C,IAAI,UAAU,CAAC,MAAM,YAAY,MAAM,EAAE;wBACrC,MAAM,eAAe,CAAC,UAAU,CAAC,UAAU,CAAC,MAAgB,CAAC,CAAC;qBACjE;yBAAM;wBACH,MAAM,eAAe,CAAC,YAAY,CAAC,UAAU,CAAC,MAAkB,CAAC,CAAC;qBACrE;iBACJ;qBAAM;oBACH,MAAM,eAAe,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;iBAChD;gBACD,IAAI,CAAC,eAAe,CAAC,yBAAyB,CAAC,eAAe,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;gBAClF,OAAO,eAAe,CAAC,GAAG,CAAC;aAC9B;YAAC,OAAO,EAAE,EAAE;gBACT,IAAI,CAAC,eAAe,CAAC,yBAAyB,CAAC,eAAe,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;aACtF;SACJ;QAED,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;IACjD,CAAC;IAED,KAAK;QACD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;YACjB,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;SAChC;QACD,KAAK,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC;;AA1IL,sDA2IC;AAvImB,8CAAwB,GAAG,CAAC,CAAC;AAyIjD,kBAAe,qBAAqB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { Client as KustoClient, KustoConnectionStringBuilder } from \"azure-kusto-data\";\n\nimport ResourceManager from \"./resourceManager\";\n\nimport IngestionBlobInfo from \"./ingestionBlobInfo\";\nimport { ContainerClient } from \"@azure/storage-blob\";\n\nimport { QueueClient } from \"@azure/storage-queue\";\n\nimport IngestionProperties, { IngestionPropertiesInput, ReportLevel, ReportMethod } from \"./ingestionProperties\";\nimport { AbstractKustoClient } from \"./abstractKustoClient\";\nimport {\n    IngestionStatus,\n    TableReportIngestionResult,\n    IngestionResult,\n    IngestionStatusInTableDescription,\n    IngestionStatusResult,\n    OperationStatus,\n    putRecordInTable,\n} from \"./ingestionResult\";\nimport { Readable } from \"stream\";\n\nimport { BlobDescriptor, StreamDescriptor } from \"./descriptors\";\n\nexport abstract class KustoIngestClientBase extends AbstractKustoClient {\n    resourceManager: ResourceManager;\n    applicationForTracing: string | null;\n    clientVersionForTracing: string | null;\n    static readonly MaxNumberOfRetryAttempts = 3;\n\n    constructor(\n        kcsb: string | KustoConnectionStringBuilder,\n        defaultProps?: IngestionPropertiesInput,\n        autoCorrectEndpoint: boolean = true,\n        isBrowser?: boolean\n    ) {\n        super(defaultProps);\n        if (typeof kcsb === \"string\") {\n            kcsb = new KustoConnectionStringBuilder(kcsb);\n        }\n        if (autoCorrectEndpoint) {\n            kcsb.dataSource = this.getIngestionEndpoint(kcsb.dataSource);\n        }\n        const kustoClient = new KustoClient(kcsb);\n        this.resourceManager = new ResourceManager(kustoClient, isBrowser);\n        this.defaultDatabase = kustoClient.defaultDatabase;\n        const clientDetails = kcsb.clientDetails();\n        this.applicationForTracing = clientDetails.applicationNameForTracing;\n        this.clientVersionForTracing = clientDetails.versionForTracing;\n    }\n\n    async ingestFromBlob(\n        blob: string | BlobDescriptor,\n        ingestionProperties?: IngestionPropertiesInput,\n        maxRetries: number = KustoIngestClientBase.MaxNumberOfRetryAttempts\n    ): Promise<IngestionResult> {\n        this.ensureOpen();\n        const props = this._getMergedProps(ingestionProperties);\n\n        const descriptor = blob instanceof BlobDescriptor ? blob : new BlobDescriptor(blob);\n\n        const authorizationContext = await this.resourceManager.getAuthorizationContext();\n        const ingestionBlobInfo = new IngestionBlobInfo(descriptor, props, authorizationContext, this.applicationForTracing, this.clientVersionForTracing);\n\n        const reportToTable = props.reportLevel !== ReportLevel.DoNotReport && props.reportMethod !== ReportMethod.Queue;\n\n        if (reportToTable) {\n            const statusTableClient = await this.resourceManager.createStatusTable();\n            const status = this.createStatusObject(props, OperationStatus.Pending, ingestionBlobInfo);\n            await putRecordInTable(statusTableClient, { ...status, partitionKey: ingestionBlobInfo.Id, rowKey: ingestionBlobInfo.Id });\n\n            const desc = new IngestionStatusInTableDescription(statusTableClient.url, ingestionBlobInfo.Id, ingestionBlobInfo.Id);\n            ingestionBlobInfo.IngestionStatusInTable = desc;\n            await this.sendQueueMessage(maxRetries, ingestionBlobInfo);\n            return new TableReportIngestionResult(desc, statusTableClient);\n        }\n\n        await this.sendQueueMessage(maxRetries, ingestionBlobInfo);\n        return new IngestionStatusResult(this.createStatusObject(props, OperationStatus.Queued, ingestionBlobInfo));\n    }\n\n    private async sendQueueMessage(maxRetries: number, blobInfo: IngestionBlobInfo) {\n        const queues = await this.resourceManager.getIngestionQueues();\n        if (queues == null) {\n            throw new Error(\"Failed to get queues\");\n        }\n        const ingestionBlobInfoJson = JSON.stringify(blobInfo);\n        const encoded = Buffer.from(ingestionBlobInfoJson).toString(\"base64\");\n        const retryCount = Math.min(maxRetries, queues.length);\n\n        for (let i = 0; i < retryCount; i++) {\n            const queueClient = new QueueClient(queues[i].uri);\n            try {\n                const queueResponse = await queueClient.sendMessage(encoded);\n                this.resourceManager.reportResourceUsageResult(queueClient.accountName, true);\n                return queueResponse;\n            } catch (_) {\n                this.resourceManager.reportResourceUsageResult(queueClient.accountName, false);\n            }\n        }\n\n        throw new Error(\"Failed to send message to queue.\");\n    }\n\n    private createStatusObject(props: IngestionProperties, status: OperationStatus, ingestionBlobInfo: IngestionBlobInfo): IngestionStatus {\n        const time = Date.now().toString();\n        return {\n            Status: status,\n            Timestamp: time,\n            IngestionSourceId: ingestionBlobInfo.Id,\n            IngestionSourcePath: ingestionBlobInfo.BlobPath.split(/[?;]/)[0],\n            Database: props.database,\n            Table: props.table,\n            UpdatedOn: time,\n            Details: \"\",\n        } as IngestionStatus;\n    }\n\n    async uploadToBlobWithRetry(\n        descriptor: string | Blob | ArrayBuffer | StreamDescriptor,\n        blobName: string,\n        maxRetries: number = KustoIngestClientBase.MaxNumberOfRetryAttempts\n    ): Promise<string> {\n        const containers = await this.resourceManager.getContainers();\n\n        if (containers == null || containers.length === 0) {\n            throw new Error(\"Failed to get containers\");\n        }\n\n        const retryCount = Math.min(maxRetries, containers.length);\n\n        // Go over all containers and try to upload the file to the first one that succeeds\n        for (let i = 0; i < retryCount; i++) {\n            const containerClient = new ContainerClient(containers[i].uri);\n            const blockBlobClient = containerClient.getBlockBlobClient(blobName);\n            try {\n                if (typeof descriptor == \"string\") {\n                    await blockBlobClient.uploadFile(descriptor);\n                } else if (descriptor instanceof StreamDescriptor) {\n                    if (descriptor.stream instanceof Buffer) {\n                        await blockBlobClient.uploadData(descriptor.stream as Buffer);\n                    } else {\n                        await blockBlobClient.uploadStream(descriptor.stream as Readable);\n                    }\n                } else {\n                    await blockBlobClient.uploadData(descriptor);\n                }\n                this.resourceManager.reportResourceUsageResult(containerClient.accountName, true);\n                return blockBlobClient.url;\n            } catch (ex) {\n                this.resourceManager.reportResourceUsageResult(containerClient.accountName, false);\n            }\n        }\n\n        throw new Error(\"Failed to upload to blob.\");\n    }\n\n    close() {\n        if (!this._isClosed) {\n            this.resourceManager.close();\n        }\n        super.close();\n    }\n}\n\nexport default KustoIngestClientBase;\n"]}

package/dist-esm/src/ingestionBlobInfo.js

@@ -5,8 +5,11 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.IngestionBlobInfo = void 0;
 const uuid_1 = require("uuid");
 class IngestionBlobInfo {
-    constructor(blobDescriptor, ingestionProperties, authContext = null) {
+    constructor(blobDescriptor, ingestionProperties, authContext = null, applicationForTracing = null, clientVersionForTracing = null) {
         var _a, _b, _c, _d, _e;
+        this.IngestionStatusInTable = null;
+        this.ApplicationForTracing = null;
+        this.ClientVersionForTracing = null;
         this.BlobPath = blobDescriptor.path;
         this.RawDataSize = blobDescriptor.size;
         this.DatabaseName = (_a = ingestionProperties.database) !== null && _a !== void 0 ? _a : null;
@@ -18,6 +21,8 @@ class IngestionBlobInfo {
         this.ReportMethod = (_e = ingestionProperties.reportMethod) !== null && _e !== void 0 ? _e : null;
         this.SourceMessageCreationTime = new Date();
         this.Id = blobDescriptor.sourceId || (0, uuid_1.v4)();
+        this.ApplicationForTracing = applicationForTracing;
+        this.ClientVersionForTracing = clientVersionForTracing;
         const additionalProperties = ingestionProperties.additionalProperties || {};
         additionalProperties.authorizationContext = authContext;
         const tags = [];