using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;

namespace VueWebCoreApi.SignalR
{
    /// <summary>
    /// 用户连接ID与用户编码映射存储（线程安全版）
    /// 解决多线程并发读写导致的字典数据错乱问题
    /// </summary>
    public static class UserIdsStore
    {
        // 替换为线程安全的并发字典
        private static readonly Dictionary<string, string> _ids = new Dictionary<string, string>();
        // 加锁保证字典操作的原子性（比ConcurrentDictionary更灵活控制批量操作）
        private static readonly object _lockObj = new object();

        /// <summary>
        /// 对外暴露的连接映射（只读，避免外部直接修改）
        /// </summary>
        public static IReadOnlyDictionary<string, string> Ids
        {
            get
            {
                lock (_lockObj)
                {
                    // 返回新字典，避免外部拿到引用后修改
                    return new Dictionary<string, string>(_ids);
                }
            }
        }

        /// <summary>
        /// 添加/更新用户连接（原子操作）
        /// </summary>
        /// <param name="connectionId">SignalR连接ID</param>
        /// <param name="userCode">用户编码</param>
        public static void AddOrUpdateUser(string connectionId, string userCode)
        {
            if (string.IsNullOrEmpty(connectionId) || string.IsNullOrEmpty(userCode))
                throw new ArgumentNullException("连接ID和用户编码不能为空");

            lock (_lockObj)
            {
                // 先移除该用户的旧连接（一个用户只保留最新的连接）
                var oldConnIds = _ids.Where(kv => kv.Value == userCode).Select(kv => kv.Key).ToList();
                foreach (var oldConnId in oldConnIds)
                {
                    _ids.Remove(oldConnId);
                }
                // 添加新连接（覆盖同名connectionId，避免重复）
                if (_ids.ContainsKey(connectionId))
                {
                    _ids[connectionId] = userCode;
                }
                else
                {
                    _ids.Add(connectionId, userCode);
                }
            }
        }

        /// <summary>
        /// 移除指定连接ID的映射
        /// </summary>
        /// <param name="connectionId">SignalR连接ID</param>
        public static void RemoveUser(string connectionId)
        {
            if (string.IsNullOrEmpty(connectionId))
                return;

            lock (_lockObj)
            {
                if (_ids.ContainsKey(connectionId))
                {
                    _ids.Remove(connectionId);
                }
            }
        }

        /// <summary>
        /// 根据用户编码获取所有在线连接ID（一个用户可能多端登录时返回多个）
        /// </summary>
        /// <param name="userCodes">用户编码列表</param>
        /// <returns>连接ID列表</returns>
        public static List<string> GetConnectionIdsByUserCodes(List<string> userCodes)
        {
            if (userCodes == null || userCodes.Count == 0)
                return new List<string>();

            lock (_lockObj)
            {
                return _ids.Where(kv => userCodes.Contains(kv.Value))
                           .Select(kv => kv.Key)
                           .Distinct() // 去重，避免重复推送
                           .ToList();
            }
        }
    }
}