Cache connections with the same jdbc url (fixes #2)

btk5h · Dec 20, 2017 · 3c485cf · 3c485cf
1 parent b3c5c36
commit 3c485cf
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/main/java/com/btk5h/skriptdb/skript/ExprDataSource.java b/src/main/java/com/btk5h/skriptdb/skript/ExprDataSource.java
@@ -4,6 +4,9 @@
 
 import org.bukkit.event.Event;
 
+import java.util.HashMap;
+import java.util.Map;
+
 import ch.njol.skript.Skript;
 import ch.njol.skript.lang.Expression;
 import ch.njol.skript.lang.ExpressionType;
@@ -30,6 +33,8 @@ public class ExprDataSource extends SimpleExpression<HikariDataSource> {
         ExpressionType.COMBINED, "[the] data(base|[ ]source) [(of|at)] %string%");
   }
 
+  private static Map<String, HikariDataSource> connectionCache = new HashMap<>();
+
   private Expression<String> url;
 
   @Override
@@ -43,9 +48,15 @@ protected HikariDataSource[] get(Event e) {
       jdbcUrl = "jdbc:" + jdbcUrl;
     }
 
+    if (connectionCache.containsKey(jdbcUrl)) {
+      return new HikariDataSource[]{connectionCache.get(jdbcUrl)};
+    }
+
     HikariDataSource ds = new HikariDataSource();
     ds.setJdbcUrl(jdbcUrl);
 
+    connectionCache.put(jdbcUrl, ds);
+
     return new HikariDataSource[]{ds};
   }