Add metrics for list databases

linkedin · Dec 10, 2024 · ab0b75d · ab0b75d
1 parent 240ba9d
commit ab0b75d
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/services/common/src/main/java/com/linkedin/openhouse/common/metrics/MetricsConstant.java b/services/common/src/main/java/com/linkedin/openhouse/common/metrics/MetricsConstant.java
@@ -44,4 +44,5 @@ private MetricsConstant() {}
       "repo_tables_search_by_database_time";
   public static final String REPO_TABLE_IDS_FIND_ALL_TIME = "repo_table_ids_find_all_time";
   public static final String REPO_TABLES_FIND_ALL_TIME = "repo_tables_find_all_time";
+  public static final String HTS_LIST_DATABASES_TIME = "hts_list_databases_time";
 }
diff --git a/...tables/src/main/java/com/linkedin/openhouse/housetables/config/MainApplicationConfig.java b/...tables/src/main/java/com/linkedin/openhouse/housetables/config/MainApplicationConfig.java
@@ -11,6 +11,8 @@
 /** Main application configuration to load cluster properties and define required beans */
 @Configuration
 public class MainApplicationConfig extends BaseApplicationConfig {
+  public static final String APP_NAME = "housetables";
+
   @Bean
   MeterRegistryCustomizer<MeterRegistry> provideMeterRegistry() {
     return registry ->

diff --git a/...bles/src/main/java/com/linkedin/openhouse/housetables/services/UserTablesServiceImpl.java b/...bles/src/main/java/com/linkedin/openhouse/housetables/services/UserTablesServiceImpl.java
@@ -108,9 +108,13 @@ public void deleteUserTable(String databaseId, String tableId) {
   }
 
   private List<UserTableDto> listDatabases() {
-    return StreamSupport.stream(htsJdbcRepository.findAllDistinctDatabaseIds().spliterator(), false)
-        .map(databaseId -> UserTableDto.builder().databaseId(databaseId).build())
-        .collect(Collectors.toList());
+    return METRICS_REPORTER.executeWithStats(
+        () ->
+            StreamSupport.stream(
+                    htsJdbcRepository.findAllDistinctDatabaseIds().spliterator(), false)
+                .map(databaseId -> UserTableDto.builder().databaseId(databaseId).build())
+                .collect(Collectors.toList()),
+        MetricsConstant.HTS_LIST_DATABASES_TIME);
   }
 
   private List<UserTableDto> listTables(UserTable userTable) {