fix: cond options

bytedance · Mar 19, 2024 · 6250b41 · 6250b41
1 parent 676a5d5
commit 6250b41
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/lang/channel/cond.go b/lang/channel/cond.go
@@ -25,7 +25,11 @@ type Cond interface {
 }
 
 func NewCond(opts ...CondOption) Cond {
-	return new(cond)
+	cd := new(cond)
+	for _, opt := range opts {
+		opt(cd)
+	}
+	return cd
 }
 
 type condSignal = chan struct{}

diff --git a/lang/channel/cond_test.go b/lang/channel/cond_test.go
@@ -43,6 +43,18 @@ func TestCond(t *testing.T) {
 	}
 }
 
+func TestCondTimeout(t *testing.T) {
+	cd := NewCond(WithCondTimeout(time.Millisecond * 200))
+	go func() {
+		time.Sleep(time.Millisecond * 500)
+		cd.Broadcast()
+	}()
+	begin := time.Now()
+	cd.Wait(context.Background())
+	cost := time.Since(begin)
+	t.Logf("cost=%dms", cost.Milliseconds())
+}
+
 func BenchmarkChanCond(b *testing.B) {
 	b.ReportAllocs()
 	b.ResetTimer()