1
0
mirror of https://github.com/IBM/fp-go.git synced 2025-08-10 22:31:32 +02:00

fix: try a more efficient implementation of standard interfaces

Signed-off-by: Dr. Carsten Leue <carsten.leue@de.ibm.com>
This commit is contained in:
Dr. Carsten Leue
2024-02-07 13:26:17 +01:00
parent d86cf55a3d
commit a774d63e66
3 changed files with 71 additions and 28 deletions

View File

@@ -20,15 +20,17 @@ import (
) )
type ( type (
// Either defines a data structure that logically holds either an E or an A. The flag discriminates the cases either struct {
Either[E, A any] struct {
isLeft bool isLeft bool
value any value any
} }
// Either defines a data structure that logically holds either an E or an A. The flag discriminates the cases
Either[E, A any] either
) )
// String prints some debug info for the object // String prints some debug info for the object
func (s Either[E, A]) String() string { func eitherString(s *either) string {
if s.isLeft { if s.isLeft {
return fmt.Sprintf("Left[%T](%v)", s.value, s.value) return fmt.Sprintf("Left[%T](%v)", s.value, s.value)
} }
@@ -36,15 +38,25 @@ func (s Either[E, A]) String() string {
} }
// Format prints some debug info for the object // Format prints some debug info for the object
func (s Either[E, A]) Format(f fmt.State, c rune) { func eitherFormat(e *either, f fmt.State, c rune) {
switch c { switch c {
case 's': case 's':
fmt.Fprint(f, s.String()) fmt.Fprint(f, eitherString(e))
default: default:
fmt.Fprint(f, s.String()) fmt.Fprint(f, eitherString(e))
} }
} }
// String prints some debug info for the object
func (s Either[E, A]) String() string {
return eitherString((*either)(&s))
}
// Format prints some debug info for the object
func (s Either[E, A]) Format(f fmt.State, c rune) {
eitherFormat((*either)(&s), f, c)
}
// IsLeft tests if the [Either] is a left value. Rather use [Fold] if you need to access the values. Inverse is [IsRight]. // IsLeft tests if the [Either] is a left value. Rather use [Fold] if you need to access the values. Inverse is [IsRight].
func IsLeft[E, A any](val Either[E, A]) bool { func IsLeft[E, A any](val Either[E, A]) bool {
return val.isLeft return val.isLeft

View File

@@ -17,6 +17,7 @@ package either
import ( import (
"errors" "errors"
"fmt"
"testing" "testing"
F "github.com/IBM/fp-go/function" F "github.com/IBM/fp-go/function"
@@ -109,3 +110,13 @@ func TestFromOption(t *testing.T) {
assert.Equal(t, Left[int]("none"), FromOption[int](F.Constant("none"))(O.None[int]())) assert.Equal(t, Left[int]("none"), FromOption[int](F.Constant("none"))(O.None[int]()))
assert.Equal(t, Right[string](1), FromOption[int](F.Constant("none"))(O.Some(1))) assert.Equal(t, Right[string](1), FromOption[int](F.Constant("none"))(O.Some(1)))
} }
func TestStringer(t *testing.T) {
e := Of[error]("foo")
exp := "Right[string](foo)"
assert.Equal(t, exp, e.String())
var s fmt.Stringer = e
assert.Equal(t, exp, s.String())
}

View File

@@ -19,52 +19,72 @@ import (
"bytes" "bytes"
"encoding/json" "encoding/json"
"fmt" "fmt"
"reflect"
) )
var ( var (
// jsonNull is the cached representation of the `null` serialization in JSON
jsonNull = []byte("null") jsonNull = []byte("null")
) )
// Option defines a data structure that logically holds a value or not // Option defines a data structure that logically holds a value or not
type Option[A any] struct { type Option[A any] struct {
isSome bool isSome bool
some A value A
}
// optString prints some debug info for the object
func optString(isSome bool, value any) string {
if isSome {
return fmt.Sprintf("Some[%T](%v)", value, value)
}
return fmt.Sprintf("None[%T]", value)
}
// optFormat prints some debug info for the object
func optFormat(isSome bool, value any, f fmt.State, c rune) {
switch c {
case 's':
fmt.Fprint(f, optString(isSome, value))
default:
fmt.Fprint(f, optString(isSome, value))
}
} }
// String prints some debug info for the object // String prints some debug info for the object
func (s Option[A]) String() string { func (s Option[A]) String() string {
if s.isSome { return optString(s.isSome, s.value)
return fmt.Sprintf("Some[%T](%v)", s.some, s.some)
}
return fmt.Sprintf("None[%T]", s.some)
} }
// Format prints some debug info for the object // Format prints some debug info for the object
func (s Option[A]) Format(f fmt.State, c rune) { func (s Option[A]) Format(f fmt.State, c rune) {
switch c { optFormat(s.isSome, s.value, f, c)
case 's':
fmt.Fprint(f, s.String())
default:
fmt.Fprint(f, s.String())
}
} }
func (s Option[A]) MarshalJSON() ([]byte, error) { func optMarshalJSON(isSome bool, value any) ([]byte, error) {
if IsSome(s) { if isSome {
return json.Marshal(s.some) return json.Marshal(value)
} }
return jsonNull, nil return jsonNull, nil
} }
func (s *Option[A]) UnmarshalJSON(data []byte) error { func (s Option[A]) MarshalJSON() ([]byte, error) {
return optMarshalJSON(s.isSome, s.value)
}
func optUnmarshalJSON(isSome *bool, value any, data []byte) error {
// decode the value // decode the value
if bytes.Equal(data, jsonNull) { if bytes.Equal(data, jsonNull) {
s.isSome = false *isSome = false
s.some = *new(A) reflect.ValueOf(value).Elem().SetZero()
return nil return nil
} }
s.isSome = true *isSome = true
return json.Unmarshal(data, &s.some) return json.Unmarshal(data, value)
}
func (s *Option[A]) UnmarshalJSON(data []byte) error {
return optUnmarshalJSON(&s.isSome, &s.value, data)
} }
func IsNone[T any](val Option[T]) bool { func IsNone[T any](val Option[T]) bool {
@@ -72,7 +92,7 @@ func IsNone[T any](val Option[T]) bool {
} }
func Some[T any](value T) Option[T] { func Some[T any](value T) Option[T] {
return Option[T]{isSome: true, some: value} return Option[T]{isSome: true, value: value}
} }
func Of[T any](value T) Option[T] { func Of[T any](value T) Option[T] {
@@ -89,11 +109,11 @@ func IsSome[T any](val Option[T]) bool {
func MonadFold[A, B any](ma Option[A], onNone func() B, onSome func(A) B) B { func MonadFold[A, B any](ma Option[A], onNone func() B, onSome func(A) B) B {
if IsSome(ma) { if IsSome(ma) {
return onSome(ma.some) return onSome(ma.value)
} }
return onNone() return onNone()
} }
func Unwrap[A any](ma Option[A]) (A, bool) { func Unwrap[A any](ma Option[A]) (A, bool) {
return ma.some, ma.isSome return ma.value, ma.isSome
} }