mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-14 02:53:50 +00:00
25 lines
351 B
Go
25 lines
351 B
Go
|
package cmdutil
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"os"
|
||
|
|
"os/signal"
|
||
|
|
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
func WaitForSignal(ctx context.Context, signals ...os.Signal) {
|
||
|
|
var sigC = make(chan os.Signal, 1)
|
||
|
|
signal.Notify(sigC, signals...)
|
||
|
|
defer signal.Stop(sigC)
|
||
|
|
|
||
|
|
select {
|
||
|
|
case sig := <-sigC:
|
||
|
|
logrus.Warnf("%v", sig)
|
||
|
|
signal.Ignore()
|
||
|
|
|
||
|
|
case <-ctx.Done():
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|