84 lines
2.9 KiB
Go
84 lines
2.9 KiB
Go
package entity
|
|
|
|
import (
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// Album 相册实体
|
|
type Album struct {
|
|
ID uint `json:"id" gorm:"primarykey"`
|
|
Title string `json:"title" gorm:"not null;size:200"`
|
|
Description string `json:"description" gorm:"type:text"`
|
|
Slug string `json:"slug" gorm:"uniqueIndex;size:255"`
|
|
CoverPhotoID *uint `json:"cover_photo_id" gorm:"index"`
|
|
UserID uint `json:"user_id" gorm:"not null;index"`
|
|
CategoryID *uint `json:"category_id" gorm:"index"`
|
|
IsPublic bool `json:"is_public" gorm:"default:true;index"`
|
|
IsFeatured bool `json:"is_featured" gorm:"default:false;index"`
|
|
Password string `json:"-" gorm:"size:255"` // 私密相册密码
|
|
ViewCount int `json:"view_count" gorm:"default:0;index"`
|
|
LikeCount int `json:"like_count" gorm:"default:0;index"`
|
|
PhotoCount int `json:"photo_count" gorm:"default:0;index"`
|
|
SortOrder int `json:"sort_order" gorm:"default:0;index"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
DeletedAt gorm.DeletedAt `json:"deleted_at" gorm:"index"`
|
|
|
|
// 关联
|
|
User User `json:"user,omitempty" gorm:"foreignKey:UserID"`
|
|
Category *Category `json:"category,omitempty" gorm:"foreignKey:CategoryID"`
|
|
CoverPhoto *Photo `json:"cover_photo,omitempty" gorm:"foreignKey:CoverPhotoID"`
|
|
Photos []Photo `json:"photos,omitempty" gorm:"foreignKey:AlbumID"`
|
|
}
|
|
|
|
// AlbumStats 相册统计信息
|
|
type AlbumStats struct {
|
|
Total int64 `json:"total"` // 总相册数
|
|
Published int64 `json:"published"` // 已发布相册数
|
|
Private int64 `json:"private"` // 私有相册数
|
|
Featured int64 `json:"featured"` // 精选相册数
|
|
TotalViews int64 `json:"total_views"` // 总浏览量
|
|
TotalLikes int64 `json:"total_likes"` // 总点赞数
|
|
TotalPhotos int64 `json:"total_photos"` // 总照片数
|
|
CategoryCounts map[string]int64 `json:"category_counts"` // 各分类相册数量
|
|
}
|
|
|
|
// TableName 指定表名
|
|
func (Album) TableName() string {
|
|
return "albums"
|
|
}
|
|
|
|
// HasPassword 检查是否设置了密码
|
|
func (a *Album) HasPassword() bool {
|
|
return a.Password != ""
|
|
}
|
|
|
|
// IsEmpty 检查相册是否为空
|
|
func (a *Album) IsEmpty() bool {
|
|
return a.PhotoCount == 0
|
|
}
|
|
|
|
// CanViewBy 检查指定用户是否可以查看相册
|
|
func (a *Album) CanViewBy(user *User) bool {
|
|
// 公开相册
|
|
if a.IsPublic && !a.HasPassword() {
|
|
return true
|
|
}
|
|
|
|
// 相册所有者或管理员
|
|
if user != nil && (user.ID == a.UserID || user.IsAdmin()) {
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
// CanEditBy 检查指定用户是否可以编辑相册
|
|
func (a *Album) CanEditBy(user *User) bool {
|
|
if user == nil {
|
|
return false
|
|
}
|
|
return user.ID == a.UserID || user.IsAdmin()
|
|
} |