2022-04-27 17:45:53 +08:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"os"
|
|
|
|
"os/signal"
|
2022-07-12 17:37:48 +08:00
|
|
|
"syscall"
|
2022-04-27 17:45:53 +08:00
|
|
|
|
|
|
|
"gitea.com/gitea/act_runner/cmd"
|
|
|
|
)
|
|
|
|
|
2022-07-13 13:44:14 +08:00
|
|
|
func withContextFunc(ctx context.Context, f func()) context.Context {
|
2022-04-27 17:45:53 +08:00
|
|
|
ctx, cancel := context.WithCancel(ctx)
|
|
|
|
go func() {
|
2022-07-13 13:44:14 +08:00
|
|
|
c := make(chan os.Signal, 1)
|
|
|
|
signal.Notify(c, syscall.SIGINT, syscall.SIGTERM)
|
|
|
|
defer signal.Stop(c)
|
|
|
|
|
2022-04-27 17:45:53 +08:00
|
|
|
select {
|
2022-07-13 13:44:14 +08:00
|
|
|
case <-ctx.Done():
|
2022-04-27 17:45:53 +08:00
|
|
|
case <-c:
|
|
|
|
cancel()
|
2022-07-13 13:44:14 +08:00
|
|
|
f()
|
2022-04-27 17:45:53 +08:00
|
|
|
}
|
|
|
|
}()
|
|
|
|
|
2022-07-13 13:44:14 +08:00
|
|
|
return ctx
|
|
|
|
}
|
|
|
|
|
|
|
|
func main() {
|
|
|
|
ctx := withContextFunc(context.Background(), func() {})
|
2022-04-27 17:45:53 +08:00
|
|
|
// run the command
|
|
|
|
cmd.Execute(ctx)
|
|
|
|
}
|