mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-23 23:35:14 +00:00
28 lines
382 B
Go
28 lines
382 B
Go
package cmdutil
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"os/signal"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func WaitForSignal(ctx context.Context, signals ...os.Signal) 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)
|
|
return sig
|
|
|
|
case <-ctx.Done():
|
|
return nil
|
|
|
|
}
|
|
|
|
return nil
|
|
}
|