forked from sin365/NoSugarNet
Merge pull request 'dev_localclient' (#1) from dev_localclient into master
Reviewed-on: sin365/NoSugarNet#1
This commit is contained in:
commit
32aaa8e931
Binary file not shown.
Binary file not shown.
@ -52,6 +52,7 @@ namespace ServerCore.Manager
|
||||
AppNoSugarNet.log.Debug($"开始监听配置 Tunnel:{cfg.Key},Port:{cfg.Value.Port}");
|
||||
listener.Init();
|
||||
listener.Start(new IPEndPoint(IPAddress.Any.Address, (int)cfg.Value.Port));
|
||||
//listener.Init((int)cfg.Value.Port);
|
||||
AddLocalListener(listener);
|
||||
}
|
||||
|
||||
@ -103,7 +104,7 @@ namespace ServerCore.Manager
|
||||
{
|
||||
LocalListener _listener = mDictTunnelID2Listeners[keys[i]];
|
||||
_listener.StopAll();
|
||||
_listener.Stop();
|
||||
//_listener.Stop();
|
||||
RemoveLocalListener(_listener);
|
||||
}
|
||||
}
|
||||
|
||||
@ -0,0 +1,290 @@
|
||||
using HaoYueNet.ClientNetwork.OtherMode;
|
||||
using HaoYueNet.ServerNetwork;
|
||||
using System.Net.Sockets;
|
||||
|
||||
namespace NoSugarNet.ClientCore
|
||||
{
|
||||
public class LocalListener_Source : NetworkHelperCore_ListenerMode
|
||||
{
|
||||
public byte mTunnelID;
|
||||
public long mReciveAllLenght;
|
||||
public long mSendAllLenght;
|
||||
public LocalListener_Source(int numConnections, int receiveBufferSize, byte TunnelID)
|
||||
: base()
|
||||
{
|
||||
OnConnected += ClientNumberChange;
|
||||
OnReceive += ReceiveData;
|
||||
OnDisconnected += OnDisconnectClient;
|
||||
OnNetLog += OnShowNetLog;
|
||||
|
||||
mTunnelID = TunnelID;
|
||||
}
|
||||
|
||||
private void ClientNumberChange(Socket socket)
|
||||
{
|
||||
Console.WriteLine("Client数发生变化");
|
||||
//增加连接数
|
||||
int Idx = AddDictSocket(socket);
|
||||
if (GetSocketByIdx(Idx, out LocalClientInfo _localClientInf))
|
||||
{
|
||||
AppNoSugarNet.local.OnClientLocalConnect(mTunnelID, (byte)Idx);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 通过下标发送
|
||||
/// </summary>
|
||||
/// <param name="Idx"></param>
|
||||
/// <param name="data"></param>
|
||||
public void SendSocketByIdx(int Idx, byte[] data)
|
||||
{
|
||||
if (GetSocketByIdx(Idx, out LocalClientInfo _localClientInfo))
|
||||
{
|
||||
mSendAllLenght += data.Length;
|
||||
SendToClient(_localClientInfo._socket, data);
|
||||
}
|
||||
//TODO连接前缓存数据
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 接受包回调
|
||||
/// </summary>
|
||||
/// <param name="CMDID">协议ID</param>
|
||||
/// <param name="ERRCODE">错误编号</param>
|
||||
/// <param name="data">业务数据</param>
|
||||
private void ReceiveData(Socket sk, byte[] data)
|
||||
{
|
||||
DataCallBack(sk, data);
|
||||
}
|
||||
|
||||
public void DataCallBack(Socket sk, byte[] data)
|
||||
{
|
||||
//AppNoSugarNet.log.Debug("收到消息 数据长度=>" + data.Length);
|
||||
//记录接受长度
|
||||
mReciveAllLenght += data.Length;
|
||||
if (!GetSocketIdxBySocket(sk, out int Idx))
|
||||
return;
|
||||
try
|
||||
{
|
||||
if (GetMsgQueueByIdx(sk.Handle, out Queue<byte[]> _queue))
|
||||
{
|
||||
lock (_queue)
|
||||
{
|
||||
_queue.Enqueue(data);
|
||||
while (_queue.Count > 0)
|
||||
{
|
||||
AppNoSugarNet.local.OnClientTunnelDataCallBack(mTunnelID, (byte)Idx, _queue.Dequeue());
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
////抛出网络数据
|
||||
//AppNoSugarNet.local.OnClientTunnelDataCallBack(mTunnelID, (byte)Idx, data);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Console.WriteLine("逻辑处理错误：" + ex.ToString());
|
||||
}
|
||||
}
|
||||
|
||||
public void CloseConnectByIdx(byte Idx)
|
||||
{
|
||||
if (GetSocketByIdx(Idx, out LocalClientInfo _localClientInf))
|
||||
{
|
||||
_localClientInf._socket.Shutdown(SocketShutdown.Both);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 断开连接
|
||||
/// </summary>
|
||||
/// <param name="sk"></param>
|
||||
public void OnDisconnectClient(Socket sk)
|
||||
{
|
||||
AppNoSugarNet.log.Debug("断开连接");
|
||||
|
||||
if (!GetSocketIdxBySocket(sk, out int Idx))
|
||||
return;
|
||||
|
||||
AppNoSugarNet.local.OnClientLocalDisconnect(mTunnelID, (byte)Idx);
|
||||
RemoveDictSocket(sk);
|
||||
}
|
||||
|
||||
public void OnShowNetLog(string msg)
|
||||
{
|
||||
AppNoSugarNet.log.Debug(msg);
|
||||
}
|
||||
|
||||
#region 一个轻量级无用户连接管理
|
||||
Dictionary<nint, int> DictSocketHandle2Idx = new Dictionary<nint, int>();
|
||||
Dictionary<nint, Queue<byte[]>> DictSocketHandle2Msg = new Dictionary<nint, Queue<byte[]>>();
|
||||
Dictionary<int, LocalClientInfo> DictIdx2LocalClientInfo = new Dictionary<int, LocalClientInfo>();
|
||||
int mSeedIdx = 0;
|
||||
List<int> FreeIdxs = new List<int>();
|
||||
public class LocalClientInfo
|
||||
{
|
||||
public Socket _socket;
|
||||
public bool bRemoteConnect;
|
||||
public bool bLocalConnect => _socket.Connected;
|
||||
public Queue<IdxWithMsg> msgQueue = new Queue<IdxWithMsg>();
|
||||
}
|
||||
|
||||
int GetNextIdx()
|
||||
{
|
||||
if (FreeIdxs.Count > 0)
|
||||
{
|
||||
int Idx = FreeIdxs[0];
|
||||
FreeIdxs.RemoveAt(0);
|
||||
return Idx;
|
||||
}
|
||||
return mSeedIdx++;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 追加Socket返回下标
|
||||
/// </summary>
|
||||
/// <param name="socket"></param>
|
||||
/// <returns></returns>
|
||||
public int AddDictSocket(Socket socket)
|
||||
{
|
||||
if (socket == null)
|
||||
return -1;
|
||||
lock (DictSocketHandle2Idx)
|
||||
{
|
||||
int Idx = GetNextIdx();
|
||||
DictSocketHandle2Idx[socket.Handle] = Idx;
|
||||
DictIdx2LocalClientInfo[Idx] = new LocalClientInfo() { _socket = socket,bRemoteConnect = false};
|
||||
DictSocketHandle2Msg[socket.Handle] = new Queue<byte[]>();
|
||||
return Idx;
|
||||
}
|
||||
}
|
||||
|
||||
public void RemoveDictSocket(Socket socket)
|
||||
{
|
||||
if (socket == null)
|
||||
return;
|
||||
lock (DictSocketHandle2Idx)
|
||||
{
|
||||
if (!DictSocketHandle2Idx.ContainsKey(socket.Handle))
|
||||
return;
|
||||
int Idx = DictSocketHandle2Idx[socket.Handle];
|
||||
FreeIdxs.Add(Idx);
|
||||
if (DictIdx2LocalClientInfo.ContainsKey(Idx))
|
||||
DictIdx2LocalClientInfo.Remove(Idx);
|
||||
|
||||
if (DictSocketHandle2Msg.ContainsKey(socket.Handle))
|
||||
DictSocketHandle2Msg.Remove(socket.Handle);
|
||||
|
||||
DictSocketHandle2Idx.Remove(socket.Handle);
|
||||
}
|
||||
}
|
||||
|
||||
bool GetSocketByIdx(int Idx, out LocalClientInfo _localClientInfo)
|
||||
{
|
||||
if (!DictIdx2LocalClientInfo.ContainsKey(Idx))
|
||||
{
|
||||
_localClientInfo = null;
|
||||
return false;
|
||||
}
|
||||
|
||||
_localClientInfo = DictIdx2LocalClientInfo[Idx];
|
||||
return true;
|
||||
}
|
||||
|
||||
bool GetMsgQueueByIdx(nint handle, out Queue<byte[]> _queue)
|
||||
{
|
||||
if (!DictSocketHandle2Msg.ContainsKey(handle))
|
||||
{
|
||||
_queue = null;
|
||||
return false;
|
||||
}
|
||||
|
||||
_queue = DictSocketHandle2Msg[handle];
|
||||
return true;
|
||||
}
|
||||
|
||||
public bool GetSocketIdxBySocket(Socket _socket, out int Idx)
|
||||
{
|
||||
if (_socket == null)
|
||||
{
|
||||
Idx = -1;
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!DictSocketHandle2Idx.ContainsKey(_socket.Handle))
|
||||
{
|
||||
Idx = -1;
|
||||
return false;
|
||||
}
|
||||
|
||||
Idx = DictSocketHandle2Idx[_socket.Handle];
|
||||
return true;
|
||||
}
|
||||
|
||||
public bool CheckRemoteConnect(int Idx)
|
||||
{
|
||||
if (!GetSocketByIdx(Idx, out LocalClientInfo _localClientInfo))
|
||||
return false;
|
||||
return _localClientInfo.bRemoteConnect;
|
||||
}
|
||||
|
||||
public void SetRemoteConnectd(int Idx,bool bConnected)
|
||||
{
|
||||
if (!GetSocketByIdx(Idx, out LocalClientInfo _localClientInfo))
|
||||
return;
|
||||
if (bConnected)
|
||||
AppNoSugarNet.log.Debug("远端本地连接已连接！！！！");
|
||||
else
|
||||
AppNoSugarNet.log.Debug("远端本地连接已断开连接！！！！");
|
||||
_localClientInfo.bRemoteConnect = bConnected;
|
||||
}
|
||||
|
||||
public void StopAll()
|
||||
{
|
||||
lock (DictIdx2LocalClientInfo)
|
||||
{
|
||||
int[] Idxs = DictIdx2LocalClientInfo.Keys.ToArray();
|
||||
for (int i = 0; i < Idxs.Length; i++)
|
||||
{
|
||||
CloseConnectByIdx((byte)Idxs[i]);
|
||||
}
|
||||
DictIdx2LocalClientInfo.Clear();
|
||||
}
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
|
||||
#region 缓存
|
||||
public void EnqueueIdxWithMsg(byte Idx, byte[] data)
|
||||
{
|
||||
if (!GetSocketByIdx(Idx, out LocalClientInfo _localClientInfo))
|
||||
return;
|
||||
|
||||
IdxWithMsg Msg = AppNoSugarNet.local._localMsgPool.Dequeue();
|
||||
Msg.Idx = Idx;
|
||||
Msg.data = data;
|
||||
_localClientInfo.msgQueue.Enqueue(Msg);
|
||||
}
|
||||
public bool GetDictMsgQueue(byte Idx,out List<IdxWithMsg> MsgList)
|
||||
{
|
||||
if (!GetSocketByIdx(Idx, out LocalClientInfo _localClientInfo) || _localClientInfo.msgQueue.Count < 1)
|
||||
{
|
||||
MsgList = null;
|
||||
return false;
|
||||
}
|
||||
|
||||
MsgList = new List<IdxWithMsg>();
|
||||
lock (_localClientInfo.msgQueue)
|
||||
{
|
||||
while (_localClientInfo.msgQueue.Count > 0)
|
||||
{
|
||||
IdxWithMsg msg = _localClientInfo.msgQueue.Dequeue();
|
||||
MsgList.Add(msg);
|
||||
}
|
||||
return true;
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user