mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-30 14:56:19 +00:00
Alex ShtinandGitHub
8e8f1b0f7c
## What changed?
<!-- Describe what has changed in this PR -->
Remove license header from every file. Because it is really hard to
follow in this PR here is the summary:
1. License header is removed from all `.go` and `.proto` files
:fireworks::fireworks:🎆.
2. `LICENSE` file in the root directory has only Temporal and Uber
copyrights.
3. 5 other `LICENSE` files added to the packages which have copyrights
different from Temporal and Uber: Datadog, Xargin, "Mat Ryer, Tyler
Bunnell and contributors".
4. `license_file` flag is removed from all code generation tools.
5. `copyright_file` flag is removed from `go:generate mockgen`
directive.
6. All copyright related targets are removed from `Makefile`.
7. Updated Temporal copyright year to 2025 everywhere.
## Why?
<!-- Tell your future self why have you made these changes -->
I double checked with legal department that it is not needed to have
license header in every file. One file per repo is enough. I put all
copyrights to the root `LICENSE` file and removed header from all other
files. Also updated tools and `Makefile`.
113 lines
2.4 KiB
Go
113 lines
2.4 KiB
Go
package queues
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"go.temporal.io/server/common/collection"
|
|
"go.temporal.io/server/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,
|
|
)
|
|
}
|