Skip to content

Commit 1c69fe7

Browse files
committed
simplify signal handler
1 parent 7519b73 commit 1c69fe7

File tree

1 file changed

+11
-15
lines changed

1 file changed

+11
-15
lines changed

cmd/crowdsec/serve.go

Lines changed: 11 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ import (
2323
)
2424

2525
//nolint:deadcode,unused // debugHandler is kept as a dev convenience: it shuts down and serialize internal state
26-
func debugHandler(sig os.Signal, cConfig *csconfig.Config) error {
26+
func debugHandler(cConfig *csconfig.Config) error {
2727
var (
2828
tmpFile string
2929
err error
@@ -48,7 +48,7 @@ func debugHandler(sig os.Signal, cConfig *csconfig.Config) error {
4848
return nil
4949
}
5050

51-
func reloadHandler(sig os.Signal) (*csconfig.Config, error) {
51+
func reloadHandler() (*csconfig.Config, error) {
5252
var tmpFile string
5353

5454
// re-initialize tombs
@@ -206,7 +206,7 @@ func shutdownCrowdsec() error {
206206
return nil
207207
}
208208

209-
func shutdown(sig os.Signal, cConfig *csconfig.Config) error {
209+
func shutdown(cConfig *csconfig.Config) error {
210210
if !cConfig.DisableAgent {
211211
if err := shutdownCrowdsec(); err != nil {
212212
return fmt.Errorf("failed to shut down crowdsec: %w", err)
@@ -262,24 +262,21 @@ func HandleSignals(cConfig *csconfig.Config) error {
262262

263263
go func() {
264264
defer trace.CatchPanic("crowdsec/HandleSignals")
265-
Loop:
266-
for {
267-
s := <-signalChan
265+
266+
for s := range signalChan {
268267
switch s {
269268
// kill -SIGHUP XXXX
270269
case syscall.SIGHUP:
271270
log.Warning("SIGHUP received, reloading")
272271

273-
if err = shutdown(s, cConfig); err != nil {
272+
if err = shutdown(cConfig); err != nil {
274273
exitChan <- fmt.Errorf("failed shutdown: %w", err)
275-
276-
break Loop
274+
return
277275
}
278276

279-
if newConfig, err = reloadHandler(s); err != nil {
277+
if newConfig, err = reloadHandler(); err != nil {
280278
exitChan <- fmt.Errorf("reload handler failure: %w", err)
281-
282-
break Loop
279+
return
283280
}
284281

285282
if newConfig != nil {
@@ -288,10 +285,9 @@ func HandleSignals(cConfig *csconfig.Config) error {
288285
// ctrl+C, kill -SIGINT XXXX, kill -SIGTERM XXXX
289286
case os.Interrupt, syscall.SIGTERM:
290287
log.Warning("SIGTERM received, shutting down")
291-
if err = shutdown(s, cConfig); err != nil {
288+
if err = shutdown(cConfig); err != nil {
292289
exitChan <- fmt.Errorf("failed shutdown: %w", err)
293-
294-
break Loop
290+
return
295291
}
296292
exitChan <- nil
297293
}

0 commit comments

Comments
 (0)