assert len > 0

2020-10-15 14:25:10 +08:00
parent 901fadb5d3
commit 8291eabc2c
4 changed files with 23 additions and 0 deletions
--- a/core/fx/fn.go
+++ b/core/fx/fn.go
@@ -162,6 +162,7 @@ func (p Stream) Head(n int64) Stream {
 	if n < 1 {
 		panic("n must be greater than 0")
 	}
+
 	source := make(chan interface{})

 	go func() {
@@ -247,6 +248,10 @@ func (p Stream) Sort(less LessFunc) Stream {
 }

 func (p Stream) Tail(n int64) Stream {
+	if n < 1 {
+		panic("n should be greater than 0")
+	}
+
 	source := make(chan interface{})

 	go func() {