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`.
85 lines
2.1 KiB
Go
85 lines
2.1 KiB
Go
package collection
|
|
|
|
import "sync"
|
|
|
|
// OnceMap is a concurrent map which lazily constructs its values. Map values are initialized on-the-fly, using a
|
|
// provided construction function only when a key is accessed for the first time.
|
|
type OnceMap[K comparable, T any] struct {
|
|
mu sync.RWMutex
|
|
inner map[K]T
|
|
construct func(K) T
|
|
}
|
|
|
|
// NewOnceMap creates a [OnceMap] from a given construct function.
|
|
// construct should be kept light as it is called while holding a lock on the entire map.
|
|
func NewOnceMap[K comparable, T any](construct func(K) T) *OnceMap[K, T] {
|
|
return &OnceMap[K, T]{
|
|
construct: construct,
|
|
inner: make(map[K]T, 0),
|
|
}
|
|
}
|
|
|
|
func (m *OnceMap[K, T]) Get(key K) T {
|
|
m.mu.RLock()
|
|
value, ok := m.inner[key]
|
|
m.mu.RUnlock()
|
|
if !ok {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
if value, ok = m.inner[key]; !ok {
|
|
value = m.construct(key)
|
|
m.inner[key] = value
|
|
}
|
|
}
|
|
|
|
return value
|
|
}
|
|
|
|
// FallibleOnceMap is a concurrent map which lazily constructs its values. Map values are initialized on-the-fly, using
|
|
// a provided construction function only when a key is accessed for the first time.
|
|
// If the construct function returns an error, the value is not cached.
|
|
type FallibleOnceMap[K comparable, T any] struct {
|
|
mu sync.RWMutex
|
|
inner map[K]T
|
|
construct func(K) (T, error)
|
|
}
|
|
|
|
// NewFallibleOnceMap creates a [FallibleOnceMap] from a given construct function.
|
|
// construct should be kept light as it is called while holding a lock on the entire map.
|
|
func NewFallibleOnceMap[K comparable, T any](construct func(K) (T, error)) *FallibleOnceMap[K, T] {
|
|
return &FallibleOnceMap[K, T]{
|
|
construct: construct,
|
|
inner: make(map[K]T, 0),
|
|
}
|
|
}
|
|
|
|
func (p *FallibleOnceMap[K, T]) Get(key K) (T, error) {
|
|
p.mu.RLock()
|
|
value, ok := p.inner[key]
|
|
p.mu.RUnlock()
|
|
if !ok {
|
|
p.mu.Lock()
|
|
defer p.mu.Unlock()
|
|
if value, ok = p.inner[key]; !ok {
|
|
var err error
|
|
value, err = p.construct(key)
|
|
if err != nil {
|
|
return value, err
|
|
}
|
|
p.inner[key] = value
|
|
}
|
|
}
|
|
|
|
return value, nil
|
|
}
|
|
|
|
func (p *FallibleOnceMap[K, T]) Pop(key K) (T, bool) {
|
|
p.mu.Lock()
|
|
defer p.mu.Unlock()
|
|
val, ok := p.inner[key]
|
|
if ok {
|
|
delete(p.inner, key)
|
|
}
|
|
return val, ok
|
|
}
|