update

go.mod

@@ -11,6 +11,7 @@ require (
 	github.com/oneclickvirt/cputest v0.0.7-20240701020012
 	github.com/oneclickvirt/defaultset v0.0.2-20240624082446
 	github.com/oneclickvirt/disktest v0.0.3-20240629152513
+	github.com/oneclickvirt/gostun v0.0.2-20240625025941
 	github.com/oneclickvirt/memorytest v0.0.1-20240624151629
 	github.com/oneclickvirt/nt3 v0.0.2-20240630152642
 	github.com/oneclickvirt/portchecker v0.0.1-20240624155429
@@ -47,7 +48,6 @@ require (
 	github.com/mitchellh/colorstring v0.0.0-20190213212951-d06e56a500db // indirect
 	github.com/mitchellh/mapstructure v1.5.0 // indirect
 	github.com/nxtrace/NTrace-core v1.3.1 // indirect
-	github.com/oneclickvirt/gostun v0.0.2-20240625025941 // indirect
 	github.com/onsi/ginkgo/v2 v2.19.0 // indirect
 	github.com/oschwald/maxminddb-golang v1.12.0 // indirect
 	github.com/pelletier/go-toml/v2 v2.2.2 // indirect

go.sum

@@ -83,8 +83,6 @@ github.com/oneclickvirt/backtrace v0.0.4-20240624090335 h1:0LP5KyA6GLlqQAtOF0uyB
 github.com/oneclickvirt/backtrace v0.0.4-20240624090335/go.mod h1:zvsC7xY/WZqs5KL2JB967OVnuqjNbxu9bW6wXRLo5h8=
 github.com/oneclickvirt/basics v0.0.3-20240625075226 h1:K9VriCHIYnXPZXBSn9PRQX+jBS6AIFH8tBVb/i8VGAw=
 github.com/oneclickvirt/basics v0.0.3-20240625075226/go.mod h1:dTB+/oyFQYfTYX55rFJVWatum5F9g62zjfmHCM6Vj1s=
-github.com/oneclickvirt/cputest v0.0.6-20240630144058 h1:bFOM4MS+uaU7slFaZR91/bJ57AFM23RWPEik+GTFS2w=
-github.com/oneclickvirt/cputest v0.0.6-20240630144058/go.mod h1:MmaHN9+XMntI3rLycwj8Ne31fG18IfNoa8N2utDK1CY=
 github.com/oneclickvirt/cputest v0.0.7-20240701020012 h1:U5cCI+6ZU3pudoAlmb1b3yB9IQNm5AnDXi1TQAZJCIA=
 github.com/oneclickvirt/cputest v0.0.7-20240701020012/go.mod h1:MmaHN9+XMntI3rLycwj8Ne31fG18IfNoa8N2utDK1CY=
 github.com/oneclickvirt/defaultset v0.0.2-20240624082446 h1:5Pg3mK/u/vQvSz7anu0nxzrNdELi/AcDAU1mMsmPzyc=

goecs.go

@@ -1,7 +1,6 @@
 package main
 
 import (
-	"bufio"
 	"flag"
 	"fmt"
 	"github.com/oneclickvirt/CommonMediaTests/commediatests"
@@ -19,7 +18,6 @@ import (
 	gostunmodel "github.com/oneclickvirt/gostun/model"
 	"github.com/oneclickvirt/portchecker/email"
 	speedtestmodel "github.com/oneclickvirt/speedtest/model"
-	"os"
 	"regexp"
 	"runtime"
 	"strings"
@@ -30,7 +28,7 @@ import (
 var (
 	ecsVersion                                                        = "2024.07.01.1"
 	menuMode                                                          bool
-	choice                                                            string
+	input, choice                                                     string
 	showVersion                                                       bool
 	enableLogger                                                      bool
 	language                                                          string
@@ -87,9 +85,6 @@ func main() {
 		basicStatus, cpuTestStatus, memoryTestStatus, diskTestStatus = false, false, false, false
 		commTestStatus, utTestStatus, securityTestStatus, emailTestStatus = false, false, false, false
 		backtraceStatus, nt3Status, speedTestStatus = false, false, false
-		// 正则表达式匹配纯数字
-		re := regexp.MustCompile(`^\d+$`)
-		reader := bufio.NewReader(os.Stdin)
 		switch language {
 		case "zh":
 			fmt.Println("1. 融合怪完全体")
@@ -113,10 +108,13 @@ func main() {
 			fmt.Println("8. Hardware-Only Test (Basic System Info + CPU + Memory + dd Disk Test + fio Disk Test)")
 			fmt.Println("9. IP Quality Test (IP Test with 15 Databases + Email Port Test)")
 		}
+	Loop:
 		for {
 			fmt.Print("请输入选项 / Please enter your choice: ")
-			input, _ := reader.ReadString('\n')
+			fmt.Scanln(&input)
 			input = strings.TrimSpace(input)
+			input = strings.TrimRight(input, "\n")
+			re := regexp.MustCompile(`^\d+$`) // 正则表达式匹配纯数字
 			if re.MatchString(input) {
 				choice = input
 				switch choice {
@@ -132,14 +130,14 @@ func main() {
 					backtraceStatus = true
 					nt3Status = true
 					speedTestStatus = true
-					break
+					break Loop
 				case "2":
 					basicStatus = true
 					cpuTestStatus = true
 					memoryTestStatus = true
 					diskTestStatus = true
 					speedTestStatus = true
-					break
+					break Loop
 				case "3":
 					basicStatus = true
 					cpuTestStatus = true
@@ -151,7 +149,7 @@ func main() {
 					backtraceStatus = true
 					nt3Status = true
 					speedTestStatus = true
-					break
+					break Loop
 				case "4":
 					basicStatus = true
 					cpuTestStatus = true
@@ -160,7 +158,7 @@ func main() {
 					backtraceStatus = true
 					nt3Status = true
 					speedTestStatus = true
-					break
+					break Loop
 				case "5":
 					basicStatus = true
 					cpuTestStatus = true
@@ -168,30 +166,30 @@ func main() {
 					diskTestStatus = true
 					securityTestStatus = true
 					speedTestStatus = true
-					break
+					break Loop
 				case "6":
 					speedTestStatus = true
 					backtraceStatus = true
 					nt3Status = true
-					break
+					break Loop
 				case "7":
 					securityTestStatus = true
 					commTestStatus = true
-					break
+					break Loop
 				case "8":
 					basicStatus = true
 					cpuTestStatus = true
 					memoryTestStatus = true
 					diskTestStatus = true
-					break
+					break Loop
 				case "9":
 					emailTestStatus = true
-					break
+					break Loop
 				case "10":
 					backtraceStatus = true
 					nt3Status = true
 					speedTestStatus = true
-					break
+					break Loop
 				default:
 					if language == "zh" {
 						fmt.Println("无效的选项")