Add prometheus metrics to inner database

skodjob · Mar 25, 2024 · 0612bbf · 0612bbf
1 parent 4b0e836
commit 0612bbf
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 5 deletions.
diff --git a/...n/java/io/skodjob/dmt/model/Database.java → ...ava/io/skodjob/dmt/database/Database.java b/...n/java/io/skodjob/dmt/model/Database.java → ...ava/io/skodjob/dmt/database/Database.java
@@ -3,13 +3,20 @@
  *
  * Licensed under the Apache Software License version 2.0, available at http://www.apache.org/licenses/LICENSE-2.0
  */
-package io.skodjob.dmt.model;
+package io.skodjob.dmt.database;
 
 import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 
+import io.micrometer.core.instrument.MeterRegistry;
+import io.micrometer.core.instrument.Tags;
+import io.micrometer.prometheus.PrometheusMeterRegistry;
+import io.skodjob.dmt.model.DatabaseColumn;
+import io.skodjob.dmt.model.DatabaseEntry;
+import io.skodjob.dmt.model.DatabaseTable;
+import io.skodjob.dmt.model.DatabaseTableMetadata;
 import jakarta.enterprise.context.ApplicationScoped;
 
 import io.skodjob.dmt.exception.InnerDatabaseException;
@@ -20,12 +27,15 @@
 @Lock
 @ApplicationScoped
 public class Database {
+    private final MeterRegistry meterRegistry;
     Map<String, DatabaseTable> tables;
 
     private static final Logger LOG = Logger.getLogger(Database.class);
 
-    public Database() {
+    public Database(PrometheusMeterRegistry meterRegistry) {
         tables = new HashMap<>();
+        this.meterRegistry = meterRegistry;
+        meterRegistry.gaugeMapSize("table.count", Tags.empty(), tables);
     }
 
     @Lock(value = Lock.Type.READ)
@@ -119,5 +129,6 @@ public List<DatabaseTable> getTables() {
 
     private void createTable(DatabaseTableMetadata tableMetadata) {
         tables.put(tableMetadata.getName(), new DatabaseTable(tableMetadata));
+        meterRegistry.gaugeMapSize(tableMetadata.getName() + ".table.size", Tags.empty(), tables.get(tableMetadata.getName()).getRowsAsMap());
     }
 }
diff --git a/database-manipulation-tool/src/main/java/io/skodjob/dmt/service/MainService.java b/database-manipulation-tool/src/main/java/io/skodjob/dmt/service/MainService.java
@@ -19,7 +19,7 @@
 import io.skodjob.dmt.dao.Dao;
 import io.skodjob.dmt.dao.DaoManager;
 import io.skodjob.dmt.exception.InnerDatabaseException;
-import io.skodjob.dmt.model.Database;
+import io.skodjob.dmt.database.Database;
 import io.skodjob.dmt.model.DatabaseColumn;
 import io.skodjob.dmt.model.DatabaseEntry;
 import io.skodjob.dmt.model.DatabaseTableMetadata;

diff --git a/database-manipulation-tool/src/main/java/io/skodjob/dmt/service/UtilityService.java b/database-manipulation-tool/src/main/java/io/skodjob/dmt/service/UtilityService.java
@@ -12,7 +12,7 @@
 import org.jboss.logging.Logger;
 
 import io.skodjob.dmt.dao.DaoManager;
-import io.skodjob.dmt.model.Database;
+import io.skodjob.dmt.database.Database;
 import io.skodjob.dmt.utils.ResponseJsonBuilder;
 
 @ApplicationScoped

diff --git a/database-manipulation-tool/src/main/java/io/skodjob/dmt/utils/ResponseJsonBuilder.java b/database-manipulation-tool/src/main/java/io/skodjob/dmt/utils/ResponseJsonBuilder.java
@@ -12,7 +12,7 @@
 import jakarta.json.JsonObject;
 import jakarta.json.JsonObjectBuilder;
 
-import io.skodjob.dmt.model.Database;
+import io.skodjob.dmt.database.Database;
 import io.skodjob.dmt.model.DatabaseColumnEntry;
 import io.skodjob.dmt.model.DatabaseTable;
 import io.skodjob.dmt.model.DatabaseColumn;