2 * ============LICENSE_START=======================================================
3 * dcaegen2-collectors-veshv
4 * ================================================================================
5 * Copyright (C) 2018 NOKIA
6 * ================================================================================
7 * Licensed under the Apache License, Version 2.0 (the "License");
8 * you may not use this file except in compliance with the License.
9 * You may obtain a copy of the License at
11 * http://www.apache.org/licenses/LICENSE-2.0
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS,
15 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
18 * ============LICENSE_END=========================================================
20 package org.onap.dcae.collectors.veshv.model
22 import arrow.core.Option
23 import org.onap.dcae.collectors.veshv.utils.logging.Logger
24 import org.onap.ves.VesEventOuterClass.CommonEventHeader
26 data class Routing(val routes: List<Route>) {
28 fun routeFor(commonHeader: CommonEventHeader): Option<Route> =
29 Option.fromNullable(routes.find { it.applies(commonHeader) })
32 data class Route(val domain: String, val targetTopic: String, val partitioning: (CommonEventHeader) -> Int) {
34 fun applies(commonHeader: CommonEventHeader) = commonHeader.domain == domain
36 operator fun invoke(message: VesMessage): RoutedMessage =
37 RoutedMessage(targetTopic, partitioning(message.header), message)
45 fun routing(init: RoutingBuilder.() -> Unit): RoutingBuilder {
46 val conf = RoutingBuilder()
51 class RoutingBuilder {
52 private val routes: MutableList<RouteBuilder> = mutableListOf()
54 fun defineRoute(init: RouteBuilder.() -> Unit): RouteBuilder {
55 val rule = RouteBuilder()
61 fun build() = Routing(routes.map { it.build() }.toList())
66 private lateinit var domain: String
67 private lateinit var targetTopic: String
68 private lateinit var partitioning: (CommonEventHeader) -> Int
70 fun fromDomain(domain: String) {
74 fun toTopic(targetTopic: String) {
75 this.targetTopic = targetTopic
78 fun withFixedPartitioning(num: Int = 0) {
79 partitioning = { num }
82 fun build() = Route(domain, targetTopic, partitioning)