cid/dao/dataengine/material_verify_log_dao.go

package dataengine

import (
	consts "cid/consts/dataengine"
	daoEntity "cid/model/entity/dataengine"
	"context"

	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/os/gtime"
)

// MaterialVerifyLogDAO 素材校验日志数据访问层
type MaterialVerifyLogDAO struct{}

// MaterialVerifyLog DAO单例
var MaterialVerifyLog = new(MaterialVerifyLogDAO)

// TableName 表名
const MaterialVerifyLogTable = "material_verify_log"

// Create 创建校验日志
func (d *MaterialVerifyLogDAO) Create(ctx context.Context, log *daoEntity.MaterialVerifyLog) (int64, error) {
	// 构建插入数据，排除主键Id（让数据库自增）
	data := g.Map{
		"tenant_id":     log.TenantID,
		"material_type": log.MaterialType,
		"material_id":   log.MaterialID,
		"source_table":  log.SourceTable,
		"source_id":     log.SourceID,
		"account_id":    log.AccountID,
		"verify_status": log.VerifyStatus,
		"created_at":    gtime.Now(),
	}
	result, err := g.DB("default").Model(MaterialVerifyLogTable).Data(data).Insert()
	if err != nil {
		g.Log().Errorf(ctx, "创建校验日志失败: %v", err)
		return 0, err
	}
	id, _ := result.LastInsertId()
	return id, nil
}

// GetByID 根据ID获取日志
func (d *MaterialVerifyLogDAO) GetByID(ctx context.Context, id int64) (*daoEntity.MaterialVerifyLog, error) {
	var result daoEntity.MaterialVerifyLog
	r, err := g.DB("default").Model(MaterialVerifyLogTable).
		Where(daoEntity.MaterialVerifyLogCols.Id, id).
		One()
	if err != nil {
		return nil, err
	}
	if r.IsEmpty() {
		return nil, nil
	}
	if err = r.Struct(&result); err != nil {
		return nil, err
	}
	return &result, nil
}

// GetByTaskID 根据任务ID获取日志
func (d *MaterialVerifyLogDAO) GetByTaskID(ctx context.Context, taskID string) (*daoEntity.MaterialVerifyLog, error) {
	var result daoEntity.MaterialVerifyLog
	r, err := g.DB("default").Model(MaterialVerifyLogTable).
		Where(daoEntity.MaterialVerifyLogCols.TaskID, taskID).
		One()
	if err != nil {
		return nil, err
	}
	if r.IsEmpty() {
		return nil, nil
	}
	if err = r.Struct(&result); err != nil {
		return nil, err
	}
	return &result, nil
}

// GetByMaterialID 根据素材ID获取日志列表
func (d *MaterialVerifyLogDAO) GetByMaterialID(ctx context.Context, materialID string) ([]daoEntity.MaterialVerifyLog, error) {
	var result []daoEntity.MaterialVerifyLog
	r, err := g.DB("default").Model(MaterialVerifyLogTable).
		Where(daoEntity.MaterialVerifyLogCols.MaterialID, materialID).
		OrderDesc(daoEntity.MaterialVerifyLogCols.CreatedAt).
		All()
	if err != nil {
		return nil, err
	}
	if err = r.Structs(&result); err != nil {
		return nil, err
	}
	return result, nil
}

// GetBySource 根据来源获取日志
func (d *MaterialVerifyLogDAO) GetBySource(ctx context.Context, sourceTable string, sourceID int64) ([]daoEntity.MaterialVerifyLog, error) {
	var result []daoEntity.MaterialVerifyLog
	r, err := g.DB("default").Model(MaterialVerifyLogTable).
		Where(daoEntity.MaterialVerifyLogCols.SourceTable, sourceTable).
		Where(daoEntity.MaterialVerifyLogCols.SourceID, sourceID).
		OrderDesc(daoEntity.MaterialVerifyLogCols.CreatedAt).
		All()
	if err != nil {
		return nil, err
	}
	if err = r.Structs(&result); err != nil {
		return nil, err
	}
	return result, nil
}

