This commit is contained in:
71
src/Application.kt
Normal file
71
src/Application.kt
Normal file
@ -0,0 +1,71 @@
|
||||
package com.kmalbz
|
||||
|
||||
import com.kmalbz.api.route.InputServiceServer
|
||||
import io.ktor.application.*
|
||||
import io.ktor.response.*
|
||||
import io.ktor.routing.*
|
||||
import io.ktor.http.*
|
||||
import io.ktor.gson.*
|
||||
import io.ktor.features.*
|
||||
import org.apache.http.HttpException
|
||||
import com.kmalbz.database.DatabaseFactory
|
||||
import com.kmalbz.database.dao.InputObjects
|
||||
import io.ktor.util.KtorExperimentalAPI
|
||||
import com.rabbitmq.client.*
|
||||
import com.typesafe.config.ConfigFactory
|
||||
import io.ktor.config.HoconApplicationConfig
|
||||
import org.jetbrains.exposed.sql.SchemaUtils
|
||||
import org.jetbrains.exposed.sql.transactions.transaction
|
||||
import org.koin.ktor.ext.Koin
|
||||
|
||||
fun main(args: Array<String>): Unit = io.ktor.server.netty.EngineMain.main(args)
|
||||
|
||||
@KtorExperimentalAPI
|
||||
@Suppress("unused") // Referenced in application.conf
|
||||
fun Application.module() {
|
||||
install(ContentNegotiation) {
|
||||
gson {
|
||||
}
|
||||
}
|
||||
|
||||
install(Koin) {
|
||||
printLogger()
|
||||
modules(com.kmalbz.di.injectionModule)
|
||||
}
|
||||
|
||||
DatabaseFactory.init()
|
||||
transaction{
|
||||
SchemaUtils.create(InputObjects)
|
||||
}
|
||||
|
||||
val appConfig = HoconApplicationConfig(ConfigFactory.load())
|
||||
val factory = ConnectionFactory()
|
||||
factory.host = appConfig.property("ktor.mq.host").getString()
|
||||
factory.username = appConfig.property("ktor.mq.username").getString()
|
||||
factory.password = appConfig.property("ktor.mq.password").getString()
|
||||
|
||||
val connection = factory.newConnection()
|
||||
val channel = connection.createChannel()
|
||||
|
||||
val rabbitExchangeName = appConfig.property("ktor.mq.exchange").getString()
|
||||
|
||||
channel.exchangeDeclare(rabbitExchangeName, BuiltinExchangeType.FANOUT)
|
||||
val queueName = channel.queueDeclare().queue
|
||||
channel.queueBind(queueName, rabbitExchangeName, "")
|
||||
|
||||
routing {
|
||||
install(StatusPages) {
|
||||
exception<HttpException> {
|
||||
call.respond(HttpStatusCode.BadRequest)
|
||||
}
|
||||
|
||||
exception<IllegalStateException> {
|
||||
call.respond(HttpStatusCode.NotAcceptable)
|
||||
}
|
||||
}
|
||||
|
||||
InputServiceServer().apply {
|
||||
registerOutput()
|
||||
}
|
||||
}
|
||||
}
|
8
src/api/model/ApiObject.kt
Normal file
8
src/api/model/ApiObject.kt
Normal file
@ -0,0 +1,8 @@
|
||||
package com.kmalbz.api.model
|
||||
|
||||
import com.google.gson.annotations.SerializedName
|
||||
|
||||
data class ApiObject(
|
||||
@SerializedName("tag") val tag: String,
|
||||
@SerializedName("probability") val probability: Double
|
||||
)
|
71
src/api/route/InputServiceServer.kt
Normal file
71
src/api/route/InputServiceServer.kt
Normal file
@ -0,0 +1,71 @@
|
||||
package com.kmalbz.api.route
|
||||
|
||||
import com.kmalbz.database.service.IInputObjectService
|
||||
import io.ktor.application.call
|
||||
import io.ktor.http.HttpStatusCode
|
||||
import io.ktor.response.respond
|
||||
import io.ktor.routing.Routing
|
||||
import io.ktor.routing.get
|
||||
import org.koin.ktor.ext.inject
|
||||
import java.time.LocalDate
|
||||
import java.time.format.DateTimeFormatter
|
||||
|
||||
/**
|
||||
* Output Service - RDB
|
||||
*
|
||||
* This is the output interface of the Birbnetes system.
|
||||
*/
|
||||
class InputServiceServer {
|
||||
/**
|
||||
* output
|
||||
*/
|
||||
fun Routing.registerOutput() {
|
||||
val resultObjectService by inject<IInputObjectService>()
|
||||
get("/output"){
|
||||
call.respond(resultObjectService.getAllResultObjects())
|
||||
}
|
||||
|
||||
get("/output/filter/negative") {
|
||||
val resultList = resultObjectService.getResultObjecLessthanProbability(0.5) ?: call.respond(HttpStatusCode.NotFound)
|
||||
|
||||
call.respond(resultList)
|
||||
}
|
||||
|
||||
get("/output/filter/positive") {
|
||||
val resultList = resultObjectService.getResultObjecGreaterthanProbability(0.5) ?: call.respond(HttpStatusCode.NotFound)
|
||||
|
||||
call.respond(resultList)
|
||||
}
|
||||
|
||||
get("/output/filter/undecided") {
|
||||
val resultList = resultObjectService.getResultObjecEqualsProbability(0.5) ?: call.respond(HttpStatusCode.NotFound)
|
||||
|
||||
call.respond(resultList)
|
||||
}
|
||||
|
||||
get("/output/after/{dateAfter}") {
|
||||
val dateAfter = call.parameters["dateAfter"] ?: error(HttpStatusCode.NotAcceptable)
|
||||
val dateTimeFormatter: DateTimeFormatter = DateTimeFormatter.ISO_LOCAL_DATE
|
||||
val localDateAfter : LocalDate = LocalDate.parse(dateAfter,dateTimeFormatter)
|
||||
val resultList = resultObjectService.getResultObjectafterDate(localDateAfter) ?: call.respond(HttpStatusCode.NotFound)
|
||||
|
||||
call.respond(resultList)
|
||||
}
|
||||
|
||||
get("/output/before/{dateBefore}") {
|
||||
val dateAfter = call.parameters["dateBefore"] ?: error(HttpStatusCode.NotAcceptable)
|
||||
val dateTimeFormatter: DateTimeFormatter = DateTimeFormatter.ISO_LOCAL_DATE
|
||||
val localDateBefore : LocalDate = LocalDate.parse(dateAfter,dateTimeFormatter)
|
||||
val resultList = resultObjectService.getResultObjectbeforeDate(localDateBefore) ?: call.respond(HttpStatusCode.NotFound)
|
||||
|
||||
call.respond(resultList)
|
||||
}
|
||||
|
||||
get("/output/{tagID}") {
|
||||
val tagID = call.parameters["tagID"] ?: error(HttpStatusCode.NotAcceptable)
|
||||
val resultObject = resultObjectService.getResultObjectbyTag(tagID) ?: call.respond(HttpStatusCode.NotFound)
|
||||
|
||||
call.respond(resultObject)
|
||||
}
|
||||
}
|
||||
}
|
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() }
|
||||
}
|
||||
}
|
13
src/database/dao/InputObjects.kt
Normal file
13
src/database/dao/InputObjects.kt
Normal file
@ -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")
|
||||
}
|
13
src/database/model/InputObject.kt
Normal file
13
src/database/model/InputObject.kt
Normal file
@ -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
|
||||
}
|
16
src/database/service/IInputObjectService.kt
Normal file
16
src/database/service/IInputObjectService.kt
Normal file
@ -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>?
|
||||
}
|
77
src/database/service/InputObjectService.kt
Normal file
77
src/database/service/InputObjectService.kt
Normal file
@ -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]
|
||||
)
|
||||
}
|
10
src/di/InjectionModule.kt
Normal file
10
src/di/InjectionModule.kt
Normal file
@ -0,0 +1,10 @@
|
||||
package com.kmalbz.di
|
||||
|
||||
import com.kmalbz.database.service.IInputObjectService
|
||||
import com.kmalbz.database.service.InputObjectService
|
||||
import org.koin.dsl.module
|
||||
import org.koin.experimental.builder.singleBy
|
||||
|
||||
val injectionModule = module(createdAtStart = true) {
|
||||
singleBy<IInputObjectService,InputObjectService>()
|
||||
}
|
Reference in New Issue
Block a user