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`.
156 lines
2.7 KiB
Go
156 lines
2.7 KiB
Go
package queues
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"maps"
|
|
"sync"
|
|
"sync/atomic"
|
|
|
|
"go.temporal.io/server/common"
|
|
)
|
|
|
|
var (
|
|
errReaderGroupStopped = errors.New("queue reader group already stopped")
|
|
)
|
|
|
|
type (
|
|
ReaderInitializer func(readerID int64, slices []Slice) Reader
|
|
|
|
ReaderGroup struct {
|
|
sync.Mutex
|
|
|
|
initializer ReaderInitializer
|
|
|
|
status int32
|
|
readerMap map[int64]Reader
|
|
}
|
|
)
|
|
|
|
func NewReaderGroup(
|
|
initializer ReaderInitializer,
|
|
) *ReaderGroup {
|
|
return &ReaderGroup{
|
|
initializer: initializer,
|
|
|
|
status: common.DaemonStatusInitialized,
|
|
readerMap: make(map[int64]Reader),
|
|
}
|
|
}
|
|
|
|
func (g *ReaderGroup) Start() {
|
|
if !atomic.CompareAndSwapInt32(&g.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {
|
|
return
|
|
}
|
|
|
|
g.Lock()
|
|
defer g.Unlock()
|
|
|
|
for _, reader := range g.readerMap {
|
|
reader.Start()
|
|
}
|
|
}
|
|
|
|
func (g *ReaderGroup) Stop() {
|
|
if !atomic.CompareAndSwapInt32(&g.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {
|
|
return
|
|
}
|
|
|
|
g.Lock()
|
|
defer g.Unlock()
|
|
|
|
for _, reader := range g.readerMap {
|
|
reader.Stop()
|
|
}
|
|
}
|
|
|
|
func (g *ReaderGroup) ForEach(f func(int64, Reader)) {
|
|
g.Lock()
|
|
defer g.Unlock()
|
|
|
|
if g.readerMap == nil {
|
|
return
|
|
}
|
|
|
|
for readerID, reader := range g.readerMap {
|
|
f(readerID, reader)
|
|
}
|
|
}
|
|
|
|
func (g *ReaderGroup) Readers() map[int64]Reader {
|
|
g.Lock()
|
|
defer g.Unlock()
|
|
|
|
readerMapCopy := make(map[int64]Reader, len(g.readerMap))
|
|
maps.Copy(readerMapCopy, g.readerMap)
|
|
|
|
return readerMapCopy
|
|
}
|
|
|
|
func (g *ReaderGroup) GetOrCreateReader(readerID int64) Reader {
|
|
g.Lock()
|
|
defer g.Unlock()
|
|
|
|
reader, ok := g.getReaderByIDLocked(readerID)
|
|
if ok {
|
|
return reader
|
|
}
|
|
|
|
return g.newReaderLocked(readerID)
|
|
}
|
|
|
|
func (g *ReaderGroup) ReaderByID(readerID int64) (Reader, bool) {
|
|
g.Lock()
|
|
defer g.Unlock()
|
|
|
|
return g.getReaderByIDLocked(readerID)
|
|
}
|
|
|
|
func (g *ReaderGroup) getReaderByIDLocked(readerID int64) (Reader, bool) {
|
|
if g.readerMap == nil {
|
|
return nil, false
|
|
}
|
|
|
|
reader, ok := g.readerMap[readerID]
|
|
return reader, ok
|
|
}
|
|
|
|
func (g *ReaderGroup) NewReader(readerID int64, slices ...Slice) Reader {
|
|
g.Lock()
|
|
defer g.Unlock()
|
|
|
|
return g.newReaderLocked(readerID, slices...)
|
|
}
|
|
|
|
func (g *ReaderGroup) newReaderLocked(readerID int64, slices ...Slice) Reader {
|
|
reader := g.initializer(readerID, slices)
|
|
|
|
if _, ok := g.readerMap[readerID]; ok {
|
|
panic(fmt.Sprintf("reader with ID %v already exists", readerID))
|
|
}
|
|
|
|
g.readerMap[readerID] = reader
|
|
|
|
if g.isStarted() {
|
|
reader.Start()
|
|
}
|
|
return reader
|
|
}
|
|
|
|
func (g *ReaderGroup) RemoveReader(readerID int64) {
|
|
g.Lock()
|
|
defer g.Unlock()
|
|
|
|
reader, ok := g.readerMap[readerID]
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
reader.Stop()
|
|
delete(g.readerMap, readerID)
|
|
}
|
|
|
|
func (g *ReaderGroup) isStarted() bool {
|
|
return atomic.LoadInt32(&g.status) == common.DaemonStatusStarted
|
|
}
|