mirror of https://github.com/caddyserver/caddy.git
replacer: use RWMutex to protect static provider
This commit is contained in:
parent
6d9a83376b
commit
73ab7478f6
11
replacer.go
11
replacer.go
|
@ -22,6 +22,7 @@ import (
|
|||
"runtime"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
|
@ -29,6 +30,7 @@ import (
|
|||
func NewReplacer() *Replacer {
|
||||
rep := &Replacer{
|
||||
static: make(map[string]any),
|
||||
mapMutex: &sync.RWMutex{},
|
||||
}
|
||||
rep.providers = []ReplacerFunc{
|
||||
globalDefaultReplacements,
|
||||
|
@ -42,6 +44,7 @@ func NewReplacer() *Replacer {
|
|||
func NewEmptyReplacer() *Replacer {
|
||||
rep := &Replacer{
|
||||
static: make(map[string]any),
|
||||
mapMutex: &sync.RWMutex{},
|
||||
}
|
||||
rep.providers = []ReplacerFunc{
|
||||
rep.fromStatic,
|
||||
|
@ -54,7 +57,9 @@ func NewEmptyReplacer() *Replacer {
|
|||
// use NewReplacer to make one.
|
||||
type Replacer struct {
|
||||
providers []ReplacerFunc
|
||||
|
||||
static map[string]any
|
||||
mapMutex *sync.RWMutex
|
||||
}
|
||||
|
||||
// Map adds mapFunc to the list of value providers.
|
||||
|
@ -65,7 +70,9 @@ func (r *Replacer) Map(mapFunc ReplacerFunc) {
|
|||
|
||||
// Set sets a custom variable to a static value.
|
||||
func (r *Replacer) Set(variable string, value any) {
|
||||
r.mapMutex.Lock()
|
||||
r.static[variable] = value
|
||||
r.mapMutex.Unlock()
|
||||
}
|
||||
|
||||
// Get gets a value from the replacer. It returns
|
||||
|
@ -89,11 +96,15 @@ func (r *Replacer) GetString(variable string) (string, bool) {
|
|||
// Delete removes a variable with a static value
|
||||
// that was created using Set.
|
||||
func (r *Replacer) Delete(variable string) {
|
||||
r.mapMutex.Lock()
|
||||
delete(r.static, variable)
|
||||
r.mapMutex.Unlock()
|
||||
}
|
||||
|
||||
// fromStatic provides values from r.static.
|
||||
func (r *Replacer) fromStatic(key string) (any, bool) {
|
||||
r.mapMutex.RLock()
|
||||
defer r.mapMutex.RUnlock()
|
||||
val, ok := r.static[key]
|
||||
return val, ok
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue