feat: add function MustGet

AuroraTea · AuroraTea · commit 1559c7d68f38 · 2024-01-13T10:28:13.000+08:00
diff --git a/internal/test/nullable_test.go b/internal/test/nullable_test.go
@@ -26,6 +26,7 @@ func TestNullable(t *testing.T) {
 	value, err := myObj.Foo.Get()
 	require.NoError(t, err)
 	require.Equal(t, "bar", value)
+	require.Equal(t, "bar", myObj.Foo.MustGet())
 	// serialize back to json: leads to the same data
 	require.Equal(t, data, serialize(myObj, t))
 
@@ -50,6 +51,7 @@ func TestNullable(t *testing.T) {
 	require.True(t, myObj.Foo.IsSpecified())
 	_, err = myObj.Foo.Get()
 	require.ErrorContains(t, err, "value is null")
+	require.Panics(t, func() { myObj.Foo.MustGet() })
 	// serialize back to json: leads to the same data
 	require.Equal(t, data, serialize(myObj, t))
 
diff --git a/nullable.go b/nullable.go
@@ -51,6 +51,15 @@ func (t Nullable[T]) Get() (T, error) {
 	return t[true], nil
 }
 
+// MustGet retrieves the underlying value, if present, and panics if the value was not present
+func (t Nullable[T]) MustGet() T {
+	v, err := t.Get()
+	if err != nil {
+		panic(err)
+	}
+	return v
+}
+
 // Set sets the underlying value to a given value
 func (t *Nullable[T]) Set(value T) {
 	*t = map[bool]T{true: value}
diff --git a/nullable_example_test.go b/nullable_example_test.go
@@ -351,6 +351,7 @@ func ExampleNullable_unmarshalOptional() {
 		return
 	}
 	fmt.Printf("obj.Name.Get(): %#v <nil>\n", val)
+	fmt.Printf("obj.Name.MustGet(): %#v\n", obj.Name.MustGet())
 	fmt.Println("---")
 
 	// when it's set explicitly to a specific value
@@ -373,6 +374,7 @@ func ExampleNullable_unmarshalOptional() {
 		return
 	}
 	fmt.Printf("obj.Name.Get(): %#v <nil>\n", val)
+	fmt.Printf("obj.Name.MustGet(): %#v\n", obj.Name.MustGet())
 	fmt.Println("---")
 
 	// Output:
@@ -388,10 +390,12 @@ func ExampleNullable_unmarshalOptional() {
 	// obj.Name.IsSpecified(): true
 	// obj.Name.IsNull(): false
 	// obj.Name.Get(): "" <nil>
+	// obj.Name.MustGet(): ""
 	// ---
 	// Value:
 	// obj.Name.IsSpecified(): true
 	// obj.Name.IsNull(): false
 	// obj.Name.Get(): "foo" <nil>
+	// obj.Name.MustGet(): "foo"
 	// ---
 }
