Skip to content

Commit

Permalink
feat: adding FromSlicePtrOr (#506)
Browse files Browse the repository at this point in the history
  • Loading branch information
samber authored Jul 26, 2024
1 parent aa609e4 commit cb150f5
Show file tree
Hide file tree
Showing 3 changed files with 33 additions and 0 deletions.
13 changes: 13 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -255,6 +255,7 @@ Type manipulation helpers:
- [FromPtrOr](#fromptror)
- [ToSlicePtr](#tosliceptr)
- [FromSlicePtr](#fromsliceptr)
- [FromSlicePtrOr](#fromsliceptror)
- [ToAnySlice](#toanyslice)
- [FromAnySlice](#fromanyslice)
- [Empty](#empty)
Expand Down Expand Up @@ -2700,6 +2701,18 @@ ptr := lo.Compact(
// []string{"hello", "world"}
```

### FromSlicePtrOr

Returns a slice with the pointer values or the fallback value.

```go
str1 := "hello"
str2 := "world"

ptr := lo.FromSlicePtrOr[string]([]*string{&str1, &str2, "fallback value"})
// []string{"hello", "world", "fallback value"}
```

### ToAnySlice

Returns a slice with all elements mapped to `any` type.
Expand Down
10 changes: 10 additions & 0 deletions type_manipulation.go
Original file line number Diff line number Diff line change
Expand Up @@ -69,6 +69,16 @@ func FromSlicePtr[T any](collection []*T) []T {
})
}

// FromSlicePtr returns a slice with the pointer values or the fallback value.
func FromSlicePtrOr[T any](collection []*T, fallback T) []T {
return Map(collection, func(x *T, _ int) T {
if x == nil {
return fallback
}
return *x
})
}

// ToAnySlice returns a slice with all elements mapped to `any` type
func ToAnySlice[T any](collection []T) []any {
result := make([]any, len(collection))
Expand Down
10 changes: 10 additions & 0 deletions type_manipulation_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -131,6 +131,16 @@ func TestFromSlicePtr(t *testing.T) {
is.Equal(result1, []string{str1, str2, ""})
}

func TestFromSlicePtrOr(t *testing.T) {
is := assert.New(t)

str1 := "foo"
str2 := "bar"
result1 := FromSlicePtrOr([]*string{&str1, &str2, nil}, "fallback")

is.Equal(result1, []string{str1, str2, "fallback"})
}

func TestToAnySlice(t *testing.T) {
t.Parallel()
is := assert.New(t)
Expand Down

0 comments on commit cb150f5

Please sign in to comment.