增加原始模式
This commit is contained in:
parent
5ab3985f78
commit
d9534a1a24
246
NetLib/HaoYueNet.ClientNetwork/NetworkHelperCore_SourceMode.cs
Normal file
246
NetLib/HaoYueNet.ClientNetwork/NetworkHelperCore_SourceMode.cs
Normal file
@ -0,0 +1,246 @@
|
|||||||
|
using System.Net;
|
||||||
|
using System.Net.Sockets;
|
||||||
|
using static HaoYueNet.ClientNetwork.BaseData;
|
||||||
|
|
||||||
|
namespace HaoYueNet.ClientNetwork
|
||||||
|
{
|
||||||
|
public class NetworkHelperCore_SourceMode
|
||||||
|
{
|
||||||
|
private Socket client;
|
||||||
|
|
||||||
|
////响应倒计时计数最大值
|
||||||
|
//private static int MaxRevIndexNum = 6;
|
||||||
|
|
||||||
|
////发送倒计时计数最大值
|
||||||
|
//private static int MaxSendIndexNum = 3;
|
||||||
|
|
||||||
|
//响应倒计时计数最大值
|
||||||
|
private static int MaxRevIndexNum = 50;
|
||||||
|
|
||||||
|
//发送倒计时计数最大值
|
||||||
|
private static int MaxSendIndexNum = 3;
|
||||||
|
|
||||||
|
//响应倒计时计数
|
||||||
|
private static int RevIndex=0;
|
||||||
|
//发送倒计时计数
|
||||||
|
private static int SendIndex=0;
|
||||||
|
|
||||||
|
//计时器间隔
|
||||||
|
private static int TimerInterval = 3000;
|
||||||
|
|
||||||
|
public static string LastConnectIP;
|
||||||
|
public static int LastConnectPort;
|
||||||
|
public bool bDetailedLog = false;
|
||||||
|
|
||||||
|
public bool Init(string IP, int port,bool isHadDetailedLog = true, bool bBindReuseAddress = false,int bBindport = 0)
|
||||||
|
{
|
||||||
|
LogOut("==>初始化网络核心");
|
||||||
|
|
||||||
|
bDetailedLog = isHadDetailedLog;
|
||||||
|
RevIndex = MaxRevIndexNum;
|
||||||
|
SendIndex = MaxSendIndexNum;
|
||||||
|
|
||||||
|
client = new Socket(SocketType.Stream, ProtocolType.Tcp);
|
||||||
|
if (bBindReuseAddress)
|
||||||
|
{
|
||||||
|
client.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.ReuseAddress, true);
|
||||||
|
IPEndPoint ipe = new IPEndPoint(IPAddress.Any, Convert.ToInt32(bBindport));
|
||||||
|
client.Bind(ipe);
|
||||||
|
}
|
||||||
|
LastConnectIP = IP;
|
||||||
|
LastConnectPort = port;
|
||||||
|
return Connect(IP, port);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool Connect(string IP, int port)
|
||||||
|
{
|
||||||
|
//带回调的
|
||||||
|
try
|
||||||
|
{
|
||||||
|
if(bDetailedLog)
|
||||||
|
LogOut("连接到远程IP " + IP + ":" + port);
|
||||||
|
else
|
||||||
|
LogOut("连接到远程服务");
|
||||||
|
|
||||||
|
client.Connect(IP, port);
|
||||||
|
Thread thread = new Thread(Recive);
|
||||||
|
thread.IsBackground = true;
|
||||||
|
thread.Start(client);
|
||||||
|
int localport = ((IPEndPoint)client.LocalEndPoint).Port;
|
||||||
|
|
||||||
|
if (bDetailedLog)
|
||||||
|
LogOut($"连接成功!连接到远程IP->{IP}:{port} | 本地端口->{localport}");
|
||||||
|
else
|
||||||
|
LogOut("连接成功!");
|
||||||
|
|
||||||
|
if (bDetailedLog)
|
||||||
|
LogOut("开启心跳包检测");
|
||||||
|
|
||||||
|
OnConnected?.Invoke(true);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
catch (Exception ex)
|
||||||
|
{
|
||||||
|
if (bDetailedLog)
|
||||||
|
LogOut("连接失败:" + ex.ToString());
|
||||||
|
else
|
||||||
|
LogOut("连接失败");
|
||||||
|
|
||||||
|
OnConnected?.Invoke(false);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
~NetworkHelperCore_SourceMode()
|
||||||
|
{
|
||||||
|
client.Close();
|
||||||
|
}
|
||||||
|
|
||||||
|
private void SendToSocket(byte[] data)
|
||||||
|
{
|
||||||
|
//已拼接包长度,这里不再需要拼接长度
|
||||||
|
//data = SendDataWithHead(data);
|
||||||
|
try
|
||||||
|
{
|
||||||
|
SendWithIndex(data);
|
||||||
|
}
|
||||||
|
catch (Exception ex)
|
||||||
|
{
|
||||||
|
//连接断开
|
||||||
|
OnCloseReady();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
//LogOut("发送消息,消息长度=> "+data.Length);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void SendHeartbeat()
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
SendWithIndex(HeartbeatData);
|
||||||
|
}
|
||||||
|
catch (Exception ex)
|
||||||
|
{
|
||||||
|
//连接断开
|
||||||
|
OnCloseReady();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
//LogOut("发送心跳包");
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 发送数据并计数
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="data"></param>
|
||||||
|
private void SendWithIndex(byte[] data)
|
||||||
|
{
|
||||||
|
//增加发送计数
|
||||||
|
SendIndex = MaxSendIndexNum;
|
||||||
|
//发送数据
|
||||||
|
client.Send(data);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 供外部调用 发送消息
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="CMDID"></param>
|
||||||
|
/// <param name="data">序列化之后的数据</param>
|
||||||
|
public void SendToServer(byte[] data)
|
||||||
|
{
|
||||||
|
//LogOut("准备数据 data=> "+data);
|
||||||
|
SendToSocket(data);
|
||||||
|
}
|
||||||
|
|
||||||
|
#region 事件定义
|
||||||
|
public delegate void OnReceiveDataHandler(byte[] data);
|
||||||
|
public delegate void OnConnectedHandler(bool IsConnected);
|
||||||
|
public delegate void OnCloseHandler();
|
||||||
|
public delegate void OnLogOutHandler(string Msg);
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
public event OnConnectedHandler OnConnected;
|
||||||
|
public event OnReceiveDataHandler OnReceiveData;
|
||||||
|
public event OnCloseHandler OnClose;
|
||||||
|
/// <summary>
|
||||||
|
/// 网络库调试日志输出
|
||||||
|
/// </summary>
|
||||||
|
public event OnLogOutHandler OnLogOut;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 做好处理的连接管理
|
||||||
|
/// </summary>
|
||||||
|
private void OnCloseReady()
|
||||||
|
{
|
||||||
|
LogOut("关闭连接");
|
||||||
|
//关闭Socket连接
|
||||||
|
client.Close();
|
||||||
|
OnClose?.Invoke();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 主动关闭连接
|
||||||
|
/// </summary>
|
||||||
|
public void CloseConntect()
|
||||||
|
{
|
||||||
|
OnCloseReady();
|
||||||
|
}
|
||||||
|
|
||||||
|
private void DataCallBackReady(byte[] data)
|
||||||
|
{
|
||||||
|
//增加接收计数
|
||||||
|
RevIndex = MaxRevIndexNum;
|
||||||
|
OnReceiveData(data);
|
||||||
|
}
|
||||||
|
|
||||||
|
MemoryStream memoryStream = new MemoryStream();//开辟一个内存流
|
||||||
|
private void Recive(object o)
|
||||||
|
{
|
||||||
|
var client = o as Socket;
|
||||||
|
//MemoryStream memoryStream = new MemoryStream();//开辟一个内存流
|
||||||
|
|
||||||
|
while (true)
|
||||||
|
{
|
||||||
|
byte[] buffer = new byte[1024 * 1024 * 2];
|
||||||
|
int effective=0;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
effective = client.Receive(buffer);
|
||||||
|
if (effective == 0)
|
||||||
|
{
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch(Exception ex)
|
||||||
|
{
|
||||||
|
//远程主机强迫关闭了一个现有的连接
|
||||||
|
OnCloseReady();
|
||||||
|
return;
|
||||||
|
//断开连接
|
||||||
|
}
|
||||||
|
memoryStream.Write(buffer, 0, effective);//将接受到的数据写入内存流中
|
||||||
|
while (true)
|
||||||
|
{
|
||||||
|
if (effective > 0)//如果接受到的消息不为0(不为空)
|
||||||
|
{
|
||||||
|
DataCallBackReady(memoryStream.ToArray());
|
||||||
|
//流复用的方式 不用重新new申请
|
||||||
|
memoryStream.Position = 0;
|
||||||
|
memoryStream.SetLength(0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
public void LogOut(string Msg)
|
||||||
|
{
|
||||||
|
//Console.WriteLine(Msg);
|
||||||
|
OnLogOut?.Invoke(Msg);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Socket GetClientSocket()
|
||||||
|
{
|
||||||
|
return client;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,727 @@
|
|||||||
|
//using HunterProtobufCore;
|
||||||
|
using System.Net;
|
||||||
|
using System.Net.Sockets;
|
||||||
|
using static HaoYueNet.ServerNetwork.BaseData;
|
||||||
|
|
||||||
|
namespace HaoYueNet.ServerNetwork
|
||||||
|
{
|
||||||
|
public class TcpSaeaServer_SourceMode
|
||||||
|
{
|
||||||
|
#region 定义属性
|
||||||
|
protected int MaxRevIndexNum = 50;//响应倒计时计数最大值
|
||||||
|
protected int MaxSendIndexNum = 3;//发送倒计时计数最大值
|
||||||
|
protected static int TimerInterval = 3000;//计时器间隔
|
||||||
|
protected System.Timers.Timer _heartTimer;//心跳包计数器
|
||||||
|
public int m_maxConnectNum; //最大连接数
|
||||||
|
public int m_revBufferSize; //最大接收字节数
|
||||||
|
protected BufferManager m_bufferManager;
|
||||||
|
protected const int opsToAlloc = 2;
|
||||||
|
Socket listenSocket; //监听Socket
|
||||||
|
protected SocketEventPool m_Receivepool;
|
||||||
|
protected SocketEventPool m_Sendpool;
|
||||||
|
protected TokenMsgPool_SourceMode msg_pool;
|
||||||
|
protected int m_clientCount; //连接的客户端数量
|
||||||
|
protected Semaphore m_maxNumberAcceptedClients;//信号量
|
||||||
|
protected Dictionary<Socket, AsyncUserToken> _DictSocketAsyncUserToken = new Dictionary<Socket, AsyncUserToken>();
|
||||||
|
List<AsyncUserToken> m_clients; //客户端列表
|
||||||
|
public List<AsyncUserToken> ClientList { private set { m_clients = value; } get { return m_clients; } } //获取客户端列表
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region 定义委托
|
||||||
|
/// <summary>
|
||||||
|
/// 客户端连接数量变化时触发
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="num">当前增加客户的个数(用户退出时为负数,增加时为正数,一般为1)</param>
|
||||||
|
/// <param name="token">增加用户的信息</param>
|
||||||
|
public delegate void OnClientNumberChangeHandler(int num, AsyncUserToken token);
|
||||||
|
/// <summary>
|
||||||
|
/// 接收到客户端的数据
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="token">客户端</param>
|
||||||
|
/// <param name="buff">客户端数据</param>
|
||||||
|
public delegate void OnReceiveDataHandler(AsyncUserToken sk, int CMDID, byte[] data);
|
||||||
|
/// <summary>
|
||||||
|
/// 断开连接
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="sk"></param>
|
||||||
|
public delegate void OnDisconnectHandler(AsyncUserToken sk);
|
||||||
|
/// <summary>
|
||||||
|
/// 日志
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="sk"></param>
|
||||||
|
public delegate void OnNetLogHandler(string msg);
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region 定义事件
|
||||||
|
/// <summary>
|
||||||
|
/// 客户端连接数量变化事件
|
||||||
|
/// </summary>
|
||||||
|
public event OnClientNumberChangeHandler OnClientNumberChange;
|
||||||
|
/// <summary>
|
||||||
|
/// 接收到客户端的数据事件
|
||||||
|
/// </summary>
|
||||||
|
public event OnReceiveDataHandler OnReceive;
|
||||||
|
/// <summary>
|
||||||
|
/// 接收到客户端的断开连接
|
||||||
|
/// </summary>
|
||||||
|
public event OnDisconnectHandler OnDisconnected;
|
||||||
|
/// <summary>
|
||||||
|
/// 网络库内部输出
|
||||||
|
/// </summary>
|
||||||
|
public event OnNetLogHandler OnNetLog;
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 构造函数
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="numConnections">最大连接数</param>
|
||||||
|
/// <param name="receiveBufferSize">缓存区大小</param>
|
||||||
|
public TcpSaeaServer_SourceMode(int numConnections, int receiveBufferSize)
|
||||||
|
{
|
||||||
|
m_clientCount = 0;
|
||||||
|
m_maxConnectNum = numConnections;
|
||||||
|
m_revBufferSize = receiveBufferSize;
|
||||||
|
// allocate buffers such that the maximum number of sockets can have one outstanding read and
|
||||||
|
//write posted to the socket simultaneously
|
||||||
|
m_bufferManager = new BufferManager(receiveBufferSize * numConnections * opsToAlloc, receiveBufferSize);
|
||||||
|
|
||||||
|
m_Receivepool = new SocketEventPool(numConnections);
|
||||||
|
m_Sendpool = new SocketEventPool(numConnections);
|
||||||
|
|
||||||
|
msg_pool = new TokenMsgPool_SourceMode(numConnections);
|
||||||
|
|
||||||
|
m_maxNumberAcceptedClients = new Semaphore(numConnections, numConnections);
|
||||||
|
}
|
||||||
|
|
||||||
|
#region Client操作
|
||||||
|
/// <summary>
|
||||||
|
/// 初始化
|
||||||
|
/// </summary>
|
||||||
|
public void Init()
|
||||||
|
{
|
||||||
|
// Allocates one large byte buffer which all I/O operations use a piece of. This gaurds
|
||||||
|
// against memory fragmentation
|
||||||
|
m_bufferManager.InitBuffer();
|
||||||
|
m_clients = new List<AsyncUserToken>();
|
||||||
|
// preallocate pool of SocketAsyncEventArgs objects
|
||||||
|
SocketAsyncEventArgs readWriteEventArg;
|
||||||
|
|
||||||
|
for (int i = 0; i < m_maxConnectNum; i++)
|
||||||
|
{
|
||||||
|
readWriteEventArg = new SocketAsyncEventArgs();
|
||||||
|
readWriteEventArg.Completed += new EventHandler<SocketAsyncEventArgs>(IO_Completed);
|
||||||
|
readWriteEventArg.UserToken = new AsyncUserToken();
|
||||||
|
// assign a byte buffer from the buffer pool to the SocketAsyncEventArg object
|
||||||
|
m_bufferManager.SetBuffer(readWriteEventArg);
|
||||||
|
// add SocketAsyncEventArg to the pool
|
||||||
|
m_Receivepool.Push(readWriteEventArg);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int i = 0; i < m_maxConnectNum; i++)
|
||||||
|
{
|
||||||
|
readWriteEventArg = new SocketAsyncEventArgs();
|
||||||
|
readWriteEventArg.Completed += new EventHandler<SocketAsyncEventArgs>(IO_Completed);
|
||||||
|
readWriteEventArg.UserToken = new AsyncUserToken();
|
||||||
|
|
||||||
|
//发送是否需要如此设置 TODO
|
||||||
|
m_bufferManager.SetBuffer(readWriteEventArg);
|
||||||
|
|
||||||
|
m_Sendpool.Push(readWriteEventArg);
|
||||||
|
}
|
||||||
|
OutNetLog("初始化完毕");
|
||||||
|
}
|
||||||
|
/// <summary>
|
||||||
|
/// 启动服务
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="localEndPoint"></param>
|
||||||
|
/// <param name="bReuseAddress">是否端口重用</param>
|
||||||
|
/// <returns></returns>
|
||||||
|
public bool Start(IPEndPoint localEndPoint, bool bReuseAddress = false)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
ClearUserToken();
|
||||||
|
listenSocket = new Socket(localEndPoint.AddressFamily, SocketType.Stream, ProtocolType.Tcp);
|
||||||
|
|
||||||
|
if (bReuseAddress)
|
||||||
|
{
|
||||||
|
listenSocket.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.ReuseAddress, true);
|
||||||
|
}
|
||||||
|
|
||||||
|
listenSocket.Bind(localEndPoint);
|
||||||
|
// start the server with a listen backlog of 100 connections
|
||||||
|
listenSocket.Listen(m_maxConnectNum);
|
||||||
|
// post accepts on the listening socket
|
||||||
|
StartAccept(null);
|
||||||
|
|
||||||
|
OutNetLog("监听:" + listenSocket.LocalEndPoint.ToString());
|
||||||
|
|
||||||
|
_heartTimer = new System.Timers.Timer();
|
||||||
|
_heartTimer.Interval = TimerInterval;
|
||||||
|
_heartTimer.Elapsed += CheckUpdatetimer_Elapsed;
|
||||||
|
_heartTimer.AutoReset = true;
|
||||||
|
_heartTimer.Enabled = true;
|
||||||
|
OutNetLog("开启定时心跳包");
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
catch (Exception)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
/// <summary>
|
||||||
|
/// 停止服务
|
||||||
|
/// </summary>
|
||||||
|
public void Stop()
|
||||||
|
{
|
||||||
|
foreach (AsyncUserToken token in m_clients)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
token.Socket.Shutdown(SocketShutdown.Both);
|
||||||
|
}
|
||||||
|
catch (Exception) { }
|
||||||
|
}
|
||||||
|
try
|
||||||
|
{
|
||||||
|
listenSocket.Shutdown(SocketShutdown.Both);
|
||||||
|
}
|
||||||
|
catch (Exception) { }
|
||||||
|
|
||||||
|
listenSocket.Close();
|
||||||
|
int c_count = m_clients.Count;
|
||||||
|
ClearUserToken();
|
||||||
|
|
||||||
|
if (OnClientNumberChange != null)
|
||||||
|
OnClientNumberChange(-c_count, null);
|
||||||
|
}
|
||||||
|
public void CloseClient(AsyncUserToken token)
|
||||||
|
{
|
||||||
|
try {token.Socket.Shutdown(SocketShutdown.Both);}
|
||||||
|
catch (Exception) { }
|
||||||
|
}
|
||||||
|
/// <summary>
|
||||||
|
/// 关闭客户端连接
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="e"></param>
|
||||||
|
void CloseClientSocket(SocketAsyncEventArgs e)
|
||||||
|
{
|
||||||
|
AsyncUserToken token = e.UserToken as AsyncUserToken;
|
||||||
|
CloseReady(token);
|
||||||
|
// 释放SocketAsyncEventArg,以便其他客户端可以重用它们
|
||||||
|
ReleaseSocketAsyncEventArgs(e);
|
||||||
|
}
|
||||||
|
void CloseReady(AsyncUserToken token)
|
||||||
|
{
|
||||||
|
OnDisconnected?.Invoke(token);
|
||||||
|
RemoveUserToken(token);
|
||||||
|
//如果有事件,则调用事件,发送客户端数量变化通知
|
||||||
|
OnClientNumberChange?.Invoke(-1, token);
|
||||||
|
// 关闭与客户端关联的套接字
|
||||||
|
try { token.Socket.Shutdown(SocketShutdown.Send); } catch (Exception) { }
|
||||||
|
token.Socket.Close();
|
||||||
|
// 递减计数器以跟踪连接到服务器的客户端总数
|
||||||
|
Interlocked.Decrement(ref m_clientCount);
|
||||||
|
m_maxNumberAcceptedClients.Release();
|
||||||
|
}
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region Token管理
|
||||||
|
public AsyncUserToken GetAsyncUserTokenForSocket(Socket sk)
|
||||||
|
{
|
||||||
|
return _DictSocketAsyncUserToken.ContainsKey(sk) ? _DictSocketAsyncUserToken[sk] : null;
|
||||||
|
}
|
||||||
|
void AddUserToken(AsyncUserToken userToken)
|
||||||
|
{
|
||||||
|
lock (_DictSocketAsyncUserToken)
|
||||||
|
{
|
||||||
|
m_clients.Add(userToken);
|
||||||
|
_DictSocketAsyncUserToken.Add(userToken.Socket, userToken);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
void RemoveUserToken(AsyncUserToken userToken)
|
||||||
|
{
|
||||||
|
lock (_DictSocketAsyncUserToken)
|
||||||
|
{
|
||||||
|
m_clients.Remove(userToken);
|
||||||
|
_DictSocketAsyncUserToken.Remove(userToken.Socket);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
void ClearUserToken()
|
||||||
|
{
|
||||||
|
lock (_DictSocketAsyncUserToken)
|
||||||
|
{
|
||||||
|
m_clients.Clear();
|
||||||
|
_DictSocketAsyncUserToken.Clear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
/// <summary>
|
||||||
|
/// 回收SocketAsyncEventArgs
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="saea"></param>
|
||||||
|
/// <exception cref="ArgumentException"></exception>
|
||||||
|
void ReleaseSocketAsyncEventArgs(SocketAsyncEventArgs saea)
|
||||||
|
{
|
||||||
|
//saea.UserToken = null;//TODO
|
||||||
|
//saea.SetBuffer(null, 0, 0);
|
||||||
|
//saea.Dispose();
|
||||||
|
//↑ 这里不要自作主张去清东西,否则回收回去不可用
|
||||||
|
|
||||||
|
switch (saea.LastOperation)
|
||||||
|
{
|
||||||
|
case SocketAsyncOperation.Receive:
|
||||||
|
m_Receivepool.Push(saea);
|
||||||
|
break;
|
||||||
|
case SocketAsyncOperation.Send:
|
||||||
|
m_Sendpool.Push(saea);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
throw new ArgumentException("ReleaseSocketAsyncEventArgs > The last operation completed on the socket was not a receive or send");
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region 监听IOCP循环
|
||||||
|
/// <summary>
|
||||||
|
/// 开始接受客户端的连接请求的操作
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="acceptEventArg">在服务器的侦听套接字上发出接受操作时要使用的上下文对象</param>
|
||||||
|
public void StartAccept(SocketAsyncEventArgs acceptEventArg)
|
||||||
|
{
|
||||||
|
if (acceptEventArg == null)
|
||||||
|
{
|
||||||
|
acceptEventArg = new SocketAsyncEventArgs();
|
||||||
|
acceptEventArg.Completed += new EventHandler<SocketAsyncEventArgs>(AcceptEventArg_Completed);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// socket must be cleared since the context object is being reused
|
||||||
|
acceptEventArg.AcceptSocket = null;
|
||||||
|
}
|
||||||
|
|
||||||
|
m_maxNumberAcceptedClients.WaitOne();
|
||||||
|
if (!listenSocket.AcceptAsync(acceptEventArg))
|
||||||
|
{
|
||||||
|
ProcessAccept(acceptEventArg);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
/// <summary>
|
||||||
|
/// 此方法是与Socket关联的回调方法。AcceptAsync操作,并在接受操作完成时调用
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="sender"></param>
|
||||||
|
/// <param name="e"></param>
|
||||||
|
void AcceptEventArg_Completed(object sender, SocketAsyncEventArgs e)
|
||||||
|
{
|
||||||
|
ProcessAccept(e);
|
||||||
|
}
|
||||||
|
private void ProcessAccept(SocketAsyncEventArgs e)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
Interlocked.Increment(ref m_clientCount);
|
||||||
|
// Get the socket for the accepted client connection and put it into the
|
||||||
|
//ReadEventArg object user token
|
||||||
|
SocketAsyncEventArgs readEventArgs = m_Receivepool.Pop();
|
||||||
|
//TODO readEventArgs.UserToken这里的 UserToken 有可能是空
|
||||||
|
AsyncUserToken userToken;
|
||||||
|
if (readEventArgs.UserToken == null)
|
||||||
|
readEventArgs.UserToken = new AsyncUserToken();
|
||||||
|
|
||||||
|
userToken = (AsyncUserToken)readEventArgs.UserToken;
|
||||||
|
userToken.Socket = e.AcceptSocket;
|
||||||
|
userToken.ConnectTime = DateTime.Now;
|
||||||
|
userToken.Remote = e.AcceptSocket.RemoteEndPoint;
|
||||||
|
userToken.IPAddress = ((IPEndPoint)(e.AcceptSocket.RemoteEndPoint)).Address;
|
||||||
|
|
||||||
|
|
||||||
|
userToken.RevIndex = MaxRevIndexNum;
|
||||||
|
userToken.SendIndex = MaxSendIndexNum;
|
||||||
|
|
||||||
|
AddUserToken(userToken);
|
||||||
|
|
||||||
|
OnClientNumberChange?.Invoke(1, userToken);
|
||||||
|
if (!e.AcceptSocket.ReceiveAsync(readEventArgs))
|
||||||
|
{
|
||||||
|
ProcessReceive(readEventArgs);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (Exception me)
|
||||||
|
{
|
||||||
|
//RuncomLib.Log.LogUtils.Info(me.Message + "\r\n" + me.StackTrace);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Accept the next connection request
|
||||||
|
if (e.SocketError == SocketError.OperationAborted) return;
|
||||||
|
StartAccept(e);
|
||||||
|
}
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region 收发IOCP循环
|
||||||
|
/// <summary>当异步接收操作完成时,会调用此方法。
|
||||||
|
/// 如果远程主机关闭了连接,则套接字关闭。
|
||||||
|
/// 如果接收到数据,则将数据回显到客户端。
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="e"></param>
|
||||||
|
private void ProcessReceive(SocketAsyncEventArgs e)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
// check if the remote host closed the connection
|
||||||
|
AsyncUserToken token = (AsyncUserToken)e.UserToken;
|
||||||
|
if (e.BytesTransferred > 0 && e.SocketError == SocketError.Success)
|
||||||
|
{
|
||||||
|
//读取数据
|
||||||
|
//byte[] data = new byte[e.BytesTransferred];
|
||||||
|
//Array.Copy(e.Buffer, e.Offset, data, 0, e.BytesTransferred);
|
||||||
|
//lock (token.Buffer)
|
||||||
|
lock(token.memoryStream)
|
||||||
|
{
|
||||||
|
//token.Buffer.AddRange(data);
|
||||||
|
token.memoryStream.Write(e.Buffer, e.Offset, e.BytesTransferred);
|
||||||
|
}
|
||||||
|
do
|
||||||
|
{
|
||||||
|
//如果包头不完整
|
||||||
|
//if (token.Buffer.Count < 4)
|
||||||
|
if (token.memoryStream.Length < 4)
|
||||||
|
break;
|
||||||
|
|
||||||
|
//判断包的长度
|
||||||
|
//byte[] lenBytes = token.Buffer.GetRange(0, 4).ToArray();
|
||||||
|
//int packageLen = BitConverter.ToInt32(lenBytes, 0) - 4;
|
||||||
|
//if (packageLen > token.Buffer.Count - 4)
|
||||||
|
//{ //长度不够时,退出循环,让程序继续接收
|
||||||
|
// break;
|
||||||
|
//}
|
||||||
|
|
||||||
|
long FristBeginPos = token.memoryStream.Position;
|
||||||
|
byte[] lenBytes = new byte[4];
|
||||||
|
token.memoryStream.Seek(0, SeekOrigin.Begin);
|
||||||
|
token.memoryStream.Read(lenBytes,0,4);
|
||||||
|
int packageLen = BitConverter.ToInt32(lenBytes, 0) - 4;
|
||||||
|
if (packageLen > token.memoryStream.Length - 4)
|
||||||
|
{ //长度不够时,退出循环,让程序继续接收
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
//包够长时,则提取出来,交给后面的程序去处理
|
||||||
|
//byte[] rev = token.Buffer.GetRange(4, packageLen).ToArray();
|
||||||
|
|
||||||
|
byte[] rev = new byte[packageLen];
|
||||||
|
token.memoryStream.Seek(4, SeekOrigin.Begin);
|
||||||
|
token.memoryStream.Read(rev, 0, packageLen);
|
||||||
|
|
||||||
|
////从数据池中移除这组数据
|
||||||
|
//lock (token.Buffer)
|
||||||
|
//{
|
||||||
|
// token.Buffer.RemoveRange(0, packageLen + 4);
|
||||||
|
//}
|
||||||
|
|
||||||
|
token.memoryStream.Seek(FristBeginPos, SeekOrigin.Begin);
|
||||||
|
//从数据池中移除这组数据
|
||||||
|
lock (token.memoryStream)
|
||||||
|
{
|
||||||
|
int numberOfBytesToRemove = packageLen + 4;
|
||||||
|
byte[] buf = token.memoryStream.GetBuffer();
|
||||||
|
Buffer.BlockCopy(buf, numberOfBytesToRemove, buf, 0, (int)token.memoryStream.Length - numberOfBytesToRemove);
|
||||||
|
token.memoryStream.SetLength(token.memoryStream.Length - numberOfBytesToRemove);
|
||||||
|
}
|
||||||
|
|
||||||
|
DataCallBackReady(token, rev);
|
||||||
|
|
||||||
|
//这里API处理完后,并没有返回结果,当然结果是要返回的,却不是在这里, 这里的代码只管接收.
|
||||||
|
//若要返回结果,可在API处理中调用此类对象的SendMessage方法,统一打包发送.不要被微软的示例给迷惑了.
|
||||||
|
//} while (token.Buffer.Count > 4);
|
||||||
|
} while (token.memoryStream.Length > 4);
|
||||||
|
|
||||||
|
//继续接收. 为什么要这么写,请看Socket.ReceiveAsync方法的说明
|
||||||
|
if (!token.Socket.ReceiveAsync(e))
|
||||||
|
{
|
||||||
|
this.ProcessReceive(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
CloseClientSocket(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (Exception xe)
|
||||||
|
{
|
||||||
|
//RuncomLib.Log.LogUtils.Info(xe.Message + "\r\n" + xe.StackTrace);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
private void ProcessSend(SocketAsyncEventArgs e)
|
||||||
|
{
|
||||||
|
if (e.SocketError == SocketError.Success)
|
||||||
|
{
|
||||||
|
//TODO
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
CloseClientSocket(e);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
ReleaseSocketAsyncEventArgs(e);
|
||||||
|
SendForMsgPool();
|
||||||
|
}
|
||||||
|
void IO_Completed(object sender, SocketAsyncEventArgs e)
|
||||||
|
{
|
||||||
|
// determine which type of operation just completed and call the associated handler
|
||||||
|
|
||||||
|
switch (e.LastOperation)
|
||||||
|
{
|
||||||
|
case SocketAsyncOperation.Receive:
|
||||||
|
ProcessReceive(e);
|
||||||
|
break;
|
||||||
|
case SocketAsyncOperation.Send:
|
||||||
|
ProcessSend(e);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
throw new ArgumentException("The last operation completed on the socket was not a receive or send");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region 发送
|
||||||
|
int sendrun = 0;
|
||||||
|
/// <summary>
|
||||||
|
/// 对外暴露的发送消息
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="CMDID"></param>
|
||||||
|
/// <param name="data">序列化之后的数据</param>
|
||||||
|
public void SendToSocket(Socket sk, int CMDID, int ERRCODE, byte[] data)
|
||||||
|
{
|
||||||
|
AsyncUserToken token = GetAsyncUserTokenForSocket(sk);
|
||||||
|
/*HunterNet_S2C _s2cdata = new HunterNet_S2C();
|
||||||
|
_s2cdata.HunterNetCoreCmdID = CMDID;
|
||||||
|
_s2cdata.HunterNetCoreData = ByteString.CopyFrom(data);
|
||||||
|
_s2cdata.HunterNetCoreERRORCode = ERRCODE;
|
||||||
|
byte[] _finaldata = Serizlize(_s2cdata);*/
|
||||||
|
|
||||||
|
//byte[] _finaldata = HunterNet_S2C.CreatePkgData((ushort)CMDID, (ushort)ERRCODE, data);
|
||||||
|
|
||||||
|
SendWithIndex(token, (ushort)CMDID, (ushort)ERRCODE, data);
|
||||||
|
}
|
||||||
|
void SendForMsgPool()
|
||||||
|
{
|
||||||
|
//if (flag_SendForMsgPool) return;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
if (sendrun < msg_pool.Count || msg_pool.Count < 1)
|
||||||
|
return;
|
||||||
|
|
||||||
|
sendrun++;
|
||||||
|
while (msg_pool.Count > 0)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
TokenWithMsg_SourceMode msg = msg_pool.Dequeue();
|
||||||
|
//OutNetLog("从信息池取出发送");
|
||||||
|
//是心跳包
|
||||||
|
if (msg.bHeartbeat)
|
||||||
|
{
|
||||||
|
SendHeartbeatMessage(msg.token);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
SendMessage(msg.token,msg.data);
|
||||||
|
}
|
||||||
|
msg = null;
|
||||||
|
}
|
||||||
|
catch
|
||||||
|
{
|
||||||
|
OutNetLog("==============================================>");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
sendrun--;
|
||||||
|
OutNetLog("!!!!!!!!!!!!!!!!!!!!!!!!!!");
|
||||||
|
}
|
||||||
|
catch (Exception ex)
|
||||||
|
{
|
||||||
|
OutNetLog(ex.ToString());
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 发送心跳包
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="token"></param>
|
||||||
|
void SendHeartbeatMessage(AsyncUserToken token)
|
||||||
|
{
|
||||||
|
if (token == null || token.Socket == null || !token.Socket.Connected)
|
||||||
|
return;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
if (m_Sendpool.Count > 0)
|
||||||
|
{
|
||||||
|
SocketAsyncEventArgs myreadEventArgs = m_Sendpool.Pop();
|
||||||
|
myreadEventArgs.UserToken = token;
|
||||||
|
myreadEventArgs.AcceptSocket = token.Socket;
|
||||||
|
//直接写入SocketAsyncEventArgs的Buff
|
||||||
|
HunterNet_Heartbeat.SetDataToSocketAsyncEventArgs(myreadEventArgs);
|
||||||
|
|
||||||
|
//若不需要等待
|
||||||
|
if (!token.Socket.SendAsync(myreadEventArgs))
|
||||||
|
{
|
||||||
|
m_Sendpool.Push(myreadEventArgs);
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
//先压入队列,等待m_Sendpool回收
|
||||||
|
msg_pool.Enqueue(new TokenWithMsg_SourceMode() { token = token, bHeartbeat = true });
|
||||||
|
//OutNetLog("!!!!压入消息发送队列MSG_Pool");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (Exception e)
|
||||||
|
{
|
||||||
|
OutNetLog(e.ToString());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 发送数据并计数
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="data"></param>
|
||||||
|
void SendWithIndex(AsyncUserToken token, UInt16 CmdID, UInt16 ERRCODE, byte[] data)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
//发送数据
|
||||||
|
SendMessage(token, CmdID, ERRCODE, data);
|
||||||
|
token.SendIndex = MaxSendIndexNum;
|
||||||
|
}
|
||||||
|
catch
|
||||||
|
{
|
||||||
|
CloseReady(token);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void SendMessage(AsyncUserToken token, byte[] data)
|
||||||
|
{
|
||||||
|
if (token == null || token.Socket == null || !token.Socket.Connected)
|
||||||
|
return;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
if (m_Sendpool.Count > 0)
|
||||||
|
{
|
||||||
|
SocketAsyncEventArgs myreadEventArgs = m_Sendpool.Pop();
|
||||||
|
myreadEventArgs.UserToken = token;
|
||||||
|
myreadEventArgs.AcceptSocket = token.Socket;
|
||||||
|
myreadEventArgs.SetBuffer(data, 0, data.Length); //将数据放置进去.
|
||||||
|
|
||||||
|
//若不需要等待
|
||||||
|
if (!token.Socket.SendAsync(myreadEventArgs))
|
||||||
|
{
|
||||||
|
m_Sendpool.Push(myreadEventArgs);
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
//先压入队列,等待m_Sendpool回收
|
||||||
|
msg_pool.Enqueue(new TokenWithMsg_SourceMode() { token = token, data = data ,bHeartbeat = false});
|
||||||
|
//OutNetLog("!!!!压入消息发送队列MSG_Pool");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (Exception e)
|
||||||
|
{
|
||||||
|
OutNetLog(e.ToString());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region 处理前预备
|
||||||
|
private void DataCallBackReady(AsyncUserToken sk, byte[] data)
|
||||||
|
{
|
||||||
|
//增加接收计数
|
||||||
|
sk.RevIndex = MaxRevIndexNum;
|
||||||
|
|
||||||
|
if (data.Length == 1 && data[0] == 0x00)//心跳包
|
||||||
|
{
|
||||||
|
//OutNetLog("收到心跳包");
|
||||||
|
//无处理
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
//将数据包交给后台处理,这里你也可以新开个线程来处理.加快速度.
|
||||||
|
/*
|
||||||
|
HunterNet_C2S _s2c = DeSerizlize<HunterNet_C2S>(data);
|
||||||
|
OnReceive?.Invoke(sk, (int)_s2c.HunterNetCoreCmdID, _s2c.HunterNetCoreData.ToArray());
|
||||||
|
//DataCallBack(sk, (int)_s2c.HunterNetCoreCmdID, _s2c.HunterNetCoreData.ToArray());
|
||||||
|
*/
|
||||||
|
HunterNet_C2S.AnalysisPkgData(data, out ushort CmdID, out byte[] resultdata);
|
||||||
|
OnReceive?.Invoke(sk, CmdID, resultdata);
|
||||||
|
}
|
||||||
|
catch (Exception ex)
|
||||||
|
{
|
||||||
|
OutNetLog("数据解析错误");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
private void OutNetLog(string msg)
|
||||||
|
{
|
||||||
|
OnNetLog?.Invoke(msg);
|
||||||
|
}
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region 心跳包
|
||||||
|
/// <summary>
|
||||||
|
/// 发送心跳包
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="sk"></param>
|
||||||
|
///
|
||||||
|
private void SendHeartbeatWithIndex(AsyncUserToken token)
|
||||||
|
{
|
||||||
|
if (token == null || token.Socket == null || !token.Socket.Connected)
|
||||||
|
return;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
//OutNetLog(DateTime.Now.ToString() + "发送心跳包");
|
||||||
|
token.SendIndex = MaxSendIndexNum;
|
||||||
|
SendHeartbeatMessage(token);
|
||||||
|
}
|
||||||
|
catch (Exception e)
|
||||||
|
{
|
||||||
|
CloseReady(token);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
/// <summary>
|
||||||
|
/// 心跳包时钟事件
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="sender"></param>
|
||||||
|
/// <param name="e"></param>
|
||||||
|
private void CheckUpdatetimer_Elapsed(object sender, System.Timers.ElapsedEventArgs e)
|
||||||
|
{
|
||||||
|
for (int i = 0; i < m_clients.Count(); i++)
|
||||||
|
{
|
||||||
|
//接收服务器数据计数
|
||||||
|
m_clients[i].RevIndex--;
|
||||||
|
if (m_clients[i].RevIndex <= 0)
|
||||||
|
{
|
||||||
|
//判定掉线
|
||||||
|
CloseReady(m_clients[i]);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
//发送计数
|
||||||
|
m_clients[i].SendIndex--;
|
||||||
|
if (m_clients[i].SendIndex <= 0)//需要发送心跳包了
|
||||||
|
{
|
||||||
|
//重置倒计时计数
|
||||||
|
m_clients[i].SendIndex = MaxSendIndexNum;
|
||||||
|
SendHeartbeatWithIndex(m_clients[i]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endregion
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,57 @@
|
|||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.Linq;
|
||||||
|
using System.Net.Sockets;
|
||||||
|
using System.Text;
|
||||||
|
|
||||||
|
namespace HaoYueNet.ServerNetwork
|
||||||
|
{
|
||||||
|
public class TokenWithMsg_SourceMode
|
||||||
|
{
|
||||||
|
public AsyncUserToken token;
|
||||||
|
public byte[] data;
|
||||||
|
public bool bHeartbeat;
|
||||||
|
}
|
||||||
|
|
||||||
|
public class TokenMsgPool_SourceMode
|
||||||
|
{
|
||||||
|
//Stack<TokenWithMsg> msg_pool;
|
||||||
|
Queue<TokenWithMsg_SourceMode> msg_pool;
|
||||||
|
|
||||||
|
public TokenMsgPool_SourceMode(int capacity)
|
||||||
|
{
|
||||||
|
msg_pool = new Queue<TokenWithMsg_SourceMode>(capacity);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 向 Queue 的末尾添加一个对象。
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="item"></param>
|
||||||
|
public void Enqueue(TokenWithMsg_SourceMode item)
|
||||||
|
{
|
||||||
|
lock (msg_pool)
|
||||||
|
{
|
||||||
|
msg_pool.Enqueue(item);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//移除并返回在 Queue 的开头的对象。
|
||||||
|
public TokenWithMsg_SourceMode Dequeue()
|
||||||
|
{
|
||||||
|
lock (msg_pool)
|
||||||
|
{
|
||||||
|
return msg_pool.Dequeue();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public int Count
|
||||||
|
{
|
||||||
|
get { return msg_pool.Count; }
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Clear()
|
||||||
|
{
|
||||||
|
msg_pool.Clear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user