45 lines
980 B
Go
45 lines
980 B
Go
package build
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/microsoft/typescript-go/internal/collections"
|
|
)
|
|
|
|
type parseCacheEntry[V comparable] struct {
|
|
value V
|
|
mu sync.Mutex
|
|
}
|
|
|
|
type parseCache[K comparable, V comparable] struct {
|
|
entries collections.SyncMap[K, *parseCacheEntry[V]]
|
|
}
|
|
|
|
func (c *parseCache[K, V]) loadOrStore(key K, parse func(K) V, allowZero bool) V {
|
|
newEntry := &parseCacheEntry[V]{}
|
|
newEntry.mu.Lock()
|
|
defer newEntry.mu.Unlock()
|
|
if entry, loaded := c.entries.LoadOrStore(key, newEntry); loaded {
|
|
entry.mu.Lock()
|
|
defer entry.mu.Unlock()
|
|
if allowZero || entry.value != *new(V) {
|
|
return entry.value
|
|
}
|
|
newEntry = entry
|
|
}
|
|
newEntry.value = parse(key)
|
|
return newEntry.value
|
|
}
|
|
|
|
func (c *parseCache[K, V]) store(key K, value V) {
|
|
c.entries.Store(key, &parseCacheEntry[V]{value: value})
|
|
}
|
|
|
|
func (c *parseCache[K, V]) delete(key K) {
|
|
c.entries.Delete(key)
|
|
}
|
|
|
|
func (c *parseCache[K, V]) reset() {
|
|
c.entries = collections.SyncMap[K, *parseCacheEntry[V]]{}
|
|
}
|