玩家管理模块接入redis，已完整实现好友、多人世界的跨服交互功能

gs/dao/dao.go

@@ -6,31 +6,55 @@ import (
 	"hk4e/common/config"
 	"hk4e/pkg/logger"
 
+	"github.com/go-redis/redis/v8"
 	"go.mongodb.org/mongo-driver/mongo"
 	"go.mongodb.org/mongo-driver/mongo/options"
+	"go.mongodb.org/mongo-driver/mongo/readpref"
 )
 
 type Dao struct {
-	client *mongo.Client
-	db     *mongo.Database
+	mongo *mongo.Client
+	db    *mongo.Database
+	redis *redis.Client
 }
 
 func NewDao() (r *Dao, err error) {
 	r = new(Dao)
-	clientOptions := options.Client().ApplyURI(config.CONF.Database.Url)
+	clientOptions := options.Client().ApplyURI(config.CONF.Database.Url).SetMinPoolSize(1).SetMaxPoolSize(10)
 	client, err := mongo.Connect(context.TODO(), clientOptions)
 	if err != nil {
 		logger.Error("mongo connect error: %v", err)
 		return nil, err
 	}
-	r.client = client
+	err = client.Ping(context.TODO(), readpref.Primary())
+	if err != nil {
+		logger.Error("mongo ping error: %v", err)
+		return nil, err
+	}
+	r.mongo = client
 	r.db = client.Database("gs_hk4e")
+	r.redis = redis.NewClient(&redis.Options{
+		Addr:         config.CONF.Redis.Addr,
+		Password:     config.CONF.Redis.Password,
+		DB:           0,
+		PoolSize:     10,
+		MinIdleConns: 1,
+	})
+	err = r.redis.Ping(context.TODO()).Err()
+	if err != nil {
+		logger.Error("redis ping error: %v", err)
+		return nil, err
+	}
 	return r, nil
 }
 
 func (d *Dao) CloseDao() {
-	err := d.client.Disconnect(context.TODO())
+	err := d.mongo.Disconnect(context.TODO())
 	if err != nil {
 		logger.Error("mongo close error: %v", err)
 	}
+	err = d.redis.Close()
+	if err != nil {
+		logger.Error("redis close error: %v", err)
+	}
 }

gs/dao/player_redis.go

@@ -0,0 +1,53 @@
+package dao
+
+import (
+	"context"
+	"strconv"
+	"time"
+
+	"hk4e/gs/model"
+	"hk4e/pkg/logger"
+
+	"github.com/vmihailenco/msgpack/v5"
+)
+
+const RedisPlayerKeyPrefix = "HK4E"
+
+func (d *Dao) GetRedisPlayerKey(userId uint32) string {
+	return RedisPlayerKeyPrefix + ":USER:" + strconv.Itoa(int(userId))
+}
+
+func (d *Dao) GetRedisPlayer(userId uint32) *model.Player {
+	playerData, err := d.redis.Get(context.TODO(), d.GetRedisPlayerKey(userId)).Result()
+	if err != nil {
+		logger.Error("get player from redis error: %v", err)
+		return nil
+	}
+	player := new(model.Player)
+	err = msgpack.Unmarshal([]byte(playerData), player)
+	if err != nil {
+		logger.Error("unmarshal player error: %v", err)
+		return nil
+	}
+	return player
+}
+
+func (d *Dao) SetRedisPlayer(player *model.Player) {
+	playerData, err := msgpack.Marshal(player)
+	if err != nil {
+		logger.Error("marshal player error: %v", err)
+		return
+	}
+	err = d.redis.Set(context.TODO(), d.GetRedisPlayerKey(player.PlayerID), playerData, time.Hour*24*30).Err()
+	if err != nil {
+		logger.Error("set player from redis error: %v", err)
+		return
+	}
+}
+
+func (d *Dao) SetRedisPlayerList(playerList []*model.Player) {
+	// TODO 换成redis批量命令执行
+	for _, player := range playerList {
+		d.SetRedisPlayer(player)
+	}
+}