Fix DataSource cache when using modifiers

EngineHub · Aug 1, 2024 · 9658a8e · 9658a8e
1 parent 3999722
commit 9658a8e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/main/java/com/laytonsmith/persistence/DataSourceFactory.java b/src/main/java/com/laytonsmith/persistence/DataSourceFactory.java
@@ -53,7 +53,8 @@ private static void init() {
 	public static DataSource GetDataSource(URI uri, ConnectionMixinFactory.ConnectionMixinOptions options)
 			throws DataSourceException {
 		init();
-		DataSource source = DATA_SOURCE_POOL.get(uri);
+		URI uriKey = uri;
+		DataSource source = DATA_SOURCE_POOL.get(uriKey);
 		if(source != null) {
 			return source;
 		}
@@ -91,7 +92,7 @@ public static DataSource GetDataSource(URI uri, ConnectionMixinFactory.Connectio
 			if(!ds.getModifiers().contains(DataSource.DataSourceModifier.TRANSIENT)) {
 				ds.populate();
 			}
-			DATA_SOURCE_POOL.put(uri, ds);
+			DATA_SOURCE_POOL.put(uriKey, ds);
 			return ds;
 		} catch (InvocationTargetException | NoSuchMethodException | SecurityException | InstantiationException
 				| IllegalAccessException | IllegalArgumentException | DataSourceException ex) {