2678a8d56588ecf73223b6a2a5d69cfd1db6e174
[dcaegen2/collectors/hv-ves.git] / sources / hv-collector-utils / src / main / kotlin / org / onap / dcae / collectors / veshv / utils / shutdown_hook.kt
1 /*
2  * ============LICENSE_START=======================================================
3  * dcaegen2-collectors-veshv
4  * ================================================================================
5  * Copyright (C) 2019 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
24 /**
25  * @author Piotr Jaszczyk <piotr.jaszczyk@nokia.com>
26  * @since January 2019
27  */
28
29 fun registerShutdownHook(job: () -> Unit) {
30     Runtime.getRuntime().addShutdownHook(object : Thread() {
31         override fun run() {
32             job()
33         }
34     })
35 }
36
37 fun registerShutdownHook(job: IO<Unit>) = IO {
38     registerShutdownHook {
39         job.unsafeRunSync()
40     }
41 }