Close KafkaSender when handling SIGINT
[dcaegen2/collectors/hv-ves.git] / sources / hv-collector-utils / src / main / kotlin / org / onap / dcae / collectors / veshv / utils / server_handle.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.utils
21
22 import arrow.effects.IO
23 import reactor.netty.DisposableServer
24 import java.time.Duration
25
26 /**
27  * @author Piotr Jaszczyk <piotr.jaszczyk@nokia.com>
28  * @since August 2018
29  */
30 abstract class ServerHandle(val host: String, val port: Int) : Closeable {
31     abstract fun await(): IO<Unit>
32 }
33
34 /**
35  * @author Piotr Jaszczyk <piotr.jaszczyk@nokia.com>
36  * @since August 2018
37  */
38 class NettyServerHandle(private val ctx: DisposableServer) : ServerHandle(ctx.host(), ctx.port()) {
39     override fun close() = IO {
40         ctx.disposeNow(SHUTDOWN_TIMEOUT)
41     }
42
43     override fun await() = IO<Unit> {
44         ctx.channel().closeFuture().sync()
45     }
46
47     companion object {
48         private val SHUTDOWN_TIMEOUT = Duration.ofSeconds(10)
49     }
50 }