mirror of
https://github.com/go-kratos/kratos.git
synced 2025-01-24 03:46:37 +02:00
sync/pipeline: using timer replace ticker (#477)
This commit is contained in:
parent
04948fd5cc
commit
f2c6751ed4
@ -169,17 +169,18 @@ func (p *Pipeline) Close() (err error) {
|
||||
func (p *Pipeline) mergeproc(mirror bool, index int, ch <-chan *message) {
|
||||
defer p.wait.Done()
|
||||
var (
|
||||
m *message
|
||||
vals = make(map[string][]interface{}, p.config.MaxSize)
|
||||
closed bool
|
||||
count int
|
||||
inteval = p.config.Interval
|
||||
oldTicker = true
|
||||
m *message
|
||||
vals = make(map[string][]interface{}, p.config.MaxSize)
|
||||
closed bool
|
||||
count int
|
||||
inteval = p.config.Interval
|
||||
timeout = false
|
||||
)
|
||||
if index > 0 {
|
||||
inteval = xtime.Duration(int64(index) * (int64(p.config.Interval) / int64(p.config.Worker)))
|
||||
}
|
||||
ticker := time.NewTicker(time.Duration(inteval))
|
||||
timer := time.NewTimer(time.Duration(inteval))
|
||||
defer timer.Stop()
|
||||
for {
|
||||
select {
|
||||
case m = <-ch:
|
||||
@ -193,12 +194,8 @@ func (p *Pipeline) mergeproc(mirror bool, index int, ch <-chan *message) {
|
||||
break
|
||||
}
|
||||
continue
|
||||
case <-ticker.C:
|
||||
if oldTicker {
|
||||
ticker.Stop()
|
||||
ticker = time.NewTicker(time.Duration(p.config.Interval))
|
||||
oldTicker = false
|
||||
}
|
||||
case <-timer.C:
|
||||
timeout = true
|
||||
}
|
||||
name := p.name
|
||||
process := count
|
||||
@ -215,8 +212,12 @@ func (p *Pipeline) mergeproc(mirror bool, index int, ch <-chan *message) {
|
||||
_metricChanLen.Set(float64(len(ch)), name, strconv.Itoa(index))
|
||||
_metricCount.Add(float64(process), name, strconv.Itoa(index))
|
||||
if closed {
|
||||
ticker.Stop()
|
||||
return
|
||||
}
|
||||
if !timer.Stop() && !timeout {
|
||||
<-timer.C
|
||||
timeout = false
|
||||
}
|
||||
timer.Reset(time.Duration(p.config.Interval))
|
||||
}
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user