initial commit
This commit is contained in:
37
cursor.go
Normal file
37
cursor.go
Normal file
@ -0,0 +1,37 @@
|
||||
package nuapi
|
||||
|
||||
import (
|
||||
"context"
|
||||
)
|
||||
|
||||
type Cursor[T any] struct {
|
||||
ctx context.Context
|
||||
cancelFunc context.CancelFunc
|
||||
Chan <-chan *T
|
||||
}
|
||||
|
||||
func (c *Cursor[T]) First() *T {
|
||||
defer c.cancelFunc()
|
||||
return <-c.Chan
|
||||
}
|
||||
|
||||
func (c *Cursor[T]) Close() {
|
||||
c.cancelFunc()
|
||||
}
|
||||
|
||||
func (c *Cursor[T]) Next() (*T, bool) {
|
||||
if c.ctx.Err() != nil {
|
||||
return nil, false
|
||||
}
|
||||
|
||||
value, ok := <-c.Chan
|
||||
return value, ok
|
||||
}
|
||||
|
||||
func (c *Cursor[T]) Slice() []T {
|
||||
s := make([]T, 0)
|
||||
for value, ok := c.Next(); ok; value, ok = c.Next() {
|
||||
s = append(s, *value)
|
||||
}
|
||||
return s
|
||||
}
|
Reference in New Issue
Block a user