com.sxtanna.database.Kuery.kt Maven / Gradle / Ivy

Go to download
package com.sxtanna.database

import com.sxtanna.database.base.Database
import com.sxtanna.database.config.DatabaseConfigManager
import com.sxtanna.database.config.KueryConfig
import com.sxtanna.database.ext.loadOrSave
import com.sxtanna.database.struct.base.Creator
import com.sxtanna.database.task.KueryTask
import com.sxtanna.database.type.base.SqlObject
import com.zaxxer.hikari.HikariConfig
import com.zaxxer.hikari.HikariDataSource
import org.slf4j.Logger
import org.slf4j.LoggerFactory
import java.io.File
import java.sql.Connection
import java.sql.ResultSet
import kotlin.reflect.KClass

class Kuery private constructor(override val config : KueryConfig) : Database() {

	override val name : String = "Kuery"
	lateinit var pool : HikariDataSource
		private set

    val debug by lazy { config.debug ?: false }
    val logger : Logger = LoggerFactory.getLogger(this::class.java)

	val creators = mutableMapOf, ResultSet.() -> SqlObject>()


	override fun load() {
		val hikariConfig = HikariConfig().apply {
			jdbcUrl = "jdbc:mysql://${config.server.address}:${config.server.port}/${config.server.database}?useSSL=false"
			username = config.user.username
			password = config.user.password
			maximumPoolSize = config.pool.size
			connectionTimeout = config.pool.timeout.toLong()
			idleTimeout = config.pool.idle.toLong()
			isAutoCommit = true

			addDataSourceProperty("cachePrepStmts", true)
			addDataSourceProperty("prepStmtCacheSize", 250)
			addDataSourceProperty("serverTimezone", "EST5EDT")
			addDataSourceProperty("prepStmtCacheSqlLimit", 2048)
			addDataSourceProperty("useServerPrepStmts", true)
			addDataSourceProperty("cacheCallableStmts", true)
			addDataSourceProperty("elideSetAutoCommits", true)
			addDataSourceProperty("useLocalSessionState", true)
			addDataSourceProperty("alwaysSendSetIsolation", true)
			addDataSourceProperty("cacheResultSetMetadata", true)
			addDataSourceProperty("cacheServerConfiguration", true)
		}

		pool = HikariDataSource(hikariConfig)
	}

	override fun poison() = pool.close().also { creators.clear() }


	override fun poolResource() : Connection? = pool.connection

	override fun createTask(resource : Connection) : KueryTask = KueryTask(this, resource)


	inline fun  addCreator(noinline creator : ResultSet.() -> T) {
		creators[T::class] = creator
	}

	fun  addCreator(creator : Creator) {
		creators[creator.clazz] = creator
	}


	companion object : DatabaseConfigManager {

		@JvmStatic
		override fun get(file : File) = super.get(file)

		@JvmStatic
		override fun get(config : KueryConfig) = Kuery(config)

		@JvmStatic
		override fun getConfig(file : File) : KueryConfig = file.loadOrSave(KueryConfig.DEFAULT)

	}

}