close conn

rayokota · Jul 4, 2024 · eb8a572 · eb8a572
1 parent 282be3b
commit eb8a572
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/main/java/io/kcache/kwack/KwackEngine.java b/src/main/java/io/kcache/kwack/KwackEngine.java
@@ -817,6 +817,11 @@ public void close() throws IOException {
                 LOG.warn("Could not close cache for {}", key);
             }
         });
+        try {
+            conn.close();
+        } catch (SQLException e) {
+            LOG.warn("Could not close DuckDB connection for {}", config.getDbUrl());
+        }
         resetSchemaRegistry(config.getSchemaRegistryUrls(), schemaRegistry);
     }
 }
diff --git a/src/test/java/io/kcache/kwack/util/LocalClusterTestHarness.java b/src/test/java/io/kcache/kwack/util/LocalClusterTestHarness.java
@@ -96,6 +96,7 @@ protected void injectKwackProperties(Properties props) {
         props.put(KwackConfig.KAFKACACHE_BOOTSTRAP_SERVERS_CONFIG, bootstrapServers);
         props.put(KwackConfig.KAFKACACHE_TOPIC_REPLICATION_FACTOR_CONFIG, 1);
         props.put(KwackConfig.SCHEMA_REGISTRY_URL_CONFIG, MOCK_URL);
+        props.put(KwackConfig.DB_CONFIG, ":memory:?cache=private");
         props.put(AbstractKafkaSchemaSerDeConfig.AUTO_REGISTER_SCHEMAS, "true");
     }