Bläddra i källkod

修改配置文件

longyu 2 år sedan
förälder
incheckning
2756ee8802
2 ändrade filer med 38 tillägg och 0 borttagningar
  1. 22 0
      utils/common.go
  2. 16 0
      utils/config.go

+ 22 - 0
utils/common.go

@@ -11,6 +11,7 @@ import (
 	"image/png"
 	"math"
 	"math/rand"
+	"net"
 	"os"
 	"os/exec"
 	"regexp"
@@ -641,3 +642,24 @@ func ArticleLastTime(strTime string) string {
 	}
 	return newTime
 }
+
+func GetLocalIP() (ip string, err error) {
+	addrs, err := net.InterfaceAddrs()
+	if err != nil {
+		return
+	}
+	for _, addr := range addrs {
+		ipAddr, ok := addr.(*net.IPNet)
+		if !ok {
+			continue
+		}
+		if ipAddr.IP.IsLoopback() {
+			continue
+		}
+		if !ipAddr.IP.IsGlobalUnicast() {
+			continue
+		}
+		return ipAddr.IP.String(), nil
+	}
+	return
+}

+ 16 - 0
utils/config.go

@@ -27,6 +27,22 @@ func init() {
 		panic("配置文件读取run_mode错误 " + err.Error())
 	}
 	RunMode = tmpRunMode
+	if RunMode == "" {
+		localIp, err := GetLocalIP()
+		fmt.Println("localIp:", localIp)
+		if localIp == "10.0.0.123" {
+			RunMode = "debug"
+		} else {
+			RunMode = "release"
+		}
+		fmt.Println("RunMode:", RunMode)
+		configPath := `/home/code/config/hongze_robot/conf/app.conf`
+		fmt.Println("configPath:", configPath)
+		err = web.LoadAppConfig("ini", configPath)
+		if err != nil {
+			fmt.Println("web.LoadAppConfig Err:" + err.Error())
+		}
+	}
 
 	config, err := web.AppConfig.GetSection(RunMode)
 	if err != nil {