68 lines
1.1 KiB
Go
68 lines
1.1 KiB
Go
package cache
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/go-redis/redis/v8"
|
|
"github.com/logrusorgru/aurora"
|
|
)
|
|
|
|
var (
|
|
c *redis.Client
|
|
)
|
|
|
|
// Init ...
|
|
func Init(uri, pwd string) {
|
|
c = redis.NewClient(&redis.Options{
|
|
Addr: uri,
|
|
Password: pwd,
|
|
DB: 0, // use default DB
|
|
})
|
|
|
|
// Test
|
|
_, err := c.Ping(context.Background()).Result()
|
|
|
|
if err != nil {
|
|
log.Fatal("Cannot connect to redis", uri, err)
|
|
}
|
|
|
|
fmt.Println(aurora.Green("*** CONNECTED TO REDIS: " + uri))
|
|
|
|
// Cache roles
|
|
Roles()
|
|
}
|
|
|
|
// GetInstance ...
|
|
func GetInstance() *redis.Client {
|
|
return c
|
|
}
|
|
|
|
// SetKeyValue ...
|
|
func SetKeyValue(key string, value interface{}, expiration time.Duration) error {
|
|
ctx := context.Background()
|
|
dataByte, _ := json.Marshal(value)
|
|
r := c.Set(ctx, key, dataByte, expiration)
|
|
return r.Err()
|
|
}
|
|
|
|
// GetValueByKey ...
|
|
func GetValueByKey(key string) ([]byte, error) {
|
|
ctx := context.Background()
|
|
return c.Get(ctx, key).Bytes()
|
|
}
|
|
|
|
func GetJSON(key string, result interface{}) bool {
|
|
v, err := GetValueByKey(key)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
if err = json.Unmarshal(v, result); err != nil {
|
|
return false
|
|
}
|
|
return true
|
|
}
|