chore: refresh gramsrv public release

This commit is contained in:
A 2026-06-30 14:37:43 +08:00
parent 75cebe8dbf
commit 70b6820474
1274 changed files with 378751 additions and 59919 deletions

View file

@ -0,0 +1,95 @@
package rpc
import (
"context"
"time"
"github.com/gotd/td/tg"
"go.uber.org/zap"
"telesrv/internal/domain"
)
// RunPremiumSweeper 周期清理到期会员并通知本人在线 session。
//
// premium 下发正确性由 hydration 即时派生(premium_expires_at > now)保证,
// 不依赖本 sweeper;这里只负责两件收尾事:把过期行清 NULL(保持索引/语义
// 干净),以及向该用户全部在线 session 推 updateUser + 最新 self user,让在线
// 客户端立即降级 UI(updateUser 无 pts,不进 update_events;离线设备重连后由
// 任意带 self user 的响应自愈)。
func (r *Router) RunPremiumSweeper(ctx context.Context, interval time.Duration, batch int) {
if interval <= 0 {
interval = time.Minute
}
if batch <= 0 {
batch = 500
}
ticker := time.NewTicker(interval)
defer ticker.Stop()
for {
select {
case <-ctx.Done():
return
case <-ticker.C:
}
r.sweepExpiredPremium(ctx, batch)
}
}
func (r *Router) sweepExpiredPremium(ctx context.Context, batch int) {
svc, ok := r.deps.Users.(UserPremiumService)
if !ok {
return
}
for {
sweepCtx, cancel := context.WithTimeout(ctx, 30*time.Second)
users, err := svc.SweepExpiredPremium(sweepCtx, r.clock.Now().Unix(), batch)
cancel()
if err != nil {
r.log.Warn("premium sweep failed", zap.Error(err))
return
}
for _, u := range users {
r.pushPremiumStatusUpdate(ctx, u)
}
// 不满一批说明已扫完当前积压;满批则继续,避免长停机后积压跨多个周期。
if len(users) < batch {
return
}
}
}
// viewerPremium 报告 viewer 当前是否有效会员(限额双档判断用,best-effort:
// 服务未接通时按非会员档处理)。
func (r *Router) viewerPremium(ctx context.Context, userID int64) bool {
svc, ok := r.deps.Users.(UserPremiumStatusService)
return ok && svc.PremiumActive(ctx, userID)
}
// NotifyUserChanged 是 Admin 用例层可调用的 domain-only hook:账号基础事实
// 变更后失效 RPC 投影缓存,并向本人在线 session 推 updateUser。它不把 tg.*
// 泄漏给 admin/domain/app,协议对象只在 rpc 边界内构造。
func (r *Router) NotifyUserChanged(ctx context.Context, u domain.User) error {
if r == nil || u.ID == 0 {
return nil
}
r.invalidateRPCProjectionForUser(u.ID)
r.pushPremiumStatusUpdate(ctx, u)
return nil
}
// pushPremiumStatusUpdate 向用户本人的全部在线 session 推送会员状态变化。
// 授予、到期与 admin 认证变更共用:updateUser 触发客户端用随附的 self user
// 对象刷新 premium/verified 等基础 flag(TDesktop processUser 按 flag 翻转)。
func (r *Router) pushPremiumStatusUpdate(ctx context.Context, u domain.User) {
if u.ID == 0 {
return
}
pushCtx, cancel := context.WithTimeout(ctx, 10*time.Second)
defer cancel()
r.pushUserUpdates(pushCtx, u.ID, &tg.Updates{
Updates: []tg.UpdateClass{&tg.UpdateUser{UserID: u.ID}},
Users: []tg.UserClass{r.tgSelfUser(u)},
Date: int(r.clock.Now().Unix()),
})
}