fix: some api bug
This commit is contained in:
@@ -1,20 +1,24 @@
|
||||
package svc
|
||||
|
||||
import (
|
||||
stdsql "database/sql"
|
||||
"juwan-backend/app/snowflake/rpc/snowflake"
|
||||
"juwan-backend/app/user_verifications/rpc/internal/config"
|
||||
"juwan-backend/app/user_verifications/rpc/internal/models"
|
||||
"juwan-backend/app/user_verifications/rpc/userverifications"
|
||||
"juwan-backend/app/users/rpc/usercenter"
|
||||
"juwan-backend/common/redisx"
|
||||
"juwan-backend/common/snowflakex"
|
||||
"juwan-backend/pkg/adapter"
|
||||
"time"
|
||||
|
||||
"entgo.io/ent/dialect"
|
||||
_ "github.com/jackc/pgx/v5/stdlib"
|
||||
"github.com/zeromicro/go-zero/zrpc"
|
||||
|
||||
"ariga.io/entcache"
|
||||
"entgo.io/ent/dialect/sql"
|
||||
"github.com/redis/go-redis/v9"
|
||||
"github.com/zeromicro/go-zero/core/logx"
|
||||
"github.com/zeromicro/go-zero/zrpc"
|
||||
)
|
||||
|
||||
type ServiceContext struct {
|
||||
@@ -22,19 +26,23 @@ type ServiceContext struct {
|
||||
UserVeriModelRW *models.UserVerificationsClient
|
||||
UserVeriModelRO *models.UserVerificationsClient
|
||||
RedisClient *redis.ClusterClient
|
||||
UserVeriRpc userverifications.UserVerificationsZrpcClient
|
||||
SnowflakeRpc snowflake.SnowflakeServiceClient
|
||||
UserRpc usercenter.Usercenter
|
||||
}
|
||||
|
||||
func NewServiceContext(c config.Config) *ServiceContext {
|
||||
RWConn, err := sql.Open("pgx", c.DB.Master)
|
||||
rawRW, err := stdsql.Open("pgx", c.DB.Master)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
ROConn, err := sql.Open("pgx", c.DB.Slave)
|
||||
rawRO, err := stdsql.Open("pgx", c.DB.Slave)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
RWConn := sql.OpenDB(dialect.Postgres, rawRW)
|
||||
ROConn := sql.OpenDB(dialect.Postgres, rawRO)
|
||||
|
||||
logx.Infof("success to connect to postgres~")
|
||||
|
||||
redisConn, err := redisx.ConnectMasterSlaveCluster(c.CacheConf, 5*time.Second)
|
||||
if err != nil || redisConn == nil {
|
||||
@@ -49,7 +57,7 @@ func NewServiceContext(c config.Config) *ServiceContext {
|
||||
UserVeriModelRW: models.NewClient(models.Driver(RWDrv)).UserVerifications,
|
||||
UserVeriModelRO: models.NewClient(models.Driver(RODrv)).UserVerifications,
|
||||
RedisClient: redisConn.Client,
|
||||
UserVeriRpc: userverifications.NewUserVerificationsZrpcClient(zrpc.MustNewClient(c.UserVeriRpcConf)),
|
||||
SnowflakeRpc: snowflakex.NewClient(c.SnowflakeRpcConf),
|
||||
UserRpc: usercenter.NewUsercenter(zrpc.MustNewClient(c.UserRpcConf)),
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user