support nim-metrics 0.1.0+ (#651)

status-im · Jan 5, 2024 · 0fc5dc0 · 0fc5dc0
1 parent 5aaeb67
commit 0fc5dc0
Showing 1 changed file with 40 additions and 32 deletions.
diff --git a/eth/db/kvstore_sqlite3.nim b/eth/db/kvstore_sqlite3.nim
@@ -694,35 +694,43 @@ proc createCustomFunction*(
   ok()
 
 when defined(metrics):
-  import locks, tables, times,
-        chronicles, metrics
-
-  type Sqlite3Info = ref object of Gauge
-
-  proc newSqlite3Info*(name: string, help: string, registry = defaultRegistry): Sqlite3Info {.raises: [Exception].} =
-    validateName(name)
-    result = Sqlite3Info(name: name,
-                        help: help,
-                        typ: "gauge",
-                        creationThreadId: getThreadId())
-    result.lock.initLock()
-    result.register(registry)
-
-  var sqlite3Info* {.global.} = newSqlite3Info("sqlite3_info", "SQLite3 info")
-
-  method collect*(collector: Sqlite3Info): Metrics =
-    result = initOrderedTable[Labels, seq[Metric]]()
-    result[@[]] = @[]
-    let timestamp = getTime().toMilliseconds()
-    var currentMem, highwaterMem: int64
-
-    if (let res = sqlite3_status64(SQLITE_STATUS_MEMORY_USED, currentMem.addr, highwaterMem.addr, 0); res != SQLITE_OK):
-      error "SQLite3 error", msg = sqlite3_errstr(res)
-    else:
-      result[@[]] = @[
-        Metric(
-          name: "sqlite3_memory_used_bytes",
-          value: currentMem.float64,
-          timestamp: timestamp,
-        ),
-      ]
+  import chronicles, metrics
+
+  type Sqlite3Info = ref object of Collector
+
+  proc newSqlite3Info*(name: string, help: string, registry = defaultRegistry): Sqlite3Info {.raises: [CatchableError].} =
+    Sqlite3Info.newCollector(name, help, registry = registry)
+
+  let sqlite3Info* = newSqlite3Info("sqlite3_info", "SQLite3 info")
+
+  when declared(MetricHandler): # nim-metrics 0.1.0+
+    method collect*(collector: Sqlite3Info, output: MetricHandler) =
+      let timestamp = collector.now()
+
+      var currentMem, highwaterMem: int64
+
+      if (let res = sqlite3_status64(SQLITE_STATUS_MEMORY_USED, currentMem.addr, highwaterMem.addr, 0); res != SQLITE_OK):
+        error "SQLite3 error", msg = sqlite3_errstr(res)
+      else:
+        output(
+          name = "sqlite3_memory_used_bytes",
+          value = currentMem.float64,
+          timestamp = timestamp,
+        )
+  else: # nim-metrics 0.0.1
+    method collect*(collector: Sqlite3Info): Metrics =
+      result = initOrderedTable[Labels, seq[Metric]]()
+      result[@[]] = @[]
+      let timestamp = getTime().toMilliseconds()
+      var currentMem, highwaterMem: int64
+
+      if (let res = sqlite3_status64(SQLITE_STATUS_MEMORY_USED, currentMem.addr, highwaterMem.addr, 0); res != SQLITE_OK):
+        error "SQLite3 error", msg = sqlite3_errstr(res)
+      else:
+        result[@[]] = @[
+          Metric(
+            name: "sqlite3_memory_used_bytes",
+            value: currentMem.float64,
+            timestamp: timestamp,
+          ),
+        ]