add: anowflake email kafka, refa: redis connectg

app/users/rpc/internal/svc/serviceContext.go

@@ -1,10 +1,12 @@
 package svc
 
 import (
-	"context"
+	"juwan-backend/app/snowflake/rpc/snowflake"
 	"juwan-backend/app/users/rpc/internal/config"
 	"juwan-backend/app/users/rpc/internal/models"
 	"juwan-backend/app/users/rpc/internal/utils"
+	"juwan-backend/common/redisx"
+	"juwan-backend/common/snowflakex"
 	"time"
 
 	"github.com/redis/go-redis/v9"
@@ -14,30 +16,30 @@ import (
 
 type ServiceContext struct {
 	Config       config.Config
-	UsersModel   models.UsersModel
+	UsersModelRW models.UsersModel
+	UsersModelRO models.UsersModel
 	RedisCluster *redis.ClusterClient
+	Snowflake    snowflake.SnowflakeServiceClient
 	JwtManager   *utils.JwtManager
 }
 
 func NewServiceContext(c config.Config) *ServiceContext {
-	conn := sqlx.NewSqlConn("postgres", c.DataSource)
+	RWDBConn := sqlx.NewSqlConn("postgres", c.DB.Master)
+	RODBConn := sqlx.NewSqlConn("postgres", c.DB.Slave)
 	logx.Infof("success to connect to postgres~")
 
 	// Initialize Redis Cluster client from CacheConf
-	var redisCluster *redis.ClusterClient
-	if len(c.CacheConf) > 0 {
-		redisCluster = redis.NewClusterClient(&redis.ClusterOptions{
-			Addrs:    []string{c.CacheConf[0].Host},
-			Password: c.CacheConf[0].Pass,
-		})
-
-		// Test Redis Cluster connection
-		ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
-		defer cancel()
-		if err := redisCluster.Ping(ctx).Err(); err != nil {
+	redisConn, err := redisx.ConnectMasterSlaveCluster(c.CacheConf, 5*time.Second)
+	redisCluster := redisConn.Client
+	if redisCluster != nil {
+		if err != nil {
 			logx.Errorf("failed to connect to redis cluster: %v", err)
 		} else {
-			logx.Infof("success to connect to redis cluster~")
+			if redisConn.HasSlave {
+				logx.Infof("success to connect to redis master/slave (M: %s, S: %s)", redisConn.MasterHost, redisConn.SlaveHost)
+			} else {
+				logx.Infof("success to connect to redis master (M: %s), slave not configured", redisConn.MasterHost)
+			}
 		}
 	}
 
@@ -46,8 +48,10 @@ func NewServiceContext(c config.Config) *ServiceContext {
 
 	return &ServiceContext{
 		Config:       c,
-		UsersModel:   models.NewUsersModel(conn, c.CacheConf),
+		UsersModelRW: models.NewUsersModel(RWDBConn, c.CacheConf),
+		UsersModelRO: models.NewUsersModel(RODBConn, c.CacheConf),
 		RedisCluster: redisCluster,
 		JwtManager:   jwtManager,
+		Snowflake:    snowflakex.NewClient(c.SnowflakeRpcConf),
 	}
 }