feat 网关鉴权

Server/common/db/redis/client.go

@@ -1,34 +1,53 @@
 package redis
 
 import (
+	"common/config"
+	"common/log"
 	"context"
-	"fmt"
 	"github.com/redis/go-redis/v9"
+	"time"
 )
 
-var (
+var instance *Client
+
+type Client struct {
 	cli *redis.Client
-	Nil = redis.Nil
-)
+}
 
-func Init(host string, port int, password string, db int) error {
-	cli = redis.NewClient(&redis.Options{
-		Addr:     fmt.Sprintf("%s:%d", host, port),
-		Password: password,
-		DB:       db,
+func Init(cfg *config.RedisConfig) error {
+	client := redis.NewClient(&redis.Options{
+		Addr:     cfg.Addr,
+		Password: cfg.Password,
+		DB:       cfg.DB,
 	})
+	instance = &Client{
+		cli: client,
+	}
+	cacheInstance = &CacheClient{
+		cli:    client,
+		logger: log.GetLogger().Named("CACHE"),
+	}
 
-	_, err := cli.Ping(context.Background()).Result()
+	_, err := client.Ping(context.Background()).Result()
 	return err
 }
 
-func Client() *redis.Client {
-	return cli
-}
-
 func Close() error {
-	if cli != nil {
-		return cli.Close()
+	if instance != nil && instance.cli != nil {
+		return instance.cli.Close()
 	}
 	return nil
 }
+
+func GetClient() *Client {
+	return instance
+}
+
+func (c *Client) Set(ctx context.Context, key string, value interface{}, expiration time.Duration) *redis.StatusCmd {
+	return c.cli.Set(ctx, key, value, expiration)
+}
+
+// Get 获取数据
+func (c *Client) Get(ctx context.Context, key string) *redis.StringCmd {
+	return c.cli.Get(ctx, key)
+}