add try catch block while create bigtable connection

caraml-dev · Sep 27, 2024 · 455e253 · 455e253
1 parent 7e7e417
commit 455e253
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/...ml-store-serving/src/main/java/dev/caraml/serving/store/bigtable/BigTableStoreConfig.java b/...ml-store-serving/src/main/java/dev/caraml/serving/store/bigtable/BigTableStoreConfig.java
@@ -36,7 +36,13 @@ public OnlineRetriever getRetriever() {
           BigtableConfiguration.configure(projectId, instanceId);
       config.set(BigtableOptionsFactory.APP_PROFILE_ID_KEY, appProfileId);
 
-      Connection connection = BigtableConfiguration.connect(config);
+      Connection connection;
+      try {
+        connection = BigtableConfiguration.connect(config);
+      } catch (IllegalStateException e) {
+        throw new RuntimeException(e);
+      }
+
       return new HBaseOnlineRetriever(connection);
     }