init

src/main/kotlin/Main.kt

@@ -0,0 +1,12 @@
+import di.databasemodule
+import org.koin.core.context.GlobalContext.startKoin
+
+fun main(vararg args: String) {
+
+    startKoin {
+        printLogger()
+        modules(databasemodule)
+    }
+
+
+}

src/main/kotlin/api/ApiObject.kt

@@ -0,0 +1,8 @@
+package api
+
+import com.google.gson.annotations.SerializedName
+
+data class ApiObject(
+    @SerializedName("tag") val tag: String,
+    @SerializedName("probability") val probability: Double
+)

src/main/kotlin/config/EnvConfig.kt

@@ -0,0 +1,10 @@
+package config
+
+data class EnvConfig (
+    var mqHost: String = System.getenv("MQ_HOST") ?: "localhost",
+    var mqUserName: String = System.getenv("MQ_USERNAME") ?: "rabbitmq",
+    var mqPassWord: String = System.getenv("MQ_PASSWORD") ?: "rabbitmq",
+    var dbJdbc: String = System.getenv("DB_JDBC") ?: "input",
+    var dbUsername: String = System.getenv("DB_USERNAME") ?: "output",
+    var dbPassowrd: String = System.getenv("DB_PASSOWRD") ?: "output"
+)

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]
+        )
+}

src/main/kotlin/di/DatabaseModule.kt

@@ -0,0 +1,9 @@
+package di
+
+import database.service.IResultObjectService
+import database.service.ResultObjectService
+import org.koin.dsl.module
+
+val databasemodule = module(createdAtStart = true) {
+    single { ResultObjectService() as IResultObjectService }
+}