2b84e3f1e6dd17abaca1b858ab04ad5183522b8e
[dcaegen2/collectors/hv-ves.git] / hv-collector-domain / src / main / kotlin / org / onap / dcae / collectors / veshv / domain / ByteData.kt
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.domain
21
22 import com.google.protobuf.MessageLite
23 import io.netty.buffer.ByteBuf
24 import java.nio.charset.Charset
25
26 /**
27  * @author Piotr Jaszczyk <piotr.jaszczyk@nokia.com>
28  * @since June 2018
29  */
30 class ByteData(private val data: ByteArray) {
31
32     fun size() = data.size
33
34     /**
35      * This will expose mutable state of the data.
36      *
37      * @return wrapped data buffer (NOT a copy)
38      */
39     fun unsafeAsArray() = data
40
41     fun writeTo(byteBuf: ByteBuf) {
42         byteBuf.writeBytes(data)
43     }
44
45     fun asString(charset: Charset = Charset.defaultCharset()) = String(data, charset)
46
47     companion object {
48         val EMPTY = ByteData(byteArrayOf())
49
50         fun readFrom(byteBuf: ByteBuf, length: Int): ByteData {
51             val dataArray = ByteArray(length)
52             byteBuf.readBytes(dataArray)
53             return ByteData(dataArray)
54         }
55     }
56 }
57
58 fun MessageLite.toByteData(): ByteData = ByteData(toByteArray())