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`.
98 lines
2.1 KiB
Go
98 lines
2.1 KiB
Go
package queues
|
|
|
|
import (
|
|
"math"
|
|
"math/rand"
|
|
"slices"
|
|
"time"
|
|
|
|
"go.temporal.io/server/common/predicates"
|
|
"go.temporal.io/server/service/history/tasks"
|
|
)
|
|
|
|
func NewRandomKey() tasks.Key {
|
|
return tasks.NewKey(time.Unix(0, rand.Int63()).UTC(), rand.Int63())
|
|
}
|
|
|
|
func NewRandomRange() Range {
|
|
maxKey := NewRandomKey()
|
|
minKey := tasks.NewKey(
|
|
time.Unix(0, rand.Int63n(maxKey.FireTime.UnixNano())).UTC(),
|
|
rand.Int63n(maxKey.TaskID),
|
|
)
|
|
return NewRange(minKey, maxKey)
|
|
}
|
|
|
|
func NewRandomKeyInRange(
|
|
r Range,
|
|
) tasks.Key {
|
|
if r.IsEmpty() {
|
|
panic("can not create key in range for an empty range")
|
|
}
|
|
|
|
minFireTimeUnixNano := r.InclusiveMin.FireTime.UnixNano()
|
|
maxFireTimeUnixNano := r.ExclusiveMax.FireTime.UnixNano()
|
|
minTaskID := r.InclusiveMin.TaskID
|
|
maxTaskID := r.ExclusiveMax.TaskID
|
|
|
|
if minFireTimeUnixNano == maxFireTimeUnixNano {
|
|
return tasks.NewKey(
|
|
r.InclusiveMin.FireTime,
|
|
rand.Int63n(1+maxTaskID-minTaskID)+minTaskID,
|
|
)
|
|
}
|
|
|
|
fireTime := time.Unix(0, rand.Int63n(1+maxFireTimeUnixNano-minFireTimeUnixNano)+minFireTimeUnixNano)
|
|
if fireTime.Equal(r.InclusiveMin.FireTime) {
|
|
return tasks.NewKey(
|
|
fireTime,
|
|
rand.Int63n(math.MaxInt64-minTaskID)+minTaskID,
|
|
)
|
|
}
|
|
|
|
if fireTime.Equal(r.ExclusiveMax.FireTime) {
|
|
return tasks.NewKey(
|
|
fireTime,
|
|
rand.Int63n(maxTaskID),
|
|
)
|
|
}
|
|
|
|
return tasks.NewKey(fireTime, rand.Int63())
|
|
}
|
|
|
|
func NewRandomOrderedRangesInRange(
|
|
r Range,
|
|
numRanges int,
|
|
) []Range {
|
|
ranges := []Range{r}
|
|
for len(ranges) < numRanges {
|
|
r := ranges[0]
|
|
left, right := r.Split(NewRandomKeyInRange(r))
|
|
left.ExclusiveMax.FireTime = left.ExclusiveMax.FireTime.Add(-time.Nanosecond)
|
|
right.InclusiveMin.FireTime = right.InclusiveMin.FireTime.Add(time.Nanosecond)
|
|
ranges = append(ranges[1:], left, right)
|
|
}
|
|
|
|
slices.SortFunc(ranges, func(a, b Range) int {
|
|
return a.InclusiveMin.CompareTo(b.InclusiveMin)
|
|
})
|
|
|
|
return ranges
|
|
}
|
|
|
|
func NewRandomScopes(
|
|
numScopes int,
|
|
) []Scope {
|
|
ranges := NewRandomOrderedRangesInRange(
|
|
NewRandomRange(),
|
|
numScopes,
|
|
)
|
|
|
|
scopes := make([]Scope, 0, 10)
|
|
for _, r := range ranges {
|
|
scopes = append(scopes, NewScope(r, predicates.Universal[tasks.Task]()))
|
|
}
|
|
|
|
return scopes
|
|
}
|