birbnetes/output-service-rdb-consumer
birbnetes/output-service-rdb-consumer
4
0
Fork 0
Code Pull Requests Releases Activity

manual ack
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Torma Kristóf 2021-08-17 17:38:09 +02:00
parent a719addee4
commit 79b68af64c
Signed by: tormakris
GPG Key ID: DC83C4F2C41B1047
2 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/main/kotlin/mq/ConsumerWrapper.kt
View File

@ -20,6 +20,6 @@ class ConsumerWrapper {
val inputQueueName = inputChannel.queueDeclare().queue
inputChannel.queueBind(inputQueueName, envConfig.mqExchange, "")
inputChannel.basicConsume(inputQueueName, true, DatabaseConsumer())
inputChannel.basicConsume(inputQueueName, false, DatabaseConsumer(inputChannel))
}
}

14
src/main/kotlin/mq/DatabaseConsumer.kt
View File

@ -2,17 +2,15 @@ package mq
import api.ApiObject
import com.google.gson.Gson
import com.rabbitmq.client.AMQP
import com.rabbitmq.client.Consumer
import com.rabbitmq.client.Envelope
import com.rabbitmq.client.ShutdownSignalException
import com.rabbitmq.client.*
import database.service.IResultObjectService
import org.koin.core.component.KoinComponent
import org.koin.core.component.inject
class DatabaseConsumer: Consumer, KoinComponent {
class DatabaseConsumer(channel: Channel): Consumer, KoinComponent {
private val resultObjectService : IResultObjectService by inject()
private val gson = Gson()
private val basicChannel = channel
override fun handleConsumeOk(consumerTag : String?) {
}
override fun handleCancelOk(p0 : String?) {
@ -30,8 +28,14 @@ class DatabaseConsumer: Consumer, KoinComponent {
}
override fun handleDelivery(consumerTag : String?, envelope : Envelope?, basicProperties : AMQP.BasicProperties?, body : ByteArray?) {
try {
val rawJson = body!!.toString(Charsets.UTF_8)
val apiObject = gson.fromJson(rawJson, ApiObject::class.java)
resultObjectService.addOne(apiObject)
basicChannel.basicAck(envelope!!.deliveryTag, false)
} catch (e: Exception) {
println(e.message)
basicChannel.basicNack(envelope!!.deliveryTag, false, true)
}
}
}