Expose some more functionality to override

linkedin · Dec 17, 2024 · 0e42c5f · 0e42c5f
1 parent 4c2ffb9
commit 0e42c5f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/hoptimator-venice/src/main/java/com/linkedin/hoptimator/venice/ClusterSchema.java b/hoptimator-venice/src/main/java/com/linkedin/hoptimator/venice/ClusterSchema.java
@@ -27,7 +27,7 @@ public class ClusterSchema extends AbstractSchema {
   private static final String SSL_FACTORY_CLASS_NAME = "ssl.factory.class.name";
   public static final String DEFAULT_SSL_FACTORY_CLASS_NAME = "com.linkedin.venice.security.DefaultSSLFactory";
 
-  private final Properties properties;
+  protected final Properties properties;
   private final Map<String, Table> tableMap = new HashMap<>();
 
   public ClusterSchema(Properties properties) {
@@ -53,7 +53,7 @@ public void populate() throws InterruptedException, ExecutionException, IOExcept
       log.info("Loaded {} Venice stores.", stores.length);
       for (String store : stores) {
         StoreSchemaFetcher storeSchemaFetcher = createStoreSchemaFetcher(store);
-        tableMap.put(store, new VeniceStore(storeSchemaFetcher));
+        tableMap.put(store, createVeniceStore(storeSchemaFetcher));
       }
     }
   }
@@ -73,6 +73,10 @@ protected StoreSchemaFetcher createStoreSchemaFetcher(String storeName) {
             .setVeniceURL(properties.getProperty("router.url")));
   }
 
+  protected VeniceStore createVeniceStore(StoreSchemaFetcher storeSchemaFetcher) {
+    return new VeniceStore(storeSchemaFetcher);
+  }
+
   @Override
   public Map<String, Table> getTableMap() {
     return tableMap;

diff --git a/hoptimator-venice/src/main/java/com/linkedin/hoptimator/venice/VeniceStore.java b/hoptimator-venice/src/main/java/com/linkedin/hoptimator/venice/VeniceStore.java
@@ -59,4 +59,8 @@ public RelDataType getRowType(RelDataTypeFactory typeFactory) {
     }
     return AvroConverter.rel(combinedSchema, typeFactory);
   }
+
+  protected RelDataType rel(Schema schema, RelDataTypeFactory typeFactory) {
+    return AvroConverter.rel(schema, typeFactory);
+  }
 }