[操作日志]操作日志可以导出为Excel

2025-11-04 05:00:25 +08:00 · 2020-11-20 17:08:10 +08:00
parent f4409363f0
commit f8b6116c73
6 changed files with 142 additions and 4 deletions
--- a/internal/web/actions/default/log/exportExcel.go
+++ b/internal/web/actions/default/log/exportExcel.go
@@ -0,0 +1,112 @@
+package log
+
+import (
+	"bytes"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/iwind/TeaGo/lists"
+	timeutil "github.com/iwind/TeaGo/utils/time"
+	"github.com/tealeg/xlsx/v3"
+	"strconv"
+	"strings"
+)
+
+type ExportExcelAction struct {
+	actionutils.ParentAction
+}
+
+func (this *ExportExcelAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *ExportExcelAction) RunGet(params struct {
+	DayFrom string
+	DayTo   string
+	Keyword string
+}) {
+	logsResp, err := this.RPC().LogRPC().ListLogs(this.AdminContext(), &pb.ListLogsRequest{
+		Offset:  0,
+		Size:    1000, // 日志最大导出1000条，TODO 将来可以配置
+		DayFrom: params.DayFrom,
+		DayTo:   params.DayTo,
+		Keyword: params.Keyword,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	wb := xlsx.NewFile()
+	sheet, err := wb.AddSheet("default")
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	// 头部
+	{
+		row := sheet.AddRow()
+		row.SetHeight(25)
+		row.AddCell().SetString("ID")
+		row.AddCell().SetString("日期")
+		row.AddCell().SetString("用户")
+		row.AddCell().SetString("描述")
+		row.AddCell().SetString("IP")
+		row.AddCell().SetString("区域")
+		row.AddCell().SetString("运营商")
+		row.AddCell().SetString("页面地址")
+	}
+
+	// 数据
+	for _, log := range logsResp.Logs {
+		regionName := ""
+		ispName := ""
+		regionResp, err := this.RPC().IPLibraryRPC().LookupIPRegion(this.AdminContext(), &pb.LookupIPRegionRequest{Ip: log.Ip})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		if regionResp.Region != nil {
+			pieces := []string{}
+			if len(regionResp.Region.Country) > 0 {
+				pieces = append(pieces, regionResp.Region.Country)
+			}
+			if len(regionResp.Region.Province) > 0 && !lists.ContainsString(pieces, regionResp.Region.Province) {
+				pieces = append(pieces, regionResp.Region.Province)
+			}
+			if len(regionResp.Region.City) > 0 && !lists.ContainsString(pieces, regionResp.Region.City) && !lists.ContainsString(pieces, strings.TrimSuffix(regionResp.Region.Province, "市")) {
+				pieces = append(pieces, regionResp.Region.City)
+			}
+			regionName = strings.Join(pieces, " ")
+
+			if len(regionResp.Region.Isp) > 0 {
+				ispName = regionResp.Region.Isp
+			}
+		}
+
+		row := sheet.AddRow()
+		row.SetHeight(25)
+		row.AddCell().SetInt64(log.Id)
+		row.AddCell().SetString(timeutil.FormatTime("Y-m-d H:i:s", log.CreatedAt))
+		row.AddCell().SetString(log.UserName)
+		row.AddCell().SetString(log.Description)
+		row.AddCell().SetString(log.Ip)
+		row.AddCell().SetString(regionName)
+		row.AddCell().SetString(ispName)
+		row.AddCell().SetString(log.Action)
+	}
+
+	this.AddHeader("Content-Type", "application/vnd.ms-excel")
+	this.AddHeader("Content-Disposition", "attachment; filename=\"LOG-"+timeutil.Format("YmdHis")+".xlsx\"")
+	this.AddHeader("Cache-Control", "max-age=0")
+
+	buf := bytes.NewBuffer([]byte{})
+	err = wb.Write(buf)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.AddHeader("Content-Length", strconv.Itoa(buf.Len()))
+	this.Write(buf.Bytes())
+}