// UpdateVerifyResult 更新校验结果
func (d *MaterialVerifyLogDAO) UpdateVerifyResult(ctx context.Context, id int64, verifyStatus string, suggestion, label, resultType int, responseResult string, checkTime int64) error {
	_, err := g.DB("default").Model(MaterialVerifyLogTable).
		Where(daoEntity.MaterialVerifyLogCols.Id, id).
		Data(map[string]interface{}{
			daoEntity.MaterialVerifyLogCols.VerifyStatus:   verifyStatus,
			daoEntity.MaterialVerifyLogCols.Suggestion:     suggestion,
			daoEntity.MaterialVerifyLogCols.Label:          label,
			daoEntity.MaterialVerifyLogCols.ResultType:     resultType,
			daoEntity.MaterialVerifyLogCols.ResponseResult: responseResult,
			daoEntity.MaterialVerifyLogCols.CheckTime:      checkTime,
		}).Update()
	if err != nil {
		g.Log().Errorf(ctx, "更新校验日志结果失败: %v", err)
		return err
	}
	return nil
}

// UpdateError 更新错误信息
func (d *MaterialVerifyLogDAO) UpdateError(ctx context.Context, id int64, verifyStatus string, errorMsg string) error {
	_, err := g.DB("default").Model(MaterialVerifyLogTable).
		Where(daoEntity.MaterialVerifyLogCols.Id, id).
		Data(map[string]interface{}{
			daoEntity.MaterialVerifyLogCols.VerifyStatus: verifyStatus,
			daoEntity.MaterialVerifyLogCols.ErrorMsg:     errorMsg,
		}).Update()
	if err != nil {
		g.Log().Errorf(ctx, "更新校验日志错误失败: %v", err)
		return err
	}
	return nil
}

// UpdateTaskID 更新任务ID
func (d *MaterialVerifyLogDAO) UpdateTaskID(ctx context.Context, id int64, taskID string) error {
	_, err := g.DB("default").Model(MaterialVerifyLogTable).
		Where(daoEntity.MaterialVerifyLogCols.Id, id).
		Data(map[string]interface{}{
			daoEntity.MaterialVerifyLogCols.TaskID: taskID,
		}).Update()
	if err != nil {
		return err
	}
	return nil
}

// UpdateDuration 更新处理耗时
func (d *MaterialVerifyLogDAO) UpdateDuration(ctx context.Context, id int64, durationMs int64) error {
	_, err := g.DB("default").Model(MaterialVerifyLogTable).
		Where(daoEntity.MaterialVerifyLogCols.Id, id).
		Data(map[string]interface{}{
			daoEntity.MaterialVerifyLogCols.DurationMs: durationMs,
		}).Update()
	if err != nil {
		return err
	}
	return nil
}

// UpdateRequestParams 更新请求参数
func (d *MaterialVerifyLogDAO) UpdateRequestParams(ctx context.Context, id int64, requestParams string) error {
	_, err := g.DB("default").Model(MaterialVerifyLogTable).
		Where(daoEntity.MaterialVerifyLogCols.Id, id).
		Data(map[string]interface{}{
			daoEntity.MaterialVerifyLogCols.RequestParams: requestParams,
		}).Update()
	if err != nil {
		return err
	}
	return nil
}

// GetByCondition 根据条件分页查询
func (d *MaterialVerifyLogDAO) GetByCondition(ctx context.Context, condition map[string]interface{}, page, pageSize int) ([]daoEntity.MaterialVerifyLog, int, error) {
	var result []daoEntity.MaterialVerifyLog
	model := g.DB("default").Model(MaterialVerifyLogTable)

	for k, v := range condition {
		model = model.Where(k, v)
	}

	total, err := model.Count()
	if err != nil {
		return nil, 0, err
	}

	r, err := model.
		OrderDesc(daoEntity.MaterialVerifyLogCols.CreatedAt).
		Page(page, pageSize).
		All()
	if err != nil {
		return nil, 0, err
	}
	if err = r.Structs(&result); err != nil {
		return nil, 0, err
	}
	return result, int(total), nil
}

