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).
61 lines
1.4 KiB
Go
61 lines
1.4 KiB
Go
package matching
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/emirpasic/gods/maps/treemap"
|
|
godsutils "github.com/emirpasic/gods/utils"
|
|
"github.com/hanzoai/tasks/common/util"
|
|
"google.golang.org/protobuf/types/known/timestamppb"
|
|
)
|
|
|
|
const emptyBacklogAge time.Duration = -1
|
|
|
|
// backlogAgeTracker is not safe for concurrent use
|
|
type backlogAgeTracker struct {
|
|
tree treemap.Map // unix nano as int64 -> int (count)
|
|
}
|
|
|
|
func newBacklogAgeTracker() backlogAgeTracker {
|
|
return backlogAgeTracker{tree: *treemap.NewWith(godsutils.Int64Comparator)}
|
|
}
|
|
|
|
// record adds or removes a task from the tracker.
|
|
func (b backlogAgeTracker) record(ts *timestamppb.Timestamp, delta int) {
|
|
if ts == nil {
|
|
return
|
|
}
|
|
|
|
createTime := ts.AsTime().UnixNano()
|
|
count := delta
|
|
if prev, ok := b.tree.Get(createTime); ok {
|
|
count += prev.(int) // nolint:revive
|
|
}
|
|
if count = max(0, count); count == 0 {
|
|
b.tree.Remove(createTime)
|
|
} else {
|
|
b.tree.Put(createTime, count)
|
|
}
|
|
}
|
|
|
|
// oldestTime returns the time of the oldest task in this backlog, or
|
|
// the zero Time if empty.
|
|
func (b backlogAgeTracker) oldestTime() time.Time {
|
|
if b.tree.Empty() {
|
|
return time.Time{}
|
|
}
|
|
k, _ := b.tree.Min()
|
|
return time.Unix(0, k.(int64)) // nolint:revive
|
|
}
|
|
|
|
// minNonZeroTime returns the minimum time of a and b, ignoring zero times.
|
|
// If both a and b are zero, it returns zero.
|
|
func minNonZeroTime(a, b time.Time) time.Time {
|
|
if a.IsZero() {
|
|
return b
|
|
} else if b.IsZero() {
|
|
return a
|
|
}
|
|
return util.MinTime(a, b)
|
|
}
|