fix repeated shutdown issue on linux

motivation: repeatedly calling shutdown in linux was broken due to multiple calls to singal() which work on darwin but not linux

changes:
* conditionalize the call to signal
* add test

Author: tomerd

Sources/Lifecycle/Lifecycle.swift

@@ -343,13 +343,15 @@ extension ServiceLifecycle {
     /// - 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(), cancelAfterTrap: Bool = false) -> DispatchSourceSignal {
         let signalSource = DispatchSource.makeSignalSource(signal: sig.rawValue, queue: queue)
-        signal(sig.rawValue, SIG_IGN)
-        signalSource.setEventHandler(handler: {
+        signalSource.setEventHandler {
             if cancelAfterTrap {
                 signalSource.cancel()
             }
             handler(sig)
-        })
+        }
+        #if canImport(Darwin)
+        signal(sig.rawValue, SIG_IGN)
+        #endif
         signalSource.resume()
         return signalSource
     }

Tests/LifecycleTests/ServiceLifecycleTests+XCTest.swift

@@ -34,6 +34,7 @@ extension ServiceLifecycleTests {
             ("testNesting2", testNesting2),
             ("testSignalDescription", testSignalDescription),
             ("testBacktracesInstalledOnce", testBacktracesInstalledOnce),
+            ("testRepeatShutdown", testRepeatShutdown),
         ]
     }
 }

Tests/LifecycleTests/ServiceLifecycleTests.swift

@@ -233,4 +233,42 @@ final class ServiceLifecycleTests: XCTestCase {
         _ = ServiceLifecycle(configuration: config)
         _ = ServiceLifecycle(configuration: config)
     }
+
+    func testRepeatShutdown() {
+        if ProcessInfo.processInfo.environment["SKIP_SIGNAL_TEST"].flatMap(Bool.init) ?? false {
+            print("skipping testRepeatShutdown")
+            return
+        }
+
+        var count = 0
+
+        struct Service {
+            static let signal = ServiceLifecycle.Signal.ALRM
+
+            let lifecycle: ServiceLifecycle
+
+            init() {
+                self.lifecycle = ServiceLifecycle(configuration: .init(shutdownSignal: [Service.signal]))
+                self.lifecycle.register(GoodItem())
+            }
+        }
+
+        func gracefulShutdown() {
+            let service = Service()
+            service.lifecycle.start { error in
+                XCTAssertNil(error, "not expecting error")
+                kill(getpid(), Service.signal.rawValue)
+            }
+
+            service.lifecycle.wait()
+            count = count + 1 // not thread safe but fine for this purpose
+        }
+
+        let attempts = Int.random(in: 2 ..< 5)
+        for _ in 0 ..< attempts {
+            gracefulShutdown()
+        }
+
+        XCTAssertEqual(attempts, count)
+    }
 }

docker/Dockerfile

@@ -20,7 +20,7 @@ RUN apt-get update && apt-get install -y lsof dnsutils netcat-openbsd net-tools
 RUN apt-get update && apt-get install -y ruby ruby-dev libsqlite3-dev build-essential
 # switch of gem docs building
 RUN echo "gem: --no-document" > ~/.gemrc
-RUN if [ "${ubuntu_version}" != "xenial" ] ; then gem install jazzy ; fi
+RUN if [ "${ubuntu_version}" != "xenial" ] ; then gem install jazzy -v 0.13.7 ; fi
 
 # tools
 RUN mkdir -p $HOME/.tools