port RedstonerPlots a bit

1 files changed, 51 insertions, 0 deletions

diff --git a/src/main/kotlin/io/dico/parcels2/storage/Hikari.kt b/src/main/kotlin/io/dico/parcels2/storage/Hikari.kt
new file mode 100644
index 0000000..7e4fb7f
--- /dev/null
+++ b/src/main/kotlin/io/dico/parcels2/storage/Hikari.kt
@@ -0,0 +1,51 @@
+package io.dico.parcels2.storage
+
+import com.zaxxer.hikari.HikariConfig
+import com.zaxxer.hikari.HikariDataSource
+import io.dico.parcels2.DataConnectionOptions
+import javax.sql.DataSource
+
+fun getHikariDataSource(dialectName: String,
+                        driver: String,
+                        dco: DataConnectionOptions): DataSource = with(HikariConfig()) {
+
+    val (address, port) = dco.splitAddressAndPort() ?: throw IllegalArgumentException("Invalid address: ${dco.address}")
+
+    poolName = "redstonerplots"
+    maximumPoolSize = dco.poolSize
+    dataSourceClassName = driver
+    username = dco.username
+    password = dco.password
+    connectionTimeout = 15000
+    leakDetectionThreshold = 10000
+    connectionTestQuery = "SELECT 1"
+
+    addDataSourceProperty("serverName", address)
+    addDataSourceProperty("port", port.toString())
+    addDataSourceProperty("databaseName", dco.database)
+
+    // copied from github.com/lucko/LuckPerms
+    if (dialectName.toLowerCase() == "mariadb") {
+        addDataSourceProperty("properties", "useUnicode=true;characterEncoding=utf8")
+    } else {
+        // doesn't exist on the MariaDB driver
+        addDataSourceProperty("cachePrepStmts", "true")
+        addDataSourceProperty("alwaysSendSetIsolation", "false")
+        addDataSourceProperty("cacheServerConfiguration", "true")
+        addDataSourceProperty("elideSetAutoCommits", "true")
+        addDataSourceProperty("useLocalSessionState", "true")
+
+        // already set as default on mariadb
+        addDataSourceProperty("useServerPrepStmts", "true")
+        addDataSourceProperty("prepStmtCacheSize", "250")
+        addDataSourceProperty("prepStmtCacheSqlLimit", "2048")
+        addDataSourceProperty("cacheCallableStmts", "true")
+
+        // make sure unicode characters can be used.
+        addDataSourceProperty("characterEncoding", "utf8")
+        addDataSourceProperty("useUnicode", "true")
+    }
+
+    HikariDataSource(this)
+
+}