feat: 8 条功能 + 身份口径改为号码布为准（含批量退赛/移出、基础数据导入导出、登录有效期）
用户给出的 8 条功能 + 一条口径变更，全部带真机/真浏览器验证：
① 录入页姓名后直接显示号码布（原「号码」列保留）
② 录入页批次进度条：`第 2 批 已录 5/8 · 3 条有错`，录完变绿；已录判据复用
entryGuards.judgeEntryRow（与发布防呆同源，不会出现"进度条说录完、发布说缺 8 条"）
⑦ 小项列表双击 → 跳到该小项的组别（编辑/删除按钮不误触）
③ 赛事开关 `standings.live_on_publish`（默认开＝保持原行为）：关闭后分批发布只上屏成绩，
名次/流水/榜单一起等该组录完再结算 —— 代理实读代码后纠正了我的前提：
奖牌榜读的是 results.rank（buildRanked）而非流水汇总，只延迟流水会出现
「有金牌但 0 分」。另提供 POST /api/results/settle 人工兜底 + GET /api/results/settlement 未结算清单
④ 组别「就地更新」PUT /api/events/{id}/categories/{categoryId}：有报名数据时也能改名/改年龄/
改性别/改权重（旧路径是整表替换，一改就被 409 拦下）；改名不动报名/成绩/流水（真机逐字节比对）；
权重真的变了才触发该组重算（容差 1e-9）
④辅 运动员列表多选+全选，POST /api/athletes/batch 一次请求完成整批停用/彻底删除（逐条回报，
dryRun 预览走同一段 SQL 但不提交、不写审计）——这才是绕开限流的正解
新增 报名项批量：POST /api/entries/batch（scratch/restore/remove）。退赛只动 registered、
跳过 dq/dns 并说明原因；移出是物理删除，必须显式清 score_ledger（该表 entry_id 是
ON DELETE SET NULL，不显式删会留下"人没了分还在"的孤儿流水，验证基线 orphan=0）；
退赛同时清 results.rank（否则出现"0 分还挂着金牌"）
⑤ GET /api/export/entries.xlsx 报名名单导出（13 列含批次，状态中文与录入页逐字一致）
⑥ 基础数据包：GET /api/export/basedata.xlsx + POST /api/export/basedata/import
（四张表、表头只有一份定义、表头行自动定位、列缺席≠写空、按编码 upsert、绝不删除、
行级中文错误带真实 Excel 行号、枚举走既有校验、新增单位一定是启用）
⑧ 登录有效期面板可配：系统设置新页签，1–720 小时，默认 12h → 24h；滑动续期 +
Cookie 同步刷新（只服务端续期是假的，浏览器侧的 Max-Age 到点仍会丢 Cookie）
身份口径：号码布为唯一标识、证件号退出匹配（号码布不同=同名的两个人，新建并单独报告）；
同一份表重复证件号不再报错；两个不同的人共用证件号仍报行级错误（唯一索引保留）
其它对齐：分项成绩表草稿行不再消失（成绩列「分数未录入」、名次/积分 `—`，没成绩的行同占位）；
大屏实时赛况单栏铺满（留白 370→78px）、字号上限 56→96px、≥7 行自动拆左右两块；
删除报名项/退赛/权重等改动都会重算并推榜单。
测试：服务端 13 包（连跑两遍）、前端 335 例、vue-tsc + build、Console 18 文件 —— 全绿。
文档：README、docs/ 四份（含 debug/使用说明 副本）、devdocs 05/08/04/13。
版本号 dev_0.1.5 → dev_0.1.6。
This commit is contained in:
AskaEth committed
1 parent
8dd599bd97
commit
631c8dd8e1
80 files changed
+12468
-519
No files matched your search
+177
-43
@@ -12,7 +12,9 @@ import (
|
||||
"fmt"
|
||||
"log/slog"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
|
||||
"golang.org/x/crypto/bcrypt"
|
||||
@@ -23,8 +25,33 @@ const (
|
||||
CookieSession = "scoreboard_session"
|
||||
CookieCSRF = "csrf_token"
|
||||
|
||||
// DefaultSessionTTL 是会话有效期：12h。
|
||||
DefaultSessionTTL = 12 * time.Hour
|
||||
// DefaultSessionTTL 是登录有效期的默认值：24h。
|
||||
// 原来是 12h：现场前一晚 20:00 登录，次日早上 8:00 整被踢出登录，
|
||||
// 而那正是比赛进行中的时间——这正是本项需求要修的。
|
||||
DefaultSessionTTL = 24 * time.Hour
|
||||
// DefaultSessionHours 与 DefaultSessionTTL 同值，供以「小时」为单位的配置读写共用。
|
||||
DefaultSessionHours = 24
|
||||
// MinSessionHours / MaxSessionHours 是面板可配的登录有效期范围：1 小时 ~ 720 小时（30 天）。
|
||||
MinSessionHours = 1
|
||||
MaxSessionHours = 720
|
||||
// SessionHoursKey 是系统库 settings 里存登录有效期的键（机器级配置，与赛事级 settings 不共用）。
|
||||
SessionHoursKey = "security.session_hours"
|
||||
// SessionHoursSourceKey 记录 SessionHoursKey 是谁写的：面板写过的值一定是操作员的明确选择，
|
||||
// 不能被下面的历史占位值规则改写。
|
||||
SessionHoursSourceKey = "security.session_hours_source"
|
||||
// SessionHoursSourcePanel 是面板写入时登记的来源标记。
|
||||
SessionHoursSourcePanel = "panel"
|
||||
// legacySessionHoursSeed 是 0005 迁移种下的历史占位值。
|
||||
// 在本项改动之前，没有任何代码读过 security.session_hours，所以库里还留着 12
|
||||
// 只可能是「没人配过」；它按「未配置」处理，用新的默认 24，否则新装的库会永远停在 12。
|
||||
legacySessionHoursSeed = "12"
|
||||
// renewThreshold 是滑动续期的写入节流窗口：剩余寿命不足「有效期 - 1 分钟」时才回写 expires_at。
|
||||
// 每个请求都续期会让 sessions 变成热点写表（现场几十个客户端轮询），
|
||||
// 而少顺延这 1 分钟对「会不会掉线」没有任何影响。
|
||||
renewThreshold = time.Minute
|
||||
// touchInterval 是 last_seen_at 的回写节流窗口：没有续期时，活动时间最快每分钟写一次。
|
||||
// 一次页面加载会打若干个请求，逐个回写纯属放大写压力；last_seen_at 只用于排障，分钟级足够。
|
||||
touchInterval = time.Minute
|
||||
// DefaultBcryptCost 满足文档「bcrypt cost ≥ 12」。
|
||||
DefaultBcryptCost = 12
|
||||
// DefaultLoginPerMinute 登录限流：5 次/分钟/IP。防爆破用，别跟着写限流一起放大。
|
||||
@@ -73,7 +100,10 @@ type Session struct {
|
||||
|
||||
// Options 是认证服务的可调参数；零值即文档默认。
|
||||
type Options struct {
|
||||
Logger *slog.Logger
|
||||
Logger *slog.Logger
|
||||
// SessionTTL 显式指定登录有效期（测试 / 嵌入用）。优先级最低的那一档默认值：
|
||||
// 只要系统库 settings 里有有效的 security.session_hours（面板写的就是它），就以面板为准；
|
||||
// 这里只在系统库里没有配置时兜底（例如测试用的空库）。
|
||||
SessionTTL time.Duration
|
||||
BcryptCost int
|
||||
LoginPerMinute int
|
||||
@@ -88,12 +118,14 @@ type Options struct {
|
||||
type Service struct {
|
||||
db *sql.DB
|
||||
log *slog.Logger
|
||||
sessionTTL time.Duration
|
||||
bcryptCost int
|
||||
secureCookies bool
|
||||
now func() time.Time
|
||||
loginLimiter *Limiter
|
||||
writeLimiter *Limiter
|
||||
// ttl 是当前生效的登录有效期；面板可以随时改（SetSessionTTL），
|
||||
// 所以每个请求都会读它，必须用原子量而不是普通字段。
|
||||
ttl atomic.Int64
|
||||
loginLimiter *Limiter
|
||||
writeLimiter *Limiter
|
||||
}
|
||||
|
||||
// New 构建认证服务；db 为系统库（data/system.db）。
|
||||
@@ -102,9 +134,14 @@ func New(db *sql.DB, opts Options) *Service {
|
||||
if logger == nil {
|
||||
logger = slog.Default()
|
||||
}
|
||||
ttl := opts.SessionTTL
|
||||
if ttl <= 0 {
|
||||
ttl = DefaultSessionTTL
|
||||
// 优先级：面板（系统库设置）> 启动参数 / 配置文件 > 内置默认。
|
||||
// 面板排在最前是为了不出现「改了面板没反应」：只要系统库里有一个有效值，
|
||||
// 它就是这个进程的有效期，别处的默认值一律让位。
|
||||
ttl, source := DefaultSessionTTL, "内置默认"
|
||||
if dbTTL, ok := sessionTTLFromSettings(db); ok {
|
||||
ttl, source = dbTTL, "系统设置"
|
||||
} else if opts.SessionTTL > 0 {
|
||||
ttl, source = opts.SessionTTL, "启动参数"
|
||||
}
|
||||
cost := opts.BcryptCost
|
||||
if cost <= 0 {
|
||||
@@ -122,20 +159,86 @@ func New(db *sql.DB, opts Options) *Service {
|
||||
if writeLimit == 0 {
|
||||
writeLimit = DefaultWritePerMinute
|
||||
}
|
||||
return &Service{
|
||||
logger.Info("登录有效期", "hours", ttl.Hours(), "source", source,
|
||||
"滑动续期", true, "节流窗口", renewThreshold.String())
|
||||
s := &Service{
|
||||
db: db,
|
||||
log: logger,
|
||||
sessionTTL: ttl,
|
||||
bcryptCost: cost,
|
||||
secureCookies: opts.Secure,
|
||||
now: now,
|
||||
loginLimiter: NewLimiter(loginLimit, time.Minute, now),
|
||||
writeLimiter: NewLimiter(writeLimit, time.Minute, now),
|
||||
}
|
||||
s.ttl.Store(int64(ttl))
|
||||
return s
|
||||
}
|
||||
|
||||
// SessionTTL 返回会话有效期。
|
||||
func (s *Service) SessionTTL() time.Duration { return s.sessionTTL }
|
||||
// SessionTTL 返回当前生效的登录有效期。
|
||||
func (s *Service) SessionTTL() time.Duration { return time.Duration(s.ttl.Load()) }
|
||||
|
||||
// SetSessionTTL 更新登录有效期。只对之后创建的会话与之后的续期生效：
|
||||
// 已存在会话的 expires_at 一律不动——改设置不该把正在录入的人踢下线。
|
||||
func (s *Service) SetSessionTTL(d time.Duration) {
|
||||
if d <= 0 {
|
||||
return
|
||||
}
|
||||
s.ttl.Store(int64(d))
|
||||
}
|
||||
|
||||
// ParseSessionHours 解析登录有效期（小时）；非法或超出 1~720 一律判为无效。
|
||||
func ParseSessionHours(raw string) (int, bool) {
|
||||
hours, err := strconv.Atoi(strings.TrimSpace(raw))
|
||||
if err != nil || hours < MinSessionHours || hours > MaxSessionHours {
|
||||
return 0, false
|
||||
}
|
||||
return hours, true
|
||||
}
|
||||
|
||||
// EffectiveSessionHours 算出实际生效的小时数。
|
||||
// 规则：面板写过的值原样生效（含操作员明确选的 12）；迁移种下的历史占位值 12
|
||||
// 因为从来没有代码读过，按「未配置」处理，回落到新默认 24；缺键或非法同样是默认值。
|
||||
func EffectiveSessionHours(values map[string]string) int {
|
||||
raw := strings.TrimSpace(values[SessionHoursKey])
|
||||
if hours, ok := ParseSessionHours(raw); ok {
|
||||
if strings.TrimSpace(values[SessionHoursSourceKey]) != "" || raw != legacySessionHoursSeed {
|
||||
return hours
|
||||
}
|
||||
}
|
||||
return DefaultSessionHours
|
||||
}
|
||||
|
||||
// sessionTTLFromSettings 从系统库 settings 读登录有效期；没有有效配置（缺键或非法）时 ok=false，
|
||||
// 由调用方回落到启动参数 / 内置默认。数据库不可用不当作 fatal：认证要能带着默认值起来。
|
||||
func sessionTTLFromSettings(db *sql.DB) (time.Duration, bool) {
|
||||
if db == nil {
|
||||
return 0, false
|
||||
}
|
||||
values := make(map[string]string, 2)
|
||||
rows, err := db.Query(`SELECT key, value FROM settings WHERE key IN (?, ?)`,
|
||||
SessionHoursKey, SessionHoursSourceKey)
|
||||
if err != nil {
|
||||
slog.Warn("读取登录有效期失败，本次用默认值", "err", err, "hours", DefaultSessionHours)
|
||||
return 0, false
|
||||
}
|
||||
defer func() { _ = rows.Close() }()
|
||||
for rows.Next() {
|
||||
var k, v string
|
||||
if err := rows.Scan(&k, &v); err != nil {
|
||||
slog.Warn("解析登录有效期失败，本次用默认值", "err", err, "hours", DefaultSessionHours)
|
||||
return 0, false
|
||||
}
|
||||
values[k] = v
|
||||
}
|
||||
if err := rows.Err(); err != nil {
|
||||
slog.Warn("读取登录有效期失败，本次用默认值", "err", err, "hours", DefaultSessionHours)
|
||||
return 0, false
|
||||
}
|
||||
if _, ok := values[SessionHoursKey]; !ok {
|
||||
return 0, false
|
||||
}
|
||||
return time.Duration(EffectiveSessionHours(values)) * time.Hour, true
|
||||
}
|
||||
|
||||
// AllowLogin 按 IP 限流登录。
|
||||
func (s *Service) AllowLogin(ip string) bool { return s.loginLimiter.Allow(ip) }
|
||||
@@ -301,20 +404,22 @@ func (s *Service) Lookup(ctx context.Context, token string) (*User, *Session, er
|
||||
hash := hashToken(token)
|
||||
|
||||
var (
|
||||
u User
|
||||
sess Session
|
||||
expiresAt string
|
||||
role string
|
||||
u User
|
||||
sess Session
|
||||
expiresAt string
|
||||
lastSeenRaw string
|
||||
role string
|
||||
)
|
||||
err := s.db.QueryRowContext(ctx,
|
||||
`SELECT s.id, s.user_id, s.expires_at, COALESCE(s.ip, ''), COALESCE(s.user_agent, ''),
|
||||
COALESCE(s.last_seen_at, ''),
|
||||
u.id, u.username, u.display_name, u.role, u.active, u.must_change_password,
|
||||
COALESCE(u.last_login_at, '')
|
||||
FROM sessions s
|
||||
JOIN users u ON u.id = s.user_id
|
||||
WHERE s.token_hash = ? AND s.expires_at > ? AND u.active = 1`,
|
||||
hash, s.now().UTC().Format(time.DateTime)).
|
||||
Scan(&sess.ID, &sess.UserID, &expiresAt, &sess.IP, &sess.UserAgent,
|
||||
Scan(&sess.ID, &sess.UserID, &expiresAt, &sess.IP, &sess.UserAgent, &lastSeenRaw,
|
||||
&u.ID, &u.Username, &u.DisplayName, &role, &u.Active, &u.MustChangePassword, &u.LastLoginAt)
|
||||
switch {
|
||||
case errors.Is(err, sql.ErrNoRows):
|
||||
@@ -323,6 +428,7 @@ func (s *Service) Lookup(ctx context.Context, token string) (*User, *Session, er
|
||||
return nil, nil, fmt.Errorf("查询会话: %w", err)
|
||||
}
|
||||
|
||||
now := s.now().UTC()
|
||||
exp, err := parseDBTime(expiresAt)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
@@ -330,8 +436,26 @@ func (s *Service) Lookup(ctx context.Context, token string) (*User, *Session, er
|
||||
sess.ExpiresAt = exp
|
||||
u.Role = ParseRole(role)
|
||||
|
||||
_, _ = s.db.ExecContext(ctx, `UPDATE sessions SET last_seen_at = ? WHERE id = ?`,
|
||||
s.now().UTC().Format(time.DateTime), sess.ID)
|
||||
// 滑动续期：会话只要还在有效期内被使用，就把到期时间顺延到 now+有效期，
|
||||
// 而不是「登录满 N 小时必死」——现场前一晚登录、比赛当天一直在用的人不该掉线。
|
||||
//
|
||||
// 写入节流：只有剩余寿命已不足「有效期 - renewThreshold（1 分钟）」时才回写 expires_at。
|
||||
// 否则每点一次按钮、每次轮询都要写一遍 sessions，把单连接 sqlite 变成热点写表；
|
||||
// 少顺延这 1 分钟不会让任何人提前掉线（前一次续期至少还剩一个有效期）。
|
||||
if ttl := s.SessionTTL(); ttl > renewThreshold && exp.Sub(now) < ttl-renewThreshold {
|
||||
renewed := now.Add(ttl)
|
||||
_, _ = s.db.ExecContext(ctx, `UPDATE sessions SET last_seen_at = ?, expires_at = ? WHERE id = ?`,
|
||||
now.Format(time.DateTime), renewed.Format(time.DateTime), sess.ID)
|
||||
sess.ExpiresAt = renewed
|
||||
return &u, &sess, nil
|
||||
}
|
||||
// 没到续期点时也不是每个请求都写：last_seen_at 最快每 touchInterval 回写一次。
|
||||
// 一个页面加载会打七八个请求，逐个写一遍纯属放大写压力，而「上次活动时间」精确到分钟就够用。
|
||||
lastSeen, err := parseDBTime(lastSeenRaw)
|
||||
if err != nil || now.Sub(lastSeen) >= touchInterval {
|
||||
_, _ = s.db.ExecContext(ctx, `UPDATE sessions SET last_seen_at = ? WHERE id = ?`,
|
||||
now.Format(time.DateTime), sess.ID)
|
||||
}
|
||||
return &u, &sess, nil
|
||||
}
|
||||
|
||||
@@ -362,7 +486,7 @@ func (s *Service) createSession(ctx context.Context, userID int64, ip, userAgent
|
||||
if err != nil {
|
||||
return "", nil, err
|
||||
}
|
||||
expiresAt := s.now().UTC().Add(s.sessionTTL)
|
||||
expiresAt := s.now().UTC().Add(s.SessionTTL())
|
||||
_, err = s.db.ExecContext(ctx,
|
||||
`INSERT INTO sessions (id, user_id, token_hash, expires_at, ip, user_agent, last_seen_at)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?)`,
|
||||
@@ -377,54 +501,64 @@ func (s *Service) createSession(ctx context.Context, userID int64, ip, userAgent
|
||||
|
||||
// SetSessionCookie 写入 HttpOnly + SameSite=Lax 的会话 Cookie。
|
||||
func (s *Service) SetSessionCookie(w http.ResponseWriter, token string) {
|
||||
setSessionCookie(w, s.secureCookies, token, int(s.SessionTTL().Seconds()))
|
||||
}
|
||||
|
||||
// RefreshSessionCookies 按会话到期时间重发会话与 CSRF Cookie。
|
||||
//
|
||||
// 滑动续期之后必须重发：Cookie 的 Max-Age 是浏览器端的硬期限，
|
||||
// 不跟着续期走的话，浏览器会在「登录时刻 + 有效期」处把 Cookie 丢掉，
|
||||
// 服务端续得再久也留不住人。CSRF 沿用原值（不轮换），避免和正在飞行的写请求打架。
|
||||
// 到期时间已过时不重发——调用方不该给一个已经失效的会话续 Cookie。
|
||||
func (s *Service) RefreshSessionCookies(w http.ResponseWriter, sessionToken, csrfToken string, expiresAt time.Time) {
|
||||
life := expiresAt.Sub(s.now())
|
||||
if life <= 0 || sessionToken == "" {
|
||||
return
|
||||
}
|
||||
seconds := int(life.Seconds())
|
||||
setSessionCookie(w, s.secureCookies, sessionToken, seconds)
|
||||
if csrfToken != "" {
|
||||
setCSRFCookie(w, s.secureCookies, csrfToken, seconds)
|
||||
}
|
||||
}
|
||||
|
||||
func setSessionCookie(w http.ResponseWriter, secure bool, token string, maxAge int) {
|
||||
http.SetCookie(w, &http.Cookie{
|
||||
Name: CookieSession,
|
||||
Value: token,
|
||||
Path: "/",
|
||||
MaxAge: int(s.sessionTTL.Seconds()),
|
||||
MaxAge: maxAge,
|
||||
HttpOnly: true,
|
||||
Secure: s.secureCookies,
|
||||
Secure: secure,
|
||||
SameSite: http.SameSiteLaxMode,
|
||||
})
|
||||
}
|
||||
|
||||
// ClearSessionCookie 删除会话 Cookie。
|
||||
func (s *Service) ClearSessionCookie(w http.ResponseWriter) {
|
||||
http.SetCookie(w, &http.Cookie{
|
||||
Name: CookieSession,
|
||||
Value: "",
|
||||
Path: "/",
|
||||
MaxAge: -1,
|
||||
HttpOnly: true,
|
||||
Secure: s.secureCookies,
|
||||
SameSite: http.SameSiteLaxMode,
|
||||
})
|
||||
setSessionCookie(w, s.secureCookies, "", -1)
|
||||
}
|
||||
|
||||
// SetCSRFCookie 写入可被前端读取的 CSRF Cookie（双提交 Cookie 方案）。
|
||||
func (s *Service) SetCSRFCookie(w http.ResponseWriter, token string) {
|
||||
setCSRFCookie(w, s.secureCookies, token, int(s.SessionTTL().Seconds()))
|
||||
}
|
||||
|
||||
func setCSRFCookie(w http.ResponseWriter, secure bool, token string, maxAge int) {
|
||||
http.SetCookie(w, &http.Cookie{
|
||||
Name: CookieCSRF,
|
||||
Value: token,
|
||||
Path: "/",
|
||||
MaxAge: int(s.sessionTTL.Seconds()),
|
||||
MaxAge: maxAge,
|
||||
HttpOnly: false,
|
||||
Secure: s.secureCookies,
|
||||
Secure: secure,
|
||||
SameSite: http.SameSiteLaxMode,
|
||||
})
|
||||
}
|
||||
|
||||
// ClearCSRFCookie 删除 CSRF Cookie。
|
||||
func (s *Service) ClearCSRFCookie(w http.ResponseWriter) {
|
||||
http.SetCookie(w, &http.Cookie{
|
||||
Name: CookieCSRF,
|
||||
Value: "",
|
||||
Path: "/",
|
||||
MaxAge: -1,
|
||||
HttpOnly: false,
|
||||
Secure: s.secureCookies,
|
||||
SameSite: http.SameSiteLaxMode,
|
||||
})
|
||||
setCSRFCookie(w, s.secureCookies, "", -1)
|
||||
}
|
||||
|
||||
// NewCSRFToken 生成新的 CSRF Token。
|
||||
|
||||
@@ -0,0 +1,343 @@
|
||||
package auth
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"io"
|
||||
"log/slog"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
_ "modernc.org/sqlite"
|
||||
)
|
||||
|
||||
// 登录有效期的解析与新默认值：迁移种下的历史占位值 12 从未被任何代码读过，
|
||||
// 按「未配置」处理，否则新装的库会永远停在 12（现场前一晚登录、次日比赛当天掉线）。
|
||||
func TestEffectiveSessionHours(t *testing.T) {
|
||||
for _, tc := range []struct {
|
||||
name string
|
||||
values map[string]string
|
||||
want int
|
||||
}{
|
||||
{"历史占位值 12（无面板标记）按未配置处理", map[string]string{SessionHoursKey: "12"}, 24},
|
||||
{"面板明确选过的 12 原样生效", map[string]string{SessionHoursKey: "12", SessionHoursSourceKey: "panel"}, 12},
|
||||
{"面板写的 72 生效", map[string]string{SessionHoursKey: "72", SessionHoursSourceKey: "panel"}, 72},
|
||||
{"未配置回默认 24", map[string]string{}, 24},
|
||||
{"上界 720", map[string]string{SessionHoursKey: "720"}, 720},
|
||||
{"下界 1", map[string]string{SessionHoursKey: "1"}, 1},
|
||||
{"0 非法回默认", map[string]string{SessionHoursKey: "0"}, 24},
|
||||
{"721 非法回默认", map[string]string{SessionHoursKey: "721"}, 24},
|
||||
{"非数字非法回默认", map[string]string{SessionHoursKey: "十二"}, 24},
|
||||
} {
|
||||
t.Run(tc.name, func(t *testing.T) {
|
||||
if got := EffectiveSessionHours(tc.values); got != tc.want {
|
||||
t.Fatalf("EffectiveSessionHours(%v) = %d，期望 %d", tc.values, got, tc.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestParseSessionHours(t *testing.T) {
|
||||
for raw, want := range map[string]int{"1": 1, "24": 24, "720": 720, " 48 ": 48} {
|
||||
got, ok := ParseSessionHours(raw)
|
||||
if !ok || got != want {
|
||||
t.Fatalf("ParseSessionHours(%q) = %d,%v，期望 %d,true", raw, got, ok, want)
|
||||
}
|
||||
}
|
||||
for _, raw := range []string{"", "0", "-1", "721", "24.5", "abc"} {
|
||||
if _, ok := ParseSessionHours(raw); ok {
|
||||
t.Fatalf("ParseSessionHours(%q) 应判为非法", raw)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 系统库里的登录有效期要真正被认成会话寿命；显式传入的 Options.SessionTTL 优先（测试 / 嵌入用）。
|
||||
func TestSessionTTLFromSystemSettings(t *testing.T) {
|
||||
for _, tc := range []struct {
|
||||
name string
|
||||
seed map[string]string
|
||||
optsTTL time.Duration
|
||||
want time.Duration
|
||||
}{
|
||||
{"历史占位值 12 视为未配置 → 24h", map[string]string{SessionHoursKey: "12"}, 0, 24 * time.Hour},
|
||||
{"面板写的 1 → 1h", map[string]string{SessionHoursKey: "1", SessionHoursSourceKey: "panel"}, 0, time.Hour},
|
||||
{"面板里的值优先于启动参数", map[string]string{SessionHoursKey: "1", SessionHoursSourceKey: "panel"}, 90 * time.Minute, time.Hour},
|
||||
{"系统库没有配置时用启动参数兜底", nil, 90 * time.Minute, 90 * time.Minute},
|
||||
} {
|
||||
t.Run(tc.name, func(t *testing.T) {
|
||||
db := newSettingsDB(t, tc.seed)
|
||||
svc := New(db, Options{Logger: quietLogger(), SessionTTL: tc.optsTTL})
|
||||
if got := svc.SessionTTL(); got != tc.want {
|
||||
t.Fatalf("SessionTTL() = %s，期望 %s", got, tc.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// 滑动续期：剩余寿命少于「有效期 - 节流窗口」时才回写，且回写后到期时间顺延到 now+有效期。
|
||||
func TestLookupSlidesExpiryWithThrottle(t *testing.T) {
|
||||
db := newSettingsDB(t, nil)
|
||||
now := time.Date(2026, 9, 11, 20, 0, 0, 0, time.UTC)
|
||||
svc := New(db, Options{Logger: quietLogger(), SessionTTL: 24 * time.Hour,
|
||||
Now: func() time.Time { return now }})
|
||||
userID := insertUser(t, db, "admin")
|
||||
token := insertSession(t, db, userID, now.Add(24*time.Hour))
|
||||
|
||||
// 刚创建的会话：剩余 = 有效期，不写 expires_at（首次请求不该产生额外写）。
|
||||
if _, sess, err := svc.Lookup(context.Background(), token); err != nil {
|
||||
t.Fatalf("查询会话: %v", err)
|
||||
} else if !sess.ExpiresAt.Equal(now.Add(24 * time.Hour)) {
|
||||
t.Fatalf("新鲜会话的到期时间不该变，得到 %s", sess.ExpiresAt)
|
||||
}
|
||||
if got := dbExpiry(t, db); got != now.Add(24*time.Hour).Format(time.DateTime) {
|
||||
t.Fatalf("新鲜会话不应回写 expires_at，库里是 %s", got)
|
||||
}
|
||||
|
||||
// 剩余 23h58m（不足 24h-1m）：下一次请求应把到期时间顺延到 now+24h。
|
||||
before := now.Add(24*time.Hour - 2*time.Minute)
|
||||
setExpiry(t, db, before)
|
||||
if _, sess, err := svc.Lookup(context.Background(), token); err != nil {
|
||||
t.Fatalf("查询会话: %v", err)
|
||||
} else if want := now.Add(24 * time.Hour); !sess.ExpiresAt.Equal(want) {
|
||||
t.Fatalf("应顺延到 %s，得到 %s", want, sess.ExpiresAt)
|
||||
}
|
||||
if got, want := dbExpiry(t, db), now.Add(24*time.Hour).Format(time.DateTime); got != want {
|
||||
t.Fatalf("库里到期时间应为 %s，得到 %s", want, got)
|
||||
}
|
||||
|
||||
// 节流：紧接着的多次请求既不写 expires_at，也不写 last_seen_at——
|
||||
// 现场一次页面加载会打七八个请求，逐个写库纯属放大写压力。
|
||||
renewed := dbExpiry(t, db)
|
||||
seen := dbLastSeen(t, db)
|
||||
for i := 0; i < 5; i++ {
|
||||
now = now.Add(time.Second)
|
||||
if _, _, err := svc.Lookup(context.Background(), token); err != nil {
|
||||
t.Fatalf("第 %d 次查询会话: %v", i, err)
|
||||
}
|
||||
if got := dbExpiry(t, db); got != renewed {
|
||||
t.Fatalf("节流窗口内不该回写 expires_at：%s → %s", renewed, got)
|
||||
}
|
||||
if got := dbLastSeen(t, db); got != seen {
|
||||
t.Fatalf("节流窗口内不该回写 last_seen_at：%s → %s", seen, got)
|
||||
}
|
||||
}
|
||||
|
||||
// 越过节流窗口（1 分钟）后再次顺延：活跃会话大约每分钟才写一次库。
|
||||
now = now.Add(5 * time.Minute)
|
||||
if _, sess, err := svc.Lookup(context.Background(), token); err != nil {
|
||||
t.Fatalf("查询会话: %v", err)
|
||||
} else if want := now.Add(24 * time.Hour); !sess.ExpiresAt.Equal(want) {
|
||||
t.Fatalf("越过节流窗口后应顺延到 %s，得到 %s", want, sess.ExpiresAt)
|
||||
}
|
||||
if got, want := dbLastSeen(t, db), now.Format(time.DateTime); got != want {
|
||||
t.Fatalf("续期时活动时间应一起回写为 %s，得到 %s", want, got)
|
||||
}
|
||||
}
|
||||
|
||||
// 活动时间的节流不会影响续期：会话剩余寿命超过当前有效期（例如有效期被调短）时，
|
||||
// 只回写 last_seen_at，expires_at 保持原值。
|
||||
func TestLookupTouchThrottleKeepsLongLivedSession(t *testing.T) {
|
||||
db := newSettingsDB(t, nil)
|
||||
now := time.Date(2026, 9, 11, 20, 0, 0, 0, time.UTC)
|
||||
svc := New(db, Options{Logger: quietLogger(), SessionTTL: time.Hour,
|
||||
Now: func() time.Time { return now }})
|
||||
userID := insertUser(t, db, "admin")
|
||||
token := insertSession(t, db, userID, now.Add(20*time.Hour))
|
||||
setLastSeen(t, db, now.Add(-time.Hour))
|
||||
stored := dbExpiry(t, db)
|
||||
|
||||
if _, _, err := svc.Lookup(context.Background(), token); err != nil {
|
||||
t.Fatalf("查询会话: %v", err)
|
||||
}
|
||||
if got, want := dbLastSeen(t, db), now.Format(time.DateTime); got != want {
|
||||
t.Fatalf("活动时间应回写为 %s，得到 %s", want, got)
|
||||
}
|
||||
if got := dbExpiry(t, db); got != stored {
|
||||
t.Fatalf("已登录会话的到期时间不该被改设置的续期逻辑改写：%s → %s", stored, got)
|
||||
}
|
||||
}
|
||||
|
||||
// 越过节流窗口后再次顺延：单独跑一遍，避免上面的用例把它和活动时间混在一起。
|
||||
func TestLookupRenewsAfterThrottleWindow(t *testing.T) {
|
||||
db := newSettingsDB(t, nil)
|
||||
now := time.Date(2026, 9, 11, 20, 0, 0, 0, time.UTC)
|
||||
svc := New(db, Options{Logger: quietLogger(), SessionTTL: 24 * time.Hour,
|
||||
Now: func() time.Time { return now }})
|
||||
userID := insertUser(t, db, "admin")
|
||||
token := insertSession(t, db, userID, now.Add(24*time.Hour-2*time.Minute))
|
||||
|
||||
if _, _, err := svc.Lookup(context.Background(), token); err != nil {
|
||||
t.Fatalf("查询会话: %v", err)
|
||||
}
|
||||
now = now.Add(2 * time.Minute)
|
||||
if _, sess, err := svc.Lookup(context.Background(), token); err != nil {
|
||||
t.Fatalf("查询会话: %v", err)
|
||||
} else if want := now.Add(24 * time.Hour); !sess.ExpiresAt.Equal(want) {
|
||||
t.Fatalf("越过节流窗口后应顺延到 %s，得到 %s", want, sess.ExpiresAt)
|
||||
}
|
||||
}
|
||||
|
||||
// 到期校验不能被续期改坏：已过期的会话仍然查不到（调用方据此回 401）。
|
||||
func TestLookupRejectsExpiredSession(t *testing.T) {
|
||||
db := newSettingsDB(t, nil)
|
||||
now := time.Date(2026, 9, 11, 20, 0, 0, 0, time.UTC)
|
||||
svc := New(db, Options{Logger: quietLogger(), SessionTTL: 12 * time.Hour,
|
||||
Now: func() time.Time { return now }})
|
||||
userID := insertUser(t, db, "admin")
|
||||
token := insertSession(t, db, userID, now.Add(-time.Minute))
|
||||
|
||||
if _, _, err := svc.Lookup(context.Background(), token); err != ErrSessionNotFound {
|
||||
t.Fatalf("过期会话应返回 ErrSessionNotFound，得到 %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// 改有效期不动已存在的会话：只换进程内的 TTL，库里的 expires_at 原样保留。
|
||||
func TestSetSessionTTLKeepsExistingExpiry(t *testing.T) {
|
||||
db := newSettingsDB(t, nil)
|
||||
now := time.Date(2026, 9, 11, 20, 0, 0, 0, time.UTC)
|
||||
svc := New(db, Options{Logger: quietLogger(), SessionTTL: time.Hour,
|
||||
Now: func() time.Time { return now }})
|
||||
userID := insertUser(t, db, "admin")
|
||||
token := insertSession(t, db, userID, now.Add(time.Hour))
|
||||
stored := dbExpiry(t, db)
|
||||
|
||||
svc.SetSessionTTL(72 * time.Hour)
|
||||
|
||||
if got := svc.SessionTTL(); got != 72*time.Hour {
|
||||
t.Fatalf("SessionTTL() = %s，期望 72h", got)
|
||||
}
|
||||
if got := dbExpiry(t, db); got != stored {
|
||||
t.Fatalf("改有效期不该动已存在会话的 expires_at：%s → %s", stored, got)
|
||||
}
|
||||
// 旧会话下一次被使用时按新的、更长的有效期顺延（是延长，不是踢下线）。
|
||||
_, sess, err := svc.Lookup(context.Background(), token)
|
||||
if err != nil {
|
||||
t.Fatalf("旧会话不该被踢下线: %v", err)
|
||||
}
|
||||
if want := now.Add(72 * time.Hour); !sess.ExpiresAt.Equal(want) {
|
||||
t.Fatalf("旧会话应顺延到 %s，得到 %s", want, sess.ExpiresAt)
|
||||
}
|
||||
}
|
||||
|
||||
// 新会话按改后的有效期创建。
|
||||
func TestCreateSessionUsesCurrentTTL(t *testing.T) {
|
||||
db := newSettingsDB(t, nil)
|
||||
now := time.Date(2026, 9, 11, 20, 0, 0, 0, time.UTC)
|
||||
svc := New(db, Options{Logger: quietLogger(), SessionTTL: 12 * time.Hour,
|
||||
Now: func() time.Time { return now }})
|
||||
userID := insertUser(t, db, "admin")
|
||||
|
||||
svc.SetSessionTTL(720 * time.Hour)
|
||||
if _, sess, err := svc.createSession(context.Background(), userID, "127.0.0.1", "test"); err != nil {
|
||||
t.Fatalf("创建会话: %v", err)
|
||||
} else if want := now.Add(720 * time.Hour); !sess.ExpiresAt.Equal(want) {
|
||||
t.Fatalf("新会话应在 %s 到期，得到 %s", want, sess.ExpiresAt)
|
||||
}
|
||||
}
|
||||
|
||||
// ---------- 测试脚手架 ----------
|
||||
|
||||
func quietLogger() *slog.Logger {
|
||||
return slog.New(slog.NewTextHandler(io.Discard, nil))
|
||||
}
|
||||
|
||||
const testSchema = `
|
||||
CREATE TABLE users (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
username TEXT NOT NULL UNIQUE,
|
||||
password_hash TEXT NOT NULL DEFAULT '',
|
||||
display_name TEXT NOT NULL DEFAULT '',
|
||||
role TEXT NOT NULL DEFAULT 'recorder',
|
||||
active INTEGER NOT NULL DEFAULT 1,
|
||||
must_change_password INTEGER NOT NULL DEFAULT 0,
|
||||
last_login_at TEXT,
|
||||
updated_at TEXT NOT NULL DEFAULT (datetime('now'))
|
||||
);
|
||||
CREATE TABLE sessions (
|
||||
id TEXT PRIMARY KEY,
|
||||
user_id INTEGER NOT NULL,
|
||||
token_hash TEXT NOT NULL UNIQUE,
|
||||
expires_at TEXT NOT NULL,
|
||||
ip TEXT,
|
||||
user_agent TEXT,
|
||||
created_at TEXT NOT NULL DEFAULT (datetime('now')),
|
||||
last_seen_at TEXT
|
||||
);
|
||||
CREATE TABLE settings (
|
||||
key TEXT PRIMARY KEY,
|
||||
value TEXT NOT NULL,
|
||||
updated_at TEXT NOT NULL DEFAULT (datetime('now'))
|
||||
);`
|
||||
|
||||
func newSettingsDB(t *testing.T, seed map[string]string) *sql.DB {
|
||||
t.Helper()
|
||||
db, err := sql.Open("sqlite", "file:"+filepath.Join(t.TempDir(), "system.db"))
|
||||
if err != nil {
|
||||
t.Fatalf("打开测试库: %v", err)
|
||||
}
|
||||
db.SetMaxOpenConns(1)
|
||||
t.Cleanup(func() { _ = db.Close() })
|
||||
if _, err := db.Exec(testSchema); err != nil {
|
||||
t.Fatalf("建表: %v", err)
|
||||
}
|
||||
for k, v := range seed {
|
||||
if _, err := db.Exec(`INSERT INTO settings (key, value) VALUES (?, ?)`, k, v); err != nil {
|
||||
t.Fatalf("写入设置 %s: %v", k, err)
|
||||
}
|
||||
}
|
||||
return db
|
||||
}
|
||||
|
||||
func insertUser(t *testing.T, db *sql.DB, username string) int64 {
|
||||
t.Helper()
|
||||
res, err := db.Exec(`INSERT INTO users (username, role) VALUES (?, 'admin')`, username)
|
||||
if err != nil {
|
||||
t.Fatalf("建用户: %v", err)
|
||||
}
|
||||
id, _ := res.LastInsertId()
|
||||
return id
|
||||
}
|
||||
|
||||
func insertSession(t *testing.T, db *sql.DB, userID int64, expiresAt time.Time) string {
|
||||
t.Helper()
|
||||
token := "token-" + expiresAt.Format(time.RFC3339Nano)
|
||||
if _, err := db.Exec(
|
||||
`INSERT INTO sessions (id, user_id, token_hash, expires_at, last_seen_at) VALUES (?, ?, ?, ?, ?)`,
|
||||
"sid", userID, hashToken(token), expiresAt.Format(time.DateTime), expiresAt.Format(time.DateTime)); err != nil {
|
||||
t.Fatalf("建会话: %v", err)
|
||||
}
|
||||
return token
|
||||
}
|
||||
|
||||
func setExpiry(t *testing.T, db *sql.DB, expiresAt time.Time) {
|
||||
t.Helper()
|
||||
if _, err := db.Exec(`UPDATE sessions SET expires_at = ?`, expiresAt.Format(time.DateTime)); err != nil {
|
||||
t.Fatalf("改到期时间: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func setLastSeen(t *testing.T, db *sql.DB, at time.Time) {
|
||||
t.Helper()
|
||||
if _, err := db.Exec(`UPDATE sessions SET last_seen_at = ?`, at.Format(time.DateTime)); err != nil {
|
||||
t.Fatalf("改活动时间: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func dbExpiry(t *testing.T, db *sql.DB) string {
|
||||
t.Helper()
|
||||
var v string
|
||||
if err := db.QueryRow(`SELECT expires_at FROM sessions`).Scan(&v); err != nil {
|
||||
t.Fatalf("读 expires_at: %v", err)
|
||||
}
|
||||
return v
|
||||
}
|
||||
|
||||
func dbLastSeen(t *testing.T, db *sql.DB) string {
|
||||
t.Helper()
|
||||
var v string
|
||||
if err := db.QueryRow(`SELECT last_seen_at FROM sessions`).Scan(&v); err != nil {
|
||||
t.Fatalf("读 last_seen_at: %v", err)
|
||||
}
|
||||
return v
|
||||
}
|
||||
Reference in new issue
Block a user