feat: add GetOrDefault for easy using

jizhuozhi · Aug 1, 2024 · 6529b0f · 6529b0f
1 parent 9ec417f
commit 6529b0f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/future.go b/future.go
@@ -136,6 +136,14 @@ func (f *Future[T]) Get() (T, error) {
 	return f.state.get()
 }
 
+func (f *Future[T]) GetOrDefault(defaultVal T) T {
+	val, err := f.state.get()
+	if err != nil {
+		return defaultVal
+	}
+	return val
+}
+
 // 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
@@ -19,6 +19,7 @@ func TestPromiseAndFuture(t *testing.T) {
 	val, err := f.Get()
 	assert.Equal(t, val, 1)
 	assert.Equal(t, err, errFoo)
+	assert.Equal(t, 2, f.GetOrDefault(2))
 }
 
 func TestPromiseAndFutureConcurrency(t *testing.T) {