fixup! #21 Add TakeLastWhile implements

meian · Feb 5, 2022 · b2a225f · b2a225f
1 parent e3b4317
commit b2a225f
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/take_last.go b/take_last.go
@@ -111,14 +111,15 @@ func (it *takeLastWhileIterator[T]) build() {
 	s := iteratorToSlice(it.it)
 	if !it.whileFunc(s[len(s)-1]) {
 		it.it = emptyIter[T]()
-	} else {
-		for i := len(s) - 2; i >= 0; i-- {
-			if !it.whileFunc(s[i]) {
-				s = s[i+1:]
-				break
-			}
+		it.built = true
+		return
+	}
+	for i := len(s) - 2; i >= 0; i-- {
+		if !it.whileFunc(s[i]) {
+			s = s[i+1:]
+			break
 		}
-		it.it = makeSliceIterator(s)
 	}
+	it.it = makeSliceIterator(s)
 	it.built = true
 }