EchoChat/backend/go-service/app/meeting/service/meeting_lifecycle_service.go

// Package service 提供 meeting 模块的业务逻辑
package service

import (
	"context"
	"encoding/json"
	"fmt"
	"sync"
	"time"

	"github.com/echochat/backend/app/constants"
	"github.com/echochat/backend/app/meeting/dao"
	"github.com/echochat/backend/config"
	"github.com/echochat/backend/pkg/logs"
	"github.com/redis/go-redis/v9"
	"go.uber.org/zap"
)

// Redis key 前缀（设计文档 §5.4 生命周期 keys；Task 8 落地）
const (
	redisKeyHostGracePrefix = "echo:meeting:host_grace:"
	redisKeyEmptyTTLPrefix  = "echo:meeting:empty_ttl:"
)

// hostGracePayload host 宽限期 Redis value 结构
type hostGracePayload struct {
	HostID     int64 `json:"host_id"`
	StartedAt  int64 `json:"started_at"`  // 秒级 unix
	GraceUntil int64 `json:"grace_until"` // 秒级 unix
}

// MeetingLifecycleService 会议生命周期状态机
// 负责设计 §6.5 的 5 类状态跃迁副作用（host 宽限期、自动转让、空房 TTL、TTL 过期销毁）
// 采用 time.AfterFunc 本地 timer + Redis key TTL 双保险：
//   - 本地 timer 负责低延迟触发；服务重启由 RescheduleFromRedis 按 Redis 剩余 PTTL 重新装载
//   - 后台定时任务 MeetingCleanupTask 每 N 秒扫描 Redis 兜底防御（timer 丢失 / 多实例部署重复触发场景）
//
// 并发保护：HandleHostGraceExpired / HandleEmptyRoomExpired 入口先 DEL Redis key
// 通过 DEL 返回值确认唯一处理权（0 = 已被其他 goroutine/节点清走，直接 return 避免重复操作）
type MeetingLifecycleService struct {
	roomDAO        *dao.MeetingRoomDAO
	participantDAO *dao.MeetingParticipantDAO
	db             dbExecutor
	redis          *redis.Client
	broadcaster    *MeetingBroadcaster
	media          MediaOrchestrator
	cfg            config.MeetingConfig

	graceTimers    sync.Map // roomCode -> *time.Timer（host 宽限期）
	emptyTTLTimers sync.Map // roomCode -> *time.Timer（空房 TTL）
}

// dbExecutor MeetingLifecycleService 对事务 DB 的最小依赖抽象
// 仅用于 HandleEmptyRoomExpired 的 MarkEnded 调用链路，避免强绑 *gorm.DB
type dbExecutor interface{}

// NewMeetingLifecycleService 构造生命周期服务
// media 由上游通过 wire 注入 HTTPMediaOrchestrator
func NewMeetingLifecycleService(
	roomDAO *dao.MeetingRoomDAO,
	participantDAO *dao.MeetingParticipantDAO,
	redis *redis.Client,
	broadcaster *MeetingBroadcaster,
	media MediaOrchestrator,
	cfg *config.Config,
) *MeetingLifecycleService {
	meetingCfg := cfg.Meeting
	// 零值兜底：防止 yaml 未配置时走 0 秒 TTL
	if meetingCfg.HostGraceSeconds <= 0 {
		meetingCfg.HostGraceSeconds = constants.MeetingHostGraceSeconds
	}
	if meetingCfg.EmptyRoomTTLSeconds <= 0 {
		meetingCfg.EmptyRoomTTLSeconds = constants.MeetingEmptyRoomTTLSeconds
	}
	if meetingCfg.CleanupIntervalSeconds <= 0 {
		meetingCfg.CleanupIntervalSeconds = 30
	}
	if meetingCfg.StaleRoomHours <= 0 {
		meetingCfg.StaleRoomHours = 4
	}
	return &MeetingLifecycleService{
		roomDAO:        roomDAO,
		participantDAO: participantDAO,
		redis:          redis,
		broadcaster:    broadcaster,
		media:          media,
		cfg:            meetingCfg,
	}
}

// Config 返回生效的生命周期配置（供定时任务读取扫描周期等）
func (s *MeetingLifecycleService) Config() config.MeetingConfig {
	return s.cfg
}

// HostGraceKey 返回 host 宽限期 Redis key
func HostGraceKey(roomCode string) string {
	return redisKeyHostGracePrefix + roomCode
}

