From 8dbe5744e5541b9f435b43fb278987a259718331 Mon Sep 17 00:00:00 2001 From: daiyb <570056418@qq.com> Date: Thu, 26 Feb 2026 15:55:33 +0800 Subject: [PATCH] =?UTF-8?q?=E5=AF=BC=E5=87=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pkg/logic/cast/work.go | 42 +++++++++++++++ pkg/model/cast/work.go | 12 +++++ pkg/router/media.go | 2 +- pkg/service/cast/publish_log.go | 94 +++++++++++++++++++++++++++++++++ 4 files changed, 149 insertions(+), 1 deletion(-) diff --git a/pkg/logic/cast/work.go b/pkg/logic/cast/work.go index 8bed0387..8d41aa77 100644 --- a/pkg/logic/cast/work.go +++ b/pkg/logic/cast/work.go @@ -1,6 +1,7 @@ package cast import ( + "fmt" "fonchain-fiee/api/cast" modelCast "fonchain-fiee/pkg/model/cast" "strings" @@ -102,3 +103,44 @@ func (w *Work) ExportExcelWorkList(data []*cast.WorkListResp_Info) (*excelize.Fi return f, nil } + +func (w *Work) ExportPublishLogList(data []*cast.PublishLogInfo, savePath string) error { + f := excelize.NewFile() + defer f.Close() + + sheetName := "Sheet1" + f.SetSheetName("Sheet1", sheetName) + + headers := []string{ + "状态", "失败原因", "提交发布时间", "发布平台", "发布方式", "作品类型", "作品标题", "艺人", "用户编号", + } + for col, h := range headers { + cell, _ := excelize.CoordinatesToCellName(col+1, 1) + f.SetCellValue(sheetName, cell, h) + } + + for rowIndex, info := range data { + row := rowIndex + 2 + platformName := modelCast.PlatformNameKv[info.PlatformID] + if platformName == "" { + platformName = fmt.Sprintf("%d", info.PlatformID) + } + values := []interface{}{ + modelCast.PublishStatusMM[int(info.PublishMediaStatus)], + info.Detail, + info.CreatedAt, + modelCast.PlatformNameKv[info.PlatformID], + modelCast.PublishSourceMM[int(info.PublishSource)], + modelCast.WorkCategoryMM[int(info.WorkCategory)], + info.Title, + info.ArtistName, + info.ArtistSubNum, + } + for col, v := range values { + cell, _ := excelize.CoordinatesToCellName(col+1, row) + f.SetCellValue(sheetName, cell, v) + } + } + + return f.SaveAs(savePath) +} diff --git a/pkg/model/cast/work.go b/pkg/model/cast/work.go index 7f54c7c2..98d69cc2 100644 --- a/pkg/model/cast/work.go +++ b/pkg/model/cast/work.go @@ -32,6 +32,18 @@ var WorkCategoryMM = map[int]string{ 1: "图文", 2: "视频", } +var PublishSourceMM = map[int]string{ + 1: "手动发布", + 2: "定时任务", + 3: "手动发布", +} + +var PublishStatusMM = map[int]string{ + 1: "进行中", + 2: "发布成功", + 3: "发布失败", + 4: "发布异常", +} var WorkCostTypeMM = map[int]string{ 1: "套餐", diff --git a/pkg/router/media.go b/pkg/router/media.go index 74ac0ff8..1dbdf234 100644 --- a/pkg/router/media.go +++ b/pkg/router/media.go @@ -51,11 +51,11 @@ func MediaRouter(r *gin.RouterGroup) { work.POST("list-published", serviceCast.WorkListPublished) work.POST("update-work-script", serviceCast.UpdateWorkScript) work.POST("publish-log-list", serviceCast.PublishLogList) + work.Any("publish-log-list-export", serviceCast.PublishLogListExport) } workNoAuth := noAuth.Group("work") { - workNoAuth.POST("cron-republish", serviceCast.CronRePublish) } diff --git a/pkg/service/cast/publish_log.go b/pkg/service/cast/publish_log.go index ff24e2c2..43b76af7 100644 --- a/pkg/service/cast/publish_log.go +++ b/pkg/service/cast/publish_log.go @@ -2,11 +2,19 @@ package cast import ( "context" + "fmt" + "strings" + "time" "fonchain-fiee/api/cast" + "fonchain-fiee/pkg/cache" + logicCast "fonchain-fiee/pkg/logic/cast" + "fonchain-fiee/pkg/model/login" "fonchain-fiee/pkg/service" + "fonchain-fiee/pkg/utils" "github.com/gin-gonic/gin" + "go.uber.org/zap" ) // PublishLogList 查询发布记录列表 @@ -23,3 +31,89 @@ func PublishLogList(ctx *gin.Context) { } service.Success(ctx, resp) } + +// PublishLogListExport 导出发布记录列表 Excel +func PublishLogListExport(ctx *gin.Context) { + var req cast.ListPublishLogReq + if err := ctx.ShouldBind(&req); err != nil { + return + } + + loginInfo := login.GetUserInfoFromC(ctx) + newCtx := NewCtxWithUserInfo(ctx) + + lockKey := "PublishLogListExport" + fmt.Sprint(loginInfo.ID) + replay := cache.RedisClient.SetNX(lockKey, time.Now().Unix(), time.Minute*30) + if !replay.Val() { + return + } + defer cache.RedisClient.Del(lockKey) + + const batchSize = 5000 + var allData []*cast.PublishLogInfo + + page := int32(1) + if req.Page > 0 { + page = req.Page + } + + originalPageSize := req.PageSize + req.PageSize = batchSize + + zap.L().Info("开始分批导出发布记录列表", zap.Int32("batchSize", batchSize)) + + for { + req.Page = page + zap.L().Info("获取第 N 页数据", zap.Int32("page", page), zap.Int32("pageSize", req.PageSize)) + + resp, err := service.CastProvider.ListPublishLog(newCtx, &req) + if err != nil { + zap.L().Error("获取发布记录失败", zap.Error(err), zap.Int32("page", page)) + return + } + + if resp == nil || len(resp.Data) == 0 { + zap.L().Info("没有更多数据", zap.Int32("page", page)) + break + } + + allData = append(allData, resp.Data...) + zap.L().Info("获取数据成功", + zap.Int32("page", page), + zap.Int("本批次数量", len(resp.Data)), + zap.Int("累计总数", len(allData))) + + if len(resp.Data) < batchSize { + zap.L().Info("已到最后一页", zap.Int32("page", page)) + break + } + page++ + } + + req.PageSize = originalPageSize + zap.L().Info("数据获取完成，开始生成Excel", zap.Int("总数据量", len(allData))) + + if len(allData) == 0 { + return + } + + fileName := fmt.Sprintf("发布记录_%s.xlsx", time.Now().Format("20060102150405")) + filePath := fmt.Sprintf("./runtime/%d/%s", loginInfo.ID, fileName) + utils.CheckDirPath("./runtime/"+fmt.Sprint(loginInfo.ID), true) + + var logicWork = new(logicCast.Work) + if err := logicWork.ExportPublishLogList(allData, filePath); err != nil { + zap.L().Error("生成Excel失败", zap.Error(err)) + service.Error(ctx, err) + return + } + + scheme := "http" + if ctx.GetHeader("X-Forwarded-Proto") == "https" { + scheme = "https" + } + exportUrl := fmt.Sprintf("%s://%s/api/fiee/static/%s", scheme, ctx.Request.Host, strings.Replace(filePath, "./runtime/", "", 1)) + + zap.L().Info("Excel导出成功", zap.String("文件名", fileName), zap.Int("记录数", len(allData))) + service.Success(ctx, gin.H{"url": exportUrl}) +}