git/lancet
1
0
Fork 0
mirror of https://github.com/duke-git/lancet.git synced 2026-02-13 09:12:28 +08:00
Code Issues Projects Releases Wiki Activity

feat: add Limit method of stream

Browse Source
This commit is contained in:
dudaodong
2023-02-06 17:08:36 +08:00
parent 8e3911833d
commit 48c7794b01
2 changed files with 36 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
stream/stream.go
View File

@@ -167,7 +167,7 @@ func (s stream[T]) Peek(consumer func(item T)) stream[T] {
return s
}
// Skip Returns a stream consisting of the remaining elements of this stream after discarding the first n elements of the stream. If this stream contains fewer than n elements then an empty stream will be returned.
// Skip returns a stream consisting of the remaining elements of this stream after discarding the first n elements of the stream. If this stream contains fewer than n elements then an empty stream will be returned.
func (s stream[T]) Skip(n int) stream[T] {
if n <= 0 {
return s
@@ -180,8 +180,6 @@ func (s stream[T]) Skip(n int) stream[T] {
return FromSlice(source)
}
// source = make([]T, l-n+1, l-n+1)
for i := n; i < l; i++ {
source = append(source, s.source[i])
}
@@ -189,6 +187,25 @@ func (s stream[T]) Skip(n int) stream[T] {
return FromSlice(source)
}
// Limit returns a stream consisting of the elements of this stream, truncated to be no longer than maxSize in length.
func (s stream[T]) Limit(maxSize int) stream[T] {
if s.source == nil {
return s
}
if maxSize < 0 {
return FromSlice([]T{})
}
source := make([]T, 0, maxSize)
for i := 0; i < len(s.source) && i < maxSize; i++ {
source = append(source, s.source[i])
}
return FromSlice(source)
}
// Count returns the count of elements in the stream.
func (s stream[T]) Count() int {
return len(s.source)