map not properties

snowflakedb · Nov 20, 2023 · 7628b75 · 7628b75
1 parent 0ce288a
commit 7628b75
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/main/java/com/snowflake/kafka/connector/internal/streaming/StreamingClientProvider.java b/src/main/java/com/snowflake/kafka/connector/internal/streaming/StreamingClientProvider.java
@@ -23,8 +23,6 @@
 import com.snowflake.kafka.connector.SnowflakeSinkConnectorConfig;
 import com.snowflake.kafka.connector.internal.KCLogger;
 import java.util.Map;
-import java.util.Properties;
-
 import net.snowflake.ingest.internal.com.github.benmanes.caffeine.cache.Caffeine;
 import net.snowflake.ingest.internal.com.github.benmanes.caffeine.cache.LoadingCache;
 import net.snowflake.ingest.internal.com.github.benmanes.caffeine.cache.RemovalCause;
@@ -65,7 +63,7 @@ public static StreamingClientProvider getStreamingClientProviderForTests(
 
   /** ONLY FOR TESTING - private constructor to inject properties for testing */
   private StreamingClientProvider(
-      LoadingCache<Properties, SnowflakeStreamingIngestClient> registeredClients,
+      LoadingCache<Map<String, String>, SnowflakeStreamingIngestClient> registeredClients,
       StreamingClientHandler streamingClientHandler) {
     this();
     this.registeredClients = registeredClients;
@@ -77,7 +75,7 @@ private StreamingClientProvider(
 
   // if the one client optimization is enabled, we cache the created clients based on corresponding
   // Streaming properties
-  private LoadingCache<Properties, SnowflakeStreamingIngestClient> registeredClients;
+  private LoadingCache<Map<String, String>, SnowflakeStreamingIngestClient> registeredClients;
 
   // private constructor for singleton
   private StreamingClientProvider() {
@@ -94,7 +92,7 @@ private StreamingClientProvider() {
                       client.getName(),
                       removalCause.toString());
                 })
-            .build();
+            .build(this.streamingClientHandler::createClient);
   }
 
   /**