add registry package
This commit is contained in:
+52
@@ -0,0 +1,52 @@
|
||||
package transport
|
||||
|
||||
import (
|
||||
"errors"
|
||||
)
|
||||
|
||||
// Registry error.
|
||||
var (
|
||||
ErrKeyLocked = errors.New("key is locked")
|
||||
)
|
||||
|
||||
// Registry is a simple interface allowing untyped common values to be shared across packages.
|
||||
// This is particularly useful for resolving import cycles.
|
||||
type Registry struct {
|
||||
Data map[string]any
|
||||
Locks map[string]bool
|
||||
}
|
||||
|
||||
// Get the value of a key.
|
||||
// This function returns nil if no value exists.
|
||||
func (reg *Registry) Get(key string) any {
|
||||
return reg.Data[key]
|
||||
}
|
||||
|
||||
// Has checks whether a key is non-nil.
|
||||
func (reg *Registry) Has(key string) bool {
|
||||
return reg.Data[key] == nil
|
||||
}
|
||||
|
||||
// Lock a key.
|
||||
// This prevents the value being overwritten by Set.
|
||||
func (reg *Registry) Lock(key string) {
|
||||
reg.Locks[key] = true
|
||||
}
|
||||
|
||||
// Set the value for a key.
|
||||
// This function returns an error if the key is locked.
|
||||
func (reg *Registry) Set(key string, value any) error {
|
||||
if reg.Locks[key] {
|
||||
return ErrKeyLocked
|
||||
}
|
||||
reg.Data[key] = value
|
||||
return nil
|
||||
}
|
||||
|
||||
// New creates a new registry.
|
||||
func New() *Registry {
|
||||
return &Registry{
|
||||
Data: map[string]any{},
|
||||
Locks: map[string]bool{},
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user