From 02a2176f7c5733a4e4c4429c2028bbb86a967ce7 Mon Sep 17 00:00:00 2001
From: jiangshuai <291802688@qq.com>
Date: 星期二, 06 二月 2024 10:13:38 +0800
Subject: [PATCH] Merge branch 'master' of http://192.168.5.5:10010/r/aps/SRM

---
 core/viper.go |   47 +++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 43 insertions(+), 4 deletions(-)

diff --git a/core/viper.go b/core/viper.go
index e3dd24c..f797dd9 100644
--- a/core/viper.go
+++ b/core/viper.go
@@ -4,15 +4,14 @@
 	"flag"
 	"fmt"
 	"github.com/gin-gonic/gin"
+	"log"
 	"os"
-	"path/filepath"
 	"srm/core/internal"
 
 	"github.com/fsnotify/fsnotify"
 	"github.com/spf13/viper"
 
 	"srm/global"
-	_ "srm/packfile"
 )
 
 // Viper //
@@ -68,7 +67,47 @@
 		fmt.Println(err)
 	}
 
-	// root 閫傞厤鎬� 鏍规嵁root浣嶇疆鍘绘壘鍒板搴旇縼绉讳綅缃�,淇濊瘉root璺緞鏈夋晥
-	global.GVA_CONFIG.AutoCode.Root, _ = filepath.Abs("..")
+	ConfigFromEnv()
 	return v
 }
+
+func ConfigFromEnv() {
+	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 {
+		global.GVA_CONFIG.Mysql.Path = DBHost
+		global.GVA_CONFIG.Mysql.Port = DBPort
+		global.GVA_CONFIG.Mysql.Dbname = DBName
+		global.GVA_CONFIG.Mysql.Username = DBUser
+		global.GVA_CONFIG.Mysql.Password = DBPasswd
+	}
+	log.Printf("mysql dsn:%v", global.GVA_CONFIG.Mysql.Dsn())
+
+	grpcPort := os.Getenv("GRPC_PORT")
+	if grpcPort != "" {
+		global.GVA_CONFIG.System.GrpcPort = grpcPort
+	}
+
+	apsGrpc := os.Getenv("APS_GRPC")
+	if apsGrpc != "" {
+		global.GVA_CONFIG.System.GrpcUrl = apsGrpc
+	}
+
+	wmsGrpc := os.Getenv("WMS_GRPC")
+	if wmsGrpc != "" {
+		global.GVA_CONFIG.System.GrpcWmsUrl = wmsGrpc
+	}
+
+	adminGrpc := os.Getenv("ADMIN_GRPC")
+	if adminGrpc != "" {
+		global.GVA_CONFIG.System.GrpcAdminUrl = adminGrpc
+	}
+	log.Printf("   System:                %+v", global.GVA_CONFIG.System)
+}

--
Gitblit v1.8.0