59 lines
986 B
Go
59 lines
986 B
Go
package services
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"sync"
|
|
)
|
|
|
|
type Config interface {
|
|
Put(key string, value string) error
|
|
Get(key string) (string, error)
|
|
Delete(key string) error
|
|
io.Closer
|
|
}
|
|
|
|
func NewConfigMemory() Config {
|
|
return &ConfigMemory{
|
|
data: make(map[string]string),
|
|
lock: sync.RWMutex{},
|
|
}
|
|
}
|
|
|
|
// ConfigMemory 一个简单的内存配置归档，仅用于测试
|
|
type ConfigMemory struct {
|
|
data map[string]string
|
|
lock sync.RWMutex
|
|
}
|
|
|
|
func (m *ConfigMemory) Put(key string, value string) error {
|
|
m.lock.Lock()
|
|
defer m.lock.Unlock()
|
|
m.data[key] = value
|
|
return nil
|
|
}
|
|
|
|
func (m *ConfigMemory) Get(key string) (string, error) {
|
|
m.lock.RLock()
|
|
defer m.lock.RUnlock()
|
|
v, ok := m.data[key]
|
|
if !ok {
|
|
return "", os.ErrNotExist
|
|
}
|
|
return v, nil
|
|
}
|
|
|
|
func (m *ConfigMemory) Delete(key string) error {
|
|
m.lock.Lock()
|
|
defer m.lock.Unlock()
|
|
delete(m.data, key)
|
|
return nil
|
|
}
|
|
|
|
func (m *ConfigMemory) Close() error {
|
|
m.lock.Lock()
|
|
defer m.lock.Unlock()
|
|
clear(m.data)
|
|
return nil
|
|
}
|