azure-kusto-ingest 6.0.2 → 7.0.0-alpha.1

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

@@ -1 +1 @@
-{"version":3,"file":"managedStreamingIngestClient.js","sourceRoot":"","sources":["../../src/managedStreamingIngestClient.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;;;AAIlC,gDAA0C;AAC1C,uDAAsF;AAEtF,+DAA4D;AAC5D,+CAAqF;AACrF,qDAAkD;AAClD,kEAA0C;AAC1C,mCAA2C;AAC3C,+CAAoF;AACpF,oFAA4D;AAG5D,MAAM,aAAa,GAAG,IAAI,GAAG,IAAI,GAAG,CAAC,CAAC;AACtC,MAAM,YAAY,GAAG,CAAC,CAAC;AACvB,MAAM,YAAY,GAAG,iBAAiB,CAAC;AAEvC,MAAM,iCAAkC,SAAQ,yCAAmB;IAM/D;;;;;;;OAOG;IACH,MAAM,CAAC,sBAAsB,CACzB,kBAAgD,EAChD,YAAuC;;QAEvC,IAAI,kBAAkB,CAAC,UAAU,IAAI,IAAI,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE;YAClG,MAAM,IAAI,KAAK,CAAC,iDAAiD,YAAY,GAAG,CAAC,CAAC;SACrF;QAED,MAAM,sBAAsB,GAAG,+CAA4B,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC;QAC7F,sBAAsB,CAAC,UAAU,GAAG,MAAA,sBAAsB,CAAC,UAAU,0CAAE,OAAO,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;QAEzG,OAAO,IAAI,iCAAiC,CAAC,sBAAsB,EAAE,kBAAkB,EAAE,YAAY,CAAC,CAAC;IAC3G,CAAC;IAED;;;;;;;OAOG;IACH,MAAM,CAAC,0BAA0B,CAC7B,sBAAoD,EACpD,YAAuC;;QAEvC,IAAI,sBAAsB,CAAC,UAAU,IAAI,IAAI,IAAI,sBAAsB,CAAC,UAAU,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE;YACzG,MAAM,IAAI,KAAK,CAAC,yDAAyD,YAAY,GAAG,CAAC,CAAC;SAC7F;QAED,MAAM,kBAAkB,GAAG,+CAA4B,CAAC,YAAY,CAAC,sBAAsB,CAAC,CAAC;QAC7F,kBAAkB,CAAC,UAAU,GAAG,MAAA,kBAAkB,CAAC,UAAU,0CAAE,OAAO,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;QAEjG,OAAO,IAAI,iCAAiC,CAAC,sBAAsB,EAAE,kBAAkB,EAAE,YAAY,CAAC,CAAC;IAC3G,CAAC;IAED,YACI,UAAiD,EACjD,MAA6C,EAC7C,YAAuC,EACvC,sBAA+B,IAAI;QAEnC,KAAK,CAAC,YAAY,CAAC,CAAC;QArDhB,sBAAiB,GAAG,CAAC,CAAC;QACtB,mBAAc,GAAG,CAAC,CAAC;QAqDvB,IAAI,CAAC,qBAAqB,GAAG,IAAI,+BAAqB,CAAC,UAAU,EAAE,YAAY,EAAE,mBAAmB,CAAC,CAAC;QACtG,IAAI,CAAC,kBAAkB,GAAG,IAAI,sBAAY,CAAC,MAAM,EAAE,YAAY,EAAE,mBAAmB,CAAC,CAAC;QAEtF,IAAI,IAAI,CAAC,qBAAqB,CAAC,eAAe,IAAI,IAAI,CAAC,qBAAqB,CAAC,eAAe,KAAK,IAAI,CAAC,kBAAkB,CAAC,eAAe,EAAE;YACtI,MAAM,IAAI,KAAK,CACX,iDAAiD,IAAI,CAAC,qBAAqB,CAAC,eAAe,2DAA2D,IAAI,CAAC,kBAAkB,CAAC,eAAe,GAAG,CACnM,CAAC;SACL;QAED,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,qBAAqB,CAAC,eAAe,CAAC;IACtE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB,CAClB,MAAiD,EACjD,mBAA8C,EAC9C,eAAwB;;QAExB,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QACxD,IAAI,UAAU,GAAG,MAAM,YAAY,8BAAgB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,8BAAgB,CAAC,MAAM,CAAC,CAAC;QAC5F,IAAI,MAAM,GAAG,kBAAM,CAAC,CAAC,CAAC,MAAM,IAAA,8BAAgB,EAAC,UAAU,CAAC,MAAkB,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC;QAC/G,UAAU,GAAG,IAAI,8BAAgB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAC5D,IAAI,eAAe,GAAwB,IAAI,CAAC;QAChD,kFAAkF;QAClF,IAAI,CAAC,kBAAM,IAAI,MAAM,YAAY,MAAM,CAAC,IAAI,CAAC,kBAAM,EAAE;YACjD,eAAe,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAC1C,kBAAM,CAAC,CAAC,CAAC,MAAA,UAAU,CAAC,IAAI,mCAAI,CAAC,CAAC,CAAC,CAAE,UAAU,CAAC,MAAsB,CAAC,UAAU,EAC7E,UAAU,EACV,KAAK,EACL,eAAe,EACf,MAAM,CACT,CAAC;YAEF,MAAM,GAAG,kBAAM,CAAC,CAAC,CAAC,IAAA,8BAAgB,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC;SAClE;QAED,OAAO,eAAe,aAAf,eAAe,cAAf,eAAe,GAAI,IAAI,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,IAAI,8BAAgB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,KAAK,CAAC,CAAC;IAC9H,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAChB,IAAoC,EACpC,mBAA8C;QAE9C,IAAI,CAAC,UAAU,EAAE,CAAC;QAElB,MAAM,MAAM,GAAG,IAAI,YAAY,+BAAc,CAAC,CAAC,CAAC,MAAM,IAAA,6BAAe,EAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,IAAA,6BAAe,EAAC,IAAI,+BAAc,CAAC,IAAI,CAAC,CAAC,CAAC;QAC9H,OAAO,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,cAAc,CAChB,IAA6B,EAC7B,mBAA8C,EAC9C,eAAwB;;QAExB,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QACxD,MAAM,UAAU,GAAG,IAAI,YAAY,4BAAc,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,4BAAc,CAAC,IAAI,CAAC,CAAC;QACpF,uEAAuE;QACvE,MAAM,UAAU,CAAC,QAAQ,EAAE,CAAC;QAE5B,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,MAAA,UAAU,CAAC,IAAI,mCAAI,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,eAAe,CAAC,CAAC;QAC/G,OAAO,eAAe,aAAf,eAAe,cAAf,eAAe,GAAI,IAAI,CAAC,kBAAkB,CAAC,cAAc,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IACxF,CAAC;IAED,KAAK,CAAC,iBAAiB,CACnB,MAAc,EACd,UAA8B,EAC9B,KAAgC,EAChC,eAAwB,EACxB,MAA+B;QAE/B,MAAM,MAAM,GAAG,UAAU,YAAY,4BAAc,CAAC;QACpD,IAAI,MAAM,IAAI,aAAa,EAAE;YACzB,6FAA6F;YAC7F,MAAM,KAAK,GAAG,IAAI,wBAAgB,CAAC,YAAY,EAAE,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;YAC9F,OAAO,KAAK,CAAC,SAAS,EAAE,EAAE;gBACtB,IAAI;oBACA,MAAM,QAAQ,GACV,eAAe,aAAf,eAAe,cAAf,eAAe,GACf,oCAAoC,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,YAAY,IAAI,UAAU,CAAC,QAAQ,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;oBAC5H,IAAI,MAAM,EAAE;wBACR,OAAO,MAAM,IAAI,CAAC,qBAAqB,CAAC,cAAc,CAAC,UAA4B,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;qBACzG;oBAED,IAAI,kBAAM,EAAE;wBACR,OAAO,MAAM,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,CACpD,IAAI,8BAAgB,CAAC,IAAA,8BAAgB,EAAC,MAAO,CAAC,CAAC,CAAC,KAAK,CAAC,UAA8B,CAAC,EACrF,KAAK,EACL,QAAQ,CACX,CAAC;qBACL;oBAED,OAAO,MAAM,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,UAA8B,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;iBAC7G;gBAAC,OAAO,GAAY,EAAE;oBACnB,MAAM,WAAW,GAAG,GAAiC,CAAC;oBACtD,IAAI,WAAW,CAAC,YAAY,CAAC,EAAE;wBAC3B,MAAM,GAAG,CAAC;qBACb;oBACD,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;iBACzB;aACJ;YAED,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,kBAAM,IAAI,MAAM,CAAC,CAAC,CAAC,IAAA,8BAAgB,EAAC,MAAM,CAAC,CAAC,CAAC,CAAE,UAA+B,CAAC,MAAM,CAAC;SACvH;QAED,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,KAAK;QACD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;YACjB,IAAI,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAC;YACnC,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAC;SACnC;QACD,KAAK,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC;CACJ;AAED,kBAAe,iCAAiC,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { IngestionPropertiesInput } from \"./ingestionProperties\";\n\nimport { isNode } from \"@azure/core-util\";\nimport { KustoConnectionStringBuilder, KustoResponseDataSet } from \"azure-kusto-data\";\nimport { Readable } from \"stream\";\nimport { AbstractKustoClient } from \"./abstractKustoClient\";\nimport { AbstractDescriptor, BlobDescriptor, StreamDescriptor } from \"./descriptors\";\nimport { FileDescriptor } from \"./fileDescriptor\";\nimport IngestClient from \"./ingestClient\";\nimport { ExponentialRetry } from \"./retry\";\nimport { readableToStream, tryFileToBuffer, tryStreamToArray } from \"./streamUtils\";\nimport StreamingIngestClient from \"./streamingIngestClient\";\nimport { IngestionResult } from \"./ingestionResult\";\n\nconst maxStreamSize = 1024 * 1024 * 4;\nconst attemptCount = 3;\nconst ingestPrefix = \"https://ingest-\";\n\nclass KustoManagedStreamingIngestClient extends AbstractKustoClient {\n    private streamingIngestClient: StreamingIngestClient;\n    private queuedIngestClient: IngestClient;\n    private baseSleepTimeSecs = 1;\n    private baseJitterSecs = 1;\n\n    /**\n     * Creates a KustoManagedStreamingIngestClient from a DM connection string.\n     * This method infers the engine connection string.\n     * For advanced usage, use the constructor that takes a DM connection string and an engine connection string.\n     *\n     * @param dmConnectionString The DM connection string.\n     * @param defaultProps The default ingestion properties.\n     */\n    static fromDmConnectionString(\n        dmConnectionString: KustoConnectionStringBuilder,\n        defaultProps?: IngestionPropertiesInput\n    ): KustoManagedStreamingIngestClient {\n        if (dmConnectionString.dataSource == null || !dmConnectionString.dataSource.startsWith(ingestPrefix)) {\n            throw new Error(`DM connection string must include the prefix '${ingestPrefix}'`);\n        }\n\n        const engineConnectionString = KustoConnectionStringBuilder.fromExisting(dmConnectionString);\n        engineConnectionString.dataSource = engineConnectionString.dataSource?.replace(ingestPrefix, \"https://\");\n\n        return new KustoManagedStreamingIngestClient(engineConnectionString, dmConnectionString, defaultProps);\n    }\n\n    /**\n     * Creates a KustoManagedStreamingIngestClient from a engine connection string.\n     * This method infers the engine connection string.\n     * For advanced usage, use the constructor that takes an engine connection string and an engine connection string.\n     *\n     * @param engineConnectionString The engine connection string.\n     * @param defaultProps The default ingestion properties.\n     */\n    static fromEngineConnectionString(\n        engineConnectionString: KustoConnectionStringBuilder,\n        defaultProps?: IngestionPropertiesInput\n    ): KustoManagedStreamingIngestClient {\n        if (engineConnectionString.dataSource == null || engineConnectionString.dataSource.startsWith(ingestPrefix)) {\n            throw new Error(`Engine connection string must not include the prefix '${ingestPrefix}'`);\n        }\n\n        const dmConnectionString = KustoConnectionStringBuilder.fromExisting(engineConnectionString);\n        dmConnectionString.dataSource = dmConnectionString.dataSource?.replace(\"https://\", ingestPrefix);\n\n        return new KustoManagedStreamingIngestClient(engineConnectionString, dmConnectionString, defaultProps);\n    }\n\n    constructor(\n        engineKcsb: string | KustoConnectionStringBuilder,\n        dmKcsb: string | KustoConnectionStringBuilder,\n        defaultProps?: IngestionPropertiesInput,\n        autoCorrectEndpoint: boolean = true\n    ) {\n        super(defaultProps);\n        this.streamingIngestClient = new StreamingIngestClient(engineKcsb, defaultProps, autoCorrectEndpoint);\n        this.queuedIngestClient = new IngestClient(dmKcsb, defaultProps, autoCorrectEndpoint);\n\n        if (this.streamingIngestClient.defaultDatabase && this.streamingIngestClient.defaultDatabase !== this.queuedIngestClient.defaultDatabase) {\n            throw new Error(\n                `Default database for streaming ingest client (${this.streamingIngestClient.defaultDatabase}) must match default database for queued ingest client (${this.queuedIngestClient.defaultDatabase})`\n            );\n        }\n\n        this.defaultDatabase = this.streamingIngestClient.defaultDatabase;\n    }\n\n    /**\n     * Use Readable for Node.js and ArrayBuffer in browser\n     */\n    async ingestFromStream(\n        stream: StreamDescriptor | Readable | ArrayBuffer,\n        ingestionProperties?: IngestionPropertiesInput,\n        clientRequestId?: string\n    ): Promise<KustoResponseDataSet | IngestionResult> {\n        this.ensureOpen();\n        const props = this._getMergedProps(ingestionProperties);\n        let descriptor = stream instanceof StreamDescriptor ? stream : new StreamDescriptor(stream);\n        let result = isNode ? await tryStreamToArray(descriptor.stream as Readable, maxStreamSize) : descriptor.stream;\n        descriptor = new StreamDescriptor(result).merge(descriptor);\n        let streamingResult: Promise<any> | null = null;\n        // tryStreamToArray returns a Buffer in NodeJS impl if stream size is small enouph\n        if ((isNode && result instanceof Buffer) || !isNode) {\n            streamingResult = await this.streamWithRetries(\n                isNode ? descriptor.size ?? 0 : (descriptor.stream as ArrayBuffer).byteLength,\n                descriptor,\n                props,\n                clientRequestId,\n                result\n            );\n\n            result = isNode ? readableToStream(result) : descriptor.stream;\n        }\n\n        return streamingResult ?? this.queuedIngestClient.ingestFromStream(new StreamDescriptor(result).merge(descriptor), props);\n    }\n\n    /**\n     * Use string for Node.js and Blob in browser\n     */\n    async ingestFromFile(\n        file: FileDescriptor | string | Blob,\n        ingestionProperties?: IngestionPropertiesInput\n    ): Promise<KustoResponseDataSet | IngestionResult> {\n        this.ensureOpen();\n\n        const stream = file instanceof FileDescriptor ? await tryFileToBuffer(file) : await tryFileToBuffer(new FileDescriptor(file));\n        return await this.ingestFromStream(stream, ingestionProperties);\n    }\n\n    async ingestFromBlob(\n        blob: string | BlobDescriptor,\n        ingestionProperties?: IngestionPropertiesInput,\n        clientRequestId?: string\n    ): Promise<KustoResponseDataSet | IngestionResult> {\n        const props = this._getMergedProps(ingestionProperties);\n        const descriptor = blob instanceof BlobDescriptor ? blob : new BlobDescriptor(blob);\n        // No need to check blob size if it was given to us that it's not empty\n        await descriptor.fillSize();\n\n        const streamingResult = await this.streamWithRetries(descriptor.size ?? 0, descriptor, props, clientRequestId);\n        return streamingResult ?? this.queuedIngestClient.ingestFromBlob(descriptor, props);\n    }\n\n    async streamWithRetries(\n        length: number,\n        descriptor: AbstractDescriptor,\n        props?: IngestionPropertiesInput,\n        clientRequestId?: string,\n        stream?: Readable | ArrayBuffer\n    ): Promise<any> {\n        const isBlob = descriptor instanceof BlobDescriptor;\n        if (length <= maxStreamSize) {\n            // If we get buffer that means it was less than the max size, so we can do streamingIngestion\n            const retry = new ExponentialRetry(attemptCount, this.baseSleepTimeSecs, this.baseJitterSecs);\n            while (retry.shouldTry()) {\n                try {\n                    const sourceId =\n                        clientRequestId ??\n                        `KNC.executeManagedStreamingIngest${isBlob ? \"FromBlob\" : \"FromStream\"};${descriptor.sourceId};${retry.currentAttempt}`;\n                    if (isBlob) {\n                        return await this.streamingIngestClient.ingestFromBlob(descriptor as BlobDescriptor, props, sourceId);\n                    }\n\n                    if (isNode) {\n                        return await this.streamingIngestClient.ingestFromStream(\n                            new StreamDescriptor(readableToStream(stream!)).merge(descriptor as StreamDescriptor),\n                            props,\n                            sourceId\n                        );\n                    }\n\n                    return await this.streamingIngestClient.ingestFromStream(descriptor as StreamDescriptor, props, sourceId);\n                } catch (err: unknown) {\n                    const oneApiError = err as { \"@permanent\"?: boolean };\n                    if (oneApiError[\"@permanent\"]) {\n                        throw err;\n                    }\n                    await retry.backoff();\n                }\n            }\n\n            stream = isBlob ? undefined : isNode && stream ? readableToStream(stream) : (descriptor as StreamDescriptor).stream;\n        }\n\n        return null;\n    }\n\n    close() {\n        if (!this._isClosed) {\n            this.streamingIngestClient.close();\n            this.queuedIngestClient.close();\n        }\n        super.close();\n    }\n}\n\nexport default KustoManagedStreamingIngestClient;\n"]}
+{"version":3,"file":"managedStreamingIngestClient.js","sourceRoot":"","sources":["../../src/managedStreamingIngestClient.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAIlC,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAC1C,OAAO,EAAE,4BAA4B,EAAwB,MAAM,kBAAkB,CAAC;AAEtF,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAC/D,OAAO,EAAsB,cAAc,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACxF,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,YAAY,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,EAAE,gBAAgB,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACvF,OAAO,qBAAqB,MAAM,4BAA4B,CAAC;AAG/D,MAAM,aAAa,GAAG,IAAI,GAAG,IAAI,GAAG,CAAC,CAAC;AACtC,MAAM,YAAY,GAAG,CAAC,CAAC;AACvB,MAAM,YAAY,GAAG,iBAAiB,CAAC;AAEvC,MAAM,iCAAkC,SAAQ,mBAAmB;IAM/D;;;;;;;OAOG;IACH,MAAM,CAAC,sBAAsB,CACzB,kBAAgD,EAChD,YAAuC;;QAEvC,IAAI,kBAAkB,CAAC,UAAU,IAAI,IAAI,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;YACnG,MAAM,IAAI,KAAK,CAAC,iDAAiD,YAAY,GAAG,CAAC,CAAC;QACtF,CAAC;QAED,MAAM,sBAAsB,GAAG,4BAA4B,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC;QAC7F,sBAAsB,CAAC,UAAU,GAAG,MAAA,sBAAsB,CAAC,UAAU,0CAAE,OAAO,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;QAEzG,OAAO,IAAI,iCAAiC,CAAC,sBAAsB,EAAE,kBAAkB,EAAE,YAAY,CAAC,CAAC;IAC3G,CAAC;IAED;;;;;;;OAOG;IACH,MAAM,CAAC,0BAA0B,CAC7B,sBAAoD,EACpD,YAAuC;;QAEvC,IAAI,sBAAsB,CAAC,UAAU,IAAI,IAAI,IAAI,sBAAsB,CAAC,UAAU,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;YAC1G,MAAM,IAAI,KAAK,CAAC,yDAAyD,YAAY,GAAG,CAAC,CAAC;QAC9F,CAAC;QAED,MAAM,kBAAkB,GAAG,4BAA4B,CAAC,YAAY,CAAC,sBAAsB,CAAC,CAAC;QAC7F,kBAAkB,CAAC,UAAU,GAAG,MAAA,kBAAkB,CAAC,UAAU,0CAAE,OAAO,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;QAEjG,OAAO,IAAI,iCAAiC,CAAC,sBAAsB,EAAE,kBAAkB,EAAE,YAAY,CAAC,CAAC;IAC3G,CAAC;IAED,YACI,UAAiD,EACjD,MAA6C,EAC7C,YAAuC,EACvC,sBAA+B,IAAI;QAEnC,KAAK,CAAC,YAAY,CAAC,CAAC;QArDhB,sBAAiB,GAAG,CAAC,CAAC;QACtB,mBAAc,GAAG,CAAC,CAAC;QAqDvB,IAAI,CAAC,qBAAqB,GAAG,IAAI,qBAAqB,CAAC,UAAU,EAAE,YAAY,EAAE,mBAAmB,CAAC,CAAC;QACtG,IAAI,CAAC,kBAAkB,GAAG,IAAI,YAAY,CAAC,MAAM,EAAE,YAAY,EAAE,mBAAmB,CAAC,CAAC;QAEtF,IAAI,IAAI,CAAC,qBAAqB,CAAC,eAAe,IAAI,IAAI,CAAC,qBAAqB,CAAC,eAAe,KAAK,IAAI,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC;YACvI,MAAM,IAAI,KAAK,CACX,iDAAiD,IAAI,CAAC,qBAAqB,CAAC,eAAe,2DAA2D,IAAI,CAAC,kBAAkB,CAAC,eAAe,GAAG,CACnM,CAAC;QACN,CAAC;QAED,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,qBAAqB,CAAC,eAAe,CAAC;IACtE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB,CAClB,MAAiD,EACjD,mBAA8C,EAC9C,eAAwB;;QAExB,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QACxD,IAAI,UAAU,GAAG,MAAM,YAAY,gBAAgB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAC5F,IAAI,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,MAAM,gBAAgB,CAAC,UAAU,CAAC,MAAkB,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC;QAC/G,UAAU,GAAG,IAAI,gBAAgB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAC5D,IAAI,eAAe,GAAwB,IAAI,CAAC;QAChD,kFAAkF;QAClF,IAAI,CAAC,MAAM,IAAI,MAAM,YAAY,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YAClD,eAAe,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAC1C,MAAM,CAAC,CAAC,CAAC,CAAC,MAAA,UAAU,CAAC,IAAI,mCAAI,CAAC,CAAC,CAAC,CAAC,CAAE,UAAU,CAAC,MAAsB,CAAC,UAAU,EAC/E,UAAU,EACV,KAAK,EACL,eAAe,EACf,MAAM,CACT,CAAC;YAEF,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC;QACnE,CAAC;QAED,OAAO,eAAe,aAAf,eAAe,cAAf,eAAe,GAAI,IAAI,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,IAAI,gBAAgB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,KAAK,CAAC,CAAC;IAC9H,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAChB,IAAoC,EACpC,mBAA8C;QAE9C,IAAI,CAAC,UAAU,EAAE,CAAC;QAElB,MAAM,MAAM,GAAG,IAAI,YAAY,cAAc,CAAC,CAAC,CAAC,MAAM,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,eAAe,CAAC,IAAI,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC;QAC9H,OAAO,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,cAAc,CAChB,IAA6B,EAC7B,mBAA8C,EAC9C,eAAwB;;QAExB,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QACxD,MAAM,UAAU,GAAG,IAAI,YAAY,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,cAAc,CAAC,IAAI,CAAC,CAAC;QACpF,uEAAuE;QACvE,MAAM,UAAU,CAAC,QAAQ,EAAE,CAAC;QAE5B,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,MAAA,UAAU,CAAC,IAAI,mCAAI,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,eAAe,CAAC,CAAC;QAC/G,OAAO,eAAe,aAAf,eAAe,cAAf,eAAe,GAAI,IAAI,CAAC,kBAAkB,CAAC,cAAc,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IACxF,CAAC;IAED,KAAK,CAAC,iBAAiB,CACnB,MAAc,EACd,UAA8B,EAC9B,KAAgC,EAChC,eAAwB,EACxB,MAA+B;QAE/B,MAAM,MAAM,GAAG,UAAU,YAAY,cAAc,CAAC;QACpD,IAAI,MAAM,IAAI,aAAa,EAAE,CAAC;YAC1B,6FAA6F;YAC7F,MAAM,KAAK,GAAG,IAAI,gBAAgB,CAAC,YAAY,EAAE,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;YAC9F,OAAO,KAAK,CAAC,SAAS,EAAE,EAAE,CAAC;gBACvB,IAAI,CAAC;oBACD,MAAM,QAAQ,GACV,eAAe,aAAf,eAAe,cAAf,eAAe,GACf,oCAAoC,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,YAAY,IAAI,UAAU,CAAC,QAAQ,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;oBAC5H,IAAI,MAAM,EAAE,CAAC;wBACT,OAAO,MAAM,IAAI,CAAC,qBAAqB,CAAC,cAAc,CAAC,UAA4B,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;oBAC1G,CAAC;oBAED,IAAI,MAAM,EAAE,CAAC;wBACT,OAAO,MAAM,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,CACpD,IAAI,gBAAgB,CAAC,gBAAgB,CAAC,MAAO,CAAC,CAAC,CAAC,KAAK,CAAC,UAA8B,CAAC,EACrF,KAAK,EACL,QAAQ,CACX,CAAC;oBACN,CAAC;oBAED,OAAO,MAAM,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,UAA8B,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;gBAC9G,CAAC;gBAAC,OAAO,GAAY,EAAE,CAAC;oBACpB,MAAM,WAAW,GAAG,GAAiC,CAAC;oBACtD,IAAI,WAAW,CAAC,YAAY,CAAC,EAAE,CAAC;wBAC5B,MAAM,GAAG,CAAC;oBACd,CAAC;oBACD,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;gBAC1B,CAAC;YACL,CAAC;YAED,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,IAAI,MAAM,CAAC,CAAC,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAE,UAA+B,CAAC,MAAM,CAAC;QACxH,CAAC;QAED,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,KAAK;QACD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YAClB,IAAI,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAC;YACnC,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAC;QACpC,CAAC;QACD,KAAK,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC;CACJ;AAED,eAAe,iCAAiC,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { IngestionPropertiesInput } from \"./ingestionProperties.js\";\n\nimport { isNode } from \"@azure/core-util\";\nimport { KustoConnectionStringBuilder, KustoResponseDataSet } from \"azure-kusto-data\";\nimport { Readable } from \"stream\";\nimport { AbstractKustoClient } from \"./abstractKustoClient.js\";\nimport { AbstractDescriptor, BlobDescriptor, StreamDescriptor } from \"./descriptors.js\";\nimport { FileDescriptor } from \"./fileDescriptor.js\";\nimport IngestClient from \"./ingestClient.js\";\nimport { ExponentialRetry } from \"./retry.js\";\nimport { readableToStream, tryFileToBuffer, tryStreamToArray } from \"./streamUtils.js\";\nimport StreamingIngestClient from \"./streamingIngestClient.js\";\nimport { IngestionResult } from \"./ingestionResult.js\";\n\nconst maxStreamSize = 1024 * 1024 * 4;\nconst attemptCount = 3;\nconst ingestPrefix = \"https://ingest-\";\n\nclass KustoManagedStreamingIngestClient extends AbstractKustoClient {\n    private streamingIngestClient: StreamingIngestClient;\n    private queuedIngestClient: IngestClient;\n    private baseSleepTimeSecs = 1;\n    private baseJitterSecs = 1;\n\n    /**\n     * Creates a KustoManagedStreamingIngestClient from a DM connection string.\n     * This method infers the engine connection string.\n     * For advanced usage, use the constructor that takes a DM connection string and an engine connection string.\n     *\n     * @param dmConnectionString The DM connection string.\n     * @param defaultProps The default ingestion properties.\n     */\n    static fromDmConnectionString(\n        dmConnectionString: KustoConnectionStringBuilder,\n        defaultProps?: IngestionPropertiesInput,\n    ): KustoManagedStreamingIngestClient {\n        if (dmConnectionString.dataSource == null || !dmConnectionString.dataSource.startsWith(ingestPrefix)) {\n            throw new Error(`DM connection string must include the prefix '${ingestPrefix}'`);\n        }\n\n        const engineConnectionString = KustoConnectionStringBuilder.fromExisting(dmConnectionString);\n        engineConnectionString.dataSource = engineConnectionString.dataSource?.replace(ingestPrefix, \"https://\");\n\n        return new KustoManagedStreamingIngestClient(engineConnectionString, dmConnectionString, defaultProps);\n    }\n\n    /**\n     * Creates a KustoManagedStreamingIngestClient from a engine connection string.\n     * This method infers the engine connection string.\n     * For advanced usage, use the constructor that takes an engine connection string and an engine connection string.\n     *\n     * @param engineConnectionString The engine connection string.\n     * @param defaultProps The default ingestion properties.\n     */\n    static fromEngineConnectionString(\n        engineConnectionString: KustoConnectionStringBuilder,\n        defaultProps?: IngestionPropertiesInput,\n    ): KustoManagedStreamingIngestClient {\n        if (engineConnectionString.dataSource == null || engineConnectionString.dataSource.startsWith(ingestPrefix)) {\n            throw new Error(`Engine connection string must not include the prefix '${ingestPrefix}'`);\n        }\n\n        const dmConnectionString = KustoConnectionStringBuilder.fromExisting(engineConnectionString);\n        dmConnectionString.dataSource = dmConnectionString.dataSource?.replace(\"https://\", ingestPrefix);\n\n        return new KustoManagedStreamingIngestClient(engineConnectionString, dmConnectionString, defaultProps);\n    }\n\n    constructor(\n        engineKcsb: string | KustoConnectionStringBuilder,\n        dmKcsb: string | KustoConnectionStringBuilder,\n        defaultProps?: IngestionPropertiesInput,\n        autoCorrectEndpoint: boolean = true,\n    ) {\n        super(defaultProps);\n        this.streamingIngestClient = new StreamingIngestClient(engineKcsb, defaultProps, autoCorrectEndpoint);\n        this.queuedIngestClient = new IngestClient(dmKcsb, defaultProps, autoCorrectEndpoint);\n\n        if (this.streamingIngestClient.defaultDatabase && this.streamingIngestClient.defaultDatabase !== this.queuedIngestClient.defaultDatabase) {\n            throw new Error(\n                `Default database for streaming ingest client (${this.streamingIngestClient.defaultDatabase}) must match default database for queued ingest client (${this.queuedIngestClient.defaultDatabase})`,\n            );\n        }\n\n        this.defaultDatabase = this.streamingIngestClient.defaultDatabase;\n    }\n\n    /**\n     * Use Readable for Node.js and ArrayBuffer in browser\n     */\n    async ingestFromStream(\n        stream: StreamDescriptor | Readable | ArrayBuffer,\n        ingestionProperties?: IngestionPropertiesInput,\n        clientRequestId?: string,\n    ): Promise<KustoResponseDataSet | IngestionResult> {\n        this.ensureOpen();\n        const props = this._getMergedProps(ingestionProperties);\n        let descriptor = stream instanceof StreamDescriptor ? stream : new StreamDescriptor(stream);\n        let result = isNode ? await tryStreamToArray(descriptor.stream as Readable, maxStreamSize) : descriptor.stream;\n        descriptor = new StreamDescriptor(result).merge(descriptor);\n        let streamingResult: Promise<any> | null = null;\n        // tryStreamToArray returns a Buffer in NodeJS impl if stream size is small enouph\n        if ((isNode && result instanceof Buffer) || !isNode) {\n            streamingResult = await this.streamWithRetries(\n                isNode ? (descriptor.size ?? 0) : (descriptor.stream as ArrayBuffer).byteLength,\n                descriptor,\n                props,\n                clientRequestId,\n                result,\n            );\n\n            result = isNode ? readableToStream(result) : descriptor.stream;\n        }\n\n        return streamingResult ?? this.queuedIngestClient.ingestFromStream(new StreamDescriptor(result).merge(descriptor), props);\n    }\n\n    /**\n     * Use string for Node.js and Blob in browser\n     */\n    async ingestFromFile(\n        file: FileDescriptor | string | Blob,\n        ingestionProperties?: IngestionPropertiesInput,\n    ): Promise<KustoResponseDataSet | IngestionResult> {\n        this.ensureOpen();\n\n        const stream = file instanceof FileDescriptor ? await tryFileToBuffer(file) : await tryFileToBuffer(new FileDescriptor(file));\n        return await this.ingestFromStream(stream, ingestionProperties);\n    }\n\n    async ingestFromBlob(\n        blob: string | BlobDescriptor,\n        ingestionProperties?: IngestionPropertiesInput,\n        clientRequestId?: string,\n    ): Promise<KustoResponseDataSet | IngestionResult> {\n        const props = this._getMergedProps(ingestionProperties);\n        const descriptor = blob instanceof BlobDescriptor ? blob : new BlobDescriptor(blob);\n        // No need to check blob size if it was given to us that it's not empty\n        await descriptor.fillSize();\n\n        const streamingResult = await this.streamWithRetries(descriptor.size ?? 0, descriptor, props, clientRequestId);\n        return streamingResult ?? this.queuedIngestClient.ingestFromBlob(descriptor, props);\n    }\n\n    async streamWithRetries(\n        length: number,\n        descriptor: AbstractDescriptor,\n        props?: IngestionPropertiesInput,\n        clientRequestId?: string,\n        stream?: Readable | ArrayBuffer,\n    ): Promise<any> {\n        const isBlob = descriptor instanceof BlobDescriptor;\n        if (length <= maxStreamSize) {\n            // If we get buffer that means it was less than the max size, so we can do streamingIngestion\n            const retry = new ExponentialRetry(attemptCount, this.baseSleepTimeSecs, this.baseJitterSecs);\n            while (retry.shouldTry()) {\n                try {\n                    const sourceId =\n                        clientRequestId ??\n                        `KNC.executeManagedStreamingIngest${isBlob ? \"FromBlob\" : \"FromStream\"};${descriptor.sourceId};${retry.currentAttempt}`;\n                    if (isBlob) {\n                        return await this.streamingIngestClient.ingestFromBlob(descriptor as BlobDescriptor, props, sourceId);\n                    }\n\n                    if (isNode) {\n                        return await this.streamingIngestClient.ingestFromStream(\n                            new StreamDescriptor(readableToStream(stream!)).merge(descriptor as StreamDescriptor),\n                            props,\n                            sourceId,\n                        );\n                    }\n\n                    return await this.streamingIngestClient.ingestFromStream(descriptor as StreamDescriptor, props, sourceId);\n                } catch (err: unknown) {\n                    const oneApiError = err as { \"@permanent\"?: boolean };\n                    if (oneApiError[\"@permanent\"]) {\n                        throw err;\n                    }\n                    await retry.backoff();\n                }\n            }\n\n            stream = isBlob ? undefined : isNode && stream ? readableToStream(stream) : (descriptor as StreamDescriptor).stream;\n        }\n\n        return null;\n    }\n\n    close() {\n        if (!this._isClosed) {\n            this.streamingIngestClient.close();\n            this.queuedIngestClient.close();\n        }\n        super.close();\n    }\n}\n\nexport default KustoManagedStreamingIngestClient;\n"]}

