StarEscort/DotNet/Hotfix/Scenes/Realm/UserSessionComponentSystem.cs

namespace ET.Server
{
    [FriendOf(typeof(UserSessionComponent))]
    public static class UserSessionComponentSystem
    {
        public class UserSessionComponentDestroySystem: DestroySystem<UserSessionComponent>
        {
            protected override void Destroy(UserSessionComponent self)
            {
                self.UserSession.Clear();
            }
        }

        public static long Get(this UserSessionComponent self, long userId)
        {
            if (!self.UserSession.TryGetValue(userId, out long instanceId))
            {
                return 0;
            }

            return instanceId;
        }

        public static void Add(this UserSessionComponent self, long userId, long instanceId)
        {
            if (self.UserSession.ContainsKey(userId))
            {
                self.UserSession[userId] = instanceId;
                return;
            }
            self.UserSession.Add(userId, instanceId);
        }

        public static void Remove(this UserSessionComponent self, long userId)
        {
            if (self.UserSession.ContainsKey(userId))
            {
                self.UserSession.Remove(userId);
            }
        }
    }
}