mirror of
https://github.com/jpillora/chisel
synced 2026-06-08 15:07:02 +00:00
Jaime Pillora
29 lines
530 B
Go
29 lines
530 B
Go
package cos
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
)
|
|
|
|
//InterruptContext returns a context which is
|
|
//cancelled on OS Interrupt
|
|
func InterruptContext() context.Context {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
go func() {
|
|
sig := make(chan os.Signal, 1)
|
|
signal.Notify(sig, os.Interrupt) //windows compatible?
|
|
<-sig
|
|
signal.Stop(sig)
|
|
cancel()
|
|
}()
|
|
return ctx
|
|
}
|
|
|
|
//SleepSignal sleeps for the given duration,
|
|
//or until a SIGHUP is received
|
|
func SleepSignal(d time.Duration) {
|
|
<-AfterSignal(d)
|
|
}
|