AAEmu.Commons/Network/Core/Server.cs - develop - NLObP/AAEmu

AAEmu.Commons/Network/Core/Server.cs

69 строк2 KB

Roger Barreto

Collection expressions check + blank lines update (#1110)

11 дек 2024, 19:38

11 дек 2024, 19:380ce5643

using System;
using System.Collections.Generic;
using System.Linq;
using System.Net;
using System.Net.Sockets;
using NetCoreServer;
using NLog;

namespace AAEmu.Commons.Network.Core;

public class Server : TcpServer
{
    private static Logger Logger { get; } = LogManager.GetCurrentClassLogger();
    private BaseProtocolHandler _protocolHandler;
    private readonly HashSet<Session> _sessions = [];

    public BaseProtocolHandler GetHandler() => _protocolHandler;

    public Server(IPAddress address, int port, BaseProtocolHandler protocolHandler)
        : base(address, port)
    {
        _protocolHandler = protocolHandler;
    }

    protected override TcpSession CreateSession() => new Session(this);

    protected override void OnStarted()
    {
        Logger.Info($"TCP server listening start on {Endpoint}");
    }

    protected override void OnStopped()
    {
        Logger.Info("TCP server listener stopped!");
    }

    protected override void OnConnected(TcpSession session)
    {
        Logger.Info(
            $"Connect from {session.Socket.RemoteEndPoint} established, session id: {session.Id}");
        _sessions.Add((Session)session);
    }

    protected override void OnDisconnected(TcpSession session)
    {
        Logger.Info($"Connect from session id: {session.Id} disconnected");
        _sessions.Remove((Session)session);
    }

    protected override void OnError(SocketError error)
    {
        Logger.Error($"TCP server SocketError: {error}");
    }

    public Session GetSession(Func<Session, bool> func)
    {
        return _sessions.SingleOrDefault(func);
    }

    public HashSet<Session> GetSessions()
    {
        return _sessions;
    }

    public IEnumerable<Session> GetSessions(Func<Session, bool> func)
    {
        return _sessions.Where(func);
    }
}