소스 검색

修改配置文件读取

longyu 2 년 전
부모
커밋
bfbeb0a01b
2개의 변경된 파일31개의 추가작업 그리고 4개의 파일을 삭제
  1. 22 0
      utils/common.go
  2. 9 4
      utils/config.go

+ 22 - 0
utils/common.go

@@ -11,6 +11,7 @@ import (
 	"image/png"
 	"math"
 	"math/rand"
+	"net"
 	"os"
 	"os/exec"
 	"regexp"
@@ -620,3 +621,24 @@ func Mp3Time(videoPlaySeconds string) string {
 	}
 	return timeStr
 }
+
+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
+}

+ 9 - 4
utils/config.go

@@ -2,7 +2,6 @@ package utils
 
 import (
 	"fmt"
-	beego "github.com/beego/beego/v2/adapter"
 	"github.com/beego/beego/v2/server/web"
 	"github.com/rdlucklib/rdluck_tools/cache"
 )
@@ -36,9 +35,15 @@ func init() {
 	}
 	RunMode = tmpRunMode
 	if RunMode == "" {
-		RunMode = "release"
+		localIp, err := GetLocalIP()
+		fmt.Println("localIp:", localIp)
+		if localIp == "10.0.0.123" {
+			RunMode = "debug"
+		} else {
+			RunMode = "release"
+		}
 		configPath := `/home/code/config/hongze_cygxzs/conf/app.conf`
-		err := web.LoadAppConfig("ini", configPath)
+		err = web.LoadAppConfig("ini", configPath)
 		if err != nil {
 			fmt.Println("web.LoadAppConfig Err:" + err.Error())
 		}
@@ -47,7 +52,7 @@ func init() {
 	if err != nil {
 		panic("配置文件读取错误 " + err.Error())
 	}
-	beego.Info(RunMode + " 模式")
+	fmt.Println(RunMode + " 模式")
 	MYSQL_URL = config["mysql_url"]
 	MYSQL_URL_RDDP = config["mysql_url_rddp"]
 	MYSQL_URL_TACTICS = config["mysql_url_tactics"]