Support mariadb

5 files changed, 51 insertions, 8 deletions

diff --git a/build.gradle.kts b/build.gradle.kts
index cd04016..878e17c 100644
--- a/build.gradle.kts
+++ b/build.gradle.kts
@@ -5,6 +5,7 @@ import org.jetbrains.kotlin.gradle.dsl.Coroutines.ENABLE
 import org.jetbrains.kotlin.gradle.plugin.KotlinPlatformJvmPlugin
 import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
 import java.io.PrintWriter
+import java.net.URL
 
 val stdout = PrintWriter("gradle-output.txt")
 
@@ -124,6 +125,17 @@ tasks {
         manifest.attributes["Class-Path"] = "../lib/kotlin-stdlib.jar"
         dependsOn(kotlinStdlibJar)
     }
+
+    val createDebugServer by creating {
+
+        val jarUrl = URL("https://yivesmirror.com/files/spigot/spigot-latest.jar")
+        val serverJarFile = file("$serverDir/lib/spigot.jar")
+
+
+        doFirst {
+
+        }
+    }
 }
 
 stdout.flush()
diff --git a/src/main/kotlin/io/dico/parcels2/storage/ExposedBacking.kt b/src/main/kotlin/io/dico/parcels2/storage/ExposedBacking.kt
index 9a26acc..cbcb6f4 100644
--- a/src/main/kotlin/io/dico/parcels2/storage/ExposedBacking.kt
+++ b/src/main/kotlin/io/dico/parcels2/storage/ExposedBacking.kt
@@ -9,6 +9,7 @@ import kotlinx.coroutines.experimental.channels.ProducerScope
 import org.jetbrains.exposed.sql.*
 import org.jetbrains.exposed.sql.SchemaUtils.create
 import org.jetbrains.exposed.sql.transactions.transaction
+import org.jetbrains.exposed.sql.vendors.DatabaseDialect
 import org.joda.time.DateTime
 import java.util.*
 import javax.sql.DataSource
@@ -62,6 +63,14 @@ class ExposedBacking(private val dataSourceFactory: () -> DataSource) : Backing
 
     override val isConnected get() = database != null
 
+    companion object {
+        init {
+            Database.registerDialect("mariadb") {
+                Class.forName("org.jetbrains.exposed.sql.vendors.MysqlDialect").newInstance() as DatabaseDialect
+            }
+        }
+    }
+
     override suspend fun init() {
         if (isShutdown) throw IllegalStateException()
         dataSource = dataSourceFactory()
diff --git a/src/main/kotlin/io/dico/parcels2/storage/ExposedExtensions.kt b/src/main/kotlin/io/dico/parcels2/storage/ExposedExtensions.kt
index 7533586..816ff97 100644
--- a/src/main/kotlin/io/dico/parcels2/storage/ExposedExtensions.kt
+++ b/src/main/kotlin/io/dico/parcels2/storage/ExposedExtensions.kt
@@ -1,8 +1,10 @@
 package io.dico.parcels2.storage
 
-import org.jetbrains.exposed.sql.*
+import org.jetbrains.exposed.sql.Column
+import org.jetbrains.exposed.sql.Index
+import org.jetbrains.exposed.sql.Table
+import org.jetbrains.exposed.sql.Transaction
 import org.jetbrains.exposed.sql.statements.InsertStatement
-import org.jetbrains.exposed.sql.statements.UpdateStatement
 import org.jetbrains.exposed.sql.transactions.TransactionManager
 
 class UpsertStatement<Key : Any>(table: Table, conflictColumn: Column<*>? = null, conflictIndex: Index? = null)
@@ -26,12 +28,23 @@ class UpsertStatement<Key : Any>(table: Table, conflictColumn: Column<*>? = null
 
     override fun prepareSQL(transaction: Transaction) = buildString {
         append(super.prepareSQL(transaction))
-        append(" ON CONFLICT(")
-        append(indexName)
-        append(") DO UPDATE SET ")
 
-        values.keys.filter { it !in indexColumns}.joinTo(this) { "${transaction.identity(it)}=EXCLUDED.${transaction.identity(it)}" }
-    }.also { println(it) }
+        val dialect = transaction.db.vendor
+        if (dialect == "postgresql") {
+
+            append(" ON CONFLICT(")
+            append(indexName)
+            append(") DO UPDATE SET ")
+
+            values.keys.filter { it !in indexColumns }.joinTo(this) { "${transaction.identity(it)}=EXCLUDED.${transaction.identity(it)}" }
+
+        } else {
+
+            append (" ON DUPLICATE KEY UPDATE ")
+            values.keys.filter { it !in indexColumns }.joinTo(this) { "${transaction.identity(it)}=VALUES(${transaction.identity(it)})" }
+
+        }
+    }
 
 }
 
diff --git a/src/main/kotlin/io/dico/parcels2/storage/Hikari.kt b/src/main/kotlin/io/dico/parcels2/storage/Hikari.kt
index e21a24e..2dfa872 100644
--- a/src/main/kotlin/io/dico/parcels2/storage/Hikari.kt
+++ b/src/main/kotlin/io/dico/parcels2/storage/Hikari.kt
@@ -15,6 +15,15 @@ fun getHikariConfig(dialectName: String,
             dataSourceProperties["portNumber"] = port.toString()
             dataSourceProperties["databaseName"] = dco.database
         }
+
+        "mariadb" -> run {
+            dataSourceClassName = "org.mariadb.jdbc.MariaDbDataSource"
+            dataSourceProperties["serverName"] = address
+            dataSourceProperties["port"] = port.toString()
+            dataSourceProperties["databaseName"] = dco.database
+            dataSourceProperties["properties"] = "useUnicode=true;characterEncoding=utf8"
+        }
+
         else -> throw IllegalArgumentException("Unsupported dialect: $dialectName")
     }
 
diff --git a/src/main/kotlin/io/dico/parcels2/storage/StorageFactory.kt b/src/main/kotlin/io/dico/parcels2/storage/StorageFactory.kt
index 13df2ed..bb5013a 100644
--- a/src/main/kotlin/io/dico/parcels2/storage/StorageFactory.kt
+++ b/src/main/kotlin/io/dico/parcels2/storage/StorageFactory.kt
@@ -27,7 +27,7 @@ interface StorageFactory {
 
 class ConnectionStorageFactory : StorageFactory {
     override val optionsClass = DataConnectionOptions::class
-    private val types: List<String> = listOf("postgresql")
+    private val types: List<String> = listOf("postgresql", "mariadb")
 
     fun register(companion: StorageFactory.StorageFactories) {
         types.forEach { companion.registerFactory(it, this) }