新增 cache

pkg/services/config.go

@@ -0,0 +1,58 @@
+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
+}