// EmptyTTLKey 返回空房 TTL Redis key
func EmptyTTLKey(roomCode string) string {
	return redisKeyEmptyTTLPrefix + roomCode
}

// OnHostDisconnect host WS 断线钩子
// 写入 host_grace key（TTL HostGraceSeconds）并启动本地 AfterFunc
// 若已有同 roomCode 的 grace 记录（重复触发），幂等刷新 TTL + 替换 timer
func (s *MeetingLifecycleService) OnHostDisconnect(ctx context.Context, roomCode string, hostID int64) {
	funcName := "service.meeting_lifecycle_service.OnHostDisconnect"

	now := time.Now()
	payload := hostGracePayload{
		HostID:     hostID,
		StartedAt:  now.Unix(),
		GraceUntil: now.Add(time.Duration(s.cfg.HostGraceSeconds) * time.Second).Unix(),
	}
	raw, _ := json.Marshal(payload)

	graceDur := time.Duration(s.cfg.HostGraceSeconds) * time.Second
	// Redis TTL 比本地 timer 多留一段 buffer（max(cleanup*2, 30s)）
	// 保证本地 timer 先触发 → DEL 命中；即便本地 timer 丢失，cleanup 也能在 buffer 内兜底扫到
	redisTTL := graceDur + s.ttlBuffer()
	if err := s.redis.Set(ctx, HostGraceKey(roomCode), string(raw), redisTTL).Err(); err != nil {
		logs.Warn(ctx, funcName, "写入 host_grace key 失败",
			zap.String("room_code", roomCode), zap.Int64("host_id", hostID), zap.Error(err))
		return
	}

	s.scheduleHostGraceTimer(roomCode, graceDur)
	logs.Info(ctx, funcName, "host 进入宽限期",
		zap.String("room_code", roomCode),
		zap.Int64("host_id", hostID),
		zap.Int("grace_seconds", s.cfg.HostGraceSeconds))
}

// OnHostReconnect host 重连钩子
// 清除 host_grace key 并取消本地 timer；用于 host 掉线后 TTL 内重新出现在 WS 层
// 幂等：key 不存在也不报错
func (s *MeetingLifecycleService) OnHostReconnect(ctx context.Context, roomCode string, hostID int64) {
	funcName := "service.meeting_lifecycle_service.OnHostReconnect"

	deleted, err := s.redis.Del(ctx, HostGraceKey(roomCode)).Result()
	if err != nil {
		logs.Warn(ctx, funcName, "清除 host_grace key 失败",
			zap.String("room_code", roomCode), zap.Error(err))
	}
	s.cancelTimer(&s.graceTimers, roomCode)
	if deleted > 0 {
		logs.Info(ctx, funcName, "host 宽限期内重连，撤销转让计划",
			zap.String("room_code", roomCode), zap.Int64("host_id", hostID))
	}
}

