add option waitForConnection in database config

This commit is contained in:
InsanusMokrassar 2022-09-04 14:37:22 +06:00
parent ab0578b312
commit dfbc7be6f2
3 changed files with 28 additions and 9 deletions

View File

@ -2,6 +2,9 @@
## 2.3.0
* `Bot`:
* Add option `waitForConnection` in database config
## 2.2.0
* `Versions`:

View File

@ -6,6 +6,7 @@ import org.jetbrains.exposed.sql.Database
import org.jetbrains.exposed.sql.transactions.transactionManager
import org.koin.core.scope.Scope
import org.sqlite.JDBC
import java.lang.Exception
import java.sql.Connection
inline val Scope.database: Database?
@ -16,10 +17,14 @@ data class DatabaseConfig(
val url: String = "jdbc:sqlite:file:test?mode=memory&cache=shared",
val driver: String = JDBC::class.qualifiedName!!,
val username: String = "",
val password: String = ""
val password: String = "",
val waitForConnection: Boolean = true
) {
@Transient
val database: Database = Database.connect(
val database: Database by lazy(LazyThreadSafetyMode.SYNCHRONIZED) {
while (true) {
return@lazy try {
Database.connect(
url,
driver,
username,
@ -27,4 +32,15 @@ data class DatabaseConfig(
).also {
it.transactionManager.defaultIsolationLevel = Connection.TRANSACTION_SERIALIZABLE // Or Connection.TRANSACTION_READ_UNCOMMITTED
}
} catch (e: Throwable) {
if (waitForConnection) {
Thread.sleep(1000L)
continue
} else {
throw e
}
}
}
error("Unable to get database by some reason")
}
}

View File

@ -4,7 +4,7 @@
"driver": "org.sqlite.JDBC",
"username": "OPTIONAL username",
"password": "OPTIONAL password",
"initAutomatically": false
"waitForConnection": true
},
"botToken": "1234567890:ABCDEFGHIJKLMNOP_qrstuvwxyz12345678",
"plugins": [