pass install.keyValueService to TKVSMF

palantir · May 2, 2024 · aaba39e · aaba39e
1 parent f7e4929
commit aaba39e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/...-api/src/main/java/com/palantir/atlasdb/spi/TransactionKeyValueServiceManagerFactory.java b/...-api/src/main/java/com/palantir/atlasdb/spi/TransactionKeyValueServiceManagerFactory.java
@@ -45,6 +45,7 @@ TransactionKeyValueServiceManager create(
             MetricsManager metricsManager,
             CoordinationService<T> coordinationService,
             KeyValueServiceManager keyValueServiceManager,
+            KeyValueServiceConfig sourceInstall,
             TransactionKeyValueServiceConfig install,
             Refreshable<TransactionKeyValueServiceRuntimeConfig> runtime,
             boolean initializeAsync);

diff --git a/atlasdb-config/src/main/java/com/palantir/atlasdb/factory/TransactionManagers.java b/atlasdb-config/src/main/java/com/palantir/atlasdb/factory/TransactionManagers.java
@@ -463,6 +463,7 @@ private TransactionManager serializableInternal(@Output List<AutoCloseable> clos
                                 lockAndTimestampServices,
                                 internalKeyValueService,
                                 keyValueServiceManager,
+                                config().keyValueService(),
                                 config().transactionKeyValueService().get(),
                                 runtimeConfig().get().map(optionalConfig -> optionalConfig
                                         .get()
@@ -656,6 +657,7 @@ private <T> TransactionKeyValueServiceManager createTransactionKeyValueServiceMa
             LockAndTimestampServices lockAndTimestampServices,
             KeyValueService internalTablesKeyValueService,
             KeyValueServiceManager keyValueServiceManager,
+            KeyValueServiceConfig sourceConfig,
             TransactionKeyValueServiceConfig config,
             Refreshable<TransactionKeyValueServiceRuntimeConfig> runtimeConfigRefreshable,
             boolean initializeAsync) {
@@ -675,6 +677,7 @@ private <T> TransactionKeyValueServiceManager createTransactionKeyValueServiceMa
                 metricsManager,
                 coordinationService,
                 keyValueServiceManager,
+                sourceConfig,
                 config,
                 runtimeConfigRefreshable,
                 initializeAsync);