Merge pull request #2 from oneclickvirt/copilot/fix-7ffd9cd7-8960-42d6-aaf2-a1832844b932

优先使用STREAM方法进行无root权限内存测试 (Prioritize STREAM method for no-root memory testing)

Author: spiritLHLS

.gitignore

@@ -1,5 +1,6 @@
 # Build artifacts
 memorytest
+main
 /cmd/cmd
 *.exe
 

README.md

@@ -8,20 +8,24 @@
 
 # 功能(Features)
 
-- [x] 支持使用```stream```进行高性能内存带宽测试 (最高优先级)
-- [x] 支持使用```dd```测试内存的读写IO (可选方式)
-- [x] 支持使用```sysbench```测试内存的顺序读写IO (兜底实现)
+- [x] 支持使用```stream```进行高性能内存带宽测试 (最高优先级，无需root权限)
+- [x] 支持使用```dd```测试内存的读写IO (需要root权限)
+- [x] 支持使用```sysbench```测试内存的顺序读写IO (需要root权限)
 - [x] 支持使用```winsat```测试内存的读写性能
 - [x] 支持Go自身静态依赖注入[dd](https://github.com/oneclickvirt/dd)，使用时无额外环境依赖需求
 - [x] 以```-l```指定输出的语言类型，可指定```zh```或```en```，默认不指定时使用中文输出
-- [x] 以```-m```指定测试的方法，可指定```stream```或```dd```或```sysbench```或```winsat```，默认不指定时按优先级自动选择测试方法
-- [x] 全平台编译支持，支持无权限测试时使用C重构或自编译的mbw程序模拟大内存COPY块测试内存性能
+- [x] 以```-m```指定测试的方法，可指定```stream```或```dd```或```sysbench```或```winsat```或```auto```，默认不指定时按优先级自动选择测试方法
+- [x] 全平台编译支持，支持无权限测试时优先尝试STREAM，若不可用再使用C重构或自编译的mbw程序模拟大内存COPY块测试内存性能
 
 ## 测试方法优先级
 当不指定`-m`参数时，程序按以下优先级自动选择测试方法：
-1. **STREAM** - 如果检测到stream二进制文件，优先使用
-2. **DD** - 如果STREAM不可用，使用DD测试
-3. **Sysbench** - 作为最终的兜底实现
+1. **STREAM** - 如果检测到stream二进制文件，优先使用(无需root权限)
+2. **DD** - 如果STREAM不可用，使用DD测试(需要root权限)
+3. **Sysbench** - 作为最终的兜底实现(需要root权限)
+
+**无root权限时的特殊处理：**
+- 当检测到系统无root权限时，DD和Sysbench测试会自动优先尝试STREAM方式
+- 如果STREAM不可用，则回退使用mbw程序进行内存测试
 
 注意：默认不自动安装```sysbench```组件，如需使用请自行安装后再使用本项目，如```apt update && apt install sysbench -y```
 

cmd/main.go

@@ -23,7 +23,7 @@ func main() {
 	memorytestFlag.BoolVar(&help, "h", false, "Show help information")
 	memorytestFlag.BoolVar(&showVersion, "v", false, "show version")
 	memorytestFlag.StringVar(&language, "l", "", "Language parameter (en or zh)")
-	memorytestFlag.StringVar(&testMethod, "m", "", "Specific Test Method (stream or dd or sysbench or winsat)")
+	memorytestFlag.StringVar(&testMethod, "m", "", "Specific Test Method (stream, dd, sysbench, winsat, or auto)")
 	memorytestFlag.BoolVar(&memory.EnableLoger, "log", false, "Enable logging")
 	memorytestFlag.Parse(os.Args[1:])
 	if help {

memory/macos_linux_mem.go

@@ -57,6 +57,11 @@ func SysBenchTest(language string) string {
 		} else {
 			fmt.Println("当前检测到系统无root权限")
 		}
+		// Try STREAM first when no root permission, fallback to mbw if not available
+		streamResult := StreamTest(language)
+		if streamResult != "" && strings.TrimSpace(streamResult) != "" {
+			return streamResult
+		}
 		return simpleMemoryTest(language)
 	}
 	var result string
@@ -366,6 +371,11 @@ func DDTest(language string) string {
 		} else {
 			fmt.Println("当前检测到系统无root权限")
 		}
+		// Try STREAM first when no root permission, fallback to mbw if not available
+		streamResult := StreamTest(language)
+		if streamResult != "" && strings.TrimSpace(streamResult) != "" {
+			return streamResult
+		}
 		return simpleMemoryTest(language)
 	}
 	// 获取内存文件系统目录