moredoc/model/documentScore.go

package model

import (
	"fmt"
	"time"

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

type DocumentScore struct {
	Id         int64      `form:"id" json:"id,omitempty" gorm:"primaryKey;autoIncrement;column:id;comment:;"`
	DocumentId int64      `form:"document_id" json:"document_id,omitempty" gorm:"column:document_id;type:bigint(20);size:20;default:0;comment:文档ID;index:idx_document_user,unique"`
	UserId     int64      `form:"user_id" json:"user_id,omitempty" gorm:"column:user_id;type:bigint(20);size:20;default:0;comment:用户ID;index:idx_document_user,unique"`
	Score      int        `form:"score" json:"score,omitempty" gorm:"column:score;type:int(11);size:11;default:0;comment:文档评分值，3位数，如500表示5分;"`
	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 (DocumentScore) TableName() string {
	return tablePrefix + "document_score"
}

// CreateDocumentScore 创建DocumentScore
func (m *DBModel) CreateDocumentScore(documentScore *DocumentScore) (err error) {
	doc, _ := m.GetDocument(documentScore.DocumentId, "id", "score_count", "score")
	if doc.Id == 0 {
		err = fmt.Errorf("文档不存在")
		return
	}

	tx := m.db.Begin()
	defer func() {
		if err != nil {
			tx.Rollback()
		} else {
			tx.Commit()
		}
	}()

	// 创建评分记录
	err = tx.Create(documentScore).Error
	if err != nil {
		m.logger.Error("CreateDocumentScore", zap.Error(err))
		return
	}

	score := (documentScore.Score + doc.Score*doc.ScoreCount) / (doc.ScoreCount + 1)
	// 对应文档，评分总数+1，评分总值+score
	err = tx.Model(&Document{}).Where("id = ?", documentScore.DocumentId).Updates(
		map[string]interface{}{
			"score_count": gorm.Expr("score_count + ?", 1),
			"score":       score,
		},
	).Error
	if err != nil {
		m.logger.Error("CreateDocumentScore", zap.Error(err))
		return
	}
	return
}

// GetDocumentScore 获取DocumentScore
func (m *DBModel) GetDocumentScore(userId, documentId int64) (documentScore DocumentScore, err error) {
	err = m.db.Where("user_id = ? and document_id = ?", userId, documentId).Find(&documentScore).Error
	return
}