act_runner/main.go

38 lines
622 B
Go
Raw Normal View History

// Copyright 2022 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT
2022-04-27 17:45:53 +08:00
package main
import (
"context"
"os"
"os/signal"
"syscall"
2022-04-27 17:45:53 +08:00
"gitea.com/gitea/act_runner/cmd"
)
func withContextFunc(ctx context.Context, f func()) context.Context {
2022-04-27 17:45:53 +08: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 17:45:53 +08:00
select {
case <-ctx.Done():
2022-04-27 17:45:53 +08:00
case <-c:
cancel()
f()
2022-04-27 17:45:53 +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)
}