mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +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`.
127 lines
2.7 KiB
Go
127 lines
2.7 KiB
Go
package metrics
|
|
|
|
import (
|
|
"math/rand"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
"go.uber.org/mock/gomock"
|
|
)
|
|
|
|
type (
|
|
// baggageBenchTest Was used to compare the behavior of sync.Map vs map+mutex in the context of usage for
|
|
// metricsContext.
|
|
// See testMapBaggage for test logic.
|
|
// As a summary, using mutex performed considerably faster than sync.Map.
|
|
// These results are to be reviewed if the usage scenario of metricsContext changes.
|
|
baggageBenchTest struct {
|
|
suite.Suite
|
|
*require.Assertions
|
|
controller *gomock.Controller
|
|
}
|
|
|
|
testBaggage interface {
|
|
Add(k string, v int64)
|
|
Get(k string) int64
|
|
}
|
|
|
|
baggageSyncMap struct {
|
|
data *sync.Map
|
|
}
|
|
|
|
baggageMutexMap struct {
|
|
sync.Mutex
|
|
data map[string]int64
|
|
}
|
|
)
|
|
|
|
func TestBaggageBenchSuite(t *testing.T) {
|
|
s := new(baggageBenchTest)
|
|
suite.Run(t, s)
|
|
}
|
|
|
|
func (s *baggageBenchTest) SetupTest() {
|
|
s.Assertions = require.New(s.T())
|
|
s.controller = gomock.NewController(s.T())
|
|
}
|
|
|
|
func (s *baggageBenchTest) TearDownTest() {}
|
|
|
|
func (b *baggageSyncMap) Add(k string, v int64) {
|
|
for done := false; !done; {
|
|
metricInterface, _ := b.data.LoadAndDelete(k)
|
|
var newValue = v
|
|
if metricInterface != nil {
|
|
newValue += metricInterface.(int64)
|
|
}
|
|
_, loaded := b.data.LoadOrStore(k, newValue)
|
|
done = !loaded
|
|
}
|
|
}
|
|
|
|
func (b *baggageSyncMap) Get(k string) int64 {
|
|
metricInterface, _ := b.data.LoadAndDelete(k)
|
|
if metricInterface == nil {
|
|
return 0
|
|
}
|
|
return metricInterface.(int64)
|
|
}
|
|
|
|
func (b *baggageMutexMap) Add(k string, v int64) {
|
|
b.Lock()
|
|
defer b.Unlock()
|
|
|
|
value := b.data[k]
|
|
value += v
|
|
b.data[k] = value
|
|
}
|
|
|
|
func (b *baggageMutexMap) Get(k string) int64 {
|
|
b.Lock()
|
|
defer b.Unlock()
|
|
return b.data[k]
|
|
}
|
|
|
|
// roughly 1.7s/7.5s for mutex/sync
|
|
// baggageCount := 1000
|
|
// threadCount := 20
|
|
// updatesPerThread := 1000
|
|
func testMapBaggage(createTestObj func() testBaggage) {
|
|
baggageCount := 10
|
|
threadCount := 10
|
|
updatesPerThread := 10
|
|
|
|
keys := []string{"k1", "k2", "k3", "k4", "k5"}
|
|
start := time.Now()
|
|
sum := int64(0)
|
|
for bag := 0; bag < baggageCount; bag++ {
|
|
testObj := createTestObj()
|
|
wg := sync.WaitGroup{}
|
|
wg.Add(threadCount)
|
|
for th := 0; th < threadCount; th++ {
|
|
go func(key string) {
|
|
for upd := 0; upd < updatesPerThread; upd++ {
|
|
testObj.Add(key, rand.Int63())
|
|
}
|
|
wg.Done()
|
|
}(keys[th%len(keys)])
|
|
}
|
|
wg.Wait()
|
|
val := testObj.Get(keys[0])
|
|
sum += val
|
|
}
|
|
println("sum: ", sum)
|
|
println("duration: ", time.Since(start))
|
|
}
|
|
|
|
func (s *baggageBenchTest) TestSyncMapBaggage() {
|
|
testMapBaggage(func() testBaggage { return &baggageSyncMap{data: &sync.Map{}} })
|
|
}
|
|
|
|
func (s *baggageBenchTest) TestMutexMapBaggage() {
|
|
testMapBaggage(func() testBaggage { return &baggageMutexMap{data: make(map[string]int64)} })
|
|
}
|