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`.
208 lines
3.7 KiB
Go
208 lines
3.7 KiB
Go
package locks
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/dgryski/go-farm"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
type (
|
|
idMutexSuite struct {
|
|
suite.Suite
|
|
|
|
numShard uint32
|
|
idMutex IDMutex
|
|
}
|
|
|
|
testIdentifier struct {
|
|
A string
|
|
B string
|
|
C string
|
|
}
|
|
)
|
|
|
|
func BenchmarkGolangMutex(b *testing.B) {
|
|
lock := &sync.Mutex{}
|
|
for i := 0; i < b.N; i++ {
|
|
lock.Lock()
|
|
func() {}()
|
|
lock.Unlock()
|
|
}
|
|
}
|
|
|
|
func BenchmarkIDMutex_String(b *testing.B) {
|
|
identifier := "random string"
|
|
idLock := NewIDMutex(32, func(key interface{}) uint32 {
|
|
id, ok := key.(string)
|
|
if !ok {
|
|
return 0
|
|
}
|
|
return farm.Fingerprint32([]byte(id))
|
|
})
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
idLock.LockID(identifier)
|
|
idLock.UnlockID(identifier)
|
|
}
|
|
}
|
|
|
|
func BenchmarkIDMutex_Struct(b *testing.B) {
|
|
identifier := testIdentifier{
|
|
A: "some random A",
|
|
B: "some random B",
|
|
C: "some random C",
|
|
}
|
|
idLock := NewIDMutex(32, func(key interface{}) uint32 {
|
|
id, ok := key.(testIdentifier)
|
|
if !ok {
|
|
return 0
|
|
}
|
|
return farm.Fingerprint32([]byte(id.A))
|
|
})
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
idLock.LockID(identifier)
|
|
idLock.UnlockID(identifier)
|
|
}
|
|
}
|
|
|
|
func BenchmarkIDMutex_StringConcurrent(b *testing.B) {
|
|
identifier := "random string"
|
|
idLock := NewIDMutex(32, func(key interface{}) uint32 {
|
|
id, ok := key.(string)
|
|
if !ok {
|
|
return 0
|
|
}
|
|
return farm.Fingerprint32([]byte(id))
|
|
})
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
counter := 0
|
|
iteration := 2000
|
|
|
|
waitGroupBegin := &sync.WaitGroup{}
|
|
waitGroupBegin.Add(1)
|
|
|
|
waitGroupEnd := &sync.WaitGroup{}
|
|
waitGroupEnd.Add(iteration)
|
|
|
|
fn := func() {
|
|
waitGroupBegin.Wait()
|
|
|
|
idLock.LockID(identifier)
|
|
counter++
|
|
idLock.UnlockID(identifier)
|
|
|
|
waitGroupEnd.Done()
|
|
}
|
|
|
|
for i := 0; i < iteration; i++ {
|
|
go fn()
|
|
}
|
|
waitGroupBegin.Done()
|
|
waitGroupEnd.Wait()
|
|
}
|
|
}
|
|
|
|
func TestIDMutexSuite(t *testing.T) {
|
|
s := new(idMutexSuite)
|
|
suite.Run(t, s)
|
|
}
|
|
|
|
func (s *idMutexSuite) SetupSuite() {
|
|
}
|
|
|
|
func (s *idMutexSuite) TearDownSuite() {
|
|
|
|
}
|
|
|
|
func (s *idMutexSuite) SetupTest() {
|
|
s.numShard = 32
|
|
s.idMutex = NewIDMutex(s.numShard, func(key interface{}) uint32 {
|
|
id, ok := key.(string)
|
|
if !ok {
|
|
return 0
|
|
}
|
|
return farm.Fingerprint32([]byte(id))
|
|
})
|
|
}
|
|
|
|
func (s *idMutexSuite) TearDownTest() {
|
|
|
|
}
|
|
|
|
func (s *idMutexSuite) TestLockOnDiffIDs() {
|
|
identifier1 := "some random identifier 1"
|
|
identifier2 := "some random identifier 2"
|
|
|
|
s.idMutex.LockID(identifier1)
|
|
s.idMutex.LockID(identifier2)
|
|
s.idMutex.UnlockID(identifier2)
|
|
s.idMutex.UnlockID(identifier1)
|
|
|
|
s.idMutex.LockID(identifier1)
|
|
s.idMutex.LockID(identifier2)
|
|
s.idMutex.UnlockID(identifier1)
|
|
s.idMutex.UnlockID(identifier2)
|
|
}
|
|
|
|
func (s *idMutexSuite) TestLockOnSameID() {
|
|
count := 0
|
|
identifier := "some random identifier"
|
|
|
|
waitGroupBegin := &sync.WaitGroup{}
|
|
waitGroupBegin.Add(1)
|
|
|
|
waitGroupEnd := &sync.WaitGroup{}
|
|
waitGroupEnd.Add(1)
|
|
|
|
s.idMutex.LockID(identifier)
|
|
go func() {
|
|
waitGroupBegin.Done()
|
|
s.idMutex.LockID(identifier)
|
|
s.Equal(1, count)
|
|
s.idMutex.UnlockID(identifier)
|
|
waitGroupEnd.Done()
|
|
}()
|
|
waitGroupBegin.Wait()
|
|
count = 1
|
|
s.idMutex.UnlockID(identifier)
|
|
waitGroupEnd.Wait()
|
|
}
|
|
|
|
func (s *idMutexSuite) TestConcurrentAccess() {
|
|
counter := 0
|
|
identifier := "some random identifier"
|
|
iteration := 2000
|
|
|
|
waitGroupBegin := &sync.WaitGroup{}
|
|
waitGroupBegin.Add(1)
|
|
|
|
waitGroupEnd := &sync.WaitGroup{}
|
|
waitGroupEnd.Add(iteration)
|
|
|
|
fn := func() {
|
|
waitGroupBegin.Wait()
|
|
|
|
s.idMutex.LockID(identifier)
|
|
counter++
|
|
s.idMutex.UnlockID(identifier)
|
|
|
|
waitGroupEnd.Done()
|
|
}
|
|
|
|
for i := 0; i < iteration; i++ {
|
|
go fn()
|
|
}
|
|
waitGroupBegin.Done()
|
|
waitGroupEnd.Wait()
|
|
|
|
s.Equal(iteration, counter)
|
|
impl := s.idMutex.(*idMutexImpl)
|
|
for i := uint32(0); i < s.numShard; i++ {
|
|
s.Equal(0, len(impl.shards[i].mutexInfos))
|
|
}
|
|
}
|