## CI 修复 (P0/P1) P0 — 编译阻塞: - interfaces.go: postUseCase ISP 接口移除不用的 Create/Update/Delete P1 — 必须修复: - ST1000: 为 13 个包添加包注释 (common/config/model/service/...) - ST1005: redis_store.go 全部错误消息改为小写开头 - errcheck (19处): defer Close()→闭包忽略, notifier.Create→_=, r.Run→检查error - errorlint (9处): switch-on-error→errors.Is 链, ==→errors.Is - ST1020/ST1022: 导出符号注释以符号名开头 P2 — 安全评审: - gosec (12处): G203/G301/G304/G306 添加 nolint 注释并附理由 P3 — 清理: - unused: 移除 hasUnicode/energyStore/current/energyAdminUseCase - gofumpt + goimports 全量格式化 (35+ 文件)
143 lines
3.8 KiB
Go
143 lines
3.8 KiB
Go
package session
|
||
|
||
import (
|
||
"log"
|
||
"time"
|
||
|
||
"metazone.cc/mce/internal/config"
|
||
"metazone.cc/mce/internal/model"
|
||
)
|
||
|
||
// userStore SessionManager 对 UserRepo 的最小接口(ISP:1 个方法)
|
||
type userStore interface {
|
||
FindByIDForAuth(userID uint) (*model.User, error)
|
||
}
|
||
|
||
// Manager 会话管理器:创建、验证、销毁、续期
|
||
type Manager struct {
|
||
store Store
|
||
userRepo userStore
|
||
cfg *config.Config
|
||
}
|
||
|
||
// NewManager 构造函数
|
||
func NewManager(store Store, userRepo userStore, cfg *config.Config) *Manager {
|
||
return &Manager{store: store, userRepo: userRepo, cfg: cfg}
|
||
}
|
||
|
||
// Create 创建会话并写入存储
|
||
func (m *Manager) Create(user *model.User, rememberMe bool, ip, userAgent string) (string, error) {
|
||
sid, err := NewID()
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
now := time.Now()
|
||
s := &Session{
|
||
ID: sid,
|
||
UserID: user.ID,
|
||
Email: user.Email,
|
||
Username: user.Username,
|
||
Avatar: user.Avatar,
|
||
Role: user.Role,
|
||
Status: user.Status,
|
||
Exp: user.Exp,
|
||
RememberMe: rememberMe,
|
||
IP: ip,
|
||
UserAgent: userAgent,
|
||
CreatedAt: now,
|
||
LastAccess: now,
|
||
}
|
||
if err := m.store.Set(s); err != nil {
|
||
return "", err
|
||
}
|
||
log.Printf("[SessionManager] Created sid=%s uid=%d rememberMe=%v ip=%s", sid[:16]+"...", user.ID, rememberMe, ip)
|
||
return sid, nil
|
||
}
|
||
|
||
// Validate 验证会话:读取 → 检查过期 → 检查用户状态 → 续期 → 返回
|
||
// 若用户不存在则返回 nil;被封禁/锁定用户仍返回 session(由上层中间件决定可访问范围)
|
||
func (m *Manager) Validate(sid string) (*Session, error) {
|
||
if sid == "" {
|
||
return nil, nil
|
||
}
|
||
|
||
s, err := m.store.Get(sid)
|
||
if err != nil || s == nil {
|
||
return nil, nil
|
||
}
|
||
|
||
// 检查用户是否仍然存在
|
||
user, err := m.userRepo.FindByIDForAuth(s.UserID)
|
||
if err != nil {
|
||
log.Printf("[SessionManager] Validate: user not found uid=%d err=%v", s.UserID, err)
|
||
_ = m.store.Delete(sid)
|
||
return nil, nil
|
||
}
|
||
|
||
// 同步 DB 最新字段(头像/用户名/角色/状态/经验值等可能已变更)
|
||
s.Avatar = user.Avatar
|
||
s.Username = user.Username
|
||
s.Role = user.Role
|
||
s.Status = user.Status
|
||
s.Exp = user.Exp
|
||
|
||
// 滑动窗口续期
|
||
s.Touch()
|
||
if err := m.store.Set(s); err != nil {
|
||
log.Printf("[SessionManager] Validate: touch failed sid=%s err=%v", sid[:16]+"...", err)
|
||
}
|
||
|
||
return s, nil
|
||
}
|
||
|
||
// Destroy 销毁单个会话(用户主动退出登录)
|
||
func (m *Manager) Destroy(sid string) error {
|
||
return m.store.Delete(sid)
|
||
}
|
||
|
||
// DestroyByUID 销毁某用户的所有会话(改密/注销/强制下线)
|
||
func (m *Manager) DestroyByUID(uid uint) error {
|
||
return m.store.DeleteByUID(uid)
|
||
}
|
||
|
||
// IdleTimeout 根据 rememberMe 返回对应的空闲超时时间
|
||
func (m *Manager) IdleTimeout(rememberMe bool) time.Duration {
|
||
if rememberMe {
|
||
return time.Duration(m.cfg.Session.RememberTimeout) * time.Minute
|
||
}
|
||
return time.Duration(m.cfg.Session.IdleTimeout) * time.Minute
|
||
}
|
||
|
||
// ListByUID 列出某用户的所有活跃会话
|
||
func (m *Manager) ListByUID(uid uint) ([]*Session, error) {
|
||
return m.store.ListByUID(uid)
|
||
}
|
||
|
||
// DestroyOtherByUID 销毁某用户除当前会话外的所有会话
|
||
func (m *Manager) DestroyOtherByUID(uid uint, currentSID string) error {
|
||
return m.store.DeleteByUIDExclude(uid, currentSID)
|
||
}
|
||
|
||
// UpdateRemark 更新指定会话的备注(需验证归属)
|
||
func (m *Manager) UpdateRemark(sid string, uid uint, remark string) error {
|
||
s, err := m.store.Get(sid)
|
||
if err != nil || s == nil {
|
||
return err
|
||
}
|
||
if s.UserID != uid {
|
||
return nil
|
||
}
|
||
s.Remark = remark
|
||
return m.store.Set(s)
|
||
}
|
||
|
||
// UpdateSession 直接更新会话存储(用于同步经验值等字段)
|
||
func (m *Manager) UpdateSession(s *Session) error {
|
||
return m.store.Set(s)
|
||
}
|
||
|
||
// GetStoreMetrics 返回当前存储的状态信息
|
||
func (m *Manager) GetStoreMetrics() StoreMetrics {
|
||
return m.store.Metrics()
|
||
}
|