Add stats for bytes metered and scanned (#110)

grafana · Nov 8, 2021 · 8da8ded · 8da8ded
1 parent cef0bf1
commit 8da8ded
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/src/DataSource.ts b/src/DataSource.ts
@@ -7,6 +7,7 @@ import {
   ScopedVars,
   QueryResultMetaStat,
   TimeRange,
+  getValueFormat,
 } from '@grafana/data';
 import { DataSourceWithBackend, getTemplateSrv } from '@grafana/runtime';
 import { Observable, of, merge } from 'rxjs';
@@ -133,6 +134,7 @@ export class DataSource extends DataSourceWithBackend<TimestreamQuery, Timestrea
           if (!meta) {
             return allData.length ? allData : data; // NOOP
           }
+          const valueFormat = getValueFormat('decbytes');
 
           // Single request
           meta.fetchStartTime = t.fetchStartTime;
@@ -226,6 +228,24 @@ export class DataSource extends DataSourceWithBackend<TimestreamQuery, Timestrea
                   });
                 }
               }
+              if (tracker.status.CumulativeBytesMetered) {
+                const v = valueFormat(tracker.status.CumulativeBytesMetered, 2, 1024);
+                stats.push({
+                  displayName: 'Cumulative bytes metered',
+                  value: Number(v.text),
+                  unit: v.suffix?.trimLeft(),
+                  decimals: 2,
+                });
+              }
+              if (tracker.status.CumulativeBytesScanned) {
+                const v = valueFormat(tracker.status.CumulativeBytesScanned, 2, 1024);
+                stats.push({
+                  displayName: 'Cumulative bytes scanned',
+                  value: Number(v.text),
+                  unit: v.suffix?.trimLeft(),
+                  decimals: 2,
+                });
+              }
               allData[0].meta!.stats = stats;
             }
           }

diff --git a/src/types.ts b/src/types.ts
@@ -53,6 +53,11 @@ export interface TimestreamCustomMeta {
   fetchEndTime?: number; // The frontend clock
   fetchTime?: number; // The frontend clock
 
+  status: {
+    CumulativeBytesMetered?: number;
+    CumulativeBytesScanned?: number;
+  };
+
   // when multiple queries exist we keep track of each request
   subs?: TimestreamCustomMeta[];
 }