feat: add Subscribe for easy using

jizhuozhi · Aug 1, 2024 · 4b35def · 4b35def
1 parent 6529b0f
commit 4b35def
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/future.go b/future.go
@@ -144,6 +144,10 @@ func (f *Future[T]) GetOrDefault(defaultVal T) T {
 	return val
 }
 
+func (f *Future[T]) Subscribe(cb func(val T, err error)) {
+	f.state.subscribe(cb)
+}
+
 // noCopy may be embedded into structs which must not be copied
 // after the first use.
 //

diff --git a/future_test.go b/future_test.go
@@ -58,6 +58,22 @@ func TestPromiseSetTwice(t *testing.T) {
 	})
 }
 
+func TestFutureSubscribe(t *testing.T) {
+	p := NewPromise[int]()
+	f := p.Future()
+	val1 := 0
+	val2 := 0
+	f.Subscribe(func(val int, err error) {
+		val1 = val + 1
+	})
+	f.Subscribe(func(val int, err error) {
+		val2 = val + 2
+	})
+	p.Set(1, nil)
+	assert.Equal(t, val1, 2)
+	assert.Equal(t, val2, 3)
+}
+
 func Benchmark(b *testing.B) {
 	b.Run("Promise", func(b *testing.B) {
 		for i := 0; i < b.N; i++ {