mirror of https://github.com/synctv-org/synctv
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
1.6 KiB
Go
76 lines
1.6 KiB
Go
package bootstrap
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"os/signal"
|
|
"sync"
|
|
"syscall"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/zijiren233/gencontainer/pqueue"
|
|
)
|
|
|
|
var (
|
|
c chan os.Signal
|
|
notifyTaskLock sync.Mutex
|
|
notifyTaskQueue = pqueue.NewMaxPriorityQueue[*SysNotifyTask]()
|
|
once sync.Once
|
|
WaitCbk func()
|
|
)
|
|
|
|
type SysNotifyTask struct {
|
|
Task func() error
|
|
Name string
|
|
}
|
|
|
|
func NewSysNotifyTask(name string, task func() error) *SysNotifyTask {
|
|
return &SysNotifyTask{
|
|
Name: name,
|
|
Task: task,
|
|
}
|
|
}
|
|
|
|
func InitSysNotify() {
|
|
c = make(chan os.Signal, 1)
|
|
signal.Notify(c, syscall.SIGHUP /*1*/, syscall.SIGINT /*2*/, syscall.SIGQUIT /*3*/, syscall.SIGTERM /*15*/)
|
|
WaitCbk = func() {
|
|
once.Do(waitCbk)
|
|
}
|
|
}
|
|
|
|
func waitCbk() {
|
|
log.Info("wait sys notify")
|
|
log.Infof("receive sys notify: %v", <-c)
|
|
notifyTaskLock.Lock()
|
|
defer notifyTaskLock.Unlock()
|
|
log.Infof("task: running...")
|
|
for notifyTaskQueue.Len() > 0 {
|
|
_, task := notifyTaskQueue.Pop()
|
|
func() {
|
|
defer func() {
|
|
if err := recover(); err != nil {
|
|
log.Errorf("task: %s panic has returned: %v", task.Name, err)
|
|
}
|
|
}()
|
|
log.Infof("task: %s running", task.Name)
|
|
if err := task.Task(); err != nil {
|
|
log.Errorf("task: %s an error occurred: %v", task.Name, err)
|
|
}
|
|
log.Infof("task: %s done", task.Name)
|
|
}()
|
|
}
|
|
log.Info("task: all done")
|
|
}
|
|
|
|
func RegisterSysNotifyTask(priority int, task *SysNotifyTask) error {
|
|
if task == nil || task.Task == nil {
|
|
return errors.New("task is nil")
|
|
}
|
|
notifyTaskLock.Lock()
|
|
defer notifyTaskLock.Unlock()
|
|
notifyTaskQueue.Push(priority, task)
|
|
return nil
|
|
}
|