zhangchao/go_typecho
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加配置文件

Browse Source
This commit is contained in:
zhangchao
2026-06-02 13:52:47 +08:00
parent 6117222625
commit aab3aeb5b2
31 changed files with 3326 additions and 198 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/models.go
+213 -57
View File
@@ -1,52 +1,103 @@
package models
import (
"log/slog"
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
// 用户模型
type User struct {
ID uint `gorm:"primarykey" json:"id"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
Username string `gorm:"uniqueIndex;size:50;not null" json:"username"`
Password string `gorm:"size:255;not null" json:"-"`
Nickname string `gorm:"size:100" json:"nickname"`
Email string `gorm:"size:100" json:"email"`
Avatar string `gorm:"size:255" json:"avatar"`
Role string `gorm:"size:20;default:'user'" json:"role"` // admin, user
Status int `gorm:"default:1" json:"status"` // 1:启用 0:禁用
}
const PasswordCryptLevel = 12
// 文章模型
type Post struct {
ID uint `gorm:"primarykey" json:"id"`
// ==================== 用户模型 ====================
type User struct {
ID uint `gorm:"primaryKey" json:"id"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
Title string `gorm:"size:200;not null" json:"title"`
Slug string `gorm:"uniqueIndex;size:200" json:"slug"`
Content string `gorm:"type:text" json:"content"`
Summary string `gorm:"size:500" json:"summary"`
Cover string `gorm:"size:255" json:"cover"`
AuthorID uint `json:"author_id"`
Author User `gorm:"foreignKey:AuthorID" json:"author"`
CategoryID uint `json:"category_id"`
Category Category `gorm:"foreignKey:CategoryID" json:"category"`
Tags []Tag `gorm:"many2many:post_tags;" json:"tags"`
Views int `gorm:"default:0" json:"views"`
Status string `gorm:"size:20;default:'draft'" json:"status"` // published, draft, private
IsTop bool `gorm:"default:false" json:"is_top"`
PublishedAt *time.Time `json:"published_at"`
Comments []Comment `json:"comments,omitempty"`
Username string `gorm:"uniqueIndex;size:50;not null" json:"username"`
Password string `gorm:"size:255;not null" json:"-"`
Nickname string `gorm:"size:100" json:"nickname"`
Email string `gorm:"size:100" json:"email"`
Avatar string `gorm:"size:255" json:"avatar"`
Role string `gorm:"size:20;default:'user'" json:"role"` // admin, user
Status int `gorm:"default:1" json:"status"` // 1:启用 0:禁用
IsActive bool `gorm:"default:true" json:"is_active"` // 账号激活状态
LastLoginAt *time.Time `json:"last_login_at"` // 最后登录时间
LoginCount int `gorm:"default:0" json:"login_count"` // 登录次数统计
}
// 分类模型
// TableName 指定表名
func (User) TableName() string {
return "users"
}
// SetPassword 设置密码加密
func (u *User) SetPassword(password string) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(password), PasswordCryptLevel)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
// CheckPassword 验证密码是否匹配
func (u *User) CheckPassword(password string) bool {
err := bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
return err == nil
}
// RecordLogin 记录登录信息
func (u *User) RecordLogin(db *gorm.DB) error {
now := time.Now()
u.LastLoginAt = &now
u.LoginCount++
return db.Model(u).Updates(map[string]interface{}{
"last_login_at": now,
"login_count": u.LoginCount,
}).Error
}
// ==================== 文章模型 ====================
type Post struct {
ID uint `gorm:"primaryKey" json:"id"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
Title string `gorm:"size:200;not null" json:"title"`
Slug string `gorm:"uniqueIndex;size:200" json:"slug"`
Content string `gorm:"type:text" json:"content"`
Summary string `gorm:"size:500" json:"summary"`
Cover string `gorm:"size:255" json:"cover"`
AuthorID uint `gorm:"index" json:"author_id"` // 增加索引
Author User `gorm:"foreignKey:AuthorID" json:"author"`
CategoryID uint `gorm:"index" json:"category_id"` // 增加索引
Category Category `gorm:"foreignKey:CategoryID" json:"category"`
Tags []Tag `gorm:"many2many:post_tags;" json:"tags"`
Views int `gorm:"default:0" json:"views"`
Status string `gorm:"index;size:20;default:'draft'" json:"status"` // 增加索引
IsTop bool `gorm:"default:false" json:"is_top"`
PublishedAt *time.Time `json:"published_at"`
Password string `gorm:"size:255" json:"password"` // 访问密码保护
AllowComment bool `gorm:"default:true" json:"allow_comment"` // 是否允许评论
AllowPing bool `gorm:"default:true" json:"allow_ping"` // 是否允许Pingback
Template string `gorm:"size:100" json:"template"` // 自定义模板文件
Comments []Comment `json:"comments,omitempty"`
}
// TableName 指定表名
func (Post) TableName() string {
return "posts"
}
// ==================== 分类模型 ====================
type Category struct {
ID uint `gorm:"primarykey" json:"id"`
ID uint `gorm:"primaryKey" json:"id"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
Name string `gorm:"size:100;not null" json:"name"`
@@ -56,9 +107,15 @@ type Category struct {
PostCount int `gorm:"default:0" json:"post_count"`
}
// 标签模型
// TableName 指定表名
func (Category) TableName() string {
return "categories"
}
// ==================== 标签模型 ====================
type Tag struct {
ID uint `gorm:"primarykey" json:"id"`
ID uint `gorm:"primaryKey" json:"id"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
Name string `gorm:"size:100;not null" json:"name"`
@@ -66,13 +123,19 @@ type Tag struct {
PostCount int `gorm:"default:0" json:"post_count"`
}
// 评论模型
// TableName 指定表名
func (Tag) TableName() string {
return "tags"
}
// ==================== 评论模型 ====================
type Comment struct {
ID uint `gorm:"primarykey" json:"id"`
ID uint `gorm:"primaryKey" json:"id"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
PostID uint `json:"post_id"`
PostID uint `gorm:"index" json:"post_id"` // 增加索引
Post Post `gorm:"foreignKey:PostID" json:"-"`
ParentID *uint `json:"parent_id"`
Author string `gorm:"size:100" json:"author"`
@@ -80,30 +143,123 @@ type Comment struct {
Website string `gorm:"size:255" json:"website"`
Content string `gorm:"type:text;not null" json:"content"`
IP string `gorm:"size:50" json:"-"`
Status string `gorm:"size:20;default:'pending'" json:"status"` // approved, pending, spam
Status string `gorm:"index;size:20;default:'pending'" json:"status"` // 增加索引
UserID *uint `json:"user_id"`
User *User `gorm:"foreignKey:UserID" json:"user,omitempty"`
Children []Comment `gorm:"foreignKey:ParentID" json:"children,omitempty"`
// 反垃圾字段
UserAgent string `gorm:"size:255" json:"-"` // 用户代理
IsSpam bool `gorm:"default:false" json:"is_spam"` // 是否为垃圾评论
SpamScore float64 `gorm:"default:0" json:"spam_score"` // 垃圾评分
}
// 设置模型
// TableName 指定表名
func (Comment) TableName() string {
return "comments"
}
// ==================== 设置模型 (优化) ====================
type Option struct {
ID uint `gorm:"primarykey"`
Key string `gorm:"uniqueIndex;size:100"`
Value string `gorm:"type:text"`
Name string `gorm:"column:name;type:varchar(100);primaryKey" json:"name"` // 配置键名
UserID uint `gorm:"column:user_id;type:int;default:0;primaryKey;index" json:"user_id"` // 用户ID(0=全局配置)
Value string `gorm:"column:value;type:text" json:"value"` // 配置值
}
// 页面模型（独立页面）
type Page struct {
ID uint `gorm:"primarykey" json:"id"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
Title string `gorm:"size:200;not null" json:"title"`
Slug string `gorm:"uniqueIndex;size:200" json:"slug"`
Content string `gorm:"type:text" json:"content"`
AuthorID uint `json:"author_id"`
Author User `gorm:"foreignKey:AuthorID" json:"author"`
Status string `gorm:"size:20;default:'draft'" json:"status"`
Order int `gorm:"default:0" json:"order"`
// TableName 指定表名
func (Option) TableName() string {
return "options"
}
// GetOptionValue 根据name和userID获取配置值
func GetOptionValue(db *gorm.DB, name string, userID uint) (string, error) {
var option Option
result := db.Where("name = ? AND user_id = ?", name, userID).First(&option)
if result.Error != nil {
return "", result.Error
}
return option.Value, nil
}
// SetOptionValue 根据name和userID设置配置值(存在则更新,不存在则新增)
func SetOptionValue(db *gorm.DB, name string, userID uint, value string) error {
// 先尝试查找
var option Option
result := db.Where("name = ? AND user_id = ?", name, userID).First(&option)
if result.Error == nil {
// 记录存在，更新
return db.Model(&option).Update("value", value).Error
}
// 记录不存在，创建新记录
option = Option{
Name: name,
UserID: userID,
Value: value,
}
return db.Create(&option).Error
}
// DeleteOption 删除配置项
func DeleteOption(db *gorm.DB, name string, userID uint) error {
result := db.Where("name = ? AND user_id = ?", name, userID).Delete(&Option{})
return result.Error
}
// GetOptionsByUser 获取用户的所有配置
func GetOptionsByUser(db *gorm.DB, userID uint) (map[string]string, error) {
var options []Option
result := db.Where("user_id = ?", userID).Find(&options)
if result.Error != nil {
return nil, result.Error
}
optionsMap := make(map[string]string)
for _, opt := range options {
optionsMap[opt.Name] = opt.Value
}
return optionsMap, nil
}
// BatchSetOptions 批量设置配置
func BatchSetOptions(db *gorm.DB, userID uint, options map[string]string) error {
return db.Transaction(func(tx *gorm.DB) error {
for name, value := range options {
opt := Option{
Name: name,
UserID: userID,
Value: value,
}
if err := tx.Save(&opt).Error; err != nil {
slog.Error("保存配置失败", "name", name, "error", err)
return err
}
}
return nil
})
}
// ==================== 页面模型(优化) ====================
type Page struct {
ID uint `gorm:"primaryKey" json:"id"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
Title string `gorm:"size:200;not null" json:"title"`
Slug string `gorm:"uniqueIndex;size:200" json:"slug"`
Content string `gorm:"type:text" json:"content"`
AuthorID uint `gorm:"index" json:"author_id"`
Author User `gorm:"foreignKey:AuthorID" json:"author"`
Status string `gorm:"index;size:20;default:'draft'" json:"status"` // 增加索引
Order int `gorm:"default:0" json:"order"`
Password string `gorm:"size:255" json:"password"` // 访问密码保护
AllowComment bool `gorm:"default:true" json:"allow_comment"` // 是否允许评论
ParentID uint `gorm:"default:0" json:"parent_id"` // 父级页面(支持层级页面)
}
// TableName 指定表名
func (Page) TableName() string {
return "pages"
}