56 lines
1.1 KiB
Go
56 lines
1.1 KiB
Go
package cache
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
|
|
"github.com/chaitin/panda-wiki/domain"
|
|
"github.com/chaitin/panda-wiki/store/cache"
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
type KBRepo struct {
|
|
cache *cache.Cache
|
|
}
|
|
|
|
func NewKBRepo(cache *cache.Cache) *KBRepo {
|
|
return &KBRepo{cache: cache}
|
|
}
|
|
|
|
func (r *KBRepo) GetKB(ctx context.Context, kbID string) (*domain.KnowledgeBase, error) {
|
|
kbStr, err := r.cache.Get(ctx, kbID).Result()
|
|
if err != nil {
|
|
if errors.Is(err, redis.Nil) {
|
|
return nil, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
if kbStr == "" {
|
|
return nil, nil
|
|
}
|
|
|
|
var kb domain.KnowledgeBase
|
|
err = json.Unmarshal([]byte(kbStr), &kb)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &kb, nil
|
|
}
|
|
|
|
func (r *KBRepo) SetKB(ctx context.Context, kbID string, kb *domain.KnowledgeBase) error {
|
|
kbStr, err := json.Marshal(kb)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return r.cache.Set(ctx, kbID, kbStr, 0).Err()
|
|
}
|
|
|
|
func (r *KBRepo) DeleteKB(ctx context.Context, kbID string) error {
|
|
return r.cache.Del(ctx, kbID).Err()
|
|
}
|
|
|
|
func (r *KBRepo) ClearSession(ctx context.Context) error {
|
|
return r.cache.DeleteKeysWithPrefix(ctx, "session_")
|
|
}
|