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:
@@ -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
|
||||||
|
@@ -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())
|
||||||
|
}
|
||||||
|
@@ -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
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user