BlockWorldGroup/Ledger-Databases
5
0
Fork 0
mirror of https://github.com/BlockWorldGroup/Ledger-Databases.git synced 2025-10-14 21:00:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improve performance

Browse Source
This commit is contained in:
Potatoboy9999
2022-06-14 12:09:37 -07:00
parent a767a78fc1
commit 92f9a172d1
4 changed files with 66 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
src/main/kotlin/net.quiltservertools.ledger.databases/LedgerDatabaseExtension.kt Normal file
View File

@@ -0,0 +1,50 @@
package net.quiltservertools.ledger.databases
import com.github.quiltservertools.ledger.Ledger
import com.github.quiltservertools.ledger.api.DatabaseExtension
import com.github.quiltservertools.libs.com.uchuhimo.konf.ConfigSpec
import com.zaxxer.hikari.HikariConfig
import com.zaxxer.hikari.HikariDataSource
import net.minecraft.util.Identifier
import org.h2.jdbcx.JdbcDataSource
import org.sqlite.SQLiteDataSource
import java.nio.file.Path
import javax.sql.DataSource
import kotlin.io.path.pathString
class LedgerDatabaseExtension : DatabaseExtension {
override fun getConfigSpecs(): List<ConfigSpec> = listOf(DatabaseExtensionSpec)
override fun getDataSource(fileLocation: Path): DataSource {
return when (Ledger.config[DatabaseExtensionSpec.database]) {
Databases.SQLITE -> SQLiteDataSource().apply {
url = "jdbc:sqlite:${fileLocation.resolve("ledger.sqlite").pathString}"
}
Databases.H2 -> JdbcDataSource().apply {
setURL("jdbc:h2:${fileLocation.resolve("ledger.h2").pathString};MODE=MySQL")
}
Databases.MYSQL -> HikariDataSource(HikariConfig().apply {
jdbcUrl = "jdbc:mysql://${Ledger.config[DatabaseExtensionSpec.url]}?rewriteBatchedStatements=true"
username = Ledger.config[DatabaseExtensionSpec.userName]
password = Ledger.config[DatabaseExtensionSpec.password]
maximumPoolSize = Ledger.config[DatabaseExtensionSpec.maxPoolSize]
})
Databases.POSTGRESQL -> HikariDataSource(HikariConfig().apply {
jdbcUrl = "jdbc:postgresql://${Ledger.config[DatabaseExtensionSpec.url]}?rewriteBatchedStatements=true"
username = Ledger.config[DatabaseExtensionSpec.userName]
password = Ledger.config[DatabaseExtensionSpec.password]
maximumPoolSize = Ledger.config[DatabaseExtensionSpec.maxPoolSize]
})
}
}
override fun getIdentifier(): Identifier = when (Ledger.config[DatabaseExtensionSpec.database]) {
Databases.MYSQL -> Ledger.identifier("mysql_extension")
Databases.H2 -> Ledger.identifier("h2_extension")
Databases.POSTGRESQL -> Ledger.identifier("postgresql_extension")
else -> Ledger.identifier(Ledger.DEFAULT_DATABASE)
}
}

4
src/main/kotlin/net/quiltservertools/ledger/databases/DatabaseExtensionSpec.kt
View File

@@ -7,5 +7,5 @@ object DatabaseExtensionSpec : ConfigSpec("database_extensions") {
val userName by optional("root", "username")
val password by optional("", "password")
val url by optional("localhost", "url")
val properties by optional(listOf<String>(), "properties")
}
val maxPoolSize by optional(10, "maxPoolSize")
}