fix: some api bug

This commit is contained in:
wwweww
2026-03-31 22:12:06 +08:00
parent c5ff4f0216
commit e7970ac25f
219 changed files with 16195 additions and 2126 deletions
+17 -12
View File
@@ -1,6 +1,7 @@
package svc
import (
stdsql "database/sql"
"juwan-backend/app/snowflake/rpc/snowflake"
"juwan-backend/app/users/rpc/internal/config"
"juwan-backend/app/users/rpc/internal/models"
@@ -11,30 +12,34 @@ import (
"time"
"ariga.io/entcache"
"entgo.io/ent/dialect"
"entgo.io/ent/dialect/sql"
_ "github.com/jackc/pgx/v5/stdlib"
"github.com/redis/go-redis/v9"
"github.com/zeromicro/go-zero/core/logx"
)
type ServiceContext struct {
Config config.Config
UsersModelRW *models.UsersClient
UsersModelRO *models.UsersClient
UsersModelRW *models.Client
UsersModelRO *models.Client
RedisCluster *redis.ClusterClient
Snowflake snowflake.SnowflakeServiceClient
JwtManager *utils.JwtManager
}
func NewServiceContext(c config.Config) *ServiceContext {
rawRW, err := stdsql.Open("pgx", c.DB.Master)
if err != nil {
panic(err)
}
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)
RWConn, err := sql.Open("pgx", c.DB.Master)
if err != nil {
panic(err)
}
ROConn, err := sql.Open("pgx", c.DB.Slave)
if err != nil {
panic(err)
}
logx.Infof("success to connect to postgres~")
// Initialize Redis Cluster client from CacheConf
@@ -59,8 +64,8 @@ func NewServiceContext(c config.Config) *ServiceContext {
RWDrv := entcache.NewDriver(RWConn, entcache.TTL(time.Second*30), entcache.Levels(adapter.NewRedisCache(redisCluster)))
return &ServiceContext{
Config: c,
UsersModelRW: models.NewClient(models.Driver(RWDrv)).Users,
UsersModelRO: models.NewClient(models.Driver(RODrv)).Users,
UsersModelRW: models.NewClient(models.Driver(RWDrv)),
UsersModelRO: models.NewClient(models.Driver(RODrv)),
RedisCluster: redisCluster,
JwtManager: jwtManager,
Snowflake: snowflakex.NewClient(c.SnowflakeRpcConf),