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).
112 lines
2.3 KiB
Go
112 lines
2.3 KiB
Go
package queues
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/hanzoai/tasks/common/predicates"
|
|
"github.com/hanzoai/tasks/service/history/tasks"
|
|
)
|
|
|
|
type (
|
|
Scope struct {
|
|
Range Range
|
|
Predicate tasks.Predicate
|
|
}
|
|
)
|
|
|
|
func NewScope(
|
|
r Range,
|
|
predicate tasks.Predicate,
|
|
) Scope {
|
|
return Scope{
|
|
Range: r,
|
|
Predicate: predicate,
|
|
}
|
|
}
|
|
|
|
func (s *Scope) Contains(task tasks.Task) bool {
|
|
return s.Range.ContainsKey(task.GetKey()) &&
|
|
s.Predicate.Test(task)
|
|
}
|
|
|
|
func (s *Scope) CanSplitByRange(
|
|
key tasks.Key,
|
|
) bool {
|
|
return s.Range.CanSplit(key)
|
|
}
|
|
|
|
func (s *Scope) SplitByRange(
|
|
key tasks.Key,
|
|
) (left Scope, right Scope) {
|
|
if !s.CanSplitByRange(key) {
|
|
panic(fmt.Sprintf("Unable to split scope with range %v at %v", s.Range, key))
|
|
}
|
|
|
|
leftRange, rightRange := s.Range.Split(key)
|
|
return NewScope(leftRange, s.Predicate), NewScope(rightRange, s.Predicate)
|
|
}
|
|
|
|
func (s *Scope) SplitByPredicate(
|
|
predicate tasks.Predicate,
|
|
) (pass Scope, fail Scope) {
|
|
passScope := NewScope(
|
|
s.Range,
|
|
tasks.AndPredicates(s.Predicate, predicate),
|
|
)
|
|
failScope := NewScope(
|
|
s.Range,
|
|
tasks.AndPredicates(s.Predicate, predicates.Not(predicate)),
|
|
)
|
|
return passScope, failScope
|
|
}
|
|
|
|
func (s *Scope) CanMergeByRange(
|
|
incomingScope Scope,
|
|
) bool {
|
|
return s.Range.CanMerge(incomingScope.Range) &&
|
|
s.Predicate.Equals(incomingScope.Predicate)
|
|
}
|
|
|
|
func (s *Scope) MergeByRange(
|
|
incomingScope Scope,
|
|
) Scope {
|
|
if !s.CanMergeByRange(incomingScope) {
|
|
panic(fmt.Sprintf("Unable to merge scope with range %v with range %v by range", s.Range, incomingScope.Range))
|
|
}
|
|
|
|
return NewScope(s.Range.Merge(incomingScope.Range), s.Predicate)
|
|
}
|
|
|
|
func (s *Scope) CanMergeByPredicate(
|
|
incomingScope Scope,
|
|
) bool {
|
|
return s.Range.Equals(incomingScope.Range)
|
|
}
|
|
|
|
func (s *Scope) MergeByPredicate(
|
|
incomingScope Scope,
|
|
) Scope {
|
|
if !s.CanMergeByPredicate(incomingScope) {
|
|
panic(fmt.Sprintf("Unable to merge scope with range %v with range %v by predicate", s.Range, incomingScope.Range))
|
|
}
|
|
|
|
return NewScope(s.Range, tasks.OrPredicates(s.Predicate, incomingScope.Predicate))
|
|
}
|
|
|
|
func (s *Scope) IsEmpty() bool {
|
|
if s.Range.IsEmpty() {
|
|
return true
|
|
}
|
|
|
|
if _, ok := s.Predicate.(*predicates.EmptyImpl[tasks.Task]); ok {
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func (s *Scope) Equals(scope Scope) bool {
|
|
return s.Range.Equals(scope.Range) &&
|
|
s.Predicate.Equals(scope.Predicate)
|
|
}
|