using DotNetty.Common.Utilities;
using DotNetty.Transport.Channels;

namespace HybirdFrameworkDriver.Session;

public static class ChannelUtils
{
    private static readonly AttributeKey<IoSession> SessionKey = AttributeKey<IoSession>.ValueOf("session");

    /**
     * 添加新的会话
     * @param channel
     * @param session
     * @return
     */
    public static void AddChannelSession(IChannel channel, IoSession session)
    {
        var attribute = channel.GetAttribute(SessionKey);
        attribute.Set(session);
    }

    public static IoSession GetSessionBy(IChannel channel)
    {
        var sessionAttr = channel.GetAttribute(SessionKey);
        return sessionAttr.Get();
    }

    public static string GetIp(IChannel channel)
    {
        return channel.RemoteAddress.Serialize().ToString();
    }
}