From aaba39e5394745b6d4a02d0718d86b5d1b0d3828 Mon Sep 17 00:00:00 2001
From: Raymond Huffman <rhuffman@palantir.com>
Date: Thu, 2 May 2024 13:22:09 -0700
Subject: [PATCH] pass install.keyValueService to TKVSMF

---
 .../atlasdb/spi/TransactionKeyValueServiceManagerFactory.java  | 1 +
 .../java/com/palantir/atlasdb/factory/TransactionManagers.java | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/atlasdb-api/src/main/java/com/palantir/atlasdb/spi/TransactionKeyValueServiceManagerFactory.java b/atlasdb-api/src/main/java/com/palantir/atlasdb/spi/TransactionKeyValueServiceManagerFactory.java
index 594e7102e57..e0db7ff37c8 100644
--- a/atlasdb-api/src/main/java/com/palantir/atlasdb/spi/TransactionKeyValueServiceManagerFactory.java
+++ b/atlasdb-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
index 83b48edf24c..c76de0e086d 100644
--- 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);