forked from Dirbaio/gominer
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Setup a signal context and use it. (#207)
- Loading branch information
Showing
7 changed files
with
109 additions
and
58 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,58 @@ | ||
// Copyright (c) 2013-2016 The btcsuite developers | ||
// Copyright (c) 2015-2016 The Decred developers | ||
// Use of this source code is governed by an ISC | ||
// license that can be found in the LICENSE file. | ||
|
||
package main | ||
|
||
import ( | ||
"context" | ||
"os" | ||
"os/signal" | ||
) | ||
|
||
// shutdownRequestChannel is used to initiate shutdown from one of the | ||
// subsystems using the same code paths as when an interrupt signal is received. | ||
var shutdownRequestChannel = make(chan struct{}) | ||
|
||
// interruptSignals defines the default signals to catch in order to do a proper | ||
// shutdown. This may be modified during init depending on the platform. | ||
var interruptSignals = []os.Signal{os.Interrupt} | ||
|
||
// shutdownListener listens for OS Signals such as SIGINT (Ctrl+C) and shutdown | ||
// requests from shutdownRequestChannel. It returns a context that is canceled | ||
// when either signal is received. | ||
func shutdownListener() context.Context { | ||
ctx, cancel := context.WithCancel(context.Background()) | ||
go func() { | ||
interruptChannel := make(chan os.Signal, 1) | ||
signal.Notify(interruptChannel, interruptSignals...) | ||
|
||
// Listen for initial shutdown signal and cancel the returned context. | ||
select { | ||
case sig := <-interruptChannel: | ||
minrLog.Infof("Received signal (%s). Shutting down...", sig) | ||
|
||
case <-shutdownRequestChannel: | ||
minrLog.Infof("Shutdown requested. Shutting down...") | ||
} | ||
cancel() | ||
|
||
// Listen for repeated signals and display a message so the user | ||
// knows the shutdown is in progress and the process is not | ||
// hung. | ||
for { | ||
select { | ||
case sig := <-interruptChannel: | ||
minrLog.Infof("Received signal (%s). Already "+ | ||
"shutting down...", sig) | ||
|
||
case <-shutdownRequestChannel: | ||
minrLog.Info("Shutdown requested. Already " + | ||
"shutting down...") | ||
} | ||
} | ||
}() | ||
|
||
return ctx | ||
} |
Oops, something went wrong.