feat: add setDiff (#1263)

Co-authored-by: jinbing.jb <jinbing.jb@alibaba-inc.com>
midwayjs · Sep 1, 2021 · 2bf7844 · 2bf7844
1 parent da58bf6
commit 2bf7844
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/packages/prometheus/src/service/dataService.ts b/packages/prometheus/src/service/dataService.ts
@@ -90,6 +90,19 @@ export class DataService {
     );
   }
 
+  @Master()
+  async setDiff(name, diff) {
+    const current = await this.userDefinedMetrics[name].get();
+    let value = diff;
+    if (current.values.length !== 0) {
+      value = current.values[0].value + diff;
+    }
+    this.userDefinedMetrics[name].set(
+      { ...this.prometheusConfig.labels },
+      value
+    );
+  }
+
   @Master()
   async getData() {
     const Register = PromClient.register;