This commit is contained in:
46
src/database/DatabaseFactory.kt
Normal file
46
src/database/DatabaseFactory.kt
Normal file
@ -0,0 +1,46 @@
|
||||
package com.kmalbz.database
|
||||
|
||||
import com.typesafe.config.ConfigFactory
|
||||
import com.zaxxer.hikari.*
|
||||
import io.ktor.config.HoconApplicationConfig
|
||||
import io.ktor.util.KtorExperimentalAPI
|
||||
import kotlinx.coroutines.Dispatchers
|
||||
import kotlinx.coroutines.withContext
|
||||
import org.jetbrains.exposed.sql.*
|
||||
import org.jetbrains.exposed.sql.transactions.transaction
|
||||
|
||||
object DatabaseFactory {
|
||||
|
||||
@KtorExperimentalAPI
|
||||
private val appConfig = HoconApplicationConfig(ConfigFactory.load())
|
||||
@KtorExperimentalAPI
|
||||
private val dbUrl = appConfig.property("ktor.db.jdbc").getString()
|
||||
@KtorExperimentalAPI
|
||||
private val dbUser = appConfig.property("ktor.db.user").getString()
|
||||
@KtorExperimentalAPI
|
||||
private val dbPassword = appConfig.property("ktor.db.password").getString()
|
||||
|
||||
@KtorExperimentalAPI
|
||||
fun init() {
|
||||
Database.connect(hikari())
|
||||
}
|
||||
|
||||
@KtorExperimentalAPI
|
||||
private fun hikari(): HikariDataSource {
|
||||
val config = HikariConfig()
|
||||
config.driverClassName = "org.postgresql.Driver"
|
||||
config.jdbcUrl = dbUrl
|
||||
config.username = dbUser
|
||||
config.password = dbPassword
|
||||
config.maximumPoolSize = 3
|
||||
config.isAutoCommit = false
|
||||
config.transactionIsolation = "TRANSACTION_REPEATABLE_READ"
|
||||
config.validate()
|
||||
return HikariDataSource(config)
|
||||
}
|
||||
|
||||
suspend fun <T> dbQuery(block: () -> T): T =
|
||||
withContext(Dispatchers.IO) {
|
||||
transaction { block() }
|
||||
}
|
||||
}
|
14
src/database/dao/SampleObjects.kt
Normal file
14
src/database/dao/SampleObjects.kt
Normal file
@ -0,0 +1,14 @@
|
||||
package com.kmalbz.database.dao
|
||||
|
||||
import org.jetbrains.exposed.dao.id.IntIdTable
|
||||
import org.jetbrains.exposed.sql.*
|
||||
import org.jetbrains.exposed.sql.`java-time`.date
|
||||
import java.time.LocalDate
|
||||
|
||||
object SampleObjects : IntIdTable() {
|
||||
val tag: Column<String> = varchar("tag", 32)
|
||||
val timestamp: Column<LocalDate> = date("timestamp").default(LocalDate.now())
|
||||
val device_id: Column<Int> = integer("device_id")
|
||||
val device_date: Column<LocalDate> = date("device_date")
|
||||
override val primaryKey = PrimaryKey(id, name = "PK_ResultObject_Id")
|
||||
}
|
14
src/database/model/SampleObject.kt
Normal file
14
src/database/model/SampleObject.kt
Normal file
@ -0,0 +1,14 @@
|
||||
package com.kmalbz.database.model
|
||||
|
||||
import com.kmalbz.database.dao.SampleObjects
|
||||
import org.jetbrains.exposed.dao.IntEntity
|
||||
import org.jetbrains.exposed.dao.IntEntityClass
|
||||
import org.jetbrains.exposed.dao.id.EntityID
|
||||
|
||||
class SampleObject(id: EntityID<Int>): IntEntity(id) {
|
||||
companion object : IntEntityClass<SampleObject>(SampleObjects)
|
||||
var tag by SampleObjects.tag
|
||||
var timestamp by SampleObjects.timestamp
|
||||
var device_id by SampleObjects.device_id
|
||||
var device_date by SampleObjects.device_date
|
||||
}
|
17
src/database/service/ISampleObjectService.kt
Normal file
17
src/database/service/ISampleObjectService.kt
Normal file
@ -0,0 +1,17 @@
|
||||
package com.kmalbz.database.service
|
||||
|
||||
import com.kmalbz.api.model.ApiObject
|
||||
import java.time.LocalDate
|
||||
import java.util.*
|
||||
|
||||
interface ISampleObjectService{
|
||||
fun addOne(apiObject: ApiObject)
|
||||
suspend fun getAllSampleObjects(): List<ApiObject>
|
||||
suspend fun getSampleObjectbyTag(tag: String): ApiObject?
|
||||
suspend fun getSampleObjectbyDate(date: LocalDate): List<ApiObject>?
|
||||
suspend fun getSampleObjectbeforeDate(date: LocalDate): List<ApiObject>?
|
||||
suspend fun getSampleObjectafterDate(date: LocalDate): List<ApiObject>?
|
||||
suspend fun getSampleObjecGreaterthanId(id: Int): List<ApiObject>?
|
||||
suspend fun getSampleObjecLessthanId(id: Int): List<ApiObject>?
|
||||
suspend fun getSampleObjecEqualsId(id: Int): List<ApiObject>?
|
||||
}
|
81
src/database/service/SampleObjectService.kt
Normal file
81
src/database/service/SampleObjectService.kt
Normal file
@ -0,0 +1,81 @@
|
||||
package com.kmalbz.database.service
|
||||
|
||||
import com.kmalbz.database.DatabaseFactory.dbQuery
|
||||
import com.kmalbz.database.model.SampleObject
|
||||
import com.kmalbz.database.dao.SampleObjects
|
||||
import com.kmalbz.api.model.ApiObject
|
||||
import org.jetbrains.exposed.sql.ResultRow
|
||||
import org.jetbrains.exposed.sql.select
|
||||
import org.jetbrains.exposed.sql.selectAll
|
||||
import org.jetbrains.exposed.sql.transactions.transaction
|
||||
import java.time.LocalDate
|
||||
import java.util.*
|
||||
|
||||
|
||||
class SampleObjectService : ISampleObjectService {
|
||||
|
||||
override fun addOne(apiObject: ApiObject) {
|
||||
transaction {
|
||||
SampleObject.new {
|
||||
tag = apiObject.tag
|
||||
device_date = apiObject.device_date
|
||||
device_id = apiObject.device_id
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
override suspend fun getAllSampleObjects(): List<ApiObject> = dbQuery {
|
||||
SampleObjects.selectAll().map { toResultObject(it) }
|
||||
}
|
||||
|
||||
override suspend fun getSampleObjectbyTag(tag: String): ApiObject? = dbQuery {
|
||||
SampleObjects.select {
|
||||
(SampleObjects.tag eq tag)
|
||||
}.mapNotNull { toResultObject(it) }
|
||||
.singleOrNull()
|
||||
}
|
||||
|
||||
override suspend fun getSampleObjectbyDate(date: LocalDate): List<ApiObject>? = dbQuery {
|
||||
SampleObjects.select {
|
||||
(SampleObjects.date eq date)
|
||||
}.mapNotNull { toResultObject(it) }
|
||||
}
|
||||
|
||||
override suspend fun getSampleObjectbeforeDate(date: LocalDate): List<ApiObject>? = dbQuery {
|
||||
SampleObjects.select {
|
||||
(SampleObjects.date less date)
|
||||
}.mapNotNull { toResultObject(it) }
|
||||
}
|
||||
|
||||
override suspend fun getSampleObjectafterDate(date: LocalDate): List<ApiObject>? = dbQuery {
|
||||
SampleObjects.select {
|
||||
(SampleObjects.date greater date)
|
||||
}.mapNotNull { toResultObject(it) }
|
||||
}
|
||||
|
||||
override suspend fun getSampleObjecGreaterthanId(probability: Double): List<ApiObject>? = dbQuery {
|
||||
SampleObjects.select {
|
||||
(SampleObjects.probability greater probability)
|
||||
}.mapNotNull { toResultObject(it) }
|
||||
}
|
||||
|
||||
override suspend fun getSampleObjecLessthanId(probability: Double): List<ApiObject>? = dbQuery {
|
||||
SampleObjects.select {
|
||||
(SampleObjects.probability less probability)
|
||||
}.mapNotNull { toResultObject(it) }
|
||||
}
|
||||
|
||||
override suspend fun getSampleObjecEqualsId(probability: Double): List<ApiObject>? = dbQuery {
|
||||
SampleObjects.select {
|
||||
(SampleObjects.probability eq probability)
|
||||
}.mapNotNull { toResultObject(it) }
|
||||
}
|
||||
|
||||
private fun toResultObject(row: ResultRow): ApiObject =
|
||||
ApiObject(
|
||||
tag = row[SampleObjects.tag],
|
||||
device_date = row[SampleObjects.device_date],
|
||||
device_id = row[SampleObjects.device_id],
|
||||
id = row[SampleObjects.id.]
|
||||
)
|
||||
}
|
Reference in New Issue
Block a user