From 21b770ba8bd32d476c45695f3ad5c2920c14fd2b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E7=A5=A5=E8=B6=85?= Date: Mon, 13 Mar 2023 21:49:41 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E8=BF=90=E8=A1=8C=E6=B5=8B?= =?UTF-8?q?=E8=AF=95=E7=94=A8=E4=BE=8B=E6=97=B6init()=E6=97=A0=E6=B3=95?= =?UTF-8?q?=E8=B5=B7=E4=BD=9C=E7=94=A8=E7=9A=84Bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/const/vars.go | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/internal/const/vars.go b/internal/const/vars.go index a562128..32b4fe3 100644 --- a/internal/const/vars.go +++ b/internal/const/vars.go @@ -5,6 +5,7 @@ package teaconst import ( "github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs" "os" + "strings" ) var ( @@ -15,7 +16,7 @@ var ( NodeId int64 = 0 NodeIdString = "" - IsMain = len(os.Args) == 1 || (len(os.Args) >= 2 && os.Args[1] == "pprof") + IsMain = checkMain() GlobalProductName = nodeconfigs.DefaultProductName @@ -24,3 +25,15 @@ var ( DiskIsFast = false // 是否为高速硬盘 ) + +// 检查是否为主程序 +func checkMain() bool { + if len(os.Args) == 1 || + (len(os.Args) >= 2 && os.Args[1] == "pprof") { + return true + } + exe, _ := os.Executable() + return strings.HasSuffix(exe, ".test") || + strings.HasSuffix(exe, ".test.exe") || + strings.Contains(exe, "___") +}