runner/main.go

35 lines
532 B
Go
Raw Normal View History

2022-04-27 12:45:53 +03:00
package main
import (
"context"
"os"
"os/signal"
"syscall"
2022-04-27 12:45:53 +03:00
"codeberg.org/forgejo/runner/cmd"
2022-04-27 12:45:53 +03:00
)
func withContextFunc(ctx context.Context, f func()) context.Context {
2022-04-27 12:45:53 +03:00
ctx, cancel := context.WithCancel(ctx)
go func() {
c := make(chan os.Signal, 1)
signal.Notify(c, syscall.SIGINT, syscall.SIGTERM)
defer signal.Stop(c)
2022-04-27 12:45:53 +03:00
select {
case <-ctx.Done():
2022-04-27 12:45:53 +03:00
case <-c:
cancel()
f()
2022-04-27 12:45:53 +03:00
}
}()
return ctx
}
func main() {
ctx := withContextFunc(context.Background(), func() {})
2022-04-27 12:45:53 +03:00
// run the command
cmd.Execute(ctx)
}