// HandleHostGraceExpired host 宽限期过期处理
// 触发时机：本地 timer 到期 或 cleanup task 扫到 key TTL <=0
// 流程：
//  1. DEL host_grace key → 返回 0 表示已被其他路径处理，直接 return 防止重复转让
//  2. 加载 room + 活跃成员；若 room 已 Ended / host 已变更，视为幂等完成
//  3. 活跃成员存在：选最早加入者转 host（事务），广播 host.changed，auto_reason=host_grace_expired
//  4. 无活跃成员：走 OnAllMembersLeft 路径（置 empty_ttl）
func (s *MeetingLifecycleService) HandleHostGraceExpired(ctx context.Context, roomCode string) {
	funcName := "service.meeting_lifecycle_service.HandleHostGraceExpired"

	deleted, err := s.redis.Del(ctx, HostGraceKey(roomCode)).Result()
	if err != nil {
		logs.Warn(ctx, funcName, "DEL host_grace key 失败（跳过本次处理）",
			zap.String("room_code", roomCode), zap.Error(err))
		return
	}
	s.cancelTimer(&s.graceTimers, roomCode)
	if deleted == 0 {
		logs.Debug(ctx, funcName, "host_grace key 不存在（已被重连/其他节点清走）",
			zap.String("room_code", roomCode))
		return
	}

	room, err := s.roomDAO.GetByCode(ctx, roomCode)
	if err != nil || room == nil {
		logs.Warn(ctx, funcName, "加载会议失败或已不存在",
			zap.String("room_code", roomCode), zap.Error(err))
		return
	}
	if room.Status == constants.MeetingStatusEnded {
		logs.Info(ctx, funcName, "会议已结束，跳过 host 宽限期过期处理",
			zap.String("room_code", roomCode))
		return
	}

	actives, err := s.participantDAO.ListActiveByRoom(ctx, room.ID)
	if err != nil {
		logs.Warn(ctx, funcName, "拉取活跃成员失败",
			zap.String("room_code", roomCode), zap.Error(err))
		return
	}
	// 排除原 host（若 left_at IS NULL 但 WS 掉线未落库）
	candidates := make([]int64, 0, len(actives))
	for _, p := range actives {
		if p.UserID == room.HostID {
			continue
		}
		candidates = append(candidates, p.UserID)
	}

	if len(candidates) == 0 {
		logs.Info(ctx, funcName, "host 宽限期过期且无其他活跃成员，转空房 TTL",
			zap.String("room_code", roomCode), zap.Int64("old_host_id", room.HostID))
		// 清掉 host 自己的记录（若 WS 掉线未落库，由兜底 LeaveRoom 处理）
		_, _ = s.participantDAO.LeaveRoom(ctx, room.ID, room.HostID, constants.MeetingLeftReasonDisconnect)
		s.OnAllMembersLeft(ctx, roomCode)
		return
	}

	newHostID := candidates[0]
	if err := s.participantDAO.TransferHost(ctx, room.ID, room.HostID, newHostID); err != nil {
		logs.Error(ctx, funcName, "host 自动转让 TransferHost 失败",
			zap.String("room_code", roomCode), zap.Error(err))
		return
	}
	if err := s.roomDAO.UpdateHost(ctx, room.ID, newHostID); err != nil {
		logs.Warn(ctx, funcName, "UpdateHost 失败（已发生参与者 role 变更）",
			zap.String("room_code", roomCode), zap.Error(err))
	}
	// 老 host 彻底离会（宽限期内未重连视为网络断开）
	_, _ = s.participantDAO.LeaveRoom(ctx, room.ID, room.HostID, constants.MeetingLeftReasonDisconnect)

	s.broadcaster.BroadcastToMeeting(ctx, room.ID, constants.MeetingWSEventHostChanged, map[string]interface{}{
		"room_code":   roomCode,
		"old_host_id": room.HostID,
		"new_host_id": newHostID,
		"auto_reason": "host_grace_expired",
	})
	logs.Info(ctx, funcName, "host 宽限期过期自动转让完成",
		zap.String("room_code", roomCode),
		zap.Int64("old_host_id", room.HostID),
		zap.Int64("new_host_id", newHostID))
}

// OnAllMembersLeft 全员退出钩子
// 设置 empty_ttl key + 启动本地 AfterFunc；期间有人重新加入可由 CancelEmptyTTL 复活房间
func (s *MeetingLifecycleService) OnAllMembersLeft(ctx context.Context, roomCode string) {
	funcName := "service.meeting_lifecycle_service.OnAllMembersLeft"

	ttl := time.Duration(s.cfg.EmptyRoomTTLSeconds) * time.Second
	// Redis TTL 同样加 buffer 避免与本地 timer 同时到期导致 DEL 无法命中
	redisTTL := ttl + s.ttlBuffer()
	if err := s.redis.Set(ctx, EmptyTTLKey(roomCode), "1", redisTTL).Err(); err != nil {
		logs.Warn(ctx, funcName, "写入 empty_ttl key 失败",
			zap.String("room_code", roomCode), zap.Error(err))
		return
	}
	s.scheduleEmptyTTLTimer(roomCode, ttl)
	logs.Info(ctx, funcName, "空房 TTL 已启动",
		zap.String("room_code", roomCode), zap.Int("ttl_seconds", s.cfg.EmptyRoomTTLSeconds))
}

// CancelEmptyTTL 取消空房 TTL（有新成员加入时调用，房间复活）
// 幂等：key 不存在也不报错
func (s *MeetingLifecycleService) CancelEmptyTTL(ctx context.Context, roomCode string) {
	funcName := "service.meeting_lifecycle_service.CancelEmptyTTL"

	deleted, err := s.redis.Del(ctx, EmptyTTLKey(roomCode)).Result()
	if err != nil {
		logs.Warn(ctx, funcName, "清除 empty_ttl key 失败",
			zap.String("room_code", roomCode), zap.Error(err))
	}
	s.cancelTimer(&s.emptyTTLTimers, roomCode)
	if deleted > 0 {
		logs.Info(ctx, funcName, "空房 TTL 已取消（有人加入）",
			zap.String("room_code", roomCode))
	}
}

