62 lines
1.7 KiB
Go
62 lines
1.7 KiB
Go
|
package data
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"github.com/redis/go-redis/v9"
|
||
|
|
)
|
||
|
|
|
||
|
|
type RedisClientConnector interface {
|
||
|
|
Ping(ctx context.Context) *redis.StatusCmd
|
||
|
|
LRange(ctx context.Context, key string, start, stop int64) *redis.StringSliceCmd
|
||
|
|
SInter(ctx context.Context, keys ...string) *redis.StringSliceCmd
|
||
|
|
Keys(ctx context.Context, pattern string) *redis.StringSliceCmd
|
||
|
|
SAdd(ctx context.Context, key string, members ...interface{}) *redis.IntCmd
|
||
|
|
SIsMember(ctx context.Context, key string, member interface{}) *redis.BoolCmd
|
||
|
|
}
|
||
|
|
|
||
|
|
type Redis struct {
|
||
|
|
client RedisClientConnector
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewRedisConnector(client RedisClientConnector) *Redis {
|
||
|
|
if client == nil {
|
||
|
|
client = redis.NewClient(&redis.Options{
|
||
|
|
Addr: "localhost:6379",
|
||
|
|
Password: "", // no password set
|
||
|
|
DB: 0, // use default DB
|
||
|
|
})
|
||
|
|
}
|
||
|
|
return &Redis{ client: client }
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Redis) Tags(context context.Context) []string {
|
||
|
|
result,_ := r.client.Keys(context, "*").Result()
|
||
|
|
return result
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Redis) Connected(context context.Context) bool {
|
||
|
|
if e := r.client.Ping(context).Err(); e != nil {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Redis) Query(context context.Context, terms []string) ([]string, error) {
|
||
|
|
result,_ := r.client.SInter(context, terms...).Result()
|
||
|
|
return result, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Redis) AddTag(context context.Context, key string, resource string) error {
|
||
|
|
if r,e := r.client.SAdd(context, key, resource).Result(); r != 1 || e != nil {
|
||
|
|
return e
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Redis) ResourceHasTag(context context.Context, resource string, tag string) bool {
|
||
|
|
if r,e := r.client.SIsMember(context, tag, resource).Result(); r == false || e != nil {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
return true
|
||
|
|
}
|