353 lines
8.7 KiB
Go
353 lines
8.7 KiB
Go
package cron
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"fmt"
|
||
account "fonchain-fiee/api/accountFiee"
|
||
"fonchain-fiee/api/cron"
|
||
"fonchain-fiee/pkg/service"
|
||
"fonchain-fiee/pkg/utils"
|
||
"log"
|
||
"os"
|
||
"path/filepath"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/gin-gonic/gin/binding"
|
||
"github.com/xuri/excelize/v2"
|
||
"google.golang.org/protobuf/types/known/emptypb"
|
||
)
|
||
|
||
func CreateScheduleTask(c *gin.Context) {
|
||
var req cron.CreateScheduleTaskRequest
|
||
if err := c.ShouldBindBodyWith(&req, binding.JSON); err != nil {
|
||
service.Error(c, errors.New("参数错误"))
|
||
return
|
||
}
|
||
res, err := service.CronProvider.CreateScheduleTask(context.Background(), &req)
|
||
if err != nil {
|
||
service.Error(c, errors.New("网络超时,请重试"))
|
||
return
|
||
}
|
||
if res.Code != 200 {
|
||
service.Error(c, errors.New(res.Message))
|
||
return
|
||
}
|
||
service.Success(c, res)
|
||
}
|
||
func UpdateScheduleTask(c *gin.Context) {
|
||
var req cron.UpdateScheduleTaskRequest
|
||
if err := c.ShouldBindBodyWith(&req, binding.JSON); err != nil {
|
||
service.Error(c, errors.New("参数错误"))
|
||
return
|
||
}
|
||
res, err := service.CronProvider.UpdateScheduleTask(context.Background(), &req)
|
||
if err != nil {
|
||
service.Error(c, errors.New("网络超时,请重试"))
|
||
return
|
||
}
|
||
if res.Code != 200 {
|
||
service.Error(c, errors.New(res.Message))
|
||
return
|
||
}
|
||
service.Success(c, res)
|
||
}
|
||
|
||
func GetListScheduleTask(c *gin.Context) {
|
||
var req cron.GetListScheduleTaskRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
service.Error(c, errors.New("参数错误"))
|
||
return
|
||
}
|
||
res, err := service.CronProvider.GetListScheduleTask(context.Background(), &req)
|
||
if err != nil {
|
||
service.Error(c, errors.New("网络超时,请重试"))
|
||
return
|
||
}
|
||
if res.Code != 200 {
|
||
service.Error(c, errors.New(res.Message))
|
||
return
|
||
}
|
||
service.Success(c, res)
|
||
}
|
||
|
||
func GetListExecutionRecord(c *gin.Context) {
|
||
var req cron.GetListExecutionRecordRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
service.Error(c, errors.New("参数错误"))
|
||
return
|
||
}
|
||
res, err := service.CronProvider.GetListExecutionRecord(context.Background(), &req)
|
||
if err != nil {
|
||
service.Error(c, errors.New("网络超时,请重试"))
|
||
return
|
||
}
|
||
if res.Code != 200 {
|
||
service.Error(c, errors.New(res.Message))
|
||
return
|
||
}
|
||
service.Success(c, res)
|
||
}
|
||
func GetListExecutionResult(c *gin.Context) {
|
||
var req cron.GetListExecutionResultRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
service.Error(c, errors.New("参数错误"))
|
||
return
|
||
}
|
||
res, err := service.CronProvider.GetListExecutionResult(context.Background(), &req)
|
||
if err != nil {
|
||
service.Error(c, errors.New("网络超时,请重试"))
|
||
return
|
||
}
|
||
if res.Code != 200 {
|
||
service.Error(c, errors.New(res.Message))
|
||
return
|
||
}
|
||
service.Success(c, res)
|
||
}
|
||
func ExportExcelExecutionResult(c *gin.Context) {
|
||
var req cron.GetListExecutionResultRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
service.Error(c, errors.New("参数错误"))
|
||
return
|
||
}
|
||
res, err := service.CronProvider.GetListExecutionResult(context.Background(), &req)
|
||
if err != nil {
|
||
service.Error(c, errors.New("网络超时,请重试"))
|
||
return
|
||
}
|
||
|
||
titleList := []string{
|
||
"任务标题", "艺人编号", "艺人姓名", "内容类型", "内容标题", "执行结果", "原因",
|
||
}
|
||
var dataList []interface{}
|
||
for _, task := range res.Data {
|
||
data := []interface{}{
|
||
task.TaskTitle,
|
||
task.ArtistId,
|
||
task.ArtistName,
|
||
task.ContentTypeDescription,
|
||
task.WorkTitle,
|
||
task.ResultDescription,
|
||
task.ResultDesc,
|
||
}
|
||
dataList = append(dataList, &data)
|
||
}
|
||
|
||
content, err := utils.ToExcelByType(titleList, dataList, "slice", "")
|
||
if err != nil {
|
||
service.Error(c, errors.New("生成excel失败"))
|
||
|
||
}
|
||
utils.ResponseXls(c, content, fmt.Sprintf("失败任务报告_%s.xlsx", time.Now().Format("20060102150405")))
|
||
return
|
||
}
|
||
|
||
func StartScheduleTask(c *gin.Context) {
|
||
var req cron.CommonIDRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
service.Error(c, errors.New("参数错误"))
|
||
return
|
||
}
|
||
res, err := service.CronProvider.StartScheduleTask(context.Background(), &req)
|
||
if err != nil {
|
||
service.Error(c, errors.New("网络超时,请重试"))
|
||
return
|
||
}
|
||
if res.Code != 200 {
|
||
service.Error(c, errors.New(res.Message))
|
||
return
|
||
}
|
||
service.Success(c, res)
|
||
}
|
||
|
||
func PauseScheduleTask(c *gin.Context) {
|
||
var req cron.CommonIDRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
service.Error(c, errors.New("参数错误"))
|
||
return
|
||
}
|
||
res, err := service.CronProvider.PauseScheduleTask(context.Background(), &req)
|
||
if err != nil {
|
||
service.Error(c, errors.New("网络超时,请重试"))
|
||
return
|
||
}
|
||
if res.Code != 200 {
|
||
service.Error(c, errors.New(res.Message))
|
||
return
|
||
}
|
||
service.Success(c, res)
|
||
}
|
||
|
||
func DeleteScheduleTask(c *gin.Context) {
|
||
var req cron.CommonIDRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
service.Error(c, errors.New("参数错误"))
|
||
return
|
||
}
|
||
res, err := service.CronProvider.DeleteScheduleTask(context.Background(), &req)
|
||
if err != nil {
|
||
service.Error(c, errors.New("网络超时,请重试"))
|
||
return
|
||
}
|
||
if res.Code != 200 {
|
||
service.Error(c, errors.New(res.Message))
|
||
return
|
||
}
|
||
service.Success(c, res)
|
||
}
|
||
|
||
func GetScheduleTaskStatus(c *gin.Context) {
|
||
res, err := service.CronProvider.GetScheduleTaskStatus(context.Background(), &emptypb.Empty{})
|
||
if err != nil {
|
||
service.Error(c, errors.New("网络超时,请重试"))
|
||
return
|
||
}
|
||
if res.Code != 200 {
|
||
service.Error(c, errors.New(res.Message))
|
||
return
|
||
}
|
||
service.Success(c, res)
|
||
}
|
||
|
||
func GetImportData(c *gin.Context) {
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
service.Error(c, errors.New("操作失败"))
|
||
}
|
||
}()
|
||
|
||
// 导入excel
|
||
excelFile, err := c.FormFile("excel")
|
||
if err != nil {
|
||
service.Error(c, errors.New("缺少excel文件"))
|
||
return
|
||
}
|
||
|
||
// 创建临时文件
|
||
tempDir := "tmp"
|
||
if err = os.MkdirAll(tempDir, 0755); err != nil {
|
||
service.Error(c, errors.New("创建临时目录失败"))
|
||
return
|
||
}
|
||
defer os.RemoveAll(tempDir)
|
||
|
||
// 保存excel
|
||
excelPath := filepath.Join(tempDir, "excel.xlsx")
|
||
if err = c.SaveUploadedFile(excelFile, excelPath); err != nil {
|
||
service.Error(c, errors.New("保存excel文件失败"))
|
||
return
|
||
}
|
||
|
||
// 读取excel
|
||
readExcelResult, err := readExcel(excelPath)
|
||
if err != nil {
|
||
service.Error(c, fmt.Errorf("读取excel失败: %v", err))
|
||
return
|
||
}
|
||
if len(readExcelResult) == 0 {
|
||
service.Error(c, errors.New("请检查excel文件"))
|
||
return
|
||
}
|
||
res, err := service.AccountFieeProvider.UserList(context.Background(), &account.UserListRequest{
|
||
Domain: "app",
|
||
})
|
||
if err != nil {
|
||
service.Error(c, errors.New("用户查询失败"))
|
||
return
|
||
}
|
||
subNameFindExcelData := make(map[string]excelData, len(readExcelResult))
|
||
for _, v := range subNameFindExcelData {
|
||
subNameFindExcelData[v.SubName] = v
|
||
}
|
||
subNameFindID := make(map[string]uint64, len(readExcelResult))
|
||
result := make([]excelDataResult, len(readExcelResult))
|
||
for _, v := range res.UserList {
|
||
subNameFindID[v.SubNum] = v.Id
|
||
//名称不一致的报错
|
||
if data, exists := subNameFindExcelData[v.SubNum]; exists {
|
||
// 校验名称是否一致
|
||
if data.Name != v.Name {
|
||
service.Error(c, errors.New(fmt.Sprintf("第 %d 行数据错误，请检查数据！", data.Line)))
|
||
return // 直接返回，停止后续执行
|
||
}
|
||
}
|
||
}
|
||
for _, v := range readExcelResult {
|
||
result = append(result, excelDataResult{
|
||
Id: subNameFindID[v.SubName],
|
||
Name: v.Name,
|
||
})
|
||
}
|
||
service.Success(c, readExcelResult)
|
||
|
||
}
|
||
|
||
type excelData struct {
|
||
Line uint `json:"line"`
|
||
SubName string `json:"subName"`
|
||
Name string `json:"name"`
|
||
}
|
||
|
||
type excelDataResult struct {
|
||
Id uint64 `json:"id"`
|
||
Name string `json:"name"`
|
||
}
|
||
|
||
func readExcel(excelPath string) ([]excelData, error) {
|
||
//打开excel
|
||
f, err := excelize.OpenFile(excelPath)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
defer f.Close()
|
||
|
||
//读取第一页
|
||
sheetName := f.GetSheetName(0)
|
||
if sheetName == "" {
|
||
return nil, errors.New("excel文件中没有工作表")
|
||
}
|
||
|
||
//读取数据
|
||
rows, err := f.GetRows(sheetName)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("读取工作表失败: %v", err)
|
||
}
|
||
|
||
if len(rows) <= 1 {
|
||
return nil, errors.New("excel文件没有数据行（只有表头或为空）")
|
||
}
|
||
var result []excelData
|
||
for i := 1; i < len(rows); i++ { // 从第2行开始（跳过表头）
|
||
row := rows[i]
|
||
if len(row) == 0 {
|
||
continue
|
||
}
|
||
|
||
subName := getCellValue(f, sheetName, i, 0)
|
||
name := getCellValue(f, sheetName, i, 1)
|
||
data := excelData{
|
||
Line: uint(i),
|
||
SubName: subName,
|
||
Name: name,
|
||
}
|
||
result = append(result, data)
|
||
}
|
||
|
||
return result, nil
|
||
}
|
||
func getCellValue(f *excelize.File, sheetName string, rowIndex, colIndex int) string {
|
||
colName, _ := excelize.ColumnNumberToName(colIndex + 1)
|
||
cell := fmt.Sprintf("%s%d", colName, rowIndex+1)
|
||
|
||
value, err := f.GetCellValue(sheetName, cell)
|
||
if err != nil {
|
||
log.Printf("读取单元格 %s 失败: %v", cell, err)
|
||
return ""
|
||
}
|
||
|
||
return strings.TrimSpace(value)
|
||
}
|