// HandleEmptyRoomExpired 空房 TTL 过期处理
// 流程：DEL empty_ttl key → 确认没被新成员清走 → MarkEnded(reason=empty_ttl) + CloseRouter 幂等
func (s *MeetingLifecycleService) HandleEmptyRoomExpired(ctx context.Context, roomCode string) {
	funcName := "service.meeting_lifecycle_service.HandleEmptyRoomExpired"

	deleted, err := s.redis.Del(ctx, EmptyTTLKey(roomCode)).Result()
	if err != nil {
		logs.Warn(ctx, funcName, "DEL empty_ttl key 失败",
			zap.String("room_code", roomCode), zap.Error(err))
		return
	}
	s.cancelTimer(&s.emptyTTLTimers, roomCode)
	if deleted == 0 {
		logs.Debug(ctx, funcName, "empty_ttl key 已消失（有人重入 或 已处理过）",
			zap.String("room_code", roomCode))
		return
	}

	room, err := s.roomDAO.GetByCode(ctx, roomCode)
	if err != nil || room == nil {
		logs.Warn(ctx, funcName, "加载会议失败或已不存在",
			zap.String("room_code", roomCode), zap.Error(err))
		return
	}
	if room.Status == constants.MeetingStatusEnded {
		return
	}

	// 二次校验：DEL 成功与 DB 校验之间可能有新成员加入（并发场景）
	activeCount, _ := s.participantDAO.CountActiveByRoom(ctx, room.ID)
	if activeCount > 0 {
		logs.Info(ctx, funcName, "检测到活跃成员，放弃销毁",
			zap.String("room_code", roomCode), zap.Int64("active_count", activeCount))
		return
	}

	if _, mErr := s.roomDAO.MarkEnded(ctx, room.ID, constants.MeetingEndedReasonEmptyTTL, time.Now()); mErr != nil {
		logs.Error(ctx, funcName, "MarkEnded 失败",
			zap.String("room_code", roomCode), zap.Error(mErr))
		return
	}
	if err := s.media.CloseRouter(ctx, roomCode); err != nil {
		logs.Warn(ctx, funcName, "CloseRouter 失败（不影响 DB 状态）",
			zap.String("room_code", roomCode), zap.Error(err))
	}

	s.broadcaster.BroadcastToMeeting(ctx, room.ID, constants.MeetingWSEventRoomEnded, map[string]interface{}{
		"room_code": roomCode,
		"reason":    constants.MeetingEndedReasonEmptyTTL,
	})
	logs.Info(ctx, funcName, "空房 TTL 过期，会议已销毁",
		zap.String("room_code", roomCode), zap.Int64("room_id", room.ID))
}

// RescheduleFromRedis 服务启动时扫描 Redis 已存在的 grace / empty_ttl key，按剩余 PTTL 补装本地 timer
// 服务重启后调用一次，避免重启期间 WS 断线事件的本地 timer 丢失导致宽限期/TTL 永远不触发
func (s *MeetingLifecycleService) RescheduleFromRedis(ctx context.Context) {
	funcName := "service.meeting_lifecycle_service.RescheduleFromRedis"

	s.rescheduleByPattern(ctx, funcName, redisKeyHostGracePrefix+"*", func(roomCode string, dur time.Duration) {
		s.scheduleHostGraceTimer(roomCode, dur)
	})
	s.rescheduleByPattern(ctx, funcName, redisKeyEmptyTTLPrefix+"*", func(roomCode string, dur time.Duration) {
		s.scheduleEmptyTTLTimer(roomCode, dur)
	})
}

// ScanExpired 扫描指定 key pattern，返回 TTL <=0 （或即将过期）对应的 roomCode 列表
// 供定时任务兜底调用：若本地 timer 已丢失，此时 Redis key 可能仍存在但 TTL 为负
// 本实现采用 SCAN + PTTL，单次扫描 limit 控制内存
func (s *MeetingLifecycleService) ScanExpired(ctx context.Context, keyPrefix string, limit int64) []string {
	funcName := "service.meeting_lifecycle_service.ScanExpired"

	expired := make([]string, 0, limit)
	var cursor uint64
	for {
		keys, next, err := s.redis.Scan(ctx, cursor, keyPrefix+"*", limit).Result()
		if err != nil {
			logs.Warn(ctx, funcName, "SCAN 失败", zap.String("pattern", keyPrefix), zap.Error(err))
			return expired
		}
		for _, k := range keys {
			pttl, err := s.redis.PTTL(ctx, k).Result()
			if err != nil {
				continue
			}
			// PTTL<=0（redis 过期延迟删除场景）；PTTL==-2 已不存在
			if pttl > 0 {
				continue
			}
			roomCode := keyToRoomCode(k, keyPrefix)
			if roomCode != "" {
				expired = append(expired, roomCode)
			}
		}
		if next == 0 {
			break
		}
		cursor = next
	}
	return expired
}

