moredoc/model/documentError.go

package model

import (
	"time"

	"go.uber.org/zap"
	"gorm.io/gorm"
)

type DocumentError struct {
	Id        int64      `form:"id" json:"id,omitempty" gorm:"primaryKey;autoIncrement;column:id;comment:;"`
	Message   string     `form:"message" json:"message,omitempty" gorm:"column:message;type:text;comment:文档转换失败原因;"`
	CreatedAt *time.Time `form:"created_at" json:"created_at,omitempty" gorm:"column:created_at;type:datetime;comment:创建时间;"`
	UpdatedAt *time.Time `form:"updated_at" json:"updated_at,omitempty" gorm:"column:updated_at;type:datetime;comment:更新时间;"`
}

func (DocumentError) TableName() string {
	return tablePrefix + "document_error"
}

// SetDocumentConvertError 设置文档转换失败的错误信息
// 如果 err 为 nil，则删除转换失败的记录
// 如果 err 不为 nil，则创建或更新转换失败的记录
func (m *DBModel) SetDocumentConvertError(documentId int64, err error) error {
	de := &DocumentError{
		Id: documentId,
	}
	if err == nil {
		// 如果没有错误，则删除转换失败的记录
		err = m.db.Delete(de).Error
		if err != nil {
			m.logger.Error("SetConvertError", zap.Error(err))
			return err
		}
		return nil
	}
	de.Message = err.Error()
	var exist DocumentError
	if err = m.db.First(&exist, documentId).Error; err != nil {
		// 如果不存在，则创建
		err = m.db.Create(de).Error
		if err != nil {
			m.logger.Error("SetConvertError", zap.Error(err))
		}
		return err
	}
	// 如果存在，则更新
	return m.db.Model(&exist).Updates(de).Error
}

func (m *DBModel) GetConvertError(documentIds ...int64) (errors map[int64]string) {
	errors = make(map[int64]string)
	if len(documentIds) == 0 {
		return
	}

	var des []DocumentError
	err := m.db.Model(&DocumentError{}).Where("id IN (?)", documentIds).Find(&des).Error
	if err != nil && err != gorm.ErrRecordNotFound {
		m.logger.Error("GetConvertError", zap.Error(err))
		return
	}

	for _, de := range des {
		errors[de.Id] = de.Message
	}
	return
}