366 lines
11 KiB
C#
366 lines
11 KiB
C#
/*
|
||
********************************************************************************
|
||
*Copyright(C),coolae.net
|
||
*Author: chenbin
|
||
*Version: 2.0
|
||
*Date: 2017-01-09
|
||
*Description: tcp
|
||
*Others:
|
||
*History:
|
||
*********************************************************************************
|
||
*/
|
||
|
||
using UnityEngine;
|
||
using System.Collections;
|
||
using System.IO;
|
||
using XLua;
|
||
|
||
namespace Coolape
|
||
{
|
||
#if !UNITY_WEBGL
|
||
public delegate void TcpDispatchDelegate(object data, Tcp tcp);
|
||
public class Tcp : MonoBehaviour
|
||
{
|
||
public string host;
|
||
public int port;
|
||
public bool connected = false;
|
||
public bool serializeInMainThread = true;
|
||
//是否连接
|
||
public bool isStopping = false;
|
||
const int MaxReConnectTimes = 0;
|
||
public static int __maxLen = 1024 * 1024;
|
||
|
||
System.Threading.Timer timer;
|
||
public USocket socket;
|
||
int reConnectTimes = 0;
|
||
public const string CONST_Connect = "connectCallback";
|
||
public const string CONST_OutofNetConnect = "outofNetConnect";
|
||
TcpDispatchDelegate mDispatcher;
|
||
byte[] tmpBuffer = new byte[__maxLen];
|
||
|
||
public virtual void init(string host, int port)
|
||
{
|
||
this.host = host;
|
||
this.port = port;
|
||
}
|
||
|
||
/// <summary>
|
||
/// Init the specified host, port and dispatcher.
|
||
/// </summary>
|
||
/// <param name="host">Host.</param>
|
||
/// <param name="port">Port.</param>
|
||
/// <param name="dispatcher">Dispatcher,当接收到数据并解析成功后将调用该方法.</param>
|
||
public virtual void init(string host, int port, TcpDispatchDelegate dispatcher)
|
||
{
|
||
mDispatcher = dispatcher;
|
||
this.host = host;
|
||
this.port = port;
|
||
}
|
||
|
||
public void connect()
|
||
{
|
||
connect(null);
|
||
}
|
||
public void connect(object obj)
|
||
{
|
||
if (socket != null)
|
||
{
|
||
stop();
|
||
}
|
||
isStopping = false;
|
||
socket = new USocket(host, port);
|
||
|
||
#if UNITY_EDITOR
|
||
Debug.Log("connect ==" + host + ":" + port);
|
||
#endif
|
||
//异步连接
|
||
socket.connectAsync(onConnectStateChg);
|
||
}
|
||
|
||
/// <summary>
|
||
/// Ons the connect state chg. 当连接状态发生变化时
|
||
/// </summary>
|
||
/// <param name="s">S.</param>
|
||
/// <param name="result">Result. 其实是bool类型，
|
||
/// 当为true表示连接成功，false时表示没有连接成功或连接断开</param>
|
||
public virtual void onConnectStateChg(USocket s, object result)
|
||
{
|
||
ArrayList list = result as ArrayList;
|
||
bool isConnected = (bool)list[0];
|
||
int retCode = (int)list[1];
|
||
string msg = (string)list[2];
|
||
if (isConnected)
|
||
{
|
||
#if UNITY_EDITOR
|
||
Debug.Log("connectCallback success");
|
||
#endif
|
||
connected = true;
|
||
reConnectTimes = 0;
|
||
socket.ReceiveAsync(onReceive);
|
||
enqueueData(CONST_Connect);
|
||
}
|
||
else
|
||
{
|
||
Debug.LogWarning("connectCallback fail" + host + ":" + port + "," + isStopping);
|
||
connected = false;
|
||
if (!isStopping)
|
||
{
|
||
outofNetConnect(retCode, msg);
|
||
}
|
||
}
|
||
}
|
||
|
||
public void outofNetConnect(int code, string msg)
|
||
{
|
||
if (isStopping)
|
||
return;
|
||
if (reConnectTimes < MaxReConnectTimes)
|
||
{
|
||
reConnectTimes++;
|
||
Debug.LogWarning("reconnect times=" + reConnectTimes);
|
||
if (timer != null)
|
||
{
|
||
timer.Dispose();
|
||
}
|
||
timer = TimerEx.schedule(connect, null, 5000);
|
||
}
|
||
else
|
||
{
|
||
if (timer != null)
|
||
{
|
||
timer.Dispose();
|
||
}
|
||
timer = null;
|
||
outofLine(socket, null);
|
||
}
|
||
}
|
||
|
||
public void outofLine(USocket s, object obj)
|
||
{
|
||
if (!isStopping)
|
||
{
|
||
stop();
|
||
//CLMainBase.self.onOffline();
|
||
enqueueData(CONST_OutofNetConnect);
|
||
}
|
||
}
|
||
|
||
public virtual void stop()
|
||
{
|
||
isStopping = true;
|
||
connected = false;
|
||
if (socket != null)
|
||
{
|
||
socket.close();
|
||
}
|
||
socket = null;
|
||
}
|
||
|
||
//==========================================
|
||
public bool send(object obj)
|
||
{
|
||
if (socket == null)
|
||
{
|
||
Debug.LogWarning("Socket is null");
|
||
return false;
|
||
}
|
||
object ret = packMessage(obj);
|
||
if (isStopping || !connected)
|
||
{
|
||
Debug.LogWarning("isStopping =" + isStopping + "|| !connected=" + !connected);
|
||
return false;
|
||
}
|
||
if (ret != null)
|
||
{
|
||
socket.SendAsync(ret as byte[]);
|
||
}
|
||
else
|
||
{
|
||
//这种情况可能是在组包的时候就已经发送了，还有种情况就是异常，不过其实不太可能异常，先不处理
|
||
}
|
||
return true;
|
||
}
|
||
|
||
public object packMessage(object obj)
|
||
{
|
||
try
|
||
{
|
||
return encodeData(obj);
|
||
}
|
||
catch (System.Exception e)
|
||
{
|
||
Debug.LogError(e);
|
||
return null;
|
||
}
|
||
}
|
||
|
||
MemoryStream os = new MemoryStream();
|
||
MemoryStream os2 = new MemoryStream();
|
||
|
||
/// <summary>
|
||
/// Encodes the data.数据组包准备发送
|
||
/// </summary>
|
||
/// <returns>The data.</returns>
|
||
/// <param name="obj">Object.</param>
|
||
public virtual byte[] encodeData(object obj)
|
||
{
|
||
os.Position = 0;
|
||
os2.Position = 0;
|
||
|
||
B2OutputStream.writeObject(os, obj);
|
||
int len = (int)os.Position;
|
||
B2OutputStream.writeInt(os2, len);
|
||
os2.Write(os.ToArray(), 0, len);
|
||
int pos = (int)os2.Position;
|
||
byte[] result = new byte[pos];
|
||
os2.Position = 0;
|
||
os2.Read(result, 0, pos);
|
||
return result;
|
||
}
|
||
|
||
//==========================================
|
||
MemoryStreamPool memorystreamPool = new MemoryStreamPool();
|
||
public void onReceive(USocket s, byte[] bytes, int len)
|
||
{
|
||
MemoryStream buffer = memorystreamPool.borrowObject();
|
||
buffer.Write(bytes, 0, len);
|
||
buffer.SetLength(len);
|
||
enqueueData(buffer);
|
||
}
|
||
|
||
object netData = null;
|
||
MemoryStream memoryBuff = null;
|
||
MemoryStream receivedBuffer = new MemoryStream();
|
||
public IEnumerator wrapBuffer2Unpack()
|
||
{
|
||
yield return null;
|
||
while (receivedDataQueue.Count > 0)
|
||
{
|
||
netData = receivedDataQueue.Dequeue();
|
||
if (netData != null)
|
||
{
|
||
if (netData is string)
|
||
{
|
||
if (mDispatcher != null)
|
||
{
|
||
mDispatcher(netData, this);
|
||
}
|
||
continue;
|
||
}
|
||
memoryBuff = netData as MemoryStream;
|
||
receivedBuffer.Write(memoryBuff.ToArray(), 0, (int)(memoryBuff.Length));
|
||
memorystreamPool.returnObject(memoryBuff);
|
||
}
|
||
}
|
||
if (receivedBuffer.Length > 0)
|
||
{
|
||
receivedBuffer.SetLength(receivedBuffer.Position);
|
||
unpackMsg(receivedBuffer);
|
||
}
|
||
}
|
||
|
||
public void unpackMsg(MemoryStream buffer)
|
||
{
|
||
bool isLoop = true;
|
||
object o = null;
|
||
long usedLen = 0;
|
||
while (isLoop)
|
||
{
|
||
long totalLen = buffer.Length;
|
||
if (totalLen > 2)
|
||
{
|
||
usedLen = 0;
|
||
o = parseRecivedData(buffer);
|
||
usedLen = buffer.Position;
|
||
if (usedLen > 0)
|
||
{
|
||
int leftLen = (int)(totalLen - usedLen);
|
||
if (leftLen > 0)
|
||
{
|
||
buffer.Read(tmpBuffer, 0, leftLen);
|
||
buffer.Position = 0;
|
||
buffer.Write(tmpBuffer, 0, leftLen);
|
||
buffer.SetLength(leftLen);
|
||
}
|
||
else
|
||
{
|
||
buffer.Position = 0;
|
||
buffer.SetLength(0);
|
||
isLoop = false;
|
||
}
|
||
}
|
||
else
|
||
{
|
||
//buffer.Position = totalLen;
|
||
isLoop = false;
|
||
}
|
||
|
||
if (o != null && mDispatcher != null)
|
||
{
|
||
mDispatcher(o, this);
|
||
}
|
||
}
|
||
else
|
||
{
|
||
isLoop = false;
|
||
}
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// Parses the recived data.解析接收的数据，解析成功后发送给dispatcher
|
||
/// </summary>
|
||
/// <returns>The recived data.</returns>
|
||
/// <param name="buffer">Buffer.</param>
|
||
public virtual object parseRecivedData(MemoryStream buffer)
|
||
{
|
||
object ret = null;
|
||
long oldPos = buffer.Position;
|
||
buffer.Position = 0;
|
||
long tatolLen = buffer.Length;
|
||
long needLen = B2InputStream.readInt(buffer);
|
||
if (needLen <= 0 || needLen > __maxLen)
|
||
{
|
||
// 网络Number据错误。断isOpen网络
|
||
outofLine(this.socket, false);
|
||
//this.stop();
|
||
return null;
|
||
}
|
||
long usedLen = buffer.Position;
|
||
if (usedLen + needLen <= tatolLen)
|
||
{
|
||
ret = B2InputStream.readObject(buffer);
|
||
}
|
||
else
|
||
{
|
||
//说明长度不够
|
||
buffer.Position = oldPos;
|
||
}
|
||
return ret;
|
||
}
|
||
|
||
//======================================================================
|
||
//======================================================================
|
||
//======================================================================
|
||
public Queue receivedDataQueue = new Queue();
|
||
|
||
public void enqueueData(object obj)
|
||
{
|
||
receivedDataQueue.Enqueue(obj);
|
||
if (!serializeInMainThread)
|
||
{
|
||
StartCoroutine(wrapBuffer2Unpack());
|
||
}
|
||
}
|
||
|
||
public virtual void Update()
|
||
{
|
||
if (serializeInMainThread && receivedDataQueue.Count > 0)
|
||
{
|
||
StartCoroutine(wrapBuffer2Unpack());
|
||
}
|
||
}
|
||
}
|
||
#endif
|
||
}
|