go/kratos
1
0
Fork 0
mirror of https://github.com/go-kratos/kratos.git synced 2025-01-10 00:29:01 +02:00
Code Issues Releases Activity

sync/pipeline: select context for timeout/cancel (#478)

Browse Source
This commit is contained in:
Astone 2020-01-17 13:12:33 +08:00 committed by Tony
parent 8691205479
commit 094ee714bd
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/sync/pipeline/pipeline.go
View File

@ -122,9 +122,14 @@ func (p *Pipeline) Start() {
}
// SyncAdd sync add a value to channal, channel shard in split method
func (p *Pipeline) SyncAdd(c context.Context, key string, value interface{}) {
func (p *Pipeline) SyncAdd(c context.Context, key string, value interface{}) (err error) {
ch, msg := p.add(c, key, value)
ch <- msg
select {
case ch <- msg:
case <-c.Done():
err = c.Err()
}
return
}
// Add async add a value to channal, channel shard in split method