bbgo_origin/pkg/cmd/cmdutil/signal.go

25 lines
351 B
Go
Raw Normal View History

2020-10-02 04:43:14 +00:00
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():
}
}