// CountByStatus 按状态统计
func (d *MaterialVerifyLogDAO) CountByStatus(ctx context.Context, verifyStatus string) (int, error) {
	count, err := g.DB("default").Model(MaterialVerifyLogTable).
		Where(daoEntity.MaterialVerifyLogCols.VerifyStatus, verifyStatus).
		Count()
	if err != nil {
		return 0, err
	}
	return int(count), nil
}

// GetStats 获取统计信息
func (d *MaterialVerifyLogDAO) GetStats(ctx context.Context) (map[string]int, error) {
	stats := make(map[string]int)

	// 使用实体中定义的正确状态值：PENDING=待校验, VERIFIED=校验通过, REJECTED=校验不通过
	statuses := []struct {
		statusKey string
		statusVal string
	}{
		{"pending", daoEntity.VerifyStatusPending},
		{"verified", daoEntity.VerifyStatusVerified},
		{"rejected", daoEntity.VerifyStatusRejected},
	}

	var totalCount int
	for _, item := range statuses {
		count, err := d.CountByStatus(ctx, item.statusVal)
		if err != nil {
			continue
		}
		stats[item.statusKey] = count
		totalCount += count
	}

	// 添加总计
	stats["total"] = totalCount

	return stats, nil
}

// GetPendingResults 获取待查询结果的日志（状态为submitting且有taskID）
func (d *MaterialVerifyLogDAO) GetPendingResults(ctx context.Context, limit int) ([]daoEntity.MaterialVerifyLog, error) {
	var result []daoEntity.MaterialVerifyLog

	// 查询状态为 pending 且有 task_id 的记录
	r, err := g.DB("default").Model(MaterialVerifyLogTable).
		Where(daoEntity.MaterialVerifyLogCols.VerifyStatus, consts.CheckStatusPending).
		WhereNotNull(daoEntity.MaterialVerifyLogCols.TaskID).
		Where(daoEntity.MaterialVerifyLogCols.TaskID + " != ''").
		OrderAsc(daoEntity.MaterialVerifyLogCols.CreatedAt).
		Limit(limit).
		All()
	if err != nil {
		g.Log().Errorf(ctx, "查询待处理结果日志失败: %v", err)
		return nil, err
	}
	if err = r.Structs(&result); err != nil {
		g.Log().Errorf(ctx, "转换待处理结果日志失败: %v", err)
		return nil, err
	}
	return result, nil
}

// GetLastRejectedLogByMaterialID 根据素材ID获取最后一条失败的校验日志
func (d *MaterialVerifyLogDAO) GetLastRejectedLogByMaterialID(ctx context.Context, materialID string, verifyStatus string) (*daoEntity.MaterialVerifyLog, error) {
	var result daoEntity.MaterialVerifyLog
	r, err := g.DB("default").Model(MaterialVerifyLogTable).
		Where(daoEntity.MaterialVerifyLogCols.MaterialID, materialID).
		Where(daoEntity.MaterialVerifyLogCols.VerifyStatus, verifyStatus).
		OrderDesc(daoEntity.MaterialVerifyLogCols.CreatedAt).
		One()
	if err != nil {
		return nil, err
	}
	if r.IsEmpty() {
		return nil, nil
	}
	if err = r.Struct(&result); err != nil {
		return nil, err
	}
	return &result, nil
}

// CountPendingResults 统计待查询结果的数量
func (d *MaterialVerifyLogDAO) CountPendingResults(ctx context.Context) (int, error) {
	count, err := g.DB("default").Model(MaterialVerifyLogTable).
		Where(daoEntity.MaterialVerifyLogCols.VerifyStatus, consts.CheckStatusPending).
		WhereNotNull(daoEntity.MaterialVerifyLogCols.TaskID).
		Where(daoEntity.MaterialVerifyLogCols.TaskID + " != ''").
		Count()
	if err != nil {
		return 0, err
	}
	return int(count), nil
}