// HostGracePrefix / EmptyTTLPrefix 暴露 key 前缀供定时任务扫描使用
func (s *MeetingLifecycleService) HostGracePrefix() string { return redisKeyHostGracePrefix }
func (s *MeetingLifecycleService) EmptyTTLPrefix() string  { return redisKeyEmptyTTLPrefix }

// ========== 内部辅助 ==========

func (s *MeetingLifecycleService) scheduleHostGraceTimer(roomCode string, d time.Duration) {
	s.cancelTimer(&s.graceTimers, roomCode)
	timer := time.AfterFunc(d, func() {
		ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
		defer cancel()
		s.HandleHostGraceExpired(ctx, roomCode)
	})
	s.graceTimers.Store(roomCode, timer)
}

func (s *MeetingLifecycleService) scheduleEmptyTTLTimer(roomCode string, d time.Duration) {
	s.cancelTimer(&s.emptyTTLTimers, roomCode)
	timer := time.AfterFunc(d, func() {
		ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
		defer cancel()
		s.HandleEmptyRoomExpired(ctx, roomCode)
	})
	s.emptyTTLTimers.Store(roomCode, timer)
}

// ttlBuffer 返回 Redis key 相对本地 timer 的额外 buffer 时长
// 用于保证本地 timer 先触发（DEL 命中），避免 Redis 自动过期后 DEL 返回 0 被误判为重复触发
// 规则：max(CleanupIntervalSeconds * 2, 30s)，确保单机 cleanup 至少能扫到一轮兜底
func (s *MeetingLifecycleService) ttlBuffer() time.Duration {
	buf := time.Duration(s.cfg.CleanupIntervalSeconds*2) * time.Second
	if buf < 30*time.Second {
		buf = 30 * time.Second
	}
	return buf
}

// cancelTimer 取消本地 timer（幂等，不存在也安全）
func (s *MeetingLifecycleService) cancelTimer(store *sync.Map, roomCode string) {
	if raw, ok := store.LoadAndDelete(roomCode); ok {
		if t, ok := raw.(*time.Timer); ok {
			t.Stop()
		}
	}
}

// rescheduleByPattern 通用"按 pattern 扫描 Redis + 按 PTTL 重建本地 timer"逻辑
func (s *MeetingLifecycleService) rescheduleByPattern(
	ctx context.Context,
	funcName string,
	pattern string,
	schedule func(roomCode string, dur time.Duration),
) {
	var cursor uint64
	total := 0
	for {
		keys, next, err := s.redis.Scan(ctx, cursor, pattern, 100).Result()
		if err != nil {
			logs.Warn(ctx, funcName, "SCAN 失败", zap.String("pattern", pattern), zap.Error(err))
			return
		}
		for _, k := range keys {
			pttl, err := s.redis.PTTL(ctx, k).Result()
			if err != nil || pttl <= 0 {
				continue
			}
			prefix := pattern[:len(pattern)-1] // 去掉末尾 *
			roomCode := keyToRoomCode(k, prefix)
			if roomCode == "" {
				continue
			}
			schedule(roomCode, pttl)
			total++
		}
		if next == 0 {
			break
		}
		cursor = next
	}
	if total > 0 {
		logs.Info(ctx, funcName, "服务启动重建 timer 完成",
			zap.String("pattern", pattern), zap.Int("count", total))
	}
}

// keyToRoomCode 从完整 Redis key 中去掉前缀得到 roomCode
// 返回空字符串表示格式异常
func keyToRoomCode(fullKey, prefix string) string {
	if len(fullKey) <= len(prefix) {
		return ""
	}
	return fullKey[len(prefix):]
}

// formatErr 构造统一带上下文的错误（用于少数需要 return 的场景，避免裸字符串拼接）
// nolint:unused
func formatErr(funcName, msg string, err error) error {
	return fmt.Errorf("%s: %s: %w", funcName, msg, err)
}