init repo

internal/storage/dao/result.go

@@ -0,0 +1,89 @@
+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
+}