mirror of
https://github.com/snowykami/neo-blog.git
synced 2025-09-26 19:16:24 +00:00
87 lines
2.3 KiB
Go
87 lines
2.3 KiB
Go
package model
|
||
|
||
import (
|
||
"github.com/snowykami/neo-blog/internal/dto"
|
||
"github.com/snowykami/neo-blog/pkg/constant"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
type PostBase struct {
|
||
Title string `gorm:"type:text;not null"`
|
||
Cover string `gorm:"type:text"`
|
||
Content string `gorm:"type:text;not null"`
|
||
Type string `gorm:"type:text;default:markdown"`
|
||
CategoryID uint `gorm:"index"`
|
||
Category Category `gorm:"foreignKey:CategoryID;references:ID"`
|
||
Labels []Label `gorm:"many2many:post_labels;constraint:OnUpdate:CASCADE,OnDelete:SET NULL;"`
|
||
IsOriginal bool `gorm:"default:true"`
|
||
IsPrivate bool `gorm:"default:false"`
|
||
}
|
||
type Post struct {
|
||
gorm.Model
|
||
UserID uint `gorm:"index"` // 发布者的用户ID
|
||
User User `gorm:"foreignKey:UserID;references:ID"` // 关联的用户
|
||
// core fields
|
||
PostBase
|
||
//
|
||
LikeCount uint64
|
||
CommentCount uint64
|
||
ViewCount uint64
|
||
Heat uint64
|
||
}
|
||
|
||
// CalculateHeat 热度计算
|
||
func (p *Post) CalculateHeat() float64 {
|
||
return float64(
|
||
p.LikeCount*constant.HeatFactorLikeWeight +
|
||
p.CommentCount*constant.HeatFactorCommentWeight +
|
||
p.ViewCount*constant.HeatFactorViewWeight,
|
||
)
|
||
}
|
||
|
||
func (p *Post) ToDto() *dto.PostDto {
|
||
return &dto.PostDto{
|
||
ID: p.ID,
|
||
UserID: p.UserID,
|
||
Title: p.Title,
|
||
Content: p.Content,
|
||
Cover: p.Cover,
|
||
Type: p.Type,
|
||
IsOriginal: p.IsOriginal,
|
||
IsPrivate: p.IsPrivate,
|
||
Labels: func() []dto.LabelDto {
|
||
labelDtos := make([]dto.LabelDto, len(p.Labels))
|
||
for i, label := range p.Labels {
|
||
labelDtos[i] = label.ToDto()
|
||
}
|
||
return labelDtos
|
||
}(),
|
||
LikeCount: p.LikeCount,
|
||
CommentCount: p.CommentCount,
|
||
ViewCount: p.ViewCount,
|
||
Heat: uint64(p.CalculateHeat()),
|
||
CreatedAt: p.CreatedAt,
|
||
UpdatedAt: p.UpdatedAt,
|
||
User: p.User.ToDto(),
|
||
}
|
||
}
|
||
|
||
// ToDtoWithShortContent 返回一个简化的 DTO，内容可以根据需要截断
|
||
func (p *Post) ToDtoWithShortContent(contentLength int) *dto.PostDto {
|
||
dtoPost := p.ToDto()
|
||
if len(p.Content) > contentLength {
|
||
dtoPost.Content = p.Content[:contentLength] + "..."
|
||
} else {
|
||
dtoPost.Content = p.Content
|
||
}
|
||
return dtoPost
|
||
}
|
||
|
||
// Draft 草稿
|
||
type Draft struct {
|
||
gorm.Model
|
||
PostID uint `gorm:"uniqueIndex"` // 关联的文章ID
|
||
Post Post `gorm:"foreignKey:PostID;references:ID"`
|
||
PostBase
|
||
}
|