Golang 编写一个简单的内存 KV 数据库-Golang-PHP中文网

不能直接用map当KV数据库，因其非并发安全，多goroutine读写必panic；sync.Map仅适用于读多写少场景，频繁增删改查时性能更差。

golang 编写一个简单的内存 kv 数据库

为什么不用 `map` 直接当 KV 数据库用

因为裸用 map 在并发场景下会 panic：Go 的 map 非线程安全，只要两个 goroutine 同时读写（哪怕只是读+写），就可能触发 fatal error: concurrent map read and map write。你不能靠“我保证只单线程用”来绕过——实际业务里稍加异步或 HTTP handler 就破防。

实操建议：

必须加锁：用 sync.RWMutex 包裹 map，读用 RLock，写用 Lock
避免锁粒度太粗：别整个数据库一把锁，按 key 做分段锁（shard）能提升并发吞吐，但简单场景一把全局锁够用
不要用 sync.Map 替代：它适合「读多写少 + key 固定」的缓存场景，不支持遍历、不支持 TTL、删除后内存不释放，做 KV 数据库反而更难控制

如何支持基本 CRUD 和线程安全的 `Get`/`Set`

核心是封装一个结构体，把 map[string]interface{} 和锁收进去，再暴露方法。注意类型擦除带来的隐患：存 int 和取 string 不报错，但运行时 panic。

实操建议：

值类型统一用 interface{}，但文档/注释里明确要求调用方负责类型一致性
Get 方法返回 (interface{}, bool)，用第二个 bool 判断 key 是否真实存在（区别于存了 nil）
Set 不需要返回值，但内部要处理 nil 赋值（Go 允许 map 存 nil，这是合法的）
示例片段：

type MemDB struct {
    mu sync.RWMutex
    data map[string]interface{}
}

func (db *MemDB) Get(key string) (interface{}, bool) {
    db.mu.RLock()
    defer db.mu.RUnlock()
    v, ok := db.data[key]
    return v, ok
}

func (db *MemDB) Set(key string, value interface{}) {
    db.mu.Lock()
    defer db.mu.Unlock()
    if db.data == nil {
        db.data = make(map[string]interface{})
    }
    db.data[key] = value
}

怎么加简单的 TTL 过期机制

Go 没有内置带过期的 map，自己实现得兼顾精度和开销。别用 time.AfterFunc 为每个 key 起 goroutine——10 万个 key 就 10 万个 goroutine，直接 OOM。

Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等

下载

实操建议：

用惰性删除（lazy expiration）：每次 Get 时检查过期时间，过期就删并返回不存在；Set 时存 value 和 expireAt 两个字段
数据结构改成 map[string]entry，其中 entry 是结构体：type entry struct { Value interface{}; ExpireAt time.Time }
不维护全局定时器，省心；但长期不访问的过期 key 会残留——如果业务能接受，这就是最简方案
若需主动清理，加一个低频后台 goroutine（比如每分钟扫一次），用 time.Now().Before(e.ExpireAt) 过滤，避免锁住整个 map 太久