@azure/storage-blob-changefeed 12.0.0-alpha.20230206.1 → 12.0.0-alpha.20230216.2

package/dist/index.js

@@ -4,7 +4,6 @@ Object.defineProperty(exports, '__esModule', { value: true });
 
 var tslib = require('tslib');
 var storageBlob = require('@azure/storage-blob');
-var coreHttp = require('@azure/core-http');
 var coreTracing = require('@azure/core-tracing');
 var abortController = require('@azure/abort-controller');
 require('@azure/core-paging');
@@ -25,8 +24,9 @@ const CHANGE_FEED_CHUNK_BLOCK_DOWNLOAD_SIZE = 16 * 1024 * 1024;
  * Creates a span using the global tracer.
  * @internal
  */
-const createSpan = coreTracing.createSpanFunction({
-    packagePrefix: "Azure.Storage.Blob.Changefeed",
+const tracingClient = coreTracing.createTracingClient({
+    packageName: "@azure/storage-blob-changefeed",
+    packageVersion: SDK_VERSION,
     namespace: "Microsoft.Storage",
 });
 
@@ -50,13 +50,12 @@ function floorToNearestHour(date) {
  * @param url - Source URL string
  */
 function getHost(url) {
-    const urlParsed = coreHttp.URLBuilder.parse(url);
-    return urlParsed.getHost();
+    const urlParsed = new URL(url);
+    return urlParsed.hostname;
 }
 async function getYearsPaths(containerClient, options = {}) {
-    var e_1, _a;
-    const { span, updatedOptions } = createSpan("getYearsPaths", options);
-    try {
+    return tracingClient.withSpan("getYearsPaths", options, async (updatedOptions) => {
+        var e_1, _a;
         const years = [];
         try {
             for (var _b = tslib.__asyncValues(containerClient.listBlobsByHierarchy("/", {
@@ -79,22 +78,11 @@ async function getYearsPaths(containerClient, options = {}) {
             finally { if (e_1) throw e_1.error; }
         }
         return years.sort((a, b) => a - b);
-    }
-    catch (e) {
-        span.setStatus({
-            code: coreTracing.SpanStatusCode.ERROR,
-            message: e.message,
-        });
-        throw e;
-    }
-    finally {
-        span.end();
-    }
+    });
 }
 async function getSegmentsInYear(containerClient, year, startTime, endTime, options = {}) {
-    var e_2, _a;
-    const { span, updatedOptions } = createSpan("getSegmentsInYear", options);
-    try {
+    return tracingClient.withSpan("getSegmentsInYear", options, async (updatedOptions) => {
+        var e_2, _a;
         const segments = [];
         const yearBeginTime = new Date(Date.UTC(year, 0));
         if (endTime && yearBeginTime >= endTime) {
@@ -123,17 +111,7 @@ async function getSegmentsInYear(containerClient, year, startTime, endTime, opti
             finally { if (e_2) throw e_2.error; }
         }
         return segments;
-    }
-    catch (e) {
-        span.setStatus({
-            code: coreTracing.SpanStatusCode.ERROR,
-            message: e.message,
-        });
-        throw e;
-    }
-    finally {
-        span.end();
-    }
+    });
 }
 function parseDateFromSegmentPath(segmentPath) {
     const splitPath = segmentPath.split("/");
@@ -257,8 +235,7 @@ class ChangeFeed {
         }
     }
     async advanceSegmentIfNecessary(options = {}) {
-        const { span, updatedOptions } = createSpan("ChangeFeed-advanceSegmentIfNecessary", options);
-        try {
+        return tracingClient.withSpan("ChangeFeed-advanceSegmentIfNecessary", options, async (updatedOptions) => {
             if (!this.currentSegment) {
                 throw new Error("Empty Change Feed shouldn't call this function.");
             }
@@ -290,17 +267,7 @@ class ChangeFeed {
                     this.currentSegment = undefined;
                 }
             }
-        }
-        catch (e) {
-            span.setStatus({
-                code: coreTracing.SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     hasNext() {
         // Empty ChangeFeed, using currentSegment as the indicator.
@@ -313,8 +280,7 @@ class ChangeFeed {
         return this.currentSegment.dateTime < this.end;
     }
     async getChange(options = {}) {
-        const { span, updatedOptions } = createSpan("ChangeFeed-getChange", options);
-        try {
+        return tracingClient.withSpan("ChangeFeed-getChange", options, async (updatedOptions) => {
             let event = undefined;
             while (event === undefined && this.hasNext()) {
                 event = await this.currentSegment.getChange({
@@ -327,17 +293,7 @@ class ChangeFeed {
                 });
             }
             return event;
-        }
-        catch (e) {
-            span.setStatus({
-                code: coreTracing.SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     getCursor() {
         var _a;
@@ -917,8 +873,7 @@ class Segment {
         return this.shards.length > this.shardDoneCount;
     }
     async getChange(options = {}) {
-        const { span, updatedOptions } = createSpan("Segment-getChange", options);
-        try {
+        return tracingClient.withSpan("Segment-getChange", options, async (updatedOptions) => {
             if (this.shardIndex >= this.shards.length || this.shardIndex < 0) {
                 throw new Error("shardIndex invalid.");
             }
@@ -941,17 +896,7 @@ class Segment {
                 this.shardIndex = (this.shardIndex + 1) % this.shards.length;
             }
             return event;
-        }
-        catch (e) {
-            span.setStatus({
-                code: coreTracing.SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     getCursor() {
         const shardCursors = [];
@@ -975,8 +920,7 @@ class SegmentFactory {
         this.shardFactory = shardFactory;
     }
     async create(containerClient, manifestPath, cursor, options = {}) {
-        const { span, updatedOptions } = createSpan("SegmentFactory-create", options);
-        try {
+        return tracingClient.withSpan("SegmentFactory-create", options, async (updatedOptions) => {
             const shards = [];
             const dateTime = parseDateFromSegmentPath(manifestPath);
             const blobClient = containerClient.getBlobClient(manifestPath);
@@ -1006,17 +950,7 @@ class SegmentFactory {
                 }
             }
             return new Segment(shards, shardIndex, dateTime, manifestPath);
-        }
-        catch (e) {
-            span.setStatus({
-                code: coreTracing.SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
 }
 
@@ -1033,8 +967,7 @@ class Shard {
         return (this.chunks.length > 0 || (this.currentChunk !== undefined && this.currentChunk.hasNext()));
     }
     async getChange(options = {}) {
-        const { span, updatedOptions } = createSpan("Shard-getChange", options);
-        try {
+        return tracingClient.withSpan("Shard-getChange", options, async (updatedOptions) => {
             let event = undefined;
             while (event === undefined && this.hasNext()) {
                 event = await this.currentChunk.getChange();
@@ -1047,17 +980,7 @@ class Shard {
                 }
             }
             return event;
-        }
-        catch (e) {
-            span.setStatus({
-                code: coreTracing.SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     getCursor() {
         return this.currentChunk === undefined
@@ -1076,9 +999,8 @@ class ShardFactory {
         this.chunkFactory = chunkFactory;
     }
     async create(containerClient, shardPath, shardCursor, options = {}) {
-        var e_1, _a;
-        const { span, updatedOptions } = createSpan("ShardFactory-create", options);
-        try {
+        return tracingClient.withSpan("ShardFactory-create", options, async (updatedOptions) => {
+            var e_1, _a;
             const chunks = [];
             const blockOffset = (shardCursor === null || shardCursor === void 0 ? void 0 : shardCursor.BlockOffset) || 0;
             const eventIndex = (shardCursor === null || shardCursor === void 0 ? void 0 : shardCursor.EventIndex) || 0;
@@ -1129,17 +1051,7 @@ class ShardFactory {
                 });
             }
             return new Shard(containerClient, this.chunkFactory, chunks, currentChunk, shardPath);
-        }
-        catch (e) {
-            span.setStatus({
-                code: coreTracing.SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
 }
 
@@ -1242,8 +1154,7 @@ class LazyLoadingBlobStream extends stream.Readable {
         this.options = options;
     }
     async downloadBlock(options = {}) {
-        const { span, updatedOptions } = createSpan("LazyLoadingBlobStream-downloadBlock", options);
-        try {
+        return tracingClient.withSpan("LazyLoadingBlobStream-downloadBlock", options, async (updatedOptions) => {
             const properties = await this.blobClient.getProperties({
                 abortSignal: options.abortSignal,
                 tracingOptions: updatedOptions.tracingOptions,
@@ -1259,17 +1170,7 @@ class LazyLoadingBlobStream extends stream.Readable {
                 tracingOptions: updatedOptions.tracingOptions,
             });
             this.offset += this.lastDownloadBytes;
-        }
-        catch (e) {
-            span.setStatus({
-                code: coreTracing.SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Internal _read() that will be called when the stream wants to pull more data in.
@@ -1277,9 +1178,9 @@ class LazyLoadingBlobStream extends stream.Readable {
      * @param size - Optional. The size of data to be read
      */
     async _read(size) {
-        var _a, _b, _c, _d;
-        const { span, updatedOptions } = createSpan("LazyLoadingBlobStream-read", this.options);
-        try {
+        var _a;
+        return tracingClient.withSpan("LazyLoadingBlobStream-read", (_a = this.options) !== null && _a !== void 0 ? _a : {}, async (updatedOptions) => {
+            var _a, _b, _c, _d;
             if (!size) {
                 size = this.readableHighWaterMark;
             }
@@ -1307,17 +1208,7 @@ class LazyLoadingBlobStream extends stream.Readable {
             if (count < size) {
                 this.push(null);
             }
-        }
-        catch (e) {
-            span.setStatus({
-                code: coreTracing.SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            this.emit("error", e);
-        }
-        finally {
-            span.end();
-        }
+        });
     }
 }
 
@@ -1356,8 +1247,7 @@ class ChangeFeedFactory {
         }
     }
     async create(blobServiceClient, continuationToken, options = {}) {
-        const { span, updatedOptions } = createSpan("ChangeFeedFactory-create", options);
-        try {
+        return tracingClient.withSpan("ChangeFeedFactory-create", options, async (updatedOptions) => {
             const containerClient = blobServiceClient.getContainerClient(CHANGE_FEED_CONTAINER_NAME);
             let cursor = undefined;
             // Create cursor.
@@ -1431,17 +1321,7 @@ class ChangeFeedFactory {
                 tracingOptions: updatedOptions.tracingOptions,
             });
             return new ChangeFeed(containerClient, this.segmentFactory, years, segments, currentSegment, lastConsumable, options.start, options.end);
-        }
-        catch (e) {
-            span.setStatus({
-                code: coreTracing.SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
 }