blob: a0b2cf4623e5694e4df3a23fced6a23e6bd596b9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
/*
* Copyright © 2018-2019 AT&T Intellectual Property.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.onap.ccsdk.cds.blueprintsprocessor.functions.message.prioritization.nats
import io.nats.streaming.MessageHandler
import io.nats.streaming.Subscription
import kotlinx.coroutines.runBlocking
import org.onap.ccsdk.cds.blueprintsprocessor.functions.message.prioritization.MessagePrioritizationService
import org.onap.ccsdk.cds.blueprintsprocessor.functions.message.prioritization.db.MessagePrioritization
import org.onap.ccsdk.cds.blueprintsprocessor.nats.asJsonType
import org.onap.ccsdk.cds.blueprintsprocessor.nats.service.BluePrintNatsLibPropertyService
import org.onap.ccsdk.cds.blueprintsprocessor.nats.service.BluePrintNatsService
import org.onap.ccsdk.cds.blueprintsprocessor.nats.utils.NatsClusterUtils
import org.onap.ccsdk.cds.blueprintsprocessor.nats.utils.SubscriptionOptionsUtils
import org.onap.ccsdk.cds.controllerblueprints.core.BluePrintProcessorException
import org.onap.ccsdk.cds.controllerblueprints.core.asType
import org.onap.ccsdk.cds.controllerblueprints.core.logger
import org.onap.ccsdk.cds.controllerblueprints.core.utils.ClusterUtils
open class NatsMessagePrioritizationConsumer(
private val bluePrintNatsLibPropertyService: BluePrintNatsLibPropertyService,
private val natsMessagePrioritizationService: MessagePrioritizationService
) {
private val log = logger(NatsMessagePrioritizationConsumer::class)
lateinit var bluePrintNatsService: BluePrintNatsService
private lateinit var subscription: Subscription
suspend fun startConsuming() {
val prioritizationConfiguration = natsMessagePrioritizationService.getConfiguration()
val natsConfiguration = prioritizationConfiguration.natsConfiguration
?: throw BluePrintProcessorException("couldn't get NATS consumer configuration")
check((natsMessagePrioritizationService is AbstractNatsMessagePrioritizationService)) {
"messagePrioritizationService is not of type AbstractNatsMessagePrioritizationService."
}
bluePrintNatsService = consumerService(natsConfiguration.connectionSelector)
natsMessagePrioritizationService.bluePrintNatsService = bluePrintNatsService
val inputSubject = NatsClusterUtils.currentApplicationSubject(natsConfiguration.inputSubject)
val loadBalanceGroup = ClusterUtils.applicationName()
val messageHandler = createMessageHandler()
val subscriptionOptions = SubscriptionOptionsUtils.durable(NatsClusterUtils.currentNodeDurable(inputSubject))
subscription = bluePrintNatsService.loadBalanceSubscribe(
inputSubject,
loadBalanceGroup,
messageHandler,
subscriptionOptions
)
log.info(
"Nats prioritization consumer listening on subject($inputSubject) on loadBalance group($loadBalanceGroup)."
)
}
suspend fun shutDown() {
if (::subscription.isInitialized) {
subscription.unsubscribe()
}
log.info("Nats prioritization consumer listener shutdown complete")
}
private fun consumerService(selector: String): BluePrintNatsService {
return bluePrintNatsLibPropertyService.bluePrintNatsService(selector)
}
private fun createMessageHandler(): MessageHandler {
return MessageHandler { message ->
try {
val messagePrioritization = message.asJsonType().asType(MessagePrioritization::class.java)
runBlocking {
natsMessagePrioritizationService.prioritize(messagePrioritization)
}
} catch (e: Exception) {
log.error("failed to process prioritize message", e)
}
}
}
}
|