This commit is contained in:
liangzai 2024-10-15 17:26:40 +08:00
parent c76451788c
commit 5a71745c97

View File

@ -6,9 +6,9 @@ import (
"os"
"tianyuan-api/apps/sentinel/internal/config"
TopUpServer "tianyuan-api/apps/sentinel/internal/server/TopUp"
productServer "tianyuan-api/apps/sentinel/internal/server/product"
secretServer "tianyuan-api/apps/sentinel/internal/server/secret"
topupServer "tianyuan-api/apps/sentinel/internal/server/topup"
userproductServer "tianyuan-api/apps/sentinel/internal/server/userproduct"
whitelistServer "tianyuan-api/apps/sentinel/internal/server/whitelist"
"tianyuan-api/apps/sentinel/internal/svc"
@ -22,7 +22,6 @@ import (
)
func main() {
// 读取环境变量 ENV默认为 "prod"
env := os.Getenv("ENV")
if env == "" {
env = "production"
@ -49,7 +48,7 @@ func main() {
sentinel.RegisterSecretServer(grpcServer, secretServer.NewSecretServer(ctx))
sentinel.RegisterProductServer(grpcServer, productServer.NewProductServer(ctx))
sentinel.RegisterUserProductServer(grpcServer, userproductServer.NewUserProductServer(ctx))
sentinel.RegisterTopUpServer(grpcServer, TopUpServer.NewTopUpServer(ctx))
sentinel.RegisterTopUpServer(grpcServer, topupServer.NewTopUpServer(ctx))
if c.Mode == service.DevMode || c.Mode == service.TestMode {
reflection.Register(grpcServer)