使用Go语言实现benchmark解析器
目录
- 简介
- 仅Go解析器
- 使用 Web UI 可视化数据
- 功能需求
- 项目结构设计(Go + html + js)
- Go 服务器(main.go)
- Web 页面(static/index.html)
- Chart 渲染脚本(static/main.js)
- 运行方式
- benchmark.txt文件示例
简介
用 Go
编写的 benchmark
输出解析器,功能如下
- 读取
go test -bench=. -benchmem
的输出文件(如benchmark.txt
) - 解析出每行数据
- 写入成
CSV
文件(如benchmark.csv
) Web UI
可视化数据
仅Go解析器
package main import编程客栈 ( "bufio" "encoding/csv" "fmt" "os" "regexp" ) type BenchmarkResult struct { Name string Runs string NsPerOp string BytesPerOp string AllocsPerOp string } func main() { inputFile := "benchmark.txt" outputFile := "benchmark.csv" file, err := os.Open(inputFile) if err != nil { fmt.Println("读取文件失败:", err) return } defer file.Close() // 匹配 benchmark 输出行 re := regexp.MustCompile(`^Benchmark(\w+)-\d+\s+(\d+)\s+(\d+)\s+ns/op\s+(\d+)\s+B/op\s+(\d+)\s+allocs/op`) var results []BenchmarkResult scanner := bufio.NewScanner(file) for scanner.Scan() { line := scanner.Text() matches := re.FindStringSubmatch(line) if len(matches) == 6 { results = append(results, BenchmarkResult{ Name: matches[1], Runs: matches[2], NsPerOp: matches[3], BytesPerOp: matches[4], AllocsPerOp: matches[5], }) } } if err := scanner.Err(); err != nil {php fmt.Println("读取行失败:", err) return } // 写入 CSV outFile, err := os.Create(outputFile) if err != nil { fmt.Println("创建 CSV 文件失败:", err) return } defer outFile.Close() writer := csv.NewWriter(outFile) defer writer.Flush() headers := []string{"name", "runs", "ns_per_op", "bytes_per_op", "allocs_per_op"} if err := writer.Write(headers); err != nil { fmt.Println("写入 CSV 头失败:", err) return } for _, r := range results { row := []string{r.Name, r.Runs, r.NsPerOp, r.BytesPerOp, r.AllocsPerOp} if err := writer.Write(row); err != nil { fmt.Println("写入行失败:", err) return } } fmt.Println("✅ 成功导出 CSV:", outputFile) }
使用方法
保存为 main.go
编译运行:
go run main.go
或编译为可执行文件:
go build -o benchparser ./benchparser
确保 benchmark.txt
在同目录,输出会生成 benchmark.csv
。
使用 Web UI 可视化数据
功能需求
- 支持输入:
- 从
benchmark.txt
文件读取 - 从标准输入读取(如管道输入)
- 从
- 解析
go test -bench=. -benchmem
输出 - 提供一个
Web UI
:- 显示
ns/op
、B/op
、allocs/op
的柱状图 - 支持切换不同维度视图
- 显示
项目结构设计(Go + HTML + JS)
benchmark-visualizer/
├── main.go ← 启动解析器 + Web Server├── static/│ ├── index.html ← Web 页面│ ├── chart.js ← Chart.js 库│ └── main.js ← 渲染图表的 JS 脚本
Go 服务器(main.go)
package main import ( "bufio" "encoding/json" "fmt" "net/http" "os" "regexp" ) type BenchmarkResult struct { Name string `json:"name"` Runs int `json:"runs"` NsPerOp float64`json:"ns_per_op"` BytesPerOp int `json:"bytes_per_op"` AllocsPerOp int `json:"allocs_per_op"` } var results []BenchmarkResult func parseBenchmark(input *os.File) ([]BenchmarkResult, error) { re := regexp.MustCompile(`^(Benchmark\S*)\s+(\d+)\s+([\d\.]+)\s+ns/op\s+(\d+)\s+B/op\s+(\d+)\s+allocs/op`) scanner := bufio.NewScanner(input) var parsed []BenchmarkResult for scanner.Scan() { line := scanner.Text() 编程客栈 matches := re.FindStringSubmatch(line) if len(matches) == 6 { var r BenchmarkResult r.Name = matches[1] fmt.Sscanf(matches[2], "%d", &r.Runs) fmt.Sscanf(matches[3], "%f", &r.NsPerOp) fmt.Sscanf(matches[4], "%d", &r.BytesPerOp) fmt.Sscanf(matches[5], "%d", &r.AllocsPerOp) parsed = append(parsed, r) } } return parsed, scanner.Err() } func dataHandler(w http.ResponseWriter, r *http.Request) { w.Header().Set("Content-Type", "application/json") json.NewEncoder(w).Encode(results) } func main() { // 判断是否有标准输入 stat, _ := os.Stdin.Stat() var err error if (stat.Mode() & os.ModeCharDevice) == 0 { // 从标准输入读取 results, err = parseBenchmark(os.Stdin) } else { // 默认读取 benchmark.txt 文件 file, err := os.Open("benchmark.txt") if err != nil { fmt.Println("打开 benchmark.txt 失败:", err) return } defer file.Close() results, err = parseBenchmark(file) } if err != nil { fmt.Println("解析失败:", err) return } fs := http.FileServer(http.Dir("./static")) http.Handle("/", fs) http.HandleFunc("/data", dataHandler) fmt.Println(" 服务器已启动:http://localhost:8080") http.ListenAndServe(":8080", nil) }
Web 页面(static/index.html)
<!DOCTYPE html> <html> <head> <meta charset="utf-8"> <title>Go Benchmark 可视化</title> <script src="https://cdn.jsdelivr.net/npm/chart.js"></script> <script src="main.js" defer></script> </head> <body> <h2>Go Benchmark 可视化结果</h2> <select id="metric-select"> <option value="ns_per_op">ns/op</option> <option value="bytes_per_op">B/op</option> <option value="allocs_per_op">allocs/op</option> </select> <canvas id="benchmarkChart" width="900" height="500"></canvas> </body> </html>
Chart 渲染脚本(static/main.js)
let chart; const ctx = document.getElementById('benchmarkChart').getContext('2d'); function renderChart(data, metric = "ns_per_op") { const labels = data.map(item => item.name); const values = data.map(item => item[metric]); if (chart) chart.destroy(); chart = new Chart(ctx, { type: 'bar', data: { labels: labels, datasets: [{ label: metric, data: values, backgroundColor: 'rgba(54, 162, 235, 0.6)' }] }, options: { scales: { y: { beginAtZero: true } } } }); } fetch('/data') .then(res => res.json()) .then(data => { renderChartjs(data); document.getElementById('metric-select').addEventListener('change', (e) => { renderChart(data, e.target.value); }); });
运行方式
方法 1:从文件读取
go run main.go # 或 go build -o benchvis && ./benchvis
方法 2:从标准输入读取
go test -bench=. -benchmem | go run main.go
然后访问:http://localhost:8080
效果如图:
benchmark.txt文件示例
goos: darwin
goarch: arm64pkg: echocpu: Apple M1BenchmarkEchoAll1-8 271656 4631 ns/op 21080 B/op &http://www.devze.comnbsp; 99 allocs/opBenchmarkEchoAll2-8 224186 5874 ns/op 31560 B/op 99 allocs/opBenchmarkEchoAll3-8 1273263 836.1 ns/op 640 B/op 1 allocs/opBenchmarkEchoAll4-8 1457311 809.8 ns/op 1912 B/op 8 allocs/opPASSok echo 9.128s
到此这篇关于使用Go语言实现benchmark解析器和Web UI数据可视化的文章就介绍到这了,更多相关Go benchmark解析内容请搜索编程客栈(www.devze.com)以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程客栈(www.devze.com)!
精彩评论