You can not 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
2.4 KiB
76 lines
2.4 KiB
package stages
|
|
|
|
import (
|
|
"gitea.anxinyun.cn/container/common_models"
|
|
"log"
|
|
"time"
|
|
)
|
|
|
|
// 阶段处理
|
|
type Stage struct {
|
|
Name string
|
|
In <-chan *common_models.ProcessData
|
|
processFuncs []func(*common_models.ProcessData) *common_models.ProcessData
|
|
Out chan *common_models.ProcessData
|
|
execOver chan bool //阶段执行完毕,用以排查超时
|
|
}
|
|
|
|
func NewStage(name string) *Stage {
|
|
return &Stage{
|
|
Name: name,
|
|
processFuncs: make([]func(*common_models.ProcessData) *common_models.ProcessData, 0),
|
|
In: make(<-chan *common_models.ProcessData, 1),
|
|
Out: make(chan *common_models.ProcessData, 1),
|
|
execOver: make(chan bool, 1),
|
|
}
|
|
}
|
|
|
|
func (s *Stage) StageRun() <-chan *common_models.ProcessData {
|
|
go func() {
|
|
defer func() {
|
|
close(s.Out)
|
|
log.Printf("[%s]关闭out", s.Name)
|
|
}()
|
|
for n := range s.In {
|
|
//log.Printf("[%s]处理数据 %v", s.Name, n.DeviceData.Name)
|
|
s.Out <- s.process(n)
|
|
}
|
|
log.Printf("%s over", s.Name)
|
|
}()
|
|
|
|
return s.Out
|
|
}
|
|
|
|
// AddProcess 添加处理者。处理函数定义 func(*ProcessData) *ProcessData
|
|
func (s *Stage) AddProcess(fun func(*common_models.ProcessData) *common_models.ProcessData) {
|
|
s.processFuncs = append(s.processFuncs, fun)
|
|
}
|
|
|
|
func (s *Stage) process(data *common_models.ProcessData) *common_models.ProcessData {
|
|
go s.handlerTimeOutCheck(s.Name, data.DeviceData.DeviceId)
|
|
for _, processFunc := range s.processFuncs {
|
|
//tag := fmt.Sprintf("%d/%d", i+1, len(s.processFuncs))
|
|
//log.Printf("stage[%s]流程处理 start=> %s", s.Name, data.DeviceData.Name)
|
|
func() {
|
|
defer timeCost(s.Name, data.DeviceData.DeviceId, time.Now())
|
|
data = processFunc(data)
|
|
}()
|
|
//log.Printf("stage[%s]流程处理 over=> %s", s.Name, data.DeviceData.Name)
|
|
}
|
|
s.execOver <- true
|
|
return data
|
|
}
|
|
func (s *Stage) handlerTimeOutCheck(stageName, deviceId string) {
|
|
defaultTimeout := 10 * time.Second
|
|
select {
|
|
case <-s.execOver:
|
|
case <-time.After(defaultTimeout):
|
|
log.Printf("=====================")
|
|
//TODO #TEST BEGIN 测试时可以注释掉下面这行，否则调试时超时,会引发一个 panic，导致程序中断。
|
|
log.Panicf("stage[%s] ->[%s] 流程处理,超时[%v]，请排查", stageName, deviceId, defaultTimeout)
|
|
}
|
|
}
|
|
func timeCost(nodeId, deviceId string, start time.Time) {
|
|
tc := time.Since(start)
|
|
log.Printf("stage[%s] ->[%s] 耗时 = %v", nodeId, deviceId, tc)
|
|
}
|
|
|