feat: 近实时远程下线 — 客户端轮询会话有效性(~15s),服务端 GET /v1/me/session

控制面无推送通道,access token 是无状态 JWT(15min),强制退出后被踢设备要等 token 过期
(≤15min)才登出。改成客户端每 15s 轮询会话是否仍有效,被强制退出即登出 → 延迟压到 ~15s。
- 服务端:sessions.HasActiveSession(user,device) + devices.SessionActive(按 UUID,fail-open)
  + GET /v1/me/session?device_id= 返回 {active}(恒 200,判据在 body)。无新迁移。
- 客户端:account_api.sessionActive + main.dart _RootFlowState 15s 轮询,active=false 即 logout
  (网络/鉴权异常不据此登出,fail-safe)。
- 测试:TestSQLite_SessionHasActiveSession(建会话=活跃→RevokeByDevice→非活跃)。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
wangjia
2026-06-30 07:25:16 +08:00
parent 7f070a0693
commit 0b33f12400
6 changed files with 143 additions and 1 deletions
+23
View File
@@ -3,6 +3,7 @@ package devices
import (
"encoding/json"
"net/http"
"strings"
"github.com/go-chi/chi/v5"
"github.com/wangjia/pangolin/server/internal/apierr"
@@ -29,6 +30,7 @@ func (h *Handler) RegisterRoutes(r chi.Router) {
r.Delete("/devices/{id}", h.DeleteDevice)
r.Post("/devices/{id}/logout", h.ForceLogout)
r.Post("/devices/{id}/rename", h.RenameDevice)
r.Get("/session", h.SessionCheck) // GET /v1/me/session?device_id= — 会话有效性轮询
}
type listDevicesResponse struct {
@@ -89,6 +91,27 @@ func (h *Handler) ForceLogout(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusNoContent)
}
// SessionCheck handles GET /v1/me/session?device_id=UUID — reports whether the
// caller's device still has a live session. false ⇒ force-logged-out elsewhere;
// the client polls this (~15s) and logs out on {"active":false} → near-instant kick.
// Always 200 (the verdict is in the body); device_id 取自 query。
func (h *Handler) SessionCheck(w http.ResponseWriter, r *http.Request) {
userID, ok := auth.UserIDFromContext(r.Context())
if !ok {
apierr.WriteJSON(w, http.StatusUnauthorized, apierr.ErrUnauthorized)
return
}
deviceUUID := strings.TrimSpace(r.URL.Query().Get("device_id"))
active, apiErr := h.svc.SessionActive(r.Context(), userID, deviceUUID)
if apiErr != nil {
apierr.WriteJSON(w, StatusForError(apiErr), apiErr)
return
}
w.Header().Set("Content-Type", "application/json; charset=utf-8")
w.WriteHeader(http.StatusOK)
_ = json.NewEncoder(w).Encode(map[string]bool{"active": active})
}
// RenameDevice handles POST /v1/me/devices/{id}/rename — set a device's name.
func (h *Handler) RenameDevice(w http.ResponseWriter, r *http.Request) {
userID, ok := auth.UserIDFromContext(r.Context())
+24
View File
@@ -36,6 +36,7 @@ func (n *NoopRevoker) RevokeDevice(_ context.Context, dpUUID string) error {
type SessionPort interface {
LastLoginByDevice(ctx context.Context, userID int64) (map[int64]time.Time, error)
RevokeByDevice(ctx context.Context, userID, deviceID int64) ([]string, error)
HasActiveSession(ctx context.Context, userID, deviceID int64) (bool, error)
}
// JTIRevoker drops a refresh JTI from the Redis whitelist. Satisfied by
@@ -342,6 +343,29 @@ func (svc *Service) RenameDevice(ctx context.Context, userID int64, deviceUUID,
return nil
}
// SessionActive reports whether the device (by UUID) still has a live login session
// for the user — false after a force-logout elsewhere. The client polls this (~15s)
// and logs out on false → near-instant remote kick. Fail-open (return true) on a
// missing session port / blank or unknown / not-own device, so transient ambiguity
// never spuriously logs a user out.
func (svc *Service) SessionActive(ctx context.Context, userID int64, deviceUUID string) (bool, *apierr.Error) {
if svc.sessions == nil || strings.TrimSpace(deviceUUID) == "" {
return true, nil
}
dev, err := svc.store.FindByUUID(ctx, deviceUUID)
if err != nil {
return false, apierr.ErrInternal
}
if dev == nil || dev.UserID != userID {
return true, nil // 未知 / 非本人设备:不据此登出
}
active, err := svc.sessions.HasActiveSession(ctx, userID, dev.ID)
if err != nil {
return false, apierr.ErrInternal
}
return active, nil
}
// revokeDeviceSessions marks all of a device's sessions revoked and drops their
// refresh JTIs from the Redis whitelist. Best-effort.
func (svc *Service) revokeDeviceSessions(ctx context.Context, userID, deviceID int64) {
+14
View File
@@ -47,6 +47,20 @@ func (s *Store) Rotate(ctx context.Context, oldJTI, newJTI string) error {
return nil
}
// HasActiveSession reports whether the (user, device) has at least one non-revoked
// session — i.e. the device has NOT been force-logged-out. Backs the client's
// periodic session-validity poll: near-instant remote logout without a push channel
// (access token is a stateless JWT, so we can't see revocation on normal requests).
func (s *Store) HasActiveSession(ctx context.Context, userID, deviceID int64) (bool, error) {
var n int
if err := s.db.QueryRowContext(ctx,
`SELECT COUNT(1) FROM sessions WHERE user_id=? AND device_id=? AND revoked_at IS NULL`,
userID, deviceID).Scan(&n); err != nil {
return false, fmt.Errorf("sessions.HasActiveSession: %w", err)
}
return n > 0, nil
}
// Revoke marks the session with the given JTI revoked (logout). Idempotent.
func (s *Store) Revoke(ctx context.Context, jti string) error {
now := time.Now().UTC()
@@ -12,6 +12,7 @@ import (
"github.com/wangjia/pangolin/server/internal/nodes"
agentv1 "github.com/wangjia/pangolin/server/internal/pb/agentv1"
"github.com/wangjia/pangolin/server/internal/provision"
"github.com/wangjia/pangolin/server/internal/sessions"
"github.com/wangjia/pangolin/server/internal/store"
"github.com/wangjia/pangolin/server/internal/usage"
)
@@ -57,6 +58,43 @@ func TestSQLite_UsageAccumulate(t *testing.T) {
}
}
// HasActiveSession 是「近实时远程下线」的服务端判据:有非吊销会话=在线,
// 强制退出(RevokeByDevice)后=离线 → 客户端轮询到即登出。
func TestSQLite_SessionHasActiveSession(t *testing.T) {
ctx := context.Background()
db := openSQLite(t)
// FK 前置:user + device。
res, err := db.ExecContext(ctx,
`INSERT INTO users (uuid, email, pw_hash, dp_uuid) VALUES ('u-1','a@b.c','h','dp-1')`)
if err != nil {
t.Fatalf("user: %v", err)
}
uid, _ := res.LastInsertId()
res, err = db.ExecContext(ctx,
`INSERT INTO devices (uuid, user_id, name, platform) VALUES ('d-1', ?, 'Mac', 'macos')`, uid)
if err != nil {
t.Fatalf("device: %v", err)
}
did, _ := res.LastInsertId()
ss := sessions.NewStore(db)
if ok, err := ss.HasActiveSession(ctx, uid, did); err != nil || ok {
t.Fatalf("无会话应为非活跃: ok=%v err=%v", ok, err)
}
if err := ss.Create(ctx, uid, did, "jti-1", "1.2.3.4", "v1.0"); err != nil {
t.Fatalf("create: %v", err)
}
if ok, err := ss.HasActiveSession(ctx, uid, did); err != nil || !ok {
t.Fatalf("建会话后应活跃: ok=%v err=%v", ok, err)
}
if _, err := ss.RevokeByDevice(ctx, uid, did); err != nil {
t.Fatalf("revoke: %v", err)
}
if ok, err := ss.HasActiveSession(ctx, uid, did); err != nil || ok {
t.Fatalf("强制退出后应非活跃: ok=%v err=%v", ok, err)
}
}
func TestSQLite_NodeAccumulateUsage(t *testing.T) {
ctx := context.Background()
db := openSQLite(t)