initial commit

src/database/DatabaseFactory.kt

@@ -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() }
+        }
+}

src/database/dao/InputObjects.kt

@@ -0,0 +1,13 @@
+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 InputObjects : IntIdTable() {
+    val tag: Column<String> = varchar("tag",32)
+    val date: Column<LocalDate> = date("date").default(LocalDate.now())
+    val probability: Column<Double> = double("probability")
+    override val primaryKey = PrimaryKey(id, name = "PK_ResultObject_Id")
+}

src/database/model/InputObject.kt

@@ -0,0 +1,13 @@
+package com.kmalbz.database.model
+
+import com.kmalbz.database.dao.InputObjects
+import org.jetbrains.exposed.dao.IntEntity
+import org.jetbrains.exposed.dao.IntEntityClass
+import org.jetbrains.exposed.dao.id.EntityID
+
+class InputObject(id: EntityID<Int>): IntEntity(id) {
+    companion object : IntEntityClass<InputObject>(InputObjects)
+    var tag by InputObjects.tag
+    var date by InputObjects.date
+    var probability by InputObjects.probability
+}

src/database/service/IInputObjectService.kt

@@ -0,0 +1,16 @@
+package com.kmalbz.database.service
+
+import com.kmalbz.api.model.ApiObject
+import java.time.LocalDate
+
+interface IInputObjectService{
+    fun addOne(apiObject: ApiObject)
+    suspend fun getAllResultObjects(): List<ApiObject>
+    suspend fun getResultObjectbyTag(tag: String): ApiObject?
+    suspend fun getResultObjectbyDate(date: LocalDate): List<ApiObject>?
+    suspend fun getResultObjectbeforeDate(date: LocalDate): List<ApiObject>?
+    suspend fun getResultObjectafterDate(date: LocalDate): List<ApiObject>?
+    suspend fun getResultObjecGreaterthanProbability(probability: Double): List<ApiObject>?
+    suspend fun getResultObjecLessthanProbability(probability: Double): List<ApiObject>?
+    suspend fun getResultObjecEqualsProbability(probability: Double): List<ApiObject>?
+}

src/database/service/InputObjectService.kt

@@ -0,0 +1,77 @@
+package com.kmalbz.database.service
+
+import com.kmalbz.database.DatabaseFactory.dbQuery
+import com.kmalbz.database.model.InputObject
+import com.kmalbz.database.dao.InputObjects
+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
+
+
+class InputObjectService : IInputObjectService {
+
+    override fun addOne(apiObject: ApiObject) {
+        transaction {
+            InputObject.new {
+                tag = apiObject.tag
+                probability = apiObject.probability
+            }
+        }
+    }
+
+    override suspend fun getAllResultObjects(): List<ApiObject> = dbQuery {
+        InputObjects.selectAll().map { toResultObject(it) }
+    }
+
+    override suspend fun getResultObjectbyTag(tag: String): ApiObject? = dbQuery {
+        InputObjects.select {
+            (InputObjects.tag eq tag)
+        }.mapNotNull { toResultObject(it) }
+            .singleOrNull()
+    }
+
+    override suspend fun getResultObjectbyDate(date: LocalDate): List<ApiObject>? = dbQuery {
+        InputObjects.select {
+            (InputObjects.date eq date)
+        }.mapNotNull { toResultObject(it) }
+    }
+
+    override suspend fun getResultObjectbeforeDate(date: LocalDate): List<ApiObject>? = dbQuery {
+        InputObjects.select {
+            (InputObjects.date less  date)
+        }.mapNotNull { toResultObject(it) }
+    }
+
+    override suspend fun getResultObjectafterDate(date: LocalDate): List<ApiObject>? = dbQuery {
+        InputObjects.select {
+            (InputObjects.date greater  date)
+        }.mapNotNull { toResultObject(it) }
+    }
+
+    override suspend fun getResultObjecGreaterthanProbability(probability: Double): List<ApiObject>? = dbQuery {
+        InputObjects.select {
+            (InputObjects.probability greater  probability)
+        }.mapNotNull { toResultObject(it) }
+    }
+
+    override suspend fun getResultObjecLessthanProbability(probability: Double): List<ApiObject>? = dbQuery {
+        InputObjects.select {
+            (InputObjects.probability less  probability)
+        }.mapNotNull { toResultObject(it) }
+    }
+
+    override suspend fun getResultObjecEqualsProbability(probability: Double): List<ApiObject>? = dbQuery {
+        InputObjects.select {
+            (InputObjects.probability eq  probability)
+        }.mapNotNull { toResultObject(it) }
+    }
+
+    private fun toResultObject(row: ResultRow): ApiObject =
+        ApiObject(
+            tag = row[InputObjects.tag],
+            probability = row[InputObjects.probability]
+        )
+}