feat 初次提交

db/redis/cache_client.go

@@ -0,0 +1,50 @@
+package redis
+
+import (
+	"context"
+	"encoding/json"
+	"github.com/panjf2000/gnet/v2/pkg/logging"
+	"github.com/redis/go-redis/v9"
+	"time"
+)
+
+var cacheInstance *CacheClient
+
+type CacheClient struct {
+	cli    *redis.Client
+	logger logging.Logger
+}
+
+func GetCacheClient() *CacheClient {
+	return cacheInstance
+}
+
+func (c *CacheClient) Set(ctx context.Context, key string, value interface{}, expiration time.Duration) {
+	bytes, err := json.Marshal(value)
+	if err != nil {
+		c.logger.Errorf("Set json.Marshal error: %v, key: %v, value: %v", err, key, value)
+		return
+	}
+	if err = c.cli.Set(ctx, key, bytes, expiration).Err(); err != nil {
+		c.logger.Errorf("Set redis.Set error: %v, key: %v, value: %v", err, key, value)
+	}
+}
+
+func (c *CacheClient) Del(ctx context.Context, keys ...string) {
+	if err := c.cli.Del(ctx, keys...).Err(); err != nil {
+		c.logger.Errorf("Set redis.Del error: %v, keys: %v", err, keys)
+	}
+}
+
+// Get 获取数据
+func (c *CacheClient) Get(ctx context.Context, key string, dst interface{}) bool {
+	data, err := c.cli.Get(ctx, key).Bytes()
+	if err != nil {
+		return false
+	}
+	if err = json.Unmarshal(data, dst); err != nil {
+		c.logger.Errorf("Get json.Unmarshal error: %v, key: %v", err, key)
+		return false
+	}
+	return true
+}

db/redis/client.go

@@ -0,0 +1,53 @@
+package redis
+
+import (
+	"common/config"
+	"common/log"
+	"context"
+	"github.com/redis/go-redis/v9"
+	"time"
+)
+
+var instance *Client
+
+type Client struct {
+	cli *redis.Client
+}
+
+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 := client.Ping(context.Background()).Result()
+	return err
+}
+
+func Close() error {
+	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)
+}