feat: Expose context method for writing custom Grafana metrics

near · Jul 13, 2023 · be3a5d1 · be3a5d1
1 parent fc4fb7e
commit be3a5d1
Show file tree

Hide file tree

Showing 3 changed files with 58 additions and 3 deletions.
diff --git a/indexer-js-queue-handler/indexer.js b/indexer-js-queue-handler/indexer.js
@@ -254,7 +254,16 @@ export default class Indexer {
             },
             log: async (log) => {
                 return await this.writeLog(functionName, block_height, log);
-            }
+            },
+            putMetric: (metricName, value) => {
+                const [accountId, name] = functionName.split('/');
+                return this.deps.metrics.putCustomMetric(
+                    accountId,
+                    name,
+                    `CUSTOM_${metricName}`,
+                    value
+                );
+            },
         };
     }
 

diff --git a/indexer-js-queue-handler/indexer.test.js b/indexer-js-queue-handler/indexer.test.js
@@ -895,6 +895,48 @@ mutation _1 { set(functionName: "buildnear.testnet/test", key: "foo2", data: "in
         ]);
     });
 
+    test('allows writing of custom metrics', async () => {
+        const mockFetch = jest.fn(() => ({
+            status: 200,
+            json: async () => ({
+                errors: null,
+            }),
+        }));
+        const block_height = 456;
+        const mockS3 = {
+            getObject: jest.fn(() => ({
+                promise: () => Promise.resolve({
+                    Body: {
+                        toString: () => JSON.stringify({
+                            chunks: [],
+                            header: {
+                                height: block_height
+                            }
+                        })
+                    }
+                })
+            })),
+        };
+        const metrics = {
+            putBlockHeight: () => {},
+            putCustomMetric: jest.fn(),
+        };
+        const indexer = new Indexer('mainnet', { fetch: mockFetch, s3: mockS3, awsXray: mockAwsXray, metrics });
+
+        const functions = {};
+        functions['buildnear.testnet/test'] = {code:`
+            context.putMetric('TEST_METRIC', 1)
+        `};
+        await indexer.runFunctions(block_height, functions, false, { imperative: true });
+
+        expect(metrics.putCustomMetric).toHaveBeenCalledWith(
+            'buildnear.testnet',
+            'test',
+            'CUSTOM_TEST_METRIC',
+            1
+        );
+    });
+
     // The unhandled promise causes problems with test reporting.
     // Note unhandled promise rejections fail to proceed to the next function on AWS Lambda
     test.skip('Indexer.runFunctions() continues despite promise rejection, unable to log rejection', async () => {

diff --git a/indexer-js-queue-handler/metrics.js b/indexer-js-queue-handler/metrics.js
@@ -8,11 +8,15 @@ export default class Metrics {
     }
 
     putBlockHeight(accountId, functionName, height) {
+        return this.putCustomMetric(accountId, functionName, "INDEXER_FUNCTION_LATEST_BLOCK_HEIGHT", height);
+    }
+
+    putCustomMetric(accountId, functionName, metricName, value) {
         return this.cloudwatch
             .putMetricData({
                 MetricData: [
                     {
-                        MetricName: "INDEXER_FUNCTION_LATEST_BLOCK_HEIGHT",
+                        MetricName: metricName,
                         Dimensions: [
                             {
                                 Name: "ACCOUNT_ID",
@@ -28,7 +32,7 @@ export default class Metrics {
                             },
                         ],
                         Unit: "None",
-                        Value: height,
+                        Value: value,
                     },
                 ],
                 Namespace: this.namespace,