开发者

使用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/opB/opallocs/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

效果如图:

使用Go语言实现benchmark解析器

benchmark.txt文件示例

goos: darwin

goarch: arm64

pkg: echo

cpu: Apple M1

BenchmarkEchoAll1-8         271656          4631 ns/op       21080 B/op &http://www.devze.comnbsp;        99 allocs/op

BenchmarkEchoAll2-8         224186          5874 ns/op       31560 B/op          99 allocs/op

BenchmarkEchoAll3-8        1273263           836.1 ns/op         640 B/op           1 allocs/op

BenchmarkEchoAll4-8        1457311           809.8 ns/op        1912 B/op           8 allocs/op

PASS

ok      echo    9.128s

到此这篇关于使用Go语言实现benchmark解析器和Web UI数据可视化的文章就介绍到这了,更多相关Go benchmark解析内容请搜索编程客栈(www.devze.com)以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程客栈(www.devze.com)!

0

上一篇:

下一篇:

精彩评论

暂无评论...
验证码 换一张
取 消

最新开发

开发排行榜