summaryrefslogtreecommitdiff
path: root/src/main/kotlin/io/dico/parcels2/storage/StorageFactory.kt
blob: c1e158a79ad4cf62f70b2c2e26f45d238803a079 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package io.dico.parcels2.storage

import io.dico.parcels2.DataConnectionOptions
import net.minecraft.server.v1_13_R1.WorldType.types
import kotlin.reflect.KClass

interface StorageFactory {
    companion object StorageFactories {
        private val map: MutableMap<String, StorageFactory> = HashMap()

        fun registerFactory(method: String, generator: StorageFactory): Boolean = map.putIfAbsent(method.toLowerCase(), generator) == null

        fun getFactory(method: String): StorageFactory? = map[method.toLowerCase()]

        init {
            // have to write the code like this in kotlin.
            // This code is absolutely disgusting
            ConnectionStorageFactory().register(this)
        }
    }

    val optionsClass: KClass<out Any>

    fun newStorageInstance(method: String, options: Any): Storage

}

class ConnectionStorageFactory : StorageFactory {
    override val optionsClass = DataConnectionOptions::class

    private val types: Map<String, String> = mutableMapOf(
            "mysql" to "com.mysql.jdbc.jdbc2.optional.MysqlDataSource",
            "h2" to "org.h2.jdbcx.JdbcDataSource"
    )

    fun register(companion: StorageFactory.StorageFactories) {
        types.keys.forEach {
            companion.registerFactory(it, this)
        }
    }

    override fun newStorageInstance(dialect: String, options: Any): Storage {
        val driverClass = types[dialect.toLowerCase()] ?: throw IllegalArgumentException("Storage dialect $dialect is not supported")
        return StorageWithCoroutineBacking(ExposedBacking(getHikariDataSource(dialect, driverClass, options as DataConnectionOptions)))
    }

}