cloud_util/
signal.rs

1// Copyright Rivtower Technologies LLC.
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use futures::stream::StreamExt;
16use signal_hook::consts::signal::*;
17use signal_hook_tokio::Signals;
18
19use crate::graceful_shutdown::ExitType;
20
21pub(crate) async fn handle_signals(tx_exit: flume::Sender<ExitType>) {
22    let mut signals = Signals::new([SIGTERM]).unwrap();
23    while let Some(signal) = signals.next().await {
24        match signal {
25            SIGTERM => {
26                let _ = tx_exit.send(ExitType::Signal(SIGTERM));
27                break;
28            }
29            _ => warn!("Received signal: {signal}"),
30        }
31    }
32}