init

src/main/kotlin/database/DatabaseFactory.kt

@@ -0,0 +1,38 @@
+package database
+
+import config.EnvConfig
+import com.zaxxer.hikari.*
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.withContext
+import org.jetbrains.exposed.sql.*
+import org.jetbrains.exposed.sql.transactions.transaction
+
+object DatabaseFactory {
+
+    private val envConfig = EnvConfig()
+    private val dbUrl = envConfig.dbJdbc
+    private val dbUser = envConfig.dbUsername
+    private val dbPassword = envConfig.dbPassowrd
+
+    fun init() {
+        Database.connect(hikari())
+    }
+
+    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/main/kotlin/database/dao/ResultObjects.kt

@@ -0,0 +1,13 @@
+package 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 ResultObjects : 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/main/kotlin/database/model/ResultObject.kt

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

src/main/kotlin/database/service/IResultObjectService.kt

@@ -0,0 +1,16 @@
+package database.service
+
+import java.time.LocalDate
+import api.ApiObject
+
+interface IResultObjectService{
+    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/main/kotlin/database/service/ResultObjectService.kt

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