split session tracker helpers

2026-10-06 05:16:38 +08:00 · 2026-06-27 03:47:02 +08:00
parent 0762f9fb5c
commit ccfa50e3a9
4 changed files with 225 additions and 206 deletions
@@ -0,0 +1,104 @@
+package service
+
+import (
+	"context"
+	"strings"
+	"time"
+
+	"go.uber.org/zap"
+)
+
+func (s *SessionTrackerService) upsert(ctx context.Context, in realtimeSessionInput) {
+	userID := strings.TrimSpace(in.UserID)
+	if userID == "" {
+		return
+	}
+	now := s.now()
+	in.DeviceID = strings.TrimSpace(in.DeviceID)
+	in.DeviceName = strings.TrimSpace(in.DeviceName)
+	in.Client = strings.TrimSpace(in.Client)
+	in.RemoteEndPoint = strings.TrimSpace(in.RemoteEndPoint)
+	if in.DeviceID == "" {
+		in.DeviceID = fallbackSessionDeviceID(in.DeviceName, in.Client, in.RemoteEndPoint)
+	}
+	key := userID + "\x00" + realtimeSessionTerminalKey(in.DeviceID, in.DeviceName, in.Client, in.RemoteEndPoint)
+	s.mu.Lock()
+	defer s.mu.Unlock()
+	s.pruneLocked(now)
+	if s.activity == nil {
+		s.activity = make(map[string]time.Time)
+	}
+	s.activity[userID] = now
+	existing, existed := s.sessions[key]
+	if strings.TrimSpace(in.UserName) == "" {
+		in.UserName = existing.UserName
+	}
+	if in.DeviceName == "" {
+		in.DeviceName = existing.DeviceName
+	}
+	if in.Client == "" {
+		in.Client = existing.Client
+	}
+	if in.RemoteEndPoint == "" {
+		in.RemoteEndPoint = existing.RemoteEndPoint
+	}
+	lastPlaybackAt := existing.LastPlaybackAt
+	itemID := existing.ItemID
+	positionTicks := existing.PositionTicks
+	runtimeTicks := existing.RuntimeTicks
+	isPlaying := existing.IsPlaying
+	isPaused := existing.IsPaused
+	if in.PlaybackUpdate {
+		itemID = firstNonEmptyString(in.ItemID, existing.ItemID)
+		if in.ItemID != "" || in.PositionTicks != 0 {
+			positionTicks = in.PositionTicks
+		}
+		if in.ItemID != "" || in.RuntimeTicks != 0 {
+			runtimeTicks = in.RuntimeTicks
+		}
+		isPlaying = in.IsPlaying
+		isPaused = in.IsPaused
+	}
+	if in.PlaybackUpdate && (in.ItemID != "" || in.IsPlaying) {
+		t := now
+		lastPlaybackAt = &t
+	}
+	s.sessions[key] = RealtimeSession{
+		ID:             key,
+		UserID:         userID,
+		UserName:       strings.TrimSpace(in.UserName),
+		DeviceID:       in.DeviceID,
+		DeviceName:     in.DeviceName,
+		Client:         in.Client,
+		RemoteEndPoint: in.RemoteEndPoint,
+		LastActivityAt: now,
+		ItemID:         itemID,
+		PositionTicks:  positionTicks,
+		RuntimeTicks:   runtimeTicks,
+		IsPlaying:      isPlaying,
+		IsPaused:       isPaused,
+		LastPlaybackAt: lastPlaybackAt,
+	}
+	if !existed && s.log != nil {
+		s.log.Debug("realtime session started",
+			zap.String("user_id", userID),
+			zap.String("device_id", in.DeviceID),
+			zap.String("client", in.Client),
+			zap.String("remote", in.RemoteEndPoint),
+		)
+	}
+}
+
+func (s *SessionTrackerService) pruneLocked(now time.Time) {
+	expiresBefore := now.Add(-realtimeSessionTTL)
+	for key, sess := range s.sessions {
+		if sess.LastActivityAt.Before(expiresBefore) {
+			delete(s.sessions, key)
+		}
+	}
+	for userID, lastActivity := range s.activity {
+		if lastActivity.Before(expiresBefore) {
+			delete(s.activity, userID)
+		}
+	}
+}