From 115bd9b51f5d8eade4658f844de37516486c60e7 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期六, 18 十一月 2023 17:25:25 +0800
Subject: [PATCH] crm获取aps项目模块信息

---
 conf/config.go |   43 ++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 42 insertions(+), 1 deletions(-)

diff --git a/conf/config.go b/conf/config.go
index 36a1dcd..96fe2b2 100644
--- a/conf/config.go
+++ b/conf/config.go
@@ -8,6 +8,7 @@
 	"flag"
 	"github.com/spf13/viper"
 	"log"
+	"os"
 )
 
 var (
@@ -48,6 +49,7 @@
 	System struct {
 		Env           string // 鐜鍊� develop test public
 		Port          int    // 绔彛
+		GrpcPort      string //grpc绔彛
 		DbType        string // 鏁版嵁搴撶被鍨�
 		UseMultipoint bool   // 澶氱偣鐧诲綍鎷︽埅
 		UseRedis      bool   // 浣跨敤redis
@@ -58,7 +60,9 @@
 	}
 
 	GrpcServiceAddr struct {
-		Aps string // jwt绛惧悕
+		Aps   string // aps鏈嶅姟鍦板潃
+		Admin string // admin鏈嶅姟鍦板潃
+		WMS   string //wms鏈嶅姟鍦板潃
 	}
 
 	config struct {
@@ -111,7 +115,38 @@
 	if err := Viper.Unmarshal(&Conf); err != nil {
 		log.Fatalf("Unmarshal err:%v", err)
 	}
+
+	DBHost := os.Getenv("DB_HOST")
+	DBName := os.Getenv("DB_NAME")
+	DBPort := os.Getenv("DB_PORT")
+	DBUser := os.Getenv("DB_USER")
+	DBPasswd := os.Getenv("DB_PASSWD")
+	if len(DBHost) > 0 &&
+		len(DBName) > 0 &&
+		len(DBPort) > 0 &&
+		len(DBUser) > 0 &&
+		len(DBPasswd) > 0 {
+		Conf.Mysql.Dsn = DBUser + ":" + DBPasswd + "@tcp(" + DBHost + ":" + DBPort + ")/" + DBName + "?charset=utf8&parseTime=True&loc=Local"
+	}
+
+	AdminGrpc := os.Getenv("ADMIN_GRPC")
+	if len(AdminGrpc) > 0 {
+		Conf.GrpcServiceAddr.Admin = AdminGrpc
+	}
+	GrpcPort := os.Getenv("WMS_GRPC")
+	if len(GrpcPort) > 0 {
+		Conf.GrpcServiceAddr.WMS = GrpcPort
+	}
+	ApsGrpc := os.Getenv("GRPC_PORT")
+	Host := os.Getenv("HOST")
+
+	if len(Host) > 0 && len(ApsGrpc) > 0 {
+		Conf.GrpcServiceAddr.Aps = Host + ":" + ApsGrpc
+		//Conf.GrpcServiceAddr.Aps = DBUser + DBUser +
+	}
+
 	ShowConfig()
+	CheckConfig()
 }
 
 // WriteConfig 鍥炲啓閰嶇疆
@@ -133,3 +168,9 @@
 	log.Printf("   GrpcServiceAddr:       %+v", Conf.GrpcServiceAddr)
 	log.Println("......................................................")
 }
+
+func CheckConfig() {
+	if Conf.GrpcServiceAddr.Aps == "" {
+		log.Fatalf("%v 灏氭湭閰嶇疆", "GrpcServiceAddr.Aps")
+	}
+}

--
Gitblit v1.8.0