@emartech/aggregate-metric-logger 6.7.1 → 6.7.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@emartech/aggregate-metric-logger",
-  "version": "6.7.1",
+  "version": "6.7.3",
   "description": "aggregate metrics in memory for logging",
   "main": "src/index.js",
   "types": "src/index.d.ts",

package/src/gcp-metric-collector/gcp-metric-collector.js

@@ -19,6 +19,7 @@ class GcpMetricCollector {
   initialized = false;
   timeoutId = null;
   projectId = null;
+  flushInProgress = false;
 
   constructor(projectId = null) {
     this.projectId = projectId ?? process.env.MONITORING_PROJECT_ID;
@@ -106,11 +107,17 @@ class GcpMetricCollector {
   }
 
   async publishPending() {
-    const timeSeries = createGcpTimeSeriesBuilder(this.projectId).buildTimeSeries(this.metrics);
-    for (const [, metric] of Object.entries(this.metrics)) {
-      metric.aggregator.advanceTimer();
+    if (this.flushInProgress) return;
+    this.flushInProgress = true;
+    try {
+      const timeSeries = createGcpTimeSeriesBuilder(this.projectId).buildTimeSeries(this.metrics);
+      for (const [, metric] of Object.entries(this.metrics)) {
+        metric.aggregator.advanceTimer();
+      }
+      await createGcpTimeSeriesPublisher(this.projectId).publishTimeSeries(timeSeries);
+    } finally {
+      this.flushInProgress = false;
     }
-    await createGcpTimeSeriesPublisher(this.projectId).publishTimeSeries(timeSeries);
   }
 }
 

package/src/gcp-metric-collector/gcp-metric-collector.spec.js

@@ -538,6 +538,47 @@ describe('gcpMetricCollector', () => {
     });
   });
 
+  describe('concurrent flush', () => {
+    it('should not publish concurrently if a flush is already in flight', async () => {
+      const firstCreateTimeSeriesPromise = resolvablePromise();
+      jest.spyOn(MetricServiceClient.prototype, 'createTimeSeries')
+        .mockReturnValueOnce(firstCreateTimeSeriesPromise)
+        .mockResolvedValue();
+
+      gcpMetricCollector.count('jolo', {}, 1);
+      jest.advanceTimersByTime(60 * 1000 + 1);
+
+      gcpMetricCollector.count('jolo', {}, 1);
+      jest.advanceTimersByTime(60 * 1000 + 1);
+
+      expect(MetricServiceClient.prototype.createTimeSeries).toHaveBeenCalledTimes(1);
+
+      firstCreateTimeSeriesPromise.resolve();
+      await flushPromises();
+    });
+
+    it('should publish the second count after the first flush resolves', async () => {
+      const firstCreateTimeSeriesPromise = resolvablePromise();
+      jest.spyOn(MetricServiceClient.prototype, 'createTimeSeries')
+        .mockReturnValueOnce(firstCreateTimeSeriesPromise)
+        .mockResolvedValue();
+
+      gcpMetricCollector.count('jolo', {}, 1);
+      jest.advanceTimersByTime(60 * 1000 + 1);
+
+      gcpMetricCollector.count('jolo', {}, 1);
+      jest.advanceTimersByTime(60 * 1000 + 1);
+
+      firstCreateTimeSeriesPromise.resolve();
+      await flushPromises();
+
+      jest.advanceTimersByTime(60 * 1000 + 1);
+      await flushPromises();
+
+      expect(MetricServiceClient.prototype.createTimeSeries).toHaveBeenCalledTimes(2);
+    });
+  });
+
   describe('shutdown', () => {
     it('should flush pending metrics immediately', async () => {
       stubCreateTimeSeries();