76 lines
2.2 KiB
Kotlin
76 lines
2.2 KiB
Kotlin
package com.kmalbz
|
|
|
|
import com.kmalbz.api.route.OutputServiceRDBServer
|
|
import com.kmalbz.consumer.DatabaseConsumer
|
|
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.SampleObjects
|
|
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(SampleObjects)
|
|
}
|
|
|
|
|
|
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, "")
|
|
|
|
channel.basicConsume(queueName, true, DatabaseConsumer())
|
|
|
|
routing {
|
|
install(StatusPages) {
|
|
exception<HttpException> {
|
|
call.respond(HttpStatusCode.BadRequest)
|
|
}
|
|
|
|
exception<IllegalStateException> {
|
|
call.respond(HttpStatusCode.NotAcceptable)
|
|
}
|
|
}
|
|
|
|
OutputServiceRDBServer().apply {
|
|
registerOutput()
|
|
}
|
|
}
|
|
}
|