d5f7c7c85c818b5a4ba0f3075fdb067c863e9048
[dcaegen2/collectors/hv-ves.git] /
1 /*
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
10  *
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  *
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=========================================================
19  */
20 package org.onap.dcae.collectors.veshv.simulators.xnf.impl
21
22 import com.google.protobuf.ByteString
23 import io.netty.buffer.ByteBuf
24 import io.netty.buffer.Unpooled
25 import org.onap.dcae.collectors.veshv.domain.WireFrame
26 import org.onap.ves.VesEventV5
27 import reactor.core.publisher.Flux
28 import reactor.core.publisher.Mono
29
30 /**
31  * @author Jakub Dudycz <jakub.dudycz@nokia.com>
32  * @since June 2018
33  */
34 class MessageFactory {
35
36     companion object {
37         const val DEFAULT_START_EPOCH: Long = 120034455
38         const val DEFAULT_LAST_EPOCH: Long = 120034455
39     }
40
41     fun createMessageFlux(amount: Long = -1): Flux<WireFrame> =
42             Mono.fromCallable(this::createMessage).let {
43                 if (amount < 0)
44                     it.repeat()
45                 else
46                     it.repeat(amount)
47             }
48
49
50     private fun createMessage(): WireFrame {
51         val commonHeader = VesEventV5.VesEvent.CommonEventHeader.newBuilder()
52                 .setVersion("1.9")
53                 .setEventName("Sample event name")
54                 .setDomain(VesEventV5.VesEvent.CommonEventHeader.Domain.HVRANMEAS)
55                 .setEventId("Sample event Id")
56                 .setSourceName("Sample Source")
57                 .setReportingEntityName(ByteString.copyFromUtf8("Sample byte String"))
58                 .setPriority(VesEventV5.VesEvent.CommonEventHeader.Priority.MEDIUM)
59                 .setStartEpochMicrosec(DEFAULT_START_EPOCH)
60                 .setLastEpochMicrosec(DEFAULT_LAST_EPOCH)
61                 .setSequence(2)
62                 .build()
63
64         val payload = vesMessageBytes(commonHeader)
65         return WireFrame(payload)
66     }
67
68     private fun vesMessageBytes(commonHeader: VesEventV5.VesEvent.CommonEventHeader): ByteBuf {
69         val msg = VesEventV5.VesEvent.newBuilder()
70                 .setCommonEventHeader(commonHeader)
71                 .setHvRanMeasFields(ByteString.copyFromUtf8("high volume data"))
72                 .build()
73
74         return Unpooled.wrappedBuffer(msg.toByteArray())
75     }
76 }