feat: 实时消息与管理端运营能力
新增群聊/私信与公共 WebSocket 总线,站长用户内容档案,并将通知与待审角标改为推送驱动;同步精简管理仪表盘并修复主题换肤 DOM 冲突。 Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
@@ -3,6 +3,7 @@ package model
|
||||
import (
|
||||
"crypto/sha256"
|
||||
"encoding/hex"
|
||||
"errors"
|
||||
"fmt"
|
||||
"log"
|
||||
|
||||
@@ -29,9 +30,15 @@ func InitDB(dsn string) error {
|
||||
return fmt.Errorf("refresh token 旧数据迁移失败: %w", err)
|
||||
}
|
||||
|
||||
// 清理前次失败迁移遗留的 FK 约束(constraint:- 标签防止新建,但不删旧的)
|
||||
if err := dropStaleChatFKConstraints(db); err != nil {
|
||||
return fmt.Errorf("清理遗留 FK 约束失败: %w", err)
|
||||
}
|
||||
|
||||
if err := db.AutoMigrate(
|
||||
&User{}, &Board{}, &Post{}, &Comment{}, &RefreshToken{}, &Like{}, &Notification{}, &Checkin{},
|
||||
&Announcement{}, &SiteSetting{}, &Attachment{}, &UserBoard{}, &LoginLog{},
|
||||
&ChatRoom{}, &ChatRoomMember{}, &ChatMessage{},
|
||||
); err != nil {
|
||||
return fmt.Errorf("自动迁移失败: %w", err)
|
||||
}
|
||||
@@ -54,7 +61,20 @@ func InitDB(dsn string) error {
|
||||
}
|
||||
|
||||
DB = db
|
||||
if err := db.Exec(`
|
||||
CREATE UNIQUE INDEX IF NOT EXISTS idx_chat_direct_key
|
||||
ON chat_rooms (direct_key)
|
||||
WHERE room_type = 'direct' AND deleted_at IS NULL AND direct_key <> ''
|
||||
`).Error; err != nil {
|
||||
return fmt.Errorf("私聊唯一索引创建失败: %w", err)
|
||||
}
|
||||
seedDefaultBoards(db)
|
||||
if err := seedDefaultChatRoom(db); err != nil {
|
||||
return fmt.Errorf("默认群聊大厅初始化失败: %w", err)
|
||||
}
|
||||
if err := ensureDefaultChatMemberships(db); err != nil {
|
||||
return fmt.Errorf("默认群聊入群补齐失败: %w", err)
|
||||
}
|
||||
log.Println("[model] PostgreSQL 数据库初始化完成")
|
||||
return nil
|
||||
}
|
||||
@@ -71,6 +91,22 @@ func PingDB() error {
|
||||
return sqlDB.Ping()
|
||||
}
|
||||
|
||||
// dropStaleChatFKConstraints 清理前次迁移失败遗留的 FK 约束。
|
||||
// GORM AutoMigrate 先建表再加约束:首次迁移时 Notification.Room 的 FK 添加失败,
|
||||
// 但 chat_rooms 表已建好且带 fk_chat_rooms_last_message 约束。
|
||||
// 模型已改用 constraint:- 防止新建,但 GORM 不会删除已存在的约束,需手动清理。
|
||||
func dropStaleChatFKConstraints(db *gorm.DB) error {
|
||||
// 对每个遗留约束名,在可能的两张表上都尝试删除(IF EXISTS 保证幂等)
|
||||
pairs := []struct{ table, constraint string }{
|
||||
{"chat_rooms", "fk_chat_rooms_last_message"},
|
||||
{"notifications", "fk_notifications_room"},
|
||||
}
|
||||
for _, p := range pairs {
|
||||
db.Exec(fmt.Sprintf(`ALTER TABLE IF EXISTS %s DROP CONSTRAINT IF EXISTS %s`, p.table, p.constraint))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// prepareRefreshTokenMigration 旧版 refresh_tokens 表把明文存在 token 列,
|
||||
// 新版改为 token_hash(SHA-256,NOT NULL+唯一索引)。在 AutoMigrate 之前:
|
||||
// 1. 新增可带默认值的 token_hash 列(避免对存量行加 NOT NULL 列失败)
|
||||
@@ -193,3 +229,92 @@ func seedDefaultBoards(db *gorm.DB) {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// seedDefaultChatRoom 确保存在唯一的全站大厅(公开默认群)
|
||||
func seedDefaultChatRoom(db *gorm.DB) error {
|
||||
var n int64
|
||||
if err := db.Model(&ChatRoom{}).Where("is_default = ?", true).Count(&n).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
if n > 0 {
|
||||
return nil
|
||||
}
|
||||
var owner User
|
||||
if err := db.Where("role = ?", RoleOwner).Order("id ASC").First(&owner).Error; err != nil {
|
||||
// 尚无用户时跳过,等有站长后再由下次启动补齐
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil
|
||||
}
|
||||
// 尝试任意用户
|
||||
if err2 := db.Order("id ASC").First(&owner).Error; err2 != nil {
|
||||
return nil
|
||||
}
|
||||
}
|
||||
room := ChatRoom{
|
||||
Name: "全站大厅",
|
||||
Description: "所有成员默认加入的公开群聊",
|
||||
OwnerID: owner.ID,
|
||||
RoomType: ChatRoomTypeGroup,
|
||||
IsPrivate: false,
|
||||
IsDefault: true,
|
||||
MemberCount: 0,
|
||||
}
|
||||
if err := db.Create(&room).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
log.Println("[model] 已创建默认群聊「全站大厅」")
|
||||
return nil
|
||||
}
|
||||
|
||||
// ensureDefaultChatMemberships 把未封禁用户补进默认大厅(幂等)
|
||||
func ensureDefaultChatMemberships(db *gorm.DB) error {
|
||||
var room ChatRoom
|
||||
if err := db.Where("is_default = ?", true).First(&room).Error; err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil
|
||||
}
|
||||
return err
|
||||
}
|
||||
var userIDs []uint
|
||||
if err := db.Model(&User{}).
|
||||
Where("deleted_at IS NULL AND banned = ?", false).
|
||||
Pluck("id", &userIDs).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
if len(userIDs) == 0 {
|
||||
return nil
|
||||
}
|
||||
var existing []uint
|
||||
if err := db.Model(&ChatRoomMember{}).
|
||||
Where("room_id = ?", room.ID).
|
||||
Pluck("user_id", &existing).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
have := make(map[uint]struct{}, len(existing))
|
||||
for _, id := range existing {
|
||||
have[id] = struct{}{}
|
||||
}
|
||||
added := 0
|
||||
for _, uid := range userIDs {
|
||||
if _, ok := have[uid]; ok {
|
||||
continue
|
||||
}
|
||||
role := ChatRoleMember
|
||||
if uid == room.OwnerID {
|
||||
role = ChatRoleOwner
|
||||
}
|
||||
m := ChatRoomMember{RoomID: room.ID, UserID: uid, Role: role}
|
||||
if err := db.Create(&m).Error; err != nil {
|
||||
continue
|
||||
}
|
||||
added++
|
||||
}
|
||||
if added > 0 {
|
||||
_ = db.Model(&ChatRoom{}).Where("id = ?", room.ID).
|
||||
UpdateColumn("member_count", gorm.Expr(
|
||||
"(SELECT count(*) FROM chat_room_members WHERE room_id = ?)", room.ID,
|
||||
)).Error
|
||||
log.Printf("[model] 默认群聊补齐入群 %d 人", added)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user