1
0
mirror of https://github.com/duke-git/lancet.git synced 2026-02-17 03:02:28 +08:00

refactor: refact Or func in channel.go

This commit is contained in:
dudaodong
2022-04-21 14:13:19 +08:00
parent d1c6c57700
commit c27ccad2b9
2 changed files with 16 additions and 5 deletions

View File

@@ -203,12 +203,17 @@ func (c *Channel) Or(channels ...<-chan any) <-chan any {
case <-channels[1]: case <-channels[1]:
} }
default: default:
m := len(channels) / 2
select { select {
case <-channels[0]: case <-c.Or(channels[:m]...):
case <-channels[1]: case <-c.Or(channels[m:]...):
case <-channels[2]:
case <-c.Or(append(channels[3:], orDone)...):
} }
// select {
// case <-channels[0]:
// case <-channels[1]:
// case <-channels[2]:
// case <-c.Or(append(channels[3:], orDone)...):
// }
} }
}() }()

View File

@@ -125,7 +125,13 @@ func TestOr(t *testing.T) {
start := time.Now() start := time.Now()
c := NewChannel() c := NewChannel()
<-c.Or(sig(2*time.Hour), sig(5*time.Minute), sig(1*time.Second), sig(1*time.Hour), sig(1*time.Minute)) <-c.Or(
sig(1*time.Second),
sig(2*time.Second),
sig(3*time.Second),
sig(4*time.Second),
sig(5*time.Second),
)
t.Logf("done after %v", time.Since(start)) t.Logf("done after %v", time.Since(start))