Shift to accept any parameter (#197)

elliotchance · Sep 7, 2023 · f1ace5f · f1ace5f
1 parent fc03ccc
commit f1ace5f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/v2/shift.go b/v2/shift.go
@@ -1,8 +1,8 @@
 package pie
 
-import "golang.org/x/exp/constraints"
-
 // Shift will return two values: the shifted value and the rest slice.
-func Shift[T constraints.Integer | constraints.Float](ss []T) (T, []T) {
-	return FirstOr(ss, 0), DropTop(ss, 1)
+// if the slice is empty then returned shifted value is the zero value of the slice elements and the rest slice is empty slice
+func Shift[T any](ss []T) (T, []T) {
+	var zeroValue T
+	return FirstOr(ss, zeroValue), DropTop(ss, 1)
 }