package/dist-esm/src/rankedStorageAccount.js

@@ -1,6 +1,3 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.RankedStorageAccount = void 0;
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
 class StorageAccountStats {
@@ -19,7 +16,7 @@ class StorageAccountStats {
         this.totalCount = 0;
     }
 }
-class RankedStorageAccount {
+export class RankedStorageAccount {
     constructor(accountName, numberOfBuckets, bucketDuration, timeProvider) {
         this.accountName = accountName;
         this.numberOfBuckets = numberOfBuckets;
@@ -69,5 +66,4 @@ class RankedStorageAccount {
         return rank / totalWeight;
     }
 }
-exports.RankedStorageAccount = RankedStorageAccount;
 //# sourceMappingURL=rankedStorageAccount.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"rankedStorageAccount.js","sourceRoot":"","sources":["../../src/rankedStorageAccount.ts"],"names":[],"mappings":";;;AAAA,uCAAuC;AACvC,kCAAkC;AAClC,MAAM,mBAAmB;IAIrB;QACI,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;QACtB,IAAI,CAAC,UAAU,GAAG,CAAC,CAAC;IACxB,CAAC;IAED,SAAS,CAAC,OAAgB;QACtB,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC;QACrB,IAAI,OAAO,EAAE;YACT,IAAI,CAAC,YAAY,IAAI,CAAC,CAAC;SAC1B;IACL,CAAC;IAED,KAAK;QACD,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;QACtB,IAAI,CAAC,UAAU,GAAG,CAAC,CAAC;IACxB,CAAC;CACJ;AAED,MAAa,oBAAoB;IAK7B,YAAoB,WAAmB,EAAU,eAAuB,EAAU,cAAsB,EAAU,YAA0B;QAAxH,gBAAW,GAAX,WAAW,CAAQ;QAAU,oBAAe,GAAf,eAAe,CAAQ;QAAU,mBAAc,GAAd,cAAc,CAAQ;QAAU,iBAAY,GAAZ,YAAY,CAAc;QACxI,IAAI,CAAC,OAAO,GAAG,IAAI,KAAK,CAAsB,eAAe,CAAC,CAAC,IAAI,CAAC,IAAI,mBAAmB,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,mBAAmB,EAAE,CAAC,CAAC;QACpI,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QAC1C,IAAI,CAAC,kBAAkB,GAAG,CAAC,CAAC;IAChC,CAAC;IAED,SAAS,CAAC,OAAgB;QACtB,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACrD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAC7D,CAAC;IAED,cAAc;QACV,OAAO,IAAI,CAAC,WAAW,CAAC;IAC5B,CAAC;IAED,mBAAmB;QACf,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QACxC,MAAM,SAAS,GAAG,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC;QACpD,IAAI,WAAW,GAAG,CAAC,CAAC;QAEpB,IAAI,SAAS,IAAI,IAAI,CAAC,cAAc,EAAE;YAClC,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC;YAClC,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;YAC1F,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE;gBACtC,MAAM,YAAY,GAAG,CAAC,IAAI,CAAC,kBAAkB,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC;gBAC1E,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,KAAK,EAAE,CAAC;aACtC;SACJ;QAED,OAAO,CAAC,IAAI,CAAC,kBAAkB,GAAG,WAAW,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC;IAC1E,CAAC;IAED,OAAO;QACH,IAAI,IAAI,GAAW,CAAC,CAAC;QACrB,IAAI,WAAW,GAAW,CAAC,CAAC;QAE5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE;YAC5C,MAAM,WAAW,GAAW,CAAC,IAAI,CAAC,kBAAkB,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC;YACjF,MAAM,MAAM,GAAwB,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAC9D,IAAI,MAAM,CAAC,UAAU,KAAK,CAAC,EAAE;gBACzB,SAAS;aACZ;YACD,MAAM,WAAW,GAAW,MAAM,CAAC,YAAY,GAAG,MAAM,CAAC,UAAU,CAAC;YACpE,IAAI,IAAI,WAAW,GAAG,CAAC,CAAC;YACxB,WAAW,IAAI,CAAC,CAAC;SACpB;QAED,IAAI,WAAW,KAAK,CAAC,EAAE;YACnB,OAAO,CAAC,CAAC;SACZ;QACD,OAAO,IAAI,GAAG,WAAW,CAAC;IAC9B,CAAC;CACJ;AAzDD,oDAyDC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nclass StorageAccountStats {\n    public successCount: number;\n    public totalCount: number;\n\n    constructor() {\n        this.successCount = 0;\n        this.totalCount = 0;\n    }\n\n    logResult(success: boolean): void {\n        this.totalCount += 1;\n        if (success) {\n            this.successCount += 1;\n        }\n    }\n\n    reset(): void {\n        this.successCount = 0;\n        this.totalCount = 0;\n    }\n}\n\nexport class RankedStorageAccount {\n    private buckets: StorageAccountStats[];\n    private lastUpdateTime: number;\n    private currentBucketIndex: number;\n\n    constructor(private accountName: string, private numberOfBuckets: number, private bucketDuration: number, private timeProvider: () => number) {\n        this.buckets = new Array<StorageAccountStats>(numberOfBuckets).fill(new StorageAccountStats()).map(() => new StorageAccountStats());\n        this.lastUpdateTime = this.timeProvider();\n        this.currentBucketIndex = 0;\n    }\n\n    logResult(success: boolean): void {\n        this.currentBucketIndex = this.adjustForTimePassed();\n        this.buckets[this.currentBucketIndex].logResult(success);\n    }\n\n    getAccountName(): string {\n        return this.accountName;\n    }\n\n    adjustForTimePassed(): number {\n        const currentTime = this.timeProvider();\n        const timeDelta = currentTime - this.lastUpdateTime;\n        let window_size = 0;\n\n        if (timeDelta >= this.bucketDuration) {\n            this.lastUpdateTime = currentTime;\n            window_size = Math.min(Math.floor(timeDelta / this.bucketDuration), this.numberOfBuckets);\n            for (let i = 1; i < window_size + 1; i++) {\n                const indexToReset = (this.currentBucketIndex + i) % this.numberOfBuckets;\n                this.buckets[indexToReset].reset();\n            }\n        }\n\n        return (this.currentBucketIndex + window_size) % this.numberOfBuckets;\n    }\n\n    getRank(): number {\n        let rank: number = 0;\n        let totalWeight: number = 0;\n\n        for (let i = 1; i <= this.numberOfBuckets; i++) {\n            const bucketIndex: number = (this.currentBucketIndex + i) % this.numberOfBuckets;\n            const bucket: StorageAccountStats = this.buckets[bucketIndex];\n            if (bucket.totalCount === 0) {\n                continue;\n            }\n            const successRate: number = bucket.successCount / bucket.totalCount;\n            rank += successRate * i;\n            totalWeight += i;\n        }\n\n        if (totalWeight === 0) {\n            return 1;\n        }\n        return rank / totalWeight;\n    }\n}\n"]}
+{"version":3,"file":"rankedStorageAccount.js","sourceRoot":"","sources":["../../src/rankedStorageAccount.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAClC,MAAM,mBAAmB;IAIrB;QACI,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;QACtB,IAAI,CAAC,UAAU,GAAG,CAAC,CAAC;IACxB,CAAC;IAED,SAAS,CAAC,OAAgB;QACtB,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC;QACrB,IAAI,OAAO,EAAE,CAAC;YACV,IAAI,CAAC,YAAY,IAAI,CAAC,CAAC;QAC3B,CAAC;IACL,CAAC;IAED,KAAK;QACD,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;QACtB,IAAI,CAAC,UAAU,GAAG,CAAC,CAAC;IACxB,CAAC;CACJ;AAED,MAAM,OAAO,oBAAoB;IAK7B,YACY,WAAmB,EACnB,eAAuB,EACvB,cAAsB,EACtB,YAA0B;QAH1B,gBAAW,GAAX,WAAW,CAAQ;QACnB,oBAAe,GAAf,eAAe,CAAQ;QACvB,mBAAc,GAAd,cAAc,CAAQ;QACtB,iBAAY,GAAZ,YAAY,CAAc;QAElC,IAAI,CAAC,OAAO,GAAG,IAAI,KAAK,CAAsB,eAAe,CAAC,CAAC,IAAI,CAAC,IAAI,mBAAmB,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,mBAAmB,EAAE,CAAC,CAAC;QACpI,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QAC1C,IAAI,CAAC,kBAAkB,GAAG,CAAC,CAAC;IAChC,CAAC;IAED,SAAS,CAAC,OAAgB;QACtB,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACrD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAC7D,CAAC;IAED,cAAc;QACV,OAAO,IAAI,CAAC,WAAW,CAAC;IAC5B,CAAC;IAED,mBAAmB;QACf,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QACxC,MAAM,SAAS,GAAG,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC;QACpD,IAAI,WAAW,GAAG,CAAC,CAAC;QAEpB,IAAI,SAAS,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACnC,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC;YAClC,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;YAC1F,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvC,MAAM,YAAY,GAAG,CAAC,IAAI,CAAC,kBAAkB,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC;gBAC1E,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,KAAK,EAAE,CAAC;YACvC,CAAC;QACL,CAAC;QAED,OAAO,CAAC,IAAI,CAAC,kBAAkB,GAAG,WAAW,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC;IAC1E,CAAC;IAED,OAAO;QACH,IAAI,IAAI,GAAW,CAAC,CAAC;QACrB,IAAI,WAAW,GAAW,CAAC,CAAC;QAE5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7C,MAAM,WAAW,GAAW,CAAC,IAAI,CAAC,kBAAkB,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC;YACjF,MAAM,MAAM,GAAwB,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAC9D,IAAI,MAAM,CAAC,UAAU,KAAK,CAAC,EAAE,CAAC;gBAC1B,SAAS;YACb,CAAC;YACD,MAAM,WAAW,GAAW,MAAM,CAAC,YAAY,GAAG,MAAM,CAAC,UAAU,CAAC;YACpE,IAAI,IAAI,WAAW,GAAG,CAAC,CAAC;YACxB,WAAW,IAAI,CAAC,CAAC;QACrB,CAAC;QAED,IAAI,WAAW,KAAK,CAAC,EAAE,CAAC;YACpB,OAAO,CAAC,CAAC;QACb,CAAC;QACD,OAAO,IAAI,GAAG,WAAW,CAAC;IAC9B,CAAC;CACJ","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nclass StorageAccountStats {\n    public successCount: number;\n    public totalCount: number;\n\n    constructor() {\n        this.successCount = 0;\n        this.totalCount = 0;\n    }\n\n    logResult(success: boolean): void {\n        this.totalCount += 1;\n        if (success) {\n            this.successCount += 1;\n        }\n    }\n\n    reset(): void {\n        this.successCount = 0;\n        this.totalCount = 0;\n    }\n}\n\nexport class RankedStorageAccount {\n    private buckets: StorageAccountStats[];\n    private lastUpdateTime: number;\n    private currentBucketIndex: number;\n\n    constructor(\n        private accountName: string,\n        private numberOfBuckets: number,\n        private bucketDuration: number,\n        private timeProvider: () => number,\n    ) {\n        this.buckets = new Array<StorageAccountStats>(numberOfBuckets).fill(new StorageAccountStats()).map(() => new StorageAccountStats());\n        this.lastUpdateTime = this.timeProvider();\n        this.currentBucketIndex = 0;\n    }\n\n    logResult(success: boolean): void {\n        this.currentBucketIndex = this.adjustForTimePassed();\n        this.buckets[this.currentBucketIndex].logResult(success);\n    }\n\n    getAccountName(): string {\n        return this.accountName;\n    }\n\n    adjustForTimePassed(): number {\n        const currentTime = this.timeProvider();\n        const timeDelta = currentTime - this.lastUpdateTime;\n        let window_size = 0;\n\n        if (timeDelta >= this.bucketDuration) {\n            this.lastUpdateTime = currentTime;\n            window_size = Math.min(Math.floor(timeDelta / this.bucketDuration), this.numberOfBuckets);\n            for (let i = 1; i < window_size + 1; i++) {\n                const indexToReset = (this.currentBucketIndex + i) % this.numberOfBuckets;\n                this.buckets[indexToReset].reset();\n            }\n        }\n\n        return (this.currentBucketIndex + window_size) % this.numberOfBuckets;\n    }\n\n    getRank(): number {\n        let rank: number = 0;\n        let totalWeight: number = 0;\n\n        for (let i = 1; i <= this.numberOfBuckets; i++) {\n            const bucketIndex: number = (this.currentBucketIndex + i) % this.numberOfBuckets;\n            const bucket: StorageAccountStats = this.buckets[bucketIndex];\n            if (bucket.totalCount === 0) {\n                continue;\n            }\n            const successRate: number = bucket.successCount / bucket.totalCount;\n            rank += successRate * i;\n            totalWeight += i;\n        }\n\n        if (totalWeight === 0) {\n            return 1;\n        }\n        return rank / totalWeight;\n    }\n}\n"]}

package/dist-esm/src/rankedStorageAccountSet.js

@@ -1,10 +1,7 @@
-"use strict";
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.RankedStorageAccountSet = void 0;
-const rankedStorageAccount_1 = require("./rankedStorageAccount");
-class RankedStorageAccountSet {
+import { RankedStorageAccount } from "./rankedStorageAccount.js";
+export class RankedStorageAccountSet {
     constructor(numberOfBuckets = RankedStorageAccountSet.DefaultNumberOfBuckets, bucketDuration = RankedStorageAccountSet.DefaultBucketDurationInSeconds, tiers = RankedStorageAccountSet.DefaultTiers, timeProvider = RankedStorageAccountSet.DefaultTimeProviderInSeconds) {
         this.numberOfBuckets = numberOfBuckets;
         this.bucketDuration = bucketDuration;
@@ -23,7 +20,7 @@ class RankedStorageAccountSet {
         if (this.accounts.has(accountName)) {
             return;
         }
-        this.accounts.set(accountName, new rankedStorageAccount_1.RankedStorageAccount(accountName, this.numberOfBuckets, this.bucketDuration, this.timeProvider));
+        this.accounts.set(accountName, new RankedStorageAccount(accountName, this.numberOfBuckets, this.bucketDuration, this.timeProvider));
     }
     getStorageAccount(accountName) {
         const account = this.accounts.get(accountName);
@@ -51,7 +48,6 @@ class RankedStorageAccountSet {
         return accountsByTier.flat();
     }
 }
-exports.RankedStorageAccountSet = RankedStorageAccountSet;
 RankedStorageAccountSet.DefaultNumberOfBuckets = 6;
 RankedStorageAccountSet.DefaultBucketDurationInSeconds = 10;
 RankedStorageAccountSet.DefaultTiers = [90, 70, 30, 0];

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

@@ -1 +1 @@
-{"version":3,"file":"rankedStorageAccountSet.js","sourceRoot":"","sources":["../../src/rankedStorageAccountSet.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAElC,iEAA8D;AAE9D,MAAa,uBAAuB;IAUhC,YACY,kBAA0B,uBAAuB,CAAC,sBAAsB,EACxE,iBAAyB,uBAAuB,CAAC,8BAA8B,EAC/E,QAAkB,uBAAuB,CAAC,YAAY,EACtD,eAA6B,uBAAuB,CAAC,4BAA4B;QAHjF,oBAAe,GAAf,eAAe,CAAyD;QACxE,mBAAc,GAAd,cAAc,CAAiE;QAC/E,UAAK,GAAL,KAAK,CAAiD;QACtD,iBAAY,GAAZ,YAAY,CAAqE;QAEzF,IAAI,CAAC,QAAQ,GAAG,IAAI,GAAG,EAAgC,CAAC;IAC5D,CAAC;IAED,kBAAkB,CAAC,WAAmB,EAAE,MAAe;;QACnD,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE;YACjC,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;SACnE;QACD,MAAA,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,0CAAE,SAAS,CAAC,MAAM,CAAC,CAAC;IACtD,CAAC;IAED,sBAAsB,CAAC,WAAmB;QACtC,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE;YAChC,OAAO;SACV;QACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,EAAE,IAAI,2CAAoB,CAAC,WAAW,EAAE,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;IACxI,CAAC;IAED,iBAAiB,CAAC,WAAmB;QACjC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAC/C,IAAI,OAAO,EAAE;YACT,OAAO,OAAO,CAAC;SAClB;QACD,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;IACpE,CAAC;IAED,yBAAyB;QACrB,MAAM,cAAc,GAA6B,IAAI,KAAK,CAAyB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAEtG,kCAAkC;QAClC,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE;YAC1C,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,EAAE,GAAG,GAAG,CAAC;YACrC,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,IAAI,IAAI,CAAC,CAAC;YAC/D,cAAc,CAAC,SAAS,CAAC,GAAG,cAAc,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;YAC5D,cAAc,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;SAC3C;QAED,oBAAoB;QACpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YACxC,IAAI,cAAc,CAAC,CAAC,CAAC,EAAE;gBACnB,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,GAAG,CAAC,CAAC;aACrD;SACJ;QAED,oBAAoB;QACpB,OAAO,cAAc,CAAC,IAAI,EAAE,CAAC;IACjC,CAAC;;AA7DL,0DA8DC;AA7D0B,8CAAsB,GAAW,CAAC,CAAC;AACnC,sDAA8B,GAAW,EAAE,CAAC;AAC5C,oCAAY,GAAa,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACzC,oDAA4B,GAAiB,GAAG,EAAE;IACrE,OAAO,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC;AACvC,CAAC,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { RankedStorageAccount } from \"./rankedStorageAccount\";\n\nexport class RankedStorageAccountSet {\n    public static readonly DefaultNumberOfBuckets: number = 6;\n    public static readonly DefaultBucketDurationInSeconds: number = 10;\n    public static readonly DefaultTiers: number[] = [90, 70, 30, 0];\n    public static readonly DefaultTimeProviderInSeconds: () => number = () => {\n        return new Date().getTime() / 1000;\n    };\n\n    private accounts: Map<string, RankedStorageAccount>;\n\n    constructor(\n        private numberOfBuckets: number = RankedStorageAccountSet.DefaultNumberOfBuckets,\n        private bucketDuration: number = RankedStorageAccountSet.DefaultBucketDurationInSeconds,\n        private tiers: number[] = RankedStorageAccountSet.DefaultTiers,\n        private timeProvider: () => number = RankedStorageAccountSet.DefaultTimeProviderInSeconds\n    ) {\n        this.accounts = new Map<string, RankedStorageAccount>();\n    }\n\n    logResultToAccount(accountName: string, result: boolean) {\n        if (!this.accounts.has(accountName)) {\n            throw new Error(\"Storage account name is not part of the set.\");\n        }\n        this.accounts.get(accountName)?.logResult(result);\n    }\n\n    registerStorageAccount(accountName: string) {\n        if (this.accounts.has(accountName)) {\n            return;\n        }\n        this.accounts.set(accountName, new RankedStorageAccount(accountName, this.numberOfBuckets, this.bucketDuration, this.timeProvider));\n    }\n\n    getStorageAccount(accountName: string): RankedStorageAccount {\n        const account = this.accounts.get(accountName);\n        if (account) {\n            return account;\n        }\n        throw new Error(\"Storage account name is not part of the set.\");\n    }\n\n    getRankedShuffledAccounts(): RankedStorageAccount[] {\n        const accountsByTier: RankedStorageAccount[][] = new Array<RankedStorageAccount[]>(this.tiers.length);\n\n        // Group accounts by tier and rank\n        for (const account of this.accounts.values()) {\n            const rank = account.getRank() * 100;\n            const tierInedx = this.tiers.findIndex((tier) => rank >= tier);\n            accountsByTier[tierInedx] = accountsByTier[tierInedx] || [];\n            accountsByTier[tierInedx].push(account);\n        }\n\n        // Shuffle each tier\n        for (let i = 0; i < this.tiers.length; i++) {\n            if (accountsByTier[i]) {\n                accountsByTier[i].sort(() => Math.random() - 0.5);\n            }\n        }\n\n        // Flatten the array\n        return accountsByTier.flat();\n    }\n}\n"]}
+{"version":3,"file":"rankedStorageAccountSet.js","sourceRoot":"","sources":["../../src/rankedStorageAccountSet.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAEjE,MAAM,OAAO,uBAAuB;IAUhC,YACY,kBAA0B,uBAAuB,CAAC,sBAAsB,EACxE,iBAAyB,uBAAuB,CAAC,8BAA8B,EAC/E,QAAkB,uBAAuB,CAAC,YAAY,EACtD,eAA6B,uBAAuB,CAAC,4BAA4B;QAHjF,oBAAe,GAAf,eAAe,CAAyD;QACxE,mBAAc,GAAd,cAAc,CAAiE;QAC/E,UAAK,GAAL,KAAK,CAAiD;QACtD,iBAAY,GAAZ,YAAY,CAAqE;QAEzF,IAAI,CAAC,QAAQ,GAAG,IAAI,GAAG,EAAgC,CAAC;IAC5D,CAAC;IAED,kBAAkB,CAAC,WAAmB,EAAE,MAAe;;QACnD,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;QACpE,CAAC;QACD,MAAA,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,0CAAE,SAAS,CAAC,MAAM,CAAC,CAAC;IACtD,CAAC;IAED,sBAAsB,CAAC,WAAmB;QACtC,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;YACjC,OAAO;QACX,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,EAAE,IAAI,oBAAoB,CAAC,WAAW,EAAE,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;IACxI,CAAC;IAED,iBAAiB,CAAC,WAAmB;QACjC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAC/C,IAAI,OAAO,EAAE,CAAC;YACV,OAAO,OAAO,CAAC;QACnB,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;IACpE,CAAC;IAED,yBAAyB;QACrB,MAAM,cAAc,GAA6B,IAAI,KAAK,CAAyB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAEtG,kCAAkC;QAClC,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,EAAE,GAAG,GAAG,CAAC;YACrC,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,IAAI,IAAI,CAAC,CAAC;YAC/D,cAAc,CAAC,SAAS,CAAC,GAAG,cAAc,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;YAC5D,cAAc,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5C,CAAC;QAED,oBAAoB;QACpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,IAAI,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC;gBACpB,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,GAAG,CAAC,CAAC;YACtD,CAAC;QACL,CAAC;QAED,oBAAoB;QACpB,OAAO,cAAc,CAAC,IAAI,EAAE,CAAC;IACjC,CAAC;;AA5DsB,8CAAsB,GAAW,CAAC,CAAC;AACnC,sDAA8B,GAAW,EAAE,CAAC;AAC5C,oCAAY,GAAa,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACzC,oDAA4B,GAAiB,GAAG,EAAE;IACrE,OAAO,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC;AACvC,CAAC,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { RankedStorageAccount } from \"./rankedStorageAccount.js\";\n\nexport class RankedStorageAccountSet {\n    public static readonly DefaultNumberOfBuckets: number = 6;\n    public static readonly DefaultBucketDurationInSeconds: number = 10;\n    public static readonly DefaultTiers: number[] = [90, 70, 30, 0];\n    public static readonly DefaultTimeProviderInSeconds: () => number = () => {\n        return new Date().getTime() / 1000;\n    };\n\n    private accounts: Map<string, RankedStorageAccount>;\n\n    constructor(\n        private numberOfBuckets: number = RankedStorageAccountSet.DefaultNumberOfBuckets,\n        private bucketDuration: number = RankedStorageAccountSet.DefaultBucketDurationInSeconds,\n        private tiers: number[] = RankedStorageAccountSet.DefaultTiers,\n        private timeProvider: () => number = RankedStorageAccountSet.DefaultTimeProviderInSeconds,\n    ) {\n        this.accounts = new Map<string, RankedStorageAccount>();\n    }\n\n    logResultToAccount(accountName: string, result: boolean) {\n        if (!this.accounts.has(accountName)) {\n            throw new Error(\"Storage account name is not part of the set.\");\n        }\n        this.accounts.get(accountName)?.logResult(result);\n    }\n\n    registerStorageAccount(accountName: string) {\n        if (this.accounts.has(accountName)) {\n            return;\n        }\n        this.accounts.set(accountName, new RankedStorageAccount(accountName, this.numberOfBuckets, this.bucketDuration, this.timeProvider));\n    }\n\n    getStorageAccount(accountName: string): RankedStorageAccount {\n        const account = this.accounts.get(accountName);\n        if (account) {\n            return account;\n        }\n        throw new Error(\"Storage account name is not part of the set.\");\n    }\n\n    getRankedShuffledAccounts(): RankedStorageAccount[] {\n        const accountsByTier: RankedStorageAccount[][] = new Array<RankedStorageAccount[]>(this.tiers.length);\n\n        // Group accounts by tier and rank\n        for (const account of this.accounts.values()) {\n            const rank = account.getRank() * 100;\n            const tierInedx = this.tiers.findIndex((tier) => rank >= tier);\n            accountsByTier[tierInedx] = accountsByTier[tierInedx] || [];\n            accountsByTier[tierInedx].push(account);\n        }\n\n        // Shuffle each tier\n        for (let i = 0; i < this.tiers.length; i++) {\n            if (accountsByTier[i]) {\n                accountsByTier[i].sort(() => Math.random() - 0.5);\n            }\n        }\n\n        // Flatten the array\n        return accountsByTier.flat();\n    }\n}\n"]}

package/dist-esm/src/resourceManager.js

@@ -1,30 +1,26 @@
-"use strict";
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createStatusTableClient = exports.ResourceManager = exports.IngestClientResources = exports.ResourceURI = exports.ResourceType = void 0;
-const azure_kusto_data_1 = require("azure-kusto-data");
-const retry_1 = require("./retry");
-const storage_blob_1 = require("@azure/storage-blob");
-const data_tables_1 = require("@azure/data-tables");
-const rankedStorageAccountSet_1 = require("./rankedStorageAccountSet");
-const storage_queue_1 = require("@azure/storage-queue");
+import { KustoDataErrors, TimeUtils } from "azure-kusto-data";
+import { ExponentialRetry } from "./retry.js";
+import { ContainerClient } from "@azure/storage-blob";
+import { TableClient } from "@azure/data-tables";
+import { RankedStorageAccountSet } from "./rankedStorageAccountSet.js";
+import { QueueClient } from "@azure/storage-queue";
 const ATTEMPT_COUNT = 4;
-var ResourceType;
+export var ResourceType;
 (function (ResourceType) {
     ResourceType[ResourceType["Queue"] = 0] = "Queue";
     ResourceType[ResourceType["Container"] = 1] = "Container";
     ResourceType[ResourceType["Table"] = 2] = "Table";
-})(ResourceType || (exports.ResourceType = ResourceType = {}));
-class ResourceURI {
+})(ResourceType || (ResourceType = {}));
+export class ResourceURI {
     constructor(uri, accountName, resourceType) {
         this.uri = uri;
         this.accountName = accountName;
         this.resourceType = resourceType;
     }
 }
-exports.ResourceURI = ResourceURI;
-class IngestClientResources {
+export class IngestClientResources {
     constructor(securedReadyForAggregationQueues = null, failedIngestionsQueues = null, successfulIngestionsQueues = null, containers = null, statusTables = null) {
         this.securedReadyForAggregationQueues = securedReadyForAggregationQueues;
         this.failedIngestionsQueues = failedIngestionsQueues;
@@ -37,20 +33,19 @@ class IngestClientResources {
         return resources.reduce((prev, current) => !!(prev && current), true);
     }
 }
-exports.IngestClientResources = IngestClientResources;
-class ResourceManager {
+export class ResourceManager {
     constructor(kustoClient, isBrowser = false) {
         this.kustoClient = kustoClient;
         this.isBrowser = isBrowser;
         this.baseSleepTimeSecs = 1;
         this.baseJitterSecs = 1;
-        this.refreshPeriod = azure_kusto_data_1.TimeUtils.toMilliseconds(1, 0, 0);
-        this.refreshPeriodOnError = azure_kusto_data_1.TimeUtils.toMilliseconds(0, 10, 0);
+        this.refreshPeriod = TimeUtils.toMilliseconds(1, 0, 0);
+        this.refreshPeriodOnError = TimeUtils.toMilliseconds(0, 10, 0);
         this.ingestClientResources = null;
         this.ingestClientResourcesLastUpdate = null;
         this.authorizationContext = null;
         this.authorizationContextLastUpdate = null;
-        this.rankedStorageAccountSet = new rankedStorageAccountSet_1.RankedStorageAccountSet();
+        this.rankedStorageAccountSet = new RankedStorageAccountSet();
     }
     async refreshIngestClientResources() {
         const error = await this.tryRefresh(false);
@@ -60,7 +55,7 @@ class ResourceManager {
         return this.ingestClientResources;
     }
     async getIngestClientResourcesFromService() {
-        const retry = new retry_1.ExponentialRetry(ATTEMPT_COUNT, this.baseSleepTimeSecs, this.baseJitterSecs);
+        const retry = new ExponentialRetry(ATTEMPT_COUNT, this.baseSleepTimeSecs, this.baseJitterSecs);
         while (retry.shouldTry()) {
             try {
                 const cmd = `.get ingestion resources ${this.isBrowser ? `with (EnableBlobCors='true', EnableQueueCors='true', EnableTableCors='true')` : ""}`;
@@ -73,7 +68,7 @@ class ResourceManager {
                 return resoures;
             }
             catch (error) {
-                if (!(error instanceof azure_kusto_data_1.KustoDataErrors.ThrottlingError)) {
+                if (!(error instanceof KustoDataErrors.ThrottlingError)) {
                     throw error;
                 }
                 await retry.backoff();
@@ -88,10 +83,10 @@ class ResourceManager {
             if (typedRow.ResourceTypeName === resourceName) {
                 let accountName = "";
                 if (resourceType === ResourceType.Queue) {
-                    accountName = new storage_queue_1.QueueClient(typedRow.StorageRoot).accountName;
+                    accountName = new QueueClient(typedRow.StorageRoot).accountName;
                 }
                 else if (resourceType === ResourceType.Container) {
-                    accountName = new storage_blob_1.ContainerClient(typedRow.StorageRoot).accountName;
+                    accountName = new ContainerClient(typedRow.StorageRoot).accountName;
                 }
                 result.push(new ResourceURI(typedRow.StorageRoot, accountName, resourceType));
             }
@@ -187,7 +182,7 @@ class ResourceManager {
         return error;
     }
     async getAuthorizationContextFromService() {
-        const retry = new retry_1.ExponentialRetry(ATTEMPT_COUNT, this.baseSleepTimeSecs, this.baseJitterSecs);
+        const retry = new ExponentialRetry(ATTEMPT_COUNT, this.baseSleepTimeSecs, this.baseJitterSecs);
         while (retry.shouldTry()) {
             try {
                 const response = await this.kustoClient.execute("NetDefaultDB", ".get kusto identity token");
@@ -198,7 +193,7 @@ class ResourceManager {
                 return next.value.toJSON().AuthorizationContext;
             }
             catch (error) {
-                if (!(error instanceof azure_kusto_data_1.KustoDataErrors.ThrottlingError)) {
+                if (!(error instanceof KustoDataErrors.ThrottlingError)) {
                     throw error;
                 }
                 await retry.backoff();
@@ -231,7 +226,7 @@ class ResourceManager {
         if (!statusTables) {
             throw new Error("Failed to get status table");
         }
-        return (0, exports.createStatusTableClient)(statusTables[0].uri);
+        return createStatusTableClient(statusTables[0].uri);
     }
     close() {
         this.kustoClient.close();
@@ -240,14 +235,12 @@ class ResourceManager {
         this.rankedStorageAccountSet.logResultToAccount(accountName, success);
     }
 }
-exports.ResourceManager = ResourceManager;
-const createStatusTableClient = (uri) => {
+export const createStatusTableClient = (uri) => {
     const tableUrl = new URL(uri);
     const origin = tableUrl.origin;
     const sasToken = tableUrl.search;
     const tableName = tableUrl.pathname.replace("/", "");
-    return new data_tables_1.TableClient(origin + sasToken, tableName);
+    return new TableClient(origin + sasToken, tableName);
 };
-exports.createStatusTableClient = createStatusTableClient;
-exports.default = ResourceManager;
+export default ResourceManager;
 //# sourceMappingURL=resourceManager.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"resourceManager.js","sourceRoot":"","sources":["../../src/resourceManager.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAElC,uDAAsE;AACtE,mCAA2C;AAC3C,sDAAsD;AACtD,oDAAiD;AACjD,uEAAoE;AACpE,wDAAmD;AAEnD,MAAM,aAAa,GAAG,CAAC,CAAC;AAExB,IAAY,YAIX;AAJD,WAAY,YAAY;IACpB,iDAAK,CAAA;IACL,yDAAS,CAAA;IACT,iDAAK,CAAA;AACT,CAAC,EAJW,YAAY,4BAAZ,YAAY,QAIvB;AAED,MAAa,WAAW;IACpB,YAAqB,GAAW,EAAW,WAAmB,EAAW,YAA0B;QAA9E,QAAG,GAAH,GAAG,CAAQ;QAAW,gBAAW,GAAX,WAAW,CAAQ;QAAW,iBAAY,GAAZ,YAAY,CAAc;IAAG,CAAC;CAC1G;AAFD,kCAEC;AAED,MAAa,qBAAqB;IAC9B,YACa,mCAAyD,IAAI,EAC7D,yBAA+C,IAAI,EACnD,6BAAmD,IAAI,EACvD,aAAmC,IAAI,EACvC,eAAqC,IAAI;QAJzC,qCAAgC,GAAhC,gCAAgC,CAA6B;QAC7D,2BAAsB,GAAtB,sBAAsB,CAA6B;QACnD,+BAA0B,GAA1B,0BAA0B,CAA6B;QACvD,eAAU,GAAV,UAAU,CAA6B;QACvC,iBAAY,GAAZ,YAAY,CAA6B;IACnD,CAAC;IAEJ,KAAK;QACD,MAAM,SAAS,GAAG,CAAC,IAAI,CAAC,gCAAgC,EAAE,IAAI,CAAC,sBAAsB,EAAE,IAAI,CAAC,sBAAsB,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QACxJ,OAAO,SAAS,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,OAAO,CAAC,EAAE,IAAI,CAAC,CAAC;IAC1E,CAAC;CACJ;AAbD,sDAaC;AAED,MAAa,eAAe;IAYxB,YAAqB,WAAmB,EAAW,YAAqB,KAAK;QAAxD,gBAAW,GAAX,WAAW,CAAQ;QAAW,cAAS,GAAT,SAAS,CAAiB;QAJrE,sBAAiB,GAAG,CAAC,CAAC;QACtB,mBAAc,GAAG,CAAC,CAAC;QAIvB,IAAI,CAAC,aAAa,GAAG,4BAAS,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QACvD,IAAI,CAAC,oBAAoB,GAAG,4BAAS,CAAC,cAAc,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAE/D,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;QAClC,IAAI,CAAC,+BAA+B,GAAG,IAAI,CAAC;QAE5C,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QACjC,IAAI,CAAC,8BAA8B,GAAG,IAAI,CAAC;QAE3C,IAAI,CAAC,uBAAuB,GAAG,IAAI,iDAAuB,EAAE,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,4BAA4B;QAC9B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE;YAC7B,MAAM,IAAI,KAAK,CAAC,sDAAsD,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,OAAO,OAAO,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,EAAE,CAAC,CAAC;SAC9G;QAED,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,mCAAmC;QACrC,MAAM,KAAK,GAAG,IAAI,wBAAgB,CAAC,aAAa,EAAE,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QAC/F,OAAO,KAAK,CAAC,SAAS,EAAE,EAAE;YACtB,IAAI;gBACA,MAAM,GAAG,GAAG,4BAA4B,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,8EAA8E,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;gBAC/I,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;gBACrE,MAAM,KAAK,GAAG,QAAQ,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;gBACzC,MAAM,QAAQ,GAAG,IAAI,qBAAqB,CACtC,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,iCAAiC,EAAE,YAAY,CAAC,KAAK,CAAC,EACpF,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,uBAAuB,EAAE,YAAY,CAAC,KAAK,CAAC,EAC1E,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,2BAA2B,EAAE,YAAY,CAAC,KAAK,CAAC,EAC9E,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,aAAa,EAAE,YAAY,CAAC,SAAS,CAAC,EACpE,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,uBAAuB,EAAE,YAAY,CAAC,KAAK,CAAC,CAC7E,CAAC;gBAEF,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,EAAE;oBACnB,MAAM,IAAI,KAAK,CAAC,mEAAmE,CAAC,CAAC;iBACxF;gBAED,OAAO,QAAQ,CAAC;aACnB;YAAC,OAAO,KAAc,EAAE;gBACrB,IAAI,CAAC,CAAC,KAAK,YAAY,kCAAe,CAAC,eAAe,CAAC,EAAE;oBACrD,MAAM,KAAK,CAAC;iBACf;gBACD,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;aACzB;SACJ;QACD,MAAM,IAAI,KAAK,CAAC,6EAA6E,aAAa,SAAS,CAAC,CAAC;IACzH,CAAC;IAED,iBAAiB,CAAC,KAA0B,EAAE,YAAoB,EAAE,YAA0B;QAC1F,MAAM,MAAM,GAAkB,EAAE,CAAC;QACjC,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,IAAI,EAAE,EAAE;YAC5B,MAAM,QAAQ,GAAG,GAGhB,CAAC;YACF,IAAI,QAAQ,CAAC,gBAAgB,KAAK,YAAY,EAAE;gBAC5C,IAAI,WAAW,GAAG,EAAE,CAAC;gBACrB,IAAI,YAAY,KAAK,YAAY,CAAC,KAAK,EAAE;oBACrC,WAAW,GAAG,IAAI,2BAAW,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,WAAW,CAAC;iBACnE;qBAAM,IAAI,YAAY,KAAK,YAAY,CAAC,SAAS,EAAE;oBAChD,WAAW,GAAG,IAAI,8BAAe,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,WAAW,CAAC;iBACvE;gBACD,MAAM,CAAC,IAAI,CAAC,IAAI,WAAW,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC,CAAC;aACjF;SACJ;QACD,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,uBAAuB;QACnB,IAAI,IAAI,CAAC,qBAAqB,IAAI,IAAI,EAAE;YACpC,OAAO;SACV;QAED,aAAa;QACb,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;QACnC,IAAI,IAAI,CAAC,qBAAqB,CAAC,UAAU,IAAI,IAAI,EAAE;YAC/C,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,qBAAqB,CAAC,UAAU,EAAE;gBAC3D,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;aACvC;SACJ;QACD,SAAS;QACT,IAAI,IAAI,CAAC,qBAAqB,CAAC,gCAAgC,IAAI,IAAI,EAAE;YACrE,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,qBAAqB,CAAC,gCAAgC,EAAE;gBAC7E,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;aACnC;SACJ;QAED,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE;YAC5B,IAAI,CAAC,uBAAuB,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC;SAChE;IACL,CAAC;IAED,8BAA8B,CAAC,SAAwB;;QACnD,MAAM,MAAM,GAAG,IAAI,GAAG,EAAyB,CAAC;QAChD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE;YAC9B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE;gBACnC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;aACxC;YACD,MAAA,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,0CAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;SACpD;QACD,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,mCAAmC,CAAC,SAAwB;QACxD,MAAM,kBAAkB,GAAG,IAAI,CAAC,8BAA8B,CAAC,SAAS,CAAC,CAAC;QAC1E,MAAM,qBAAqB,GAAG,IAAI,CAAC,uBAAuB,CAAC,yBAAyB,EAAE,CAAC;QACvF,MAAM,MAAM,GAAG,IAAI,KAAK,EAAiB,CAAC;QAC1C,KAAK,MAAM,OAAO,IAAI,qBAAqB,EAAE;YACzC,MAAM,WAAW,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;YAC7C,IAAI,kBAAkB,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE;gBACrC,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,WAAW,CAAkB,CAAC,CAAC;aACrE;SACJ;QACD,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,uCAAuC,CAAC,SAAwB;QAC5D,MAAM,cAAc,GAAG,IAAI,CAAC,mCAAmC,CAAC,SAAS,CAAC,CAAC;QAC3E,MAAM,MAAM,GAAG,IAAI,KAAK,EAAe,CAAC;QACxC,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,OAAO,MAAM,CAAC,MAAM,GAAG,SAAS,CAAC,MAAM,EAAE;YACrC,MAAM,OAAO,GAAG,cAAc,CAAC,KAAK,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;YAC9D,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE;gBACpB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAiB,CAAC,CAAC;aAC/C;YACD,KAAK,EAAE,CAAC;SACX;QACD,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,2BAA2B;QAC7B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAE1C,IAAI,IAAI,CAAC,oBAAoB,IAAI,IAAI,EAAE;YACnC,MAAM,IAAI,KAAK,CAAC,wDAAwD,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,OAAO,OAAO,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,EAAE,CAAC,CAAC;SAChH;QAED,OAAO,IAAI,CAAC,oBAAoB,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,kBAA2B;;QACxC,MAAM,QAAQ,GAAG,kBAAkB,CAAC,CAAC,CAAC,MAAA,IAAI,CAAC,oBAAoB,0CAAE,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC;QACrG,MAAM,WAAW,GAAG,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC,CAAC,IAAI,CAAC,+BAA+B,CAAC;QACpH,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,KAAK,GAAiB,IAAI,CAAC;QAC/B,IAAI,CAAC,QAAQ,IAAI,CAAC,WAAW,IAAI,WAAW,GAAG,IAAI,CAAC,aAAa,IAAI,GAAG,EAAE;YACtE,IAAI;gBACA,IAAI,kBAAkB,EAAE;oBACpB,IAAI,CAAC,oBAAoB,GAAG,MAAM,IAAI,CAAC,kCAAkC,EAAE,CAAC;oBAC5E,IAAI,CAAC,8BAA8B,GAAG,GAAG,CAAC;iBAC7C;qBAAM;oBACH,IAAI,CAAC,qBAAqB,GAAG,MAAM,IAAI,CAAC,mCAAmC,EAAE,CAAC;oBAC9E,IAAI,CAAC,+BAA+B,GAAG,GAAG,CAAC;oBAC3C,IAAI,CAAC,uBAAuB,EAAE,CAAC;iBAClC;aACJ;YAAC,OAAO,CAAC,EAAE;gBACR,KAAK,GAAG,CAAU,CAAC;aACtB;SACJ;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,kCAAkC;QACpC,MAAM,KAAK,GAAG,IAAI,wBAAgB,CAAC,aAAa,EAAE,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QAC/F,OAAO,KAAK,CAAC,SAAS,EAAE,EAAE;YACtB,IAAI;gBACA,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,cAAc,EAAE,2BAA2B,CAAC,CAAC;gBAC7F,MAAM,IAAI,GAAG,QAAQ,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;gBACtD,IAAI,IAAI,CAAC,IAAI,EAAE;oBACX,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;iBAC9E;gBACD,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,EAAoC,CAAC,oBAAoB,CAAC;aACrF;YAAC,OAAO,KAAc,EAAE;gBACrB,IAAI,CAAC,CAAC,KAAK,YAAY,kCAAe,CAAC,eAAe,CAAC,EAAE;oBACrD,MAAM,KAAK,CAAC;iBACf;gBACD,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;aACzB;SACJ;QACD,MAAM,IAAI,KAAK,CAAC,wEAAwE,aAAa,SAAS,CAAC,CAAC;IACpH,CAAC;IAED,KAAK,CAAC,kBAAkB;QACpB,MAAM,MAAM,GAAG,CAAC,MAAM,IAAI,CAAC,4BAA4B,EAAE,CAAC,CAAC,gCAAgC,CAAC;QAC5F,OAAO,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,uCAAuC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAChF,CAAC;IAED,KAAK,CAAC,yBAAyB;QAC3B,OAAO,CAAC,MAAM,IAAI,CAAC,4BAA4B,EAAE,CAAC,CAAC,sBAAsB,CAAC;IAC9E,CAAC;IAED,KAAK,CAAC,6BAA6B;QAC/B,OAAO,CAAC,MAAM,IAAI,CAAC,4BAA4B,EAAE,CAAC,CAAC,0BAA0B,CAAC;IAClF,CAAC;IAED,KAAK,CAAC,aAAa;QACf,MAAM,UAAU,GAAG,CAAC,MAAM,IAAI,CAAC,4BAA4B,EAAE,CAAC,CAAC,UAAU,CAAC;QAC1E,OAAO,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,uCAAuC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IACxF,CAAC;IAED,KAAK,CAAC,uBAAuB;QACzB,OAAO,IAAI,CAAC,2BAA2B,EAAE,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,eAAe;QACjB,OAAO,CAAC,MAAM,IAAI,CAAC,4BAA4B,EAAE,CAAC,CAAC,YAAY,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,iBAAiB;QACnB,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,IAAI,CAAC,YAAY,EAAE;YACf,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;SACjD;QAED,OAAO,IAAA,+BAAuB,EAAC,YAAa,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IACzD,CAAC;IAED,KAAK;QACD,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;IAC7B,CAAC;IAED,yBAAyB,CAAC,WAAmB,EAAE,OAAgB;QAC3D,IAAI,CAAC,uBAAuB,CAAC,kBAAkB,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;IAC1E,CAAC;CACJ;AAjPD,0CAiPC;AAEM,MAAM,uBAAuB,GAAG,CAAC,GAAW,EAAe,EAAE;IAChE,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;IAC9B,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC;IAC/B,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC;IACjC,MAAM,SAAS,GAAG,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IACrD,OAAO,IAAI,yBAAW,CAAC,MAAM,GAAG,QAAQ,EAAE,SAAS,CAAC,CAAC;AACzD,CAAC,CAAC;AANW,QAAA,uBAAuB,2BAMlC;AAEF,kBAAe,eAAe,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { Client, KustoDataErrors, TimeUtils } from \"azure-kusto-data\";\nimport { ExponentialRetry } from \"./retry\";\nimport { ContainerClient } from \"@azure/storage-blob\";\nimport { TableClient } from \"@azure/data-tables\";\nimport { RankedStorageAccountSet } from \"./rankedStorageAccountSet\";\nimport { QueueClient } from \"@azure/storage-queue\";\n\nconst ATTEMPT_COUNT = 4;\n\nexport enum ResourceType {\n    Queue,\n    Container,\n    Table,\n}\n\nexport class ResourceURI {\n    constructor(readonly uri: string, readonly accountName: string, readonly resourceType: ResourceType) {}\n}\n\nexport class IngestClientResources {\n    constructor(\n        readonly securedReadyForAggregationQueues: ResourceURI[] | null = null,\n        readonly failedIngestionsQueues: ResourceURI[] | null = null,\n        readonly successfulIngestionsQueues: ResourceURI[] | null = null,\n        readonly containers: ResourceURI[] | null = null,\n        readonly statusTables: ResourceURI[] | null = null\n    ) {}\n\n    valid() {\n        const resources = [this.securedReadyForAggregationQueues, this.failedIngestionsQueues, this.failedIngestionsQueues, this.containers, this.statusTables];\n        return resources.reduce((prev, current) => !!(prev && current), true);\n    }\n}\n\nexport class ResourceManager {\n    public readonly refreshPeriod: number;\n    public refreshPeriodOnError: number;\n    public ingestClientResources: IngestClientResources | null;\n    public ingestClientResourcesLastUpdate: number | null;\n    public authorizationContext: string | null;\n    public authorizationContextLastUpdate: number | null;\n\n    private baseSleepTimeSecs = 1;\n    private baseJitterSecs = 1;\n    private rankedStorageAccountSet: RankedStorageAccountSet;\n\n    constructor(readonly kustoClient: Client, readonly isBrowser: boolean = false) {\n        this.refreshPeriod = TimeUtils.toMilliseconds(1, 0, 0);\n        this.refreshPeriodOnError = TimeUtils.toMilliseconds(0, 10, 0);\n\n        this.ingestClientResources = null;\n        this.ingestClientResourcesLastUpdate = null;\n\n        this.authorizationContext = null;\n        this.authorizationContextLastUpdate = null;\n\n        this.rankedStorageAccountSet = new RankedStorageAccountSet();\n    }\n\n    async refreshIngestClientResources(): Promise<IngestClientResources> {\n        const error = await this.tryRefresh(false);\n        if (!this.ingestClientResources) {\n            throw new Error(`Failed to fetch ingestion resources from service.  ${error?.message}.\\n ${error?.stack}`);\n        }\n\n        return this.ingestClientResources;\n    }\n\n    async getIngestClientResourcesFromService(): Promise<IngestClientResources> {\n        const retry = new ExponentialRetry(ATTEMPT_COUNT, this.baseSleepTimeSecs, this.baseJitterSecs);\n        while (retry.shouldTry()) {\n            try {\n                const cmd = `.get ingestion resources ${this.isBrowser ? `with (EnableBlobCors='true', EnableQueueCors='true', EnableTableCors='true')` : \"\"}`;\n                const response = await this.kustoClient.execute(\"NetDefaultDB\", cmd);\n                const table = response.primaryResults[0];\n                const resoures = new IngestClientResources(\n                    this.getResourceByName(table, \"SecuredReadyForAggregationQueue\", ResourceType.Queue),\n                    this.getResourceByName(table, \"FailedIngestionsQueue\", ResourceType.Queue),\n                    this.getResourceByName(table, \"SuccessfulIngestionsQueue\", ResourceType.Queue),\n                    this.getResourceByName(table, \"TempStorage\", ResourceType.Container),\n                    this.getResourceByName(table, \"IngestionsStatusTable\", ResourceType.Table)\n                );\n\n                if (!resoures.valid()) {\n                    throw new Error(\"Unexpected error occured - fetched data returned missing resource\");\n                }\n\n                return resoures;\n            } catch (error: unknown) {\n                if (!(error instanceof KustoDataErrors.ThrottlingError)) {\n                    throw error;\n                }\n                await retry.backoff();\n            }\n        }\n        throw new Error(`Failed to get ingestion resources from server - the request was throttled ${ATTEMPT_COUNT} times.`);\n    }\n\n    getResourceByName(table: { rows: () => any }, resourceName: string, resourceType: ResourceType): ResourceURI[] {\n        const result: ResourceURI[] = [];\n        for (const row of table.rows()) {\n            const typedRow = row as {\n                ResourceTypeName: string;\n                StorageRoot: string;\n            };\n            if (typedRow.ResourceTypeName === resourceName) {\n                let accountName = \"\";\n                if (resourceType === ResourceType.Queue) {\n                    accountName = new QueueClient(typedRow.StorageRoot).accountName;\n                } else if (resourceType === ResourceType.Container) {\n                    accountName = new ContainerClient(typedRow.StorageRoot).accountName;\n                }\n                result.push(new ResourceURI(typedRow.StorageRoot, accountName, resourceType));\n            }\n        }\n        return result;\n    }\n\n    pupulateStorageAccounts(): void {\n        if (this.ingestClientResources == null) {\n            return;\n        }\n\n        // containers\n        const accounts = new Set<string>();\n        if (this.ingestClientResources.containers != null) {\n            for (const container of this.ingestClientResources.containers) {\n                accounts.add(container.accountName);\n            }\n        }\n        // queues\n        if (this.ingestClientResources.securedReadyForAggregationQueues != null) {\n            for (const queue of this.ingestClientResources.securedReadyForAggregationQueues) {\n                accounts.add(queue.accountName);\n            }\n        }\n\n        for (const account of accounts) {\n            this.rankedStorageAccountSet.registerStorageAccount(account);\n        }\n    }\n\n    groupResourcesByStorageAccount(resources: ResourceURI[]): Map<string, ResourceURI[]> {\n        const result = new Map<string, ResourceURI[]>();\n        for (const resource of resources) {\n            if (!result.has(resource.accountName)) {\n                result.set(resource.accountName, []);\n            }\n            result.get(resource.accountName)?.push(resource);\n        }\n        return result;\n    }\n\n    getRankedAndShuffledStorageAccounts(resources: ResourceURI[]): ResourceURI[][] {\n        const resourcesByAccount = this.groupResourcesByStorageAccount(resources);\n        const rankedStorageAccounts = this.rankedStorageAccountSet.getRankedShuffledAccounts();\n        const result = new Array<ResourceURI[]>();\n        for (const account of rankedStorageAccounts) {\n            const accountName = account.getAccountName();\n            if (resourcesByAccount.has(accountName)) {\n                result.push(resourcesByAccount.get(accountName) as ResourceURI[]);\n            }\n        }\n        return result;\n    }\n\n    getRoundRobinRankedAndShuffledResources(resources: ResourceURI[]): ResourceURI[] {\n        const rankedAccounts = this.getRankedAndShuffledStorageAccounts(resources);\n        const result = new Array<ResourceURI>();\n        let index = 0;\n        while (result.length < resources.length) {\n            const account = rankedAccounts[index % rankedAccounts.length];\n            if (account.length > 0) {\n                result.push(account.shift() as ResourceURI);\n            }\n            index++;\n        }\n        return result;\n    }\n\n    async refreshAuthorizationContext(): Promise<string> {\n        const error = await this.tryRefresh(true);\n\n        if (this.authorizationContext == null) {\n            throw new Error(`Failed to fetch Authorization context from service.  ${error?.message}.\\n ${error?.stack}`);\n        }\n\n        return this.authorizationContext;\n    }\n\n    async tryRefresh(isAuthContextFetch: boolean): Promise<Error | null> {\n        const resource = isAuthContextFetch ? this.authorizationContext?.trim() : this.ingestClientResources;\n        const lastRefresh = isAuthContextFetch ? this.authorizationContextLastUpdate : this.ingestClientResourcesLastUpdate;\n        const now = Date.now();\n        let error: Error | null = null;\n        if (!resource || !lastRefresh || lastRefresh + this.refreshPeriod <= now) {\n            try {\n                if (isAuthContextFetch) {\n                    this.authorizationContext = await this.getAuthorizationContextFromService();\n                    this.authorizationContextLastUpdate = now;\n                } else {\n                    this.ingestClientResources = await this.getIngestClientResourcesFromService();\n                    this.ingestClientResourcesLastUpdate = now;\n                    this.pupulateStorageAccounts();\n                }\n            } catch (e) {\n                error = e as Error;\n            }\n        }\n\n        return error;\n    }\n\n    async getAuthorizationContextFromService() {\n        const retry = new ExponentialRetry(ATTEMPT_COUNT, this.baseSleepTimeSecs, this.baseJitterSecs);\n        while (retry.shouldTry()) {\n            try {\n                const response = await this.kustoClient.execute(\"NetDefaultDB\", \".get kusto identity token\");\n                const next = response.primaryResults[0].rows().next();\n                if (next.done) {\n                    throw new Error(\"Failed to get authorization context - got empty results\");\n                }\n                return next.value.toJSON<{ AuthorizationContext: string }>().AuthorizationContext;\n            } catch (error: unknown) {\n                if (!(error instanceof KustoDataErrors.ThrottlingError)) {\n                    throw error;\n                }\n                await retry.backoff();\n            }\n        }\n        throw new Error(`Failed to get identity token from server - the request was throttled ${ATTEMPT_COUNT} times.`);\n    }\n\n    async getIngestionQueues(): Promise<ResourceURI[] | null> {\n        const queues = (await this.refreshIngestClientResources()).securedReadyForAggregationQueues;\n        return queues ? this.getRoundRobinRankedAndShuffledResources(queues) : null;\n    }\n\n    async getFailedIngestionsQueues(): Promise<ResourceURI[] | null> {\n        return (await this.refreshIngestClientResources()).failedIngestionsQueues;\n    }\n\n    async getSuccessfulIngestionsQueues(): Promise<ResourceURI[] | null> {\n        return (await this.refreshIngestClientResources()).successfulIngestionsQueues;\n    }\n\n    async getContainers(): Promise<ResourceURI[] | null> {\n        const containers = (await this.refreshIngestClientResources()).containers;\n        return containers ? this.getRoundRobinRankedAndShuffledResources(containers) : null;\n    }\n\n    async getAuthorizationContext(): Promise<string> {\n        return this.refreshAuthorizationContext();\n    }\n\n    async getStatusTables(): Promise<ResourceURI[] | null> {\n        return (await this.refreshIngestClientResources()).statusTables;\n    }\n\n    async createStatusTable() {\n        const statusTables = await this.getStatusTables();\n        if (!statusTables) {\n            throw new Error(\"Failed to get status table\");\n        }\n\n        return createStatusTableClient(statusTables![0].uri);\n    }\n\n    close(): void {\n        this.kustoClient.close();\n    }\n\n    reportResourceUsageResult(accountName: string, success: boolean): void {\n        this.rankedStorageAccountSet.logResultToAccount(accountName, success);\n    }\n}\n\nexport const createStatusTableClient = (uri: string): TableClient => {\n    const tableUrl = new URL(uri);\n    const origin = tableUrl.origin;\n    const sasToken = tableUrl.search;\n    const tableName = tableUrl.pathname.replace(\"/\", \"\");\n    return new TableClient(origin + sasToken, tableName);\n};\n\nexport default ResourceManager;\n"]}
+{"version":3,"file":"resourceManager.js","sourceRoot":"","sources":["../../src/resourceManager.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAU,eAAe,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACtE,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,uBAAuB,EAAE,MAAM,8BAA8B,CAAC;AACvE,OAAO,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAEnD,MAAM,aAAa,GAAG,CAAC,CAAC;AAExB,MAAM,CAAN,IAAY,YAIX;AAJD,WAAY,YAAY;IACpB,iDAAK,CAAA;IACL,yDAAS,CAAA;IACT,iDAAK,CAAA;AACT,CAAC,EAJW,YAAY,KAAZ,YAAY,QAIvB;AAED,MAAM,OAAO,WAAW;IACpB,YACa,GAAW,EACX,WAAmB,EACnB,YAA0B;QAF1B,QAAG,GAAH,GAAG,CAAQ;QACX,gBAAW,GAAX,WAAW,CAAQ;QACnB,iBAAY,GAAZ,YAAY,CAAc;IACpC,CAAC;CACP;AAED,MAAM,OAAO,qBAAqB;IAC9B,YACa,mCAAyD,IAAI,EAC7D,yBAA+C,IAAI,EACnD,6BAAmD,IAAI,EACvD,aAAmC,IAAI,EACvC,eAAqC,IAAI;QAJzC,qCAAgC,GAAhC,gCAAgC,CAA6B;QAC7D,2BAAsB,GAAtB,sBAAsB,CAA6B;QACnD,+BAA0B,GAA1B,0BAA0B,CAA6B;QACvD,eAAU,GAAV,UAAU,CAA6B;QACvC,iBAAY,GAAZ,YAAY,CAA6B;IACnD,CAAC;IAEJ,KAAK;QACD,MAAM,SAAS,GAAG,CAAC,IAAI,CAAC,gCAAgC,EAAE,IAAI,CAAC,sBAAsB,EAAE,IAAI,CAAC,sBAAsB,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QACxJ,OAAO,SAAS,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,OAAO,CAAC,EAAE,IAAI,CAAC,CAAC;IAC1E,CAAC;CACJ;AAED,MAAM,OAAO,eAAe;IAYxB,YACa,WAAmB,EACnB,YAAqB,KAAK;QAD1B,gBAAW,GAAX,WAAW,CAAQ;QACnB,cAAS,GAAT,SAAS,CAAiB;QAN/B,sBAAiB,GAAG,CAAC,CAAC;QACtB,mBAAc,GAAG,CAAC,CAAC;QAOvB,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QACvD,IAAI,CAAC,oBAAoB,GAAG,SAAS,CAAC,cAAc,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAE/D,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;QAClC,IAAI,CAAC,+BAA+B,GAAG,IAAI,CAAC;QAE5C,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QACjC,IAAI,CAAC,8BAA8B,GAAG,IAAI,CAAC;QAE3C,IAAI,CAAC,uBAAuB,GAAG,IAAI,uBAAuB,EAAE,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,4BAA4B;QAC9B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,sDAAsD,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,OAAO,OAAO,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,EAAE,CAAC,CAAC;QAC/G,CAAC;QAED,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,mCAAmC;QACrC,MAAM,KAAK,GAAG,IAAI,gBAAgB,CAAC,aAAa,EAAE,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QAC/F,OAAO,KAAK,CAAC,SAAS,EAAE,EAAE,CAAC;YACvB,IAAI,CAAC;gBACD,MAAM,GAAG,GAAG,4BAA4B,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,8EAA8E,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;gBAC/I,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;gBACrE,MAAM,KAAK,GAAG,QAAQ,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;gBACzC,MAAM,QAAQ,GAAG,IAAI,qBAAqB,CACtC,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,iCAAiC,EAAE,YAAY,CAAC,KAAK,CAAC,EACpF,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,uBAAuB,EAAE,YAAY,CAAC,KAAK,CAAC,EAC1E,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,2BAA2B,EAAE,YAAY,CAAC,KAAK,CAAC,EAC9E,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,aAAa,EAAE,YAAY,CAAC,SAAS,CAAC,EACpE,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,uBAAuB,EAAE,YAAY,CAAC,KAAK,CAAC,CAC7E,CAAC;gBAEF,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,EAAE,CAAC;oBACpB,MAAM,IAAI,KAAK,CAAC,mEAAmE,CAAC,CAAC;gBACzF,CAAC;gBAED,OAAO,QAAQ,CAAC;YACpB,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACtB,IAAI,CAAC,CAAC,KAAK,YAAY,eAAe,CAAC,eAAe,CAAC,EAAE,CAAC;oBACtD,MAAM,KAAK,CAAC;gBAChB,CAAC;gBACD,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;YAC1B,CAAC;QACL,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,6EAA6E,aAAa,SAAS,CAAC,CAAC;IACzH,CAAC;IAED,iBAAiB,CAAC,KAA0B,EAAE,YAAoB,EAAE,YAA0B;QAC1F,MAAM,MAAM,GAAkB,EAAE,CAAC;QACjC,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAG,GAGhB,CAAC;YACF,IAAI,QAAQ,CAAC,gBAAgB,KAAK,YAAY,EAAE,CAAC;gBAC7C,IAAI,WAAW,GAAG,EAAE,CAAC;gBACrB,IAAI,YAAY,KAAK,YAAY,CAAC,KAAK,EAAE,CAAC;oBACtC,WAAW,GAAG,IAAI,WAAW,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,WAAW,CAAC;gBACpE,CAAC;qBAAM,IAAI,YAAY,KAAK,YAAY,CAAC,SAAS,EAAE,CAAC;oBACjD,WAAW,GAAG,IAAI,eAAe,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,WAAW,CAAC;gBACxE,CAAC;gBACD,MAAM,CAAC,IAAI,CAAC,IAAI,WAAW,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC,CAAC;YAClF,CAAC;QACL,CAAC;QACD,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,uBAAuB;QACnB,IAAI,IAAI,CAAC,qBAAqB,IAAI,IAAI,EAAE,CAAC;YACrC,OAAO;QACX,CAAC;QAED,aAAa;QACb,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;QACnC,IAAI,IAAI,CAAC,qBAAqB,CAAC,UAAU,IAAI,IAAI,EAAE,CAAC;YAChD,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,qBAAqB,CAAC,UAAU,EAAE,CAAC;gBAC5D,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YACxC,CAAC;QACL,CAAC;QACD,SAAS;QACT,IAAI,IAAI,CAAC,qBAAqB,CAAC,gCAAgC,IAAI,IAAI,EAAE,CAAC;YACtE,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,qBAAqB,CAAC,gCAAgC,EAAE,CAAC;gBAC9E,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YACpC,CAAC;QACL,CAAC;QAED,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC7B,IAAI,CAAC,uBAAuB,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC;QACjE,CAAC;IACL,CAAC;IAED,8BAA8B,CAAC,SAAwB;;QACnD,MAAM,MAAM,GAAG,IAAI,GAAG,EAAyB,CAAC;QAChD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YAC/B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;gBACpC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YACzC,CAAC;YACD,MAAA,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,0CAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,mCAAmC,CAAC,SAAwB;QACxD,MAAM,kBAAkB,GAAG,IAAI,CAAC,8BAA8B,CAAC,SAAS,CAAC,CAAC;QAC1E,MAAM,qBAAqB,GAAG,IAAI,CAAC,uBAAuB,CAAC,yBAAyB,EAAE,CAAC;QACvF,MAAM,MAAM,GAAG,IAAI,KAAK,EAAiB,CAAC;QAC1C,KAAK,MAAM,OAAO,IAAI,qBAAqB,EAAE,CAAC;YAC1C,MAAM,WAAW,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;YAC7C,IAAI,kBAAkB,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;gBACtC,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,WAAW,CAAkB,CAAC,CAAC;YACtE,CAAC;QACL,CAAC;QACD,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,uCAAuC,CAAC,SAAwB;QAC5D,MAAM,cAAc,GAAG,IAAI,CAAC,mCAAmC,CAAC,SAAS,CAAC,CAAC;QAC3E,MAAM,MAAM,GAAG,IAAI,KAAK,EAAe,CAAC;QACxC,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,OAAO,MAAM,CAAC,MAAM,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC;YACtC,MAAM,OAAO,GAAG,cAAc,CAAC,KAAK,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;YAC9D,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACrB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAiB,CAAC,CAAC;YAChD,CAAC;YACD,KAAK,EAAE,CAAC;QACZ,CAAC;QACD,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,2BAA2B;QAC7B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAE1C,IAAI,IAAI,CAAC,oBAAoB,IAAI,IAAI,EAAE,CAAC;YACpC,MAAM,IAAI,KAAK,CAAC,wDAAwD,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,OAAO,OAAO,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,EAAE,CAAC,CAAC;QACjH,CAAC;QAED,OAAO,IAAI,CAAC,oBAAoB,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,kBAA2B;;QACxC,MAAM,QAAQ,GAAG,kBAAkB,CAAC,CAAC,CAAC,MAAA,IAAI,CAAC,oBAAoB,0CAAE,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC;QACrG,MAAM,WAAW,GAAG,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC,CAAC,IAAI,CAAC,+BAA+B,CAAC;QACpH,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,KAAK,GAAiB,IAAI,CAAC;QAC/B,IAAI,CAAC,QAAQ,IAAI,CAAC,WAAW,IAAI,WAAW,GAAG,IAAI,CAAC,aAAa,IAAI,GAAG,EAAE,CAAC;YACvE,IAAI,CAAC;gBACD,IAAI,kBAAkB,EAAE,CAAC;oBACrB,IAAI,CAAC,oBAAoB,GAAG,MAAM,IAAI,CAAC,kCAAkC,EAAE,CAAC;oBAC5E,IAAI,CAAC,8BAA8B,GAAG,GAAG,CAAC;gBAC9C,CAAC;qBAAM,CAAC;oBACJ,IAAI,CAAC,qBAAqB,GAAG,MAAM,IAAI,CAAC,mCAAmC,EAAE,CAAC;oBAC9E,IAAI,CAAC,+BAA+B,GAAG,GAAG,CAAC;oBAC3C,IAAI,CAAC,uBAAuB,EAAE,CAAC;gBACnC,CAAC;YACL,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACT,KAAK,GAAG,CAAU,CAAC;YACvB,CAAC;QACL,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,kCAAkC;QACpC,MAAM,KAAK,GAAG,IAAI,gBAAgB,CAAC,aAAa,EAAE,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QAC/F,OAAO,KAAK,CAAC,SAAS,EAAE,EAAE,CAAC;YACvB,IAAI,CAAC;gBACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,cAAc,EAAE,2BAA2B,CAAC,CAAC;gBAC7F,MAAM,IAAI,GAAG,QAAQ,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;gBACtD,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;oBACZ,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;gBAC/E,CAAC;gBACD,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,EAAoC,CAAC,oBAAoB,CAAC;YACtF,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACtB,IAAI,CAAC,CAAC,KAAK,YAAY,eAAe,CAAC,eAAe,CAAC,EAAE,CAAC;oBACtD,MAAM,KAAK,CAAC;gBAChB,CAAC;gBACD,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;YAC1B,CAAC;QACL,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,wEAAwE,aAAa,SAAS,CAAC,CAAC;IACpH,CAAC;IAED,KAAK,CAAC,kBAAkB;QACpB,MAAM,MAAM,GAAG,CAAC,MAAM,IAAI,CAAC,4BAA4B,EAAE,CAAC,CAAC,gCAAgC,CAAC;QAC5F,OAAO,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,uCAAuC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAChF,CAAC;IAED,KAAK,CAAC,yBAAyB;QAC3B,OAAO,CAAC,MAAM,IAAI,CAAC,4BAA4B,EAAE,CAAC,CAAC,sBAAsB,CAAC;IAC9E,CAAC;IAED,KAAK,CAAC,6BAA6B;QAC/B,OAAO,CAAC,MAAM,IAAI,CAAC,4BAA4B,EAAE,CAAC,CAAC,0BAA0B,CAAC;IAClF,CAAC;IAED,KAAK,CAAC,aAAa;QACf,MAAM,UAAU,GAAG,CAAC,MAAM,IAAI,CAAC,4BAA4B,EAAE,CAAC,CAAC,UAAU,CAAC;QAC1E,OAAO,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,uCAAuC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IACxF,CAAC;IAED,KAAK,CAAC,uBAAuB;QACzB,OAAO,IAAI,CAAC,2BAA2B,EAAE,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,eAAe;QACjB,OAAO,CAAC,MAAM,IAAI,CAAC,4BAA4B,EAAE,CAAC,CAAC,YAAY,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,iBAAiB;QACnB,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,IAAI,CAAC,YAAY,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAClD,CAAC;QAED,OAAO,uBAAuB,CAAC,YAAa,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IACzD,CAAC;IAED,KAAK;QACD,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;IAC7B,CAAC;IAED,yBAAyB,CAAC,WAAmB,EAAE,OAAgB;QAC3D,IAAI,CAAC,uBAAuB,CAAC,kBAAkB,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;IAC1E,CAAC;CACJ;AAED,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,GAAW,EAAe,EAAE;IAChE,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;IAC9B,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC;IAC/B,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC;IACjC,MAAM,SAAS,GAAG,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IACrD,OAAO,IAAI,WAAW,CAAC,MAAM,GAAG,QAAQ,EAAE,SAAS,CAAC,CAAC;AACzD,CAAC,CAAC;AAEF,eAAe,eAAe,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { Client, KustoDataErrors, TimeUtils } from \"azure-kusto-data\";\nimport { ExponentialRetry } from \"./retry.js\";\nimport { ContainerClient } from \"@azure/storage-blob\";\nimport { TableClient } from \"@azure/data-tables\";\nimport { RankedStorageAccountSet } from \"./rankedStorageAccountSet.js\";\nimport { QueueClient } from \"@azure/storage-queue\";\n\nconst ATTEMPT_COUNT = 4;\n\nexport enum ResourceType {\n    Queue,\n    Container,\n    Table,\n}\n\nexport class ResourceURI {\n    constructor(\n        readonly uri: string,\n        readonly accountName: string,\n        readonly resourceType: ResourceType,\n    ) {}\n}\n\nexport class IngestClientResources {\n    constructor(\n        readonly securedReadyForAggregationQueues: ResourceURI[] | null = null,\n        readonly failedIngestionsQueues: ResourceURI[] | null = null,\n        readonly successfulIngestionsQueues: ResourceURI[] | null = null,\n        readonly containers: ResourceURI[] | null = null,\n        readonly statusTables: ResourceURI[] | null = null,\n    ) {}\n\n    valid() {\n        const resources = [this.securedReadyForAggregationQueues, this.failedIngestionsQueues, this.failedIngestionsQueues, this.containers, this.statusTables];\n        return resources.reduce((prev, current) => !!(prev && current), true);\n    }\n}\n\nexport class ResourceManager {\n    public readonly refreshPeriod: number;\n    public refreshPeriodOnError: number;\n    public ingestClientResources: IngestClientResources | null;\n    public ingestClientResourcesLastUpdate: number | null;\n    public authorizationContext: string | null;\n    public authorizationContextLastUpdate: number | null;\n\n    private baseSleepTimeSecs = 1;\n    private baseJitterSecs = 1;\n    private rankedStorageAccountSet: RankedStorageAccountSet;\n\n    constructor(\n        readonly kustoClient: Client,\n        readonly isBrowser: boolean = false,\n    ) {\n        this.refreshPeriod = TimeUtils.toMilliseconds(1, 0, 0);\n        this.refreshPeriodOnError = TimeUtils.toMilliseconds(0, 10, 0);\n\n        this.ingestClientResources = null;\n        this.ingestClientResourcesLastUpdate = null;\n\n        this.authorizationContext = null;\n        this.authorizationContextLastUpdate = null;\n\n        this.rankedStorageAccountSet = new RankedStorageAccountSet();\n    }\n\n    async refreshIngestClientResources(): Promise<IngestClientResources> {\n        const error = await this.tryRefresh(false);\n        if (!this.ingestClientResources) {\n            throw new Error(`Failed to fetch ingestion resources from service.  ${error?.message}.\\n ${error?.stack}`);\n        }\n\n        return this.ingestClientResources;\n    }\n\n    async getIngestClientResourcesFromService(): Promise<IngestClientResources> {\n        const retry = new ExponentialRetry(ATTEMPT_COUNT, this.baseSleepTimeSecs, this.baseJitterSecs);\n        while (retry.shouldTry()) {\n            try {\n                const cmd = `.get ingestion resources ${this.isBrowser ? `with (EnableBlobCors='true', EnableQueueCors='true', EnableTableCors='true')` : \"\"}`;\n                const response = await this.kustoClient.execute(\"NetDefaultDB\", cmd);\n                const table = response.primaryResults[0];\n                const resoures = new IngestClientResources(\n                    this.getResourceByName(table, \"SecuredReadyForAggregationQueue\", ResourceType.Queue),\n                    this.getResourceByName(table, \"FailedIngestionsQueue\", ResourceType.Queue),\n                    this.getResourceByName(table, \"SuccessfulIngestionsQueue\", ResourceType.Queue),\n                    this.getResourceByName(table, \"TempStorage\", ResourceType.Container),\n                    this.getResourceByName(table, \"IngestionsStatusTable\", ResourceType.Table),\n                );\n\n                if (!resoures.valid()) {\n                    throw new Error(\"Unexpected error occured - fetched data returned missing resource\");\n                }\n\n                return resoures;\n            } catch (error: unknown) {\n                if (!(error instanceof KustoDataErrors.ThrottlingError)) {\n                    throw error;\n                }\n                await retry.backoff();\n            }\n        }\n        throw new Error(`Failed to get ingestion resources from server - the request was throttled ${ATTEMPT_COUNT} times.`);\n    }\n\n    getResourceByName(table: { rows: () => any }, resourceName: string, resourceType: ResourceType): ResourceURI[] {\n        const result: ResourceURI[] = [];\n        for (const row of table.rows()) {\n            const typedRow = row as {\n                ResourceTypeName: string;\n                StorageRoot: string;\n            };\n            if (typedRow.ResourceTypeName === resourceName) {\n                let accountName = \"\";\n                if (resourceType === ResourceType.Queue) {\n                    accountName = new QueueClient(typedRow.StorageRoot).accountName;\n                } else if (resourceType === ResourceType.Container) {\n                    accountName = new ContainerClient(typedRow.StorageRoot).accountName;\n                }\n                result.push(new ResourceURI(typedRow.StorageRoot, accountName, resourceType));\n            }\n        }\n        return result;\n    }\n\n    pupulateStorageAccounts(): void {\n        if (this.ingestClientResources == null) {\n            return;\n        }\n\n        // containers\n        const accounts = new Set<string>();\n        if (this.ingestClientResources.containers != null) {\n            for (const container of this.ingestClientResources.containers) {\n                accounts.add(container.accountName);\n            }\n        }\n        // queues\n        if (this.ingestClientResources.securedReadyForAggregationQueues != null) {\n            for (const queue of this.ingestClientResources.securedReadyForAggregationQueues) {\n                accounts.add(queue.accountName);\n            }\n        }\n\n        for (const account of accounts) {\n            this.rankedStorageAccountSet.registerStorageAccount(account);\n        }\n    }\n\n    groupResourcesByStorageAccount(resources: ResourceURI[]): Map<string, ResourceURI[]> {\n        const result = new Map<string, ResourceURI[]>();\n        for (const resource of resources) {\n            if (!result.has(resource.accountName)) {\n                result.set(resource.accountName, []);\n            }\n            result.get(resource.accountName)?.push(resource);\n        }\n        return result;\n    }\n\n    getRankedAndShuffledStorageAccounts(resources: ResourceURI[]): ResourceURI[][] {\n        const resourcesByAccount = this.groupResourcesByStorageAccount(resources);\n        const rankedStorageAccounts = this.rankedStorageAccountSet.getRankedShuffledAccounts();\n        const result = new Array<ResourceURI[]>();\n        for (const account of rankedStorageAccounts) {\n            const accountName = account.getAccountName();\n            if (resourcesByAccount.has(accountName)) {\n                result.push(resourcesByAccount.get(accountName) as ResourceURI[]);\n            }\n        }\n        return result;\n    }\n\n    getRoundRobinRankedAndShuffledResources(resources: ResourceURI[]): ResourceURI[] {\n        const rankedAccounts = this.getRankedAndShuffledStorageAccounts(resources);\n        const result = new Array<ResourceURI>();\n        let index = 0;\n        while (result.length < resources.length) {\n            const account = rankedAccounts[index % rankedAccounts.length];\n            if (account.length > 0) {\n                result.push(account.shift() as ResourceURI);\n            }\n            index++;\n        }\n        return result;\n    }\n\n    async refreshAuthorizationContext(): Promise<string> {\n        const error = await this.tryRefresh(true);\n\n        if (this.authorizationContext == null) {\n            throw new Error(`Failed to fetch Authorization context from service.  ${error?.message}.\\n ${error?.stack}`);\n        }\n\n        return this.authorizationContext;\n    }\n\n    async tryRefresh(isAuthContextFetch: boolean): Promise<Error | null> {\n        const resource = isAuthContextFetch ? this.authorizationContext?.trim() : this.ingestClientResources;\n        const lastRefresh = isAuthContextFetch ? this.authorizationContextLastUpdate : this.ingestClientResourcesLastUpdate;\n        const now = Date.now();\n        let error: Error | null = null;\n        if (!resource || !lastRefresh || lastRefresh + this.refreshPeriod <= now) {\n            try {\n                if (isAuthContextFetch) {\n                    this.authorizationContext = await this.getAuthorizationContextFromService();\n                    this.authorizationContextLastUpdate = now;\n                } else {\n                    this.ingestClientResources = await this.getIngestClientResourcesFromService();\n                    this.ingestClientResourcesLastUpdate = now;\n                    this.pupulateStorageAccounts();\n                }\n            } catch (e) {\n                error = e as Error;\n            }\n        }\n\n        return error;\n    }\n\n    async getAuthorizationContextFromService() {\n        const retry = new ExponentialRetry(ATTEMPT_COUNT, this.baseSleepTimeSecs, this.baseJitterSecs);\n        while (retry.shouldTry()) {\n            try {\n                const response = await this.kustoClient.execute(\"NetDefaultDB\", \".get kusto identity token\");\n                const next = response.primaryResults[0].rows().next();\n                if (next.done) {\n                    throw new Error(\"Failed to get authorization context - got empty results\");\n                }\n                return next.value.toJSON<{ AuthorizationContext: string }>().AuthorizationContext;\n            } catch (error: unknown) {\n                if (!(error instanceof KustoDataErrors.ThrottlingError)) {\n                    throw error;\n                }\n                await retry.backoff();\n            }\n        }\n        throw new Error(`Failed to get identity token from server - the request was throttled ${ATTEMPT_COUNT} times.`);\n    }\n\n    async getIngestionQueues(): Promise<ResourceURI[] | null> {\n        const queues = (await this.refreshIngestClientResources()).securedReadyForAggregationQueues;\n        return queues ? this.getRoundRobinRankedAndShuffledResources(queues) : null;\n    }\n\n    async getFailedIngestionsQueues(): Promise<ResourceURI[] | null> {\n        return (await this.refreshIngestClientResources()).failedIngestionsQueues;\n    }\n\n    async getSuccessfulIngestionsQueues(): Promise<ResourceURI[] | null> {\n        return (await this.refreshIngestClientResources()).successfulIngestionsQueues;\n    }\n\n    async getContainers(): Promise<ResourceURI[] | null> {\n        const containers = (await this.refreshIngestClientResources()).containers;\n        return containers ? this.getRoundRobinRankedAndShuffledResources(containers) : null;\n    }\n\n    async getAuthorizationContext(): Promise<string> {\n        return this.refreshAuthorizationContext();\n    }\n\n    async getStatusTables(): Promise<ResourceURI[] | null> {\n        return (await this.refreshIngestClientResources()).statusTables;\n    }\n\n    async createStatusTable() {\n        const statusTables = await this.getStatusTables();\n        if (!statusTables) {\n            throw new Error(\"Failed to get status table\");\n        }\n\n        return createStatusTableClient(statusTables![0].uri);\n    }\n\n    close(): void {\n        this.kustoClient.close();\n    }\n\n    reportResourceUsageResult(accountName: string, success: boolean): void {\n        this.rankedStorageAccountSet.logResultToAccount(accountName, success);\n    }\n}\n\nexport const createStatusTableClient = (uri: string): TableClient => {\n    const tableUrl = new URL(uri);\n    const origin = tableUrl.origin;\n    const sasToken = tableUrl.search;\n    const tableName = tableUrl.pathname.replace(\"/\", \"\");\n    return new TableClient(origin + sasToken, tableName);\n};\n\nexport default ResourceManager;\n"]}

package/dist-esm/src/retry.js

@@ -1,15 +1,11 @@
-"use strict";
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ExponentialRetry = exports.sleep = void 0;
-const sleep = (ms) => {
+export const sleep = (ms) => {
     return new Promise((resolve) => {
         setTimeout(resolve, ms);
     });
 };
-exports.sleep = sleep;
-class ExponentialRetry {
+export class ExponentialRetry {
     constructor(attemptCount, sleepBaseSecs, maxJitterSecs) {
         this.attemptCount = attemptCount;
         this.sleepBaseSecs = sleepBaseSecs;
@@ -27,11 +23,10 @@ class ExponentialRetry {
         }
         const base = this.sleepBaseSecs * Math.pow(2, this.currentAttempt - 1);
         const jitter = Math.floor(this.maxJitterSecs * Math.random());
-        await (0, exports.sleep)(1000 * (base + jitter));
+        await sleep(1000 * (base + jitter));
     }
     shouldTry() {
         return this.currentAttempt < this.attemptCount;
     }
 }
-exports.ExponentialRetry = ExponentialRetry;
 //# sourceMappingURL=retry.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"retry.js","sourceRoot":"","sources":["../../src/retry.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAE3B,MAAM,KAAK,GAAG,CAAC,EAAU,EAAiB,EAAE;IAC/C,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC3B,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACP,CAAC,CAAC;AAJW,QAAA,KAAK,SAIhB;AAEF,MAAa,gBAAgB;IAGzB,YAAmB,YAAoB,EAAS,aAAqB,EAAS,aAAqB;QAAhF,iBAAY,GAAZ,YAAY,CAAQ;QAAS,kBAAa,GAAb,aAAa,CAAQ;QAAS,kBAAa,GAAb,aAAa,CAAQ;QAC/F,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC;IAC5B,CAAC;IAEM,KAAK,CAAC,OAAO;QAChB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE;YACnB,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;SAC3C;QAED,IAAI,CAAC,cAAc,EAAE,CAAC;QAEtB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE;YACnB,6CAA6C;YAC7C,OAAO;SACV;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9D,MAAM,IAAA,aAAK,EAAC,IAAI,GAAG,CAAC,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC;IACxC,CAAC;IAEM,SAAS;QACZ,OAAO,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,YAAY,CAAC;IACnD,CAAC;CACJ;AA3BD,4CA2BC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nexport const sleep = (ms: number): Promise<void> => {\n    return new Promise((resolve) => {\n        setTimeout(resolve, ms);\n    });\n};\n\nexport class ExponentialRetry {\n    public currentAttempt: number;\n\n    constructor(public attemptCount: number, public sleepBaseSecs: number, public maxJitterSecs: number) {\n        this.currentAttempt = 0;\n    }\n\n    public async backoff(): Promise<void> {\n        if (!this.shouldTry()) {\n            throw new Error(\"Max retries exceeded\");\n        }\n\n        this.currentAttempt++;\n\n        if (!this.shouldTry()) {\n            // This was the last retry - no need to sleep\n            return;\n        }\n\n        const base = this.sleepBaseSecs * Math.pow(2, this.currentAttempt - 1);\n        const jitter = Math.floor(this.maxJitterSecs * Math.random());\n        await sleep(1000 * (base + jitter));\n    }\n\n    public shouldTry(): boolean {\n        return this.currentAttempt < this.attemptCount;\n    }\n}\n"]}
+{"version":3,"file":"retry.js","sourceRoot":"","sources":["../../src/retry.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,MAAM,CAAC,MAAM,KAAK,GAAG,CAAC,EAAU,EAAiB,EAAE;IAC/C,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC3B,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACP,CAAC,CAAC;AAEF,MAAM,OAAO,gBAAgB;IAGzB,YACW,YAAoB,EACpB,aAAqB,EACrB,aAAqB;QAFrB,iBAAY,GAAZ,YAAY,CAAQ;QACpB,kBAAa,GAAb,aAAa,CAAQ;QACrB,kBAAa,GAAb,aAAa,CAAQ;QAE5B,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC;IAC5B,CAAC;IAEM,KAAK,CAAC,OAAO;QAChB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC5C,CAAC;QAED,IAAI,CAAC,cAAc,EAAE,CAAC;QAEtB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,CAAC;YACpB,6CAA6C;YAC7C,OAAO;QACX,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9D,MAAM,KAAK,CAAC,IAAI,GAAG,CAAC,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC;IACxC,CAAC;IAEM,SAAS;QACZ,OAAO,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,YAAY,CAAC;IACnD,CAAC;CACJ","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nexport const sleep = (ms: number): Promise<void> => {\n    return new Promise((resolve) => {\n        setTimeout(resolve, ms);\n    });\n};\n\nexport class ExponentialRetry {\n    public currentAttempt: number;\n\n    constructor(\n        public attemptCount: number,\n        public sleepBaseSecs: number,\n        public maxJitterSecs: number,\n    ) {\n        this.currentAttempt = 0;\n    }\n\n    public async backoff(): Promise<void> {\n        if (!this.shouldTry()) {\n            throw new Error(\"Max retries exceeded\");\n        }\n\n        this.currentAttempt++;\n\n        if (!this.shouldTry()) {\n            // This was the last retry - no need to sleep\n            return;\n        }\n\n        const base = this.sleepBaseSecs * Math.pow(2, this.currentAttempt - 1);\n        const jitter = Math.floor(this.maxJitterSecs * Math.random());\n        await sleep(1000 * (base + jitter));\n    }\n\n    public shouldTry(): boolean {\n        return this.currentAttempt < this.attemptCount;\n    }\n}\n"]}

package/dist-esm/src/status.js

@@ -1,10 +1,7 @@
-"use strict";
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.KustoIngestStatusQueues = exports.StatusMessage = void 0;
-const statusQ_1 = require("./statusQ");
-class StatusMessage {
+import { StatusQueue } from "./statusQ.js";
+export class StatusMessage {
     constructor(raw, obj, extraProps) {
         let props = ["OperationId", "Database", "Table", "IngestionSourceId", "IngestionSourcePath", "RootActivityId"];
         if (extraProps && extraProps.length > 0) {
@@ -16,7 +13,6 @@ class StatusMessage {
         }
     }
 }
-exports.StatusMessage = StatusMessage;
 class SuccessMessage extends StatusMessage {
     constructor(raw, obj) {
         super(raw, obj, ["SucceededOn"]);
@@ -27,12 +23,11 @@ class FailureMessage extends StatusMessage {
         super(raw, obj, ["FailedOn", "Details", "ErrorCode", "FailureStatus", "OriginatesFromUpdatePolicy", "ShouldRetry"]);
     }
 }
-class KustoIngestStatusQueues {
+export class KustoIngestStatusQueues {
     constructor(kustoIngestClient) {
-        this.success = new statusQ_1.StatusQueue(() => kustoIngestClient.resourceManager.getSuccessfulIngestionsQueues().then((r) => r), SuccessMessage);
-        this.failure = new statusQ_1.StatusQueue(() => kustoIngestClient.resourceManager.getFailedIngestionsQueues().then((r) => r), FailureMessage);
+        this.success = new StatusQueue(() => kustoIngestClient.resourceManager.getSuccessfulIngestionsQueues().then((r) => r), SuccessMessage);
+        this.failure = new StatusQueue(() => kustoIngestClient.resourceManager.getFailedIngestionsQueues().then((r) => r), FailureMessage);
     }
 }
-exports.KustoIngestStatusQueues = KustoIngestStatusQueues;
-exports.default = KustoIngestStatusQueues;
+export default KustoIngestStatusQueues;
 //# sourceMappingURL=status.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"status.js","sourceRoot":"","sources":["../../src/status.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAElC,uCAAwC;AAIxC,MAAa,aAAa;IAStB,YAAY,GAAQ,EAAE,GAAQ,EAAE,UAA2B;QACvD,IAAI,KAAK,GAAa,CAAC,aAAa,EAAE,UAAU,EAAE,OAAO,EAAE,mBAAmB,EAAE,qBAAqB,EAAE,gBAAgB,CAAC,CAAC;QAEzH,IAAI,UAAU,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE;YACrC,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;SACpC;QAED,MAAM,IAAI,GAAwB,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;QAEhF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;YACtB,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;SAC3B;IACL,CAAC;CACJ;AAtBD,sCAsBC;AAED,MAAM,cAAe,SAAQ,aAAa;IAGtC,YAAY,GAAQ,EAAE,GAAQ;QAC1B,KAAK,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC;IACrC,CAAC;CACJ;AAED,MAAM,cAAe,SAAQ,aAAa;IAOtC,YAAY,GAAQ,EAAE,GAAQ;QAC1B,KAAK,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE,eAAe,EAAE,4BAA4B,EAAE,aAAa,CAAC,CAAC,CAAC;IACxH,CAAC;CACJ;AAED,MAAa,uBAAuB;IAGhC,YAAY,iBAAoC;QAC5C,IAAI,CAAC,OAAO,GAAG,IAAI,qBAAW,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC,6BAA6B,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAkB,CAAC,EAAE,cAAc,CAAC,CAAC;QACxJ,IAAI,CAAC,OAAO,GAAG,IAAI,qBAAW,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC,yBAAyB,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAkB,CAAC,EAAE,cAAc,CAAC,CAAC;IACxJ,CAAC;CACJ;AAPD,0DAOC;AAED,kBAAe,uBAAuB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { StatusQueue } from \"./statusQ\";\nimport KustoIngestClient from \"./ingestClient\";\nimport { ResourceURI } from \"./resourceManager\";\n\nexport class StatusMessage {\n    OperationId?: string;\n    Database?: string;\n    Table?: string;\n    IngestionSourceId?: string;\n    IngestionSourcePath?: string;\n    RootActivityId?: string;\n\n    [other: string]: any;\n    constructor(raw: any, obj: any, extraProps: string[] | null) {\n        let props: string[] = [\"OperationId\", \"Database\", \"Table\", \"IngestionSourceId\", \"IngestionSourcePath\", \"RootActivityId\"];\n\n        if (extraProps && extraProps.length > 0) {\n            props = props.concat(extraProps);\n        }\n\n        const _obj: Record<string, any> = obj || JSON.parse(raw || JSON.stringify(raw));\n\n        for (const prop of props) {\n            this[prop] = _obj[prop];\n        }\n    }\n}\n\nclass SuccessMessage extends StatusMessage {\n    SucceededOn?: string;\n\n    constructor(raw: any, obj: any) {\n        super(raw, obj, [\"SucceededOn\"]);\n    }\n}\n\nclass FailureMessage extends StatusMessage {\n    FailedOn?: string;\n    Details?: string;\n    ErrorCode?: string;\n    FailureStatus?: string;\n    OriginatesFromUpdatePolicy?: string;\n    ShouldRetry?: string;\n    constructor(raw: any, obj: any) {\n        super(raw, obj, [\"FailedOn\", \"Details\", \"ErrorCode\", \"FailureStatus\", \"OriginatesFromUpdatePolicy\", \"ShouldRetry\"]);\n    }\n}\n\nexport class KustoIngestStatusQueues {\n    success: StatusQueue;\n    failure: StatusQueue;\n    constructor(kustoIngestClient: KustoIngestClient) {\n        this.success = new StatusQueue(() => kustoIngestClient.resourceManager.getSuccessfulIngestionsQueues().then((r) => r as ResourceURI[]), SuccessMessage);\n        this.failure = new StatusQueue(() => kustoIngestClient.resourceManager.getFailedIngestionsQueues().then((r) => r as ResourceURI[]), FailureMessage);\n    }\n}\n\nexport default KustoIngestStatusQueues;\n"]}
+{"version":3,"file":"status.js","sourceRoot":"","sources":["../../src/status.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAI3C,MAAM,OAAO,aAAa;IAStB,YAAY,GAAQ,EAAE,GAAQ,EAAE,UAA2B;QACvD,IAAI,KAAK,GAAa,CAAC,aAAa,EAAE,UAAU,EAAE,OAAO,EAAE,mBAAmB,EAAE,qBAAqB,EAAE,gBAAgB,CAAC,CAAC;QAEzH,IAAI,UAAU,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtC,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACrC,CAAC;QAED,MAAM,IAAI,GAAwB,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;QAEhF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACvB,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;IACL,CAAC;CACJ;AAED,MAAM,cAAe,SAAQ,aAAa;IAGtC,YAAY,GAAQ,EAAE,GAAQ;QAC1B,KAAK,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC;IACrC,CAAC;CACJ;AAED,MAAM,cAAe,SAAQ,aAAa;IAOtC,YAAY,GAAQ,EAAE,GAAQ;QAC1B,KAAK,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE,eAAe,EAAE,4BAA4B,EAAE,aAAa,CAAC,CAAC,CAAC;IACxH,CAAC;CACJ;AAED,MAAM,OAAO,uBAAuB;IAGhC,YAAY,iBAAoC;QAC5C,IAAI,CAAC,OAAO,GAAG,IAAI,WAAW,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC,6BAA6B,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAkB,CAAC,EAAE,cAAc,CAAC,CAAC;QACxJ,IAAI,CAAC,OAAO,GAAG,IAAI,WAAW,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC,yBAAyB,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAkB,CAAC,EAAE,cAAc,CAAC,CAAC;IACxJ,CAAC;CACJ;AAED,eAAe,uBAAuB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { StatusQueue } from \"./statusQ.js\";\nimport KustoIngestClient from \"./ingestClient.js\";\nimport { ResourceURI } from \"./resourceManager.js\";\n\nexport class StatusMessage {\n    OperationId?: string;\n    Database?: string;\n    Table?: string;\n    IngestionSourceId?: string;\n    IngestionSourcePath?: string;\n    RootActivityId?: string;\n\n    [other: string]: any;\n    constructor(raw: any, obj: any, extraProps: string[] | null) {\n        let props: string[] = [\"OperationId\", \"Database\", \"Table\", \"IngestionSourceId\", \"IngestionSourcePath\", \"RootActivityId\"];\n\n        if (extraProps && extraProps.length > 0) {\n            props = props.concat(extraProps);\n        }\n\n        const _obj: Record<string, any> = obj || JSON.parse(raw || JSON.stringify(raw));\n\n        for (const prop of props) {\n            this[prop] = _obj[prop];\n        }\n    }\n}\n\nclass SuccessMessage extends StatusMessage {\n    SucceededOn?: string;\n\n    constructor(raw: any, obj: any) {\n        super(raw, obj, [\"SucceededOn\"]);\n    }\n}\n\nclass FailureMessage extends StatusMessage {\n    FailedOn?: string;\n    Details?: string;\n    ErrorCode?: string;\n    FailureStatus?: string;\n    OriginatesFromUpdatePolicy?: string;\n    ShouldRetry?: string;\n    constructor(raw: any, obj: any) {\n        super(raw, obj, [\"FailedOn\", \"Details\", \"ErrorCode\", \"FailureStatus\", \"OriginatesFromUpdatePolicy\", \"ShouldRetry\"]);\n    }\n}\n\nexport class KustoIngestStatusQueues {\n    success: StatusQueue;\n    failure: StatusQueue;\n    constructor(kustoIngestClient: KustoIngestClient) {\n        this.success = new StatusQueue(() => kustoIngestClient.resourceManager.getSuccessfulIngestionsQueues().then((r) => r as ResourceURI[]), SuccessMessage);\n        this.failure = new StatusQueue(() => kustoIngestClient.resourceManager.getFailedIngestionsQueues().then((r) => r as ResourceURI[]), FailureMessage);\n    }\n}\n\nexport default KustoIngestStatusQueues;\n"]}

package/dist-esm/src/statusQ.js

@@ -1,9 +1,6 @@
-"use strict";
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.StatusQueue = void 0;
-const storage_queue_1 = require("@azure/storage-queue");
+import { QueueClient } from "@azure/storage-queue";
 class QueueDetails {
     constructor(name, service) {
         this.name = name;
@@ -19,7 +16,7 @@ const shuffle = (a) => {
     }
     return a;
 };
-class StatusQueue {
+export class StatusQueue {
     constructor(getQueuesFunc, messageCls) {
         this.getQueuesFunc = getQueuesFunc;
         this.messageCls = messageCls;
@@ -33,7 +30,7 @@ class StatusQueue {
             // chop off sas
             const indexOfSas = q.uri.indexOf("?");
             const name = indexOfSas > 0 ? q.uri.substring(0, indexOfSas) : q.uri;
-            return new QueueDetails(name, new storage_queue_1.QueueClient(fullUri));
+            return new QueueDetails(name, new QueueClient(fullUri));
         });
     }
     async isEmpty() {
@@ -115,5 +112,4 @@ class StatusQueue {
         return partial.result.concat(final.result);
     }
 }
-exports.StatusQueue = StatusQueue;
 //# sourceMappingURL=statusQ.js.map