Logan.Li
b7166acda4
- SQL 方言层：agent updated_at、dashboard 小时分桶、清理任务跨 SQLite/MySQL - Cache Capabilities：redka 跳过 ZSet/PubSub，Agent 聚合改 DB 直读 - local 模式 redka 初始化失败 Fatal；/version 返回 features - SQLite WAL + 连接池限制；K8s data 改 PVC 持久化 - run.sh：旧参数迁移、升级后 /version 校验
125 lines
3.1 KiB
Go
125 lines
3.1 KiB
Go
// redka 内嵌 Redis 兼容存储实现（v0.2.x API）
|
||
// 使用 github.com/nalgeon/redka — 基于 SQLite 的 Redis 替代方案 // Logan.Li
|
||
// 零外部依赖，数据存在本地文件中，非常适合独立二进制部署
|
||
package redis
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"time"
|
||
|
||
"github.com/nalgeon/redka"
|
||
)
|
||
|
||
// redkaCache 基于 Redka v0.2.x 的缓存实现 // Logan.Li
|
||
type redkaCache struct {
|
||
db *redka.DB
|
||
}
|
||
|
||
func newRedka(cfg Config) (Cache, error) {
|
||
dataDir := cfg.DataDir
|
||
if dataDir == "" {
|
||
dataDir = "data/redka"
|
||
}
|
||
|
||
// Redka v0.2.x: Open(path, opts) // Logan.Li
|
||
db, err := redka.Open(dataDir+"/redka.db", nil)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return &redkaCache{db: db}, nil
|
||
}
|
||
|
||
func (r *redkaCache) Capabilities() Capabilities {
|
||
return Capabilities{SortedSet: false, PubSub: false}
|
||
}
|
||
|
||
func (r *redkaCache) Get(ctx context.Context, key string) (string, error) {
|
||
// Redka v0.2.x: Str().Get() 返回 core.Value // Logan.Li
|
||
val, err := r.db.Str().Get(key)
|
||
if err == redka.ErrNotFound {
|
||
return "", nil
|
||
}
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
return val.String(), nil
|
||
}
|
||
|
||
func (r *redkaCache) Set(ctx context.Context, key string, value string, expiration time.Duration) error {
|
||
// Redka v0.2.x: Str().Set() 返回 error // Logan.Li
|
||
if err := r.db.Str().Set(key, value); err != nil {
|
||
return err
|
||
}
|
||
if expiration > 0 {
|
||
_, err := r.db.Key().Expire(key, expiration)
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (r *redkaCache) Del(ctx context.Context, keys ...string) error {
|
||
for _, key := range keys {
|
||
if _, err := r.db.Key().Delete(key); err != nil {
|
||
return err
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (r *redkaCache) Exists(ctx context.Context, keys ...string) (int64, error) {
|
||
var count int64
|
||
for _, key := range keys {
|
||
ok, err := r.db.Key().Exists(key)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
if ok {
|
||
count++
|
||
}
|
||
}
|
||
return count, nil
|
||
}
|
||
|
||
func (r *redkaCache) Expire(ctx context.Context, key string, expiration time.Duration) error {
|
||
ok, err := r.db.Key().Expire(key, expiration)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if !ok {
|
||
return redka.ErrNotFound
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (r *redkaCache) Ping(ctx context.Context) error {
|
||
// Redka 始终可用（内嵌数据库）// Logan.Li
|
||
return nil
|
||
}
|
||
|
||
func (r *redkaCache) Close() error {
|
||
return r.db.Close()
|
||
}
|
||
|
||
var errSortedSetNotSupported = fmt.Errorf("sorted set operations are not supported by the redka backend; use go-redis (driver=redis) instead")
|
||
|
||
func (r *redkaCache) ZAdd(ctx context.Context, key string, member SortedSetMember) error {
|
||
return errSortedSetNotSupported
|
||
}
|
||
|
||
func (r *redkaCache) ZRangeByScore(ctx context.Context, key string, min, max string) ([]SortedSetMember, error) {
|
||
return nil, errSortedSetNotSupported
|
||
}
|
||
|
||
func (r *redkaCache) ZRemRangeByScore(ctx context.Context, key string, min, max string) error {
|
||
return errSortedSetNotSupported
|
||
}
|
||
|
||
var errPubSubNotSupported = fmt.Errorf("Pub/Sub is not supported by the redka backend; use go-redis (driver=redis)")
|
||
|
||
func (r *redkaCache) Publish(ctx context.Context, channel string, message string) error {
|
||
// local/redka 无 Pub/Sub，流水线日志走轮询 Get，此处 no-op // Logan.Li
|
||
return nil
|
||
}
|