mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
zeekay
5a601d081b
Full module path rebrand from Temporal to Hanzo Tasks: - go.mod module: github.com/hanzoai/tasks - All internal Go imports updated (1906 files) - cmd/server renamed to cmd/tasksd - Binary name: tasksd (was temporal-server) - Docker images: hanzoai/tasks (was temporalio/server) - README rewritten with Hanzo Tasks branding - LLM.md created with integration notes - Makefile, goreleaser, CI workflows updated - go.temporal.io/sdk and go.temporal.io/api imports preserved (separate repos) Fork of Temporal.io (MIT license).
113 lines
2.4 KiB
Go
113 lines
2.4 KiB
Go
package queues
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/hanzoai/tasks/common/collection"
|
|
"github.com/hanzoai/tasks/service/history/tasks"
|
|
)
|
|
|
|
type (
|
|
Iterator interface {
|
|
collection.Iterator[tasks.Task]
|
|
|
|
Range() Range
|
|
CanSplit(tasks.Key) bool
|
|
Split(key tasks.Key) (left Iterator, right Iterator)
|
|
CanMerge(Iterator) bool
|
|
Merge(Iterator) Iterator
|
|
Remaining() Iterator
|
|
}
|
|
|
|
PaginationFnProvider func(Range) collection.PaginationFn[tasks.Task]
|
|
|
|
IteratorImpl struct {
|
|
paginationFnProvider PaginationFnProvider
|
|
remainingRange Range
|
|
|
|
pagingIterator collection.Iterator[tasks.Task]
|
|
}
|
|
)
|
|
|
|
func NewIterator(
|
|
paginationFnProvider PaginationFnProvider,
|
|
r Range,
|
|
) *IteratorImpl {
|
|
return &IteratorImpl{
|
|
paginationFnProvider: paginationFnProvider,
|
|
remainingRange: r,
|
|
|
|
// lazy initialized to prevent task pre-fetching on creating the iterator
|
|
pagingIterator: nil,
|
|
}
|
|
}
|
|
|
|
func (i *IteratorImpl) HasNext() bool {
|
|
if i.pagingIterator == nil {
|
|
i.pagingIterator = collection.NewPagingIterator(i.paginationFnProvider(i.remainingRange))
|
|
}
|
|
|
|
return i.pagingIterator.HasNext()
|
|
}
|
|
|
|
func (i *IteratorImpl) Next() (tasks.Task, error) {
|
|
if !i.HasNext() {
|
|
panic("Iterator encountered Next call when there is no next item")
|
|
}
|
|
|
|
task, err := i.pagingIterator.Next()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
i.remainingRange.InclusiveMin = task.GetKey().Next()
|
|
return task, nil
|
|
}
|
|
|
|
func (i *IteratorImpl) Range() Range {
|
|
return i.remainingRange
|
|
}
|
|
|
|
func (i *IteratorImpl) CanSplit(key tasks.Key) bool {
|
|
return i.remainingRange.CanSplit(key)
|
|
}
|
|
|
|
func (i *IteratorImpl) Split(key tasks.Key) (left Iterator, right Iterator) {
|
|
if !i.CanSplit(key) {
|
|
panic(fmt.Sprintf("Unable to split iterator with range %v at %v", i.remainingRange, key))
|
|
}
|
|
|
|
leftRange, rightRange := i.remainingRange.Split(key)
|
|
left = NewIterator(
|
|
i.paginationFnProvider,
|
|
leftRange,
|
|
)
|
|
right = NewIterator(
|
|
i.paginationFnProvider,
|
|
rightRange,
|
|
)
|
|
return left, right
|
|
}
|
|
|
|
func (i *IteratorImpl) CanMerge(iter Iterator) bool {
|
|
return i.remainingRange.CanMerge(iter.Range())
|
|
}
|
|
|
|
func (i *IteratorImpl) Merge(iter Iterator) Iterator {
|
|
if !i.CanMerge(iter) {
|
|
panic(fmt.Sprintf("Unable to merge iterator range %v with incoming iterator range %v", i.remainingRange, iter.Range()))
|
|
}
|
|
|
|
return NewIterator(
|
|
i.paginationFnProvider,
|
|
i.remainingRange.Merge(iter.Range()),
|
|
)
|
|
}
|
|
|
|
func (i *IteratorImpl) Remaining() Iterator {
|
|
return NewIterator(
|
|
i.paginationFnProvider,
|
|
i.remainingRange,
|
|
)
|
|
}
|