单独拎出来放到独立仓库

Author: leiyang23

.gitignore

@@ -0,0 +1,2 @@
+.idea
+uploads

README.md

@@ -0,0 +1,10 @@
+## 项目
+解析 nmon 导出数据，以图表的形式进行展示，同时支持导出 Excel 文件。
+
+## 使用
+### 源码编译
+将代码下载到本地后，执行 `go build -o nmon-parser main.go`，然后运行 `nmon-parser web --port=8081`，在浏览器打开地址：http://localhost:2233  
+
+### 或则直接下载编译好的可执行文件
+Linux 系统：`nmon-parser web --port=8081`  
+Windows 系统：`nmon-parser.exe web --port=8081`

core/chart.go

@@ -0,0 +1,62 @@
+package core
+
+import (
+	"github.com/go-echarts/go-echarts/v2/charts"
+	"github.com/go-echarts/go-echarts/v2/opts"
+	"github.com/go-echarts/go-echarts/v2/render"
+	"github.com/go-echarts/go-echarts/v2/types"
+)
+
+func LineChart(srcFile string, category Category) (chart render.ChartSnippet, err error) {
+	line := charts.NewLine()
+	line.SetGlobalOptions(
+		charts.WithInitializationOpts(opts.Initialization{Theme: types.ThemeWesteros, Width: "1800px", Height: "800px"}),
+		charts.WithTitleOpts(
+			opts.Title{
+				Title:    category.UniqueID,
+				Subtitle: "Line chart",
+				Bottom:   "0",
+			}),
+		charts.WithDataZoomOpts(opts.DataZoom{}),
+		charts.WithToolboxOpts(
+			opts.Toolbox{
+				Show:   opts.Bool(true),
+				Orient: "horizontal",
+				Feature: &opts.ToolBoxFeature{
+					SaveAsImage: &opts.ToolBoxFeatureSaveAsImage{Show: opts.Bool(true), Title: "Save"},
+				},
+			},
+		),
+	)
+
+	var categoryData CategoryData
+
+	categoryData, err = GetCategoryData(srcFile, category)
+	if err != nil {
+		return
+	}
+
+	// X 轴为时间轴
+	line.SetXAxis(categoryData.TimeSeries)
+
+	selectedLegend := make(map[string]bool)
+	for i, flag := range category.Metrics {
+		lineData := make([]opts.LineData, 0, len(categoryData.MetricLinesData))
+		for _, metricLineData := range categoryData.MetricLinesData {
+			// fixme: 当出现采集行数据数量小于指标数时，该如何处理？直接丢弃本行吗？还是补0？
+			if len(metricLineData) < len(category.Metrics) {
+				metricLineData = append(metricLineData, make([]float64, len(category.Metrics)-len(metricLineData))...)
+			}
+
+			lineData = append(lineData, opts.LineData{Value: metricLineData[i]})
+		}
+		line.AddSeries(flag, lineData)
+		selectedLegend[flag] = i == 0
+	}
+
+	line.SetSeriesOptions(charts.WithLineChartOpts(opts.LineChart{Smooth: opts.Bool(true)}))
+	line.SetGlobalOptions(charts.WithLegendOpts(opts.Legend{Selected: selectedLegend}))
+
+	chart = line.RenderSnippet()
+	return
+}

core/export.go

@@ -0,0 +1,121 @@
+package core
+
+import (
+	"fmt"
+	"github.com/leiyang23/nmon-parser/util"
+	"github.com/xuri/excelize/v2"
+	"log/slog"
+	"runtime/debug"
+	"sort"
+)
+
+func AddToExcel(excelFile string, categories map[string]Category, data map[string]*CategoryData) (err error) {
+
+	f, err := excelize.OpenFile(excelFile)
+	if err != nil {
+		f = excelize.NewFile()
+	}
+	defer util.Close(f)
+
+	// 对 sheet 名称进行排序
+	catList := make([]string, 0)
+	for category := range data {
+		catList = append(catList, category)
+	}
+	sort.Strings(catList)
+
+	// 事先创建好需要的 sheet，方便下一步使用 写入流 写入大规模数据
+	for _, category := range catList {
+		_, err = f.NewSheet(category)
+	}
+
+	var sw *excelize.StreamWriter
+	for _, category := range catList {
+		categoryData := data[category]
+
+		sw, err = f.NewStreamWriter(category)
+		if err != nil {
+			slog.Error(err.Error())
+			return
+		}
+
+		header := make([]interface{}, 0)
+		header = append(header, "time")
+		for _, i := range categories[category].Metrics {
+			header = append(header, i)
+		}
+		err = sw.SetRow("A1", header)
+		if err != nil {
+			return
+		}
+
+		for i, row := range categoryData.MetricLinesData {
+			rowData := make([]interface{}, 0, len(header))
+			rowData = append(rowData, categoryData.TimeSeries[i])
+			for _, j := range row {
+				rowData = append(rowData, j)
+			}
+			err = sw.SetRow(fmt.Sprintf("A%d", i+2), rowData)
+			if err != nil {
+				return
+			}
+		}
+
+		// 写入磁盘
+		err = sw.Flush()
+		if err != nil {
+			return err
+		}
+	}
+
+	f.SetActiveSheet(0)
+	_ = f.DeleteSheet("Sheet1")
+	return f.SaveAs(excelFile)
+}
+
+func AddAAAToExcel(excelFile string, data []AAA) (err error) {
+	f, err := excelize.OpenFile(excelFile)
+	if err != nil {
+		f = excelize.NewFile()
+	}
+	defer util.Close(f)
+
+	_, err = f.NewSheet("AAA")
+	if err != nil {
+		return
+	}
+
+	for i, row := range data {
+		err = f.SetSheetRow("AAA", fmt.Sprintf("A%d", i+1), &[]string{row.Item1, row.Item2})
+		if err != nil {
+			return
+		}
+	}
+
+	return f.SaveAs(excelFile)
+}
+
+func AddBBBPToExcel(excelFile string, data []BBBP) (err error) {
+	f, err := excelize.OpenFile(excelFile)
+	if err != nil {
+		f = excelize.NewFile()
+	}
+	defer util.Close(f)
+
+	_, err = f.NewSheet("BBBP")
+	if err != nil {
+		slog.Error(err.Error())
+		return
+	}
+
+	for i, row := range data {
+		err = f.SetSheetRow("BBBP", fmt.Sprintf("A%d", i+1), &[]string{row.Cmd, row.Output})
+		if err != nil {
+			return
+		}
+	}
+
+	debug.FreeOSMemory()
+
+	return f.SaveAs(excelFile)
+}

core/export_test.go

@@ -0,0 +1,34 @@
+package core
+
+import (
+	"testing"
+)
+
+func TestExcel(t *testing.T) {
+	categories, err := GetAllCategory(file1)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	data, err := GetCategoryAllData(file1)
+
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = AddToExcel("nmon.xlsx", categories, data)
+	if err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestExcel2(t *testing.T) {
+	BBBPData, err := GetBBBP(file1)
+	if err != nil {
+		t.Fatal(err)
+	}
+	err = AddBBBPToExcel("kk.xlsx", BBBPData)
+	if err != nil {
+		t.Fatal(err)
+	}
+}