Accept Thenables as a collect() hander.

siimon · Aug 8, 2024 · f037e96 · f037e96
1 parent c1d76c5
commit f037e96
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/lib/counter.js b/lib/counter.js
@@ -109,7 +109,7 @@ class Counter extends Metric {
 	async get() {
 		if (this.collect) {
 			const v = this.collect();
-			if (v instanceof Promise) await v;
+			if ((v instanceof Promise) || (v.then instanceof Function)) await v;
 		}
 
 		return {

diff --git a/lib/gauge.js b/lib/gauge.js
@@ -108,7 +108,7 @@ class Gauge extends Metric {
 	async get() {
 		if (this.collect) {
 			const v = this.collect();
-			if (v instanceof Promise) await v;
+			if ((v instanceof Promise) || (v.then instanceof Function)) await v;
 		}
 		return {
 			help: this.help,

diff --git a/lib/histogram.js b/lib/histogram.js
@@ -108,7 +108,7 @@ class Histogram extends Metric {
 	async getForPromString() {
 		if (this.collect) {
 			const v = this.collect();
-			if (v instanceof Promise) await v;
+			if ((v instanceof Promise) || (v.then instanceof Function)) await v;
 		}
 		const data = Object.values(this.hashMap);
 		const values = data

diff --git a/lib/summary.js b/lib/summary.js
@@ -51,7 +51,7 @@ class Summary extends Metric {
 	async get() {
 		if (this.collect) {
 			const v = this.collect();
-			if (v instanceof Promise) await v;
+			if ((v instanceof Promise) || (v.then instanceof Function)) await v;
 		}
 		const hashKeys = Object.keys(this.hashMap);
 		const values = [];