Skip to content

Add support for HUP, USR1, USR2 signals #86

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 11 commits into from
Mar 13, 2021
Merged

Add support for HUP, USR1, USR2 signals #86

Changes from 10 commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
a76f7fb
Updated target dependency example
hassila Mar 12, 2021
6ae9879
Merge pull request #1 from hassila/hassila-readme-patch-1
hassila Mar 12, 2021
64c7fe4
Added USR1/USR2 signals
hassila Mar 12, 2021
222475c
Optionally allow to let signal handler run multiple times, useful for…
hassila Mar 12, 2021
a33fdc8
Added HUP signal support.
hassila Mar 12, 2021
8d379dc
Trying to fix soundness failures, adding documentation.
hassila Mar 12, 2021
921c30c
Remove tab.
hassila Mar 12, 2021
4dec033
Merge branch 'main' into hassila-usr-signals
hassila Mar 12, 2021
7fc68b3
Whitespace fix.
hassila Mar 12, 2021
1843fb9
Fix final whitespace (I hope).
hassila Mar 12, 2021
265fb15
Remove superfluous paranthesis in if statement
hassila Mar 13, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 14 additions & 3 deletions Sources/Lifecycle/Lifecycle.swift
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -158,7 +158,7 @@ public struct ServiceLifecycle {
let signalSource = ServiceLifecycle.trap(signal: signal, handler: { signal in
self.log("intercepted signal: \(signal)")
self.shutdown()
})
}, cancelAfterTrap: true)
self.underlying.shutdownGroup.notify(queue: .global()) {
signalSource.cancel()
}
Expand All @@ -176,12 +176,16 @@ extension ServiceLifecycle {
/// - parameters:
/// - signal: The signal to trap.
/// - handler: closure to invoke when the signal is captured.
/// - on: DispatchQueue to run the signal handler on (default global dispatch queue)
/// - cancelAfterTrap: Defaults to false, which means the signal handler can be run multiple times. If true, the DispatchSignalSource will be cancelled after being trapped once.
/// - returns: a `DispatchSourceSignal` for the given trap. The source must be cancelled by the caller.
public static func trap(signal sig: Signal, handler: @escaping (Signal) -> Void, on queue: DispatchQueue = .global()) -> DispatchSourceSignal {
public static func trap(signal sig: Signal, handler: @escaping (Signal) -> Void, on queue: DispatchQueue = .global(), cancelAfterTrap: Bool = false) -> DispatchSourceSignal {
let signalSource = DispatchSource.makeSignalSource(signal: sig.rawValue, queue: queue)
signal(sig.rawValue, SIG_IGN)
signalSource.setEventHandler(handler: {
signalSource.cancel()
if (cancelAfterTrap) {
signalSource.cancel()
}
handler(sig)
})
signalSource.resume()
Expand All @@ -194,6 +198,10 @@ extension ServiceLifecycle {

public static let TERM = Signal(rawValue: SIGTERM)
public static let INT = Signal(rawValue: SIGINT)
public static let USR1 = Signal(rawValue: SIGUSR1)
public static let USR2 = Signal(rawValue: SIGUSR2)
public static let HUP = Signal(rawValue: SIGHUP)

// for testing
internal static let ALRM = Signal(rawValue: SIGALRM)

Expand All @@ -203,6 +211,9 @@ extension ServiceLifecycle {
case Signal.TERM: result += "TERM, "
case Signal.INT: result += "INT, "
case Signal.ALRM: result += "ALRM, "
case Signal.USR1: result += "USR1, "
case Signal.USR2: result += "USR2, "
case Signal.HUP: result += "HUP, "
default: () // ok to ignore
}
result += "rawValue: \(self.rawValue))"
Expand Down