go_process_manager/internal/app/logic/wait_cond.go

package logic

import (
	"sync"
	"sync/atomic"
	"time"

	"github.com/lzh-1625/go_process_manager/config"
)

type WaitCond struct {
	Cond        *sync.Cond
	Version     *atomic.Int64
	TriggerChan chan struct{}
}

var (
	ProcessWaitCond *WaitCond
	TaskWaitCond    *WaitCond
)

func InitWaitCond() {
	ProcessWaitCond = newWaitCond()
	TaskWaitCond = newWaitCond()
}

func newWaitCond() *WaitCond {
	wc := &WaitCond{
		Cond:        sync.NewCond(&sync.Mutex{}),
		Version:     &atomic.Int64{},
		TriggerChan: make(chan struct{}),
	}
	go wc.timing()
	return wc
}

func (p *WaitCond) Trigger() {
	p.Version.Add(1)
	p.TriggerChan <- struct{}{}
}

func (p *WaitCond) timing() { // 添加定时信号清理阻塞协程
	ticker := time.NewTicker(time.Second * time.Duration(config.CF.CondWaitTime))
	defer ticker.Stop()
	for {
		select {
		case <-ticker.C:
		case <-p.TriggerChan:
		}
		ticker.Reset(time.Second * time.Duration(config.CF.CondWaitTime))
		p.Cond.Broadcast()
	}
}