90 lines
2.2 KiB
Go
90 lines
2.2 KiB
Go
package dao
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type JSON []byte
|
|
|
|
// ContentCodec 定义内容编解码接口
|
|
type ContentCodec interface {
|
|
Encode() (JSON, error)
|
|
Decode() error
|
|
GetContent() interface{} // 更明确的方法名
|
|
}
|
|
|
|
type FormulaRecognitionContent struct {
|
|
content JSON
|
|
Latex string `json:"latex"`
|
|
AdjustLatex string `json:"adjust_latex"`
|
|
EnhanceLatex string `json:"enhance_latex"`
|
|
}
|
|
|
|
func (c *FormulaRecognitionContent) Encode() (JSON, error) {
|
|
b, err := json.Marshal(c)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return b, nil
|
|
}
|
|
|
|
func (c *FormulaRecognitionContent) Decode() error {
|
|
return json.Unmarshal(c.content, c)
|
|
}
|
|
|
|
// GetPreferredContent 按优先级返回公式内容
|
|
func (c *FormulaRecognitionContent) GetContent() interface{} {
|
|
c.Decode()
|
|
if c.EnhanceLatex != "" {
|
|
return c.EnhanceLatex
|
|
} else if c.AdjustLatex != "" {
|
|
return c.AdjustLatex
|
|
} else {
|
|
return c.Latex
|
|
}
|
|
}
|
|
|
|
type RecognitionResult struct {
|
|
BaseModel
|
|
TaskID int64 `gorm:"column:task_id;bigint;not null;default:0;comment:任务ID" json:"task_id"`
|
|
TaskType TaskType `gorm:"column:task_type;varchar(16);not null;comment:任务类型;default:''" json:"task_type"`
|
|
Content JSON `gorm:"column:content;type:json;not null;comment:识别内容" json:"content"`
|
|
}
|
|
|
|
// NewContentCodec 创建对应任务类型的内容编解码器
|
|
func (r *RecognitionResult) NewContentCodec() ContentCodec {
|
|
switch r.TaskType {
|
|
case TaskTypeFormula:
|
|
return &FormulaRecognitionContent{content: r.Content}
|
|
default:
|
|
return nil
|
|
}
|
|
}
|
|
|
|
type RecognitionResultDao struct {
|
|
}
|
|
|
|
func NewRecognitionResultDao() *RecognitionResultDao {
|
|
return &RecognitionResultDao{}
|
|
}
|
|
|
|
// 模型方法
|
|
func (dao *RecognitionResultDao) Create(tx *gorm.DB, data RecognitionResult) error {
|
|
return tx.Create(&data).Error
|
|
}
|
|
|
|
func (dao *RecognitionResultDao) GetByTaskID(tx *gorm.DB, taskID int64) (result *RecognitionResult, err error) {
|
|
result = &RecognitionResult{}
|
|
err = tx.Where("task_id = ?", taskID).First(result).Error
|
|
if err != nil && err == gorm.ErrRecordNotFound {
|
|
return nil, nil
|
|
}
|
|
return
|
|
}
|
|
|
|
func (dao *RecognitionResultDao) Update(tx *gorm.DB, id int64, updates map[string]interface{}) error {
|
|
return tx.Model(&RecognitionResult{}).Where("id = ?", id).Updates(updates).Error
|
|
}
|