hondacrx
122 lines
3.3 KiB
C#
122 lines
3.3 KiB
C#
/*
|
|
* Copyright (C) 2012-2017 CypherCore <http://github.com/CypherCore>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
|
|
namespace Framework.Networking
|
|
{
|
|
public class NetworkThread<SocketType> where SocketType : ISocket
|
|
{
|
|
public void Stop()
|
|
{
|
|
_stopped = true;
|
|
}
|
|
|
|
public bool Start()
|
|
{
|
|
if (_thread != null)
|
|
return false;
|
|
|
|
_thread = new Thread(Run);
|
|
_thread.Start();
|
|
return true;
|
|
}
|
|
|
|
public int GetConnectionCount()
|
|
{
|
|
return _connections;
|
|
}
|
|
|
|
public virtual void AddSocket(SocketType sock)
|
|
{
|
|
Interlocked.Increment(ref _connections);
|
|
_newSockets.Add(sock);
|
|
SocketAdded(sock);
|
|
}
|
|
|
|
public virtual void SocketAdded(SocketType sock) { }
|
|
public virtual void SocketRemoved(SocketType sock) { }
|
|
|
|
void AddNewSockets()
|
|
{
|
|
if (_newSockets.Empty())
|
|
return;
|
|
|
|
foreach (var socket in _newSockets)
|
|
{
|
|
if (!socket.IsOpen())
|
|
{
|
|
SocketRemoved(socket);
|
|
|
|
Interlocked.Decrement(ref _connections);
|
|
}
|
|
else
|
|
_Sockets.Add(socket);
|
|
}
|
|
|
|
_newSockets.Clear();
|
|
}
|
|
|
|
void Run()
|
|
{
|
|
Log.outDebug(LogFilter.Network, "Network Thread Starting");
|
|
|
|
int sleepTime = 10;
|
|
uint tickStart = 0, diff = 0;
|
|
while (!_stopped)
|
|
{
|
|
Thread.Sleep(sleepTime);
|
|
|
|
tickStart = Time.GetMSTime();
|
|
|
|
AddNewSockets();
|
|
|
|
foreach (var socket in _Sockets.ToList())
|
|
{
|
|
if (!socket.Update())
|
|
{
|
|
if (socket.IsOpen())
|
|
socket.CloseSocket();
|
|
|
|
SocketRemoved(socket);
|
|
|
|
--_connections;
|
|
_Sockets.Remove(socket);
|
|
}
|
|
}
|
|
|
|
diff = Time.GetMSTimeDiffToNow(tickStart);
|
|
sleepTime = (int)(diff > 10 ? 0 : 10 - diff);
|
|
}
|
|
|
|
Log.outDebug(LogFilter.Misc, "Network Thread exits");
|
|
_newSockets.Clear();
|
|
_Sockets.Clear();
|
|
}
|
|
|
|
volatile int _connections;
|
|
volatile bool _stopped;
|
|
|
|
Thread _thread;
|
|
|
|
List<SocketType> _Sockets = new List<SocketType>();
|
|
List<SocketType> _newSockets = new List<SocketType>();
|
|
}
|
|
}
|