using System;
using System.Net;
using System.Net.Sockets;
using System.Threading;
using NewLife.Data;
using NewLife.Threading;
namespace NewLife.Net
{
/// <summary>增强TCP客户端</summary>
public class TcpSession : SessionBase, ISocketSession
{
#region 属性
/// <summary>会话编号</summary>
public Int32 ID { get; internal set; }
/// <summary>收到空数据时抛出异常并断开连接。默认true</summary>
public Boolean DisconnectWhenEmptyData { get; set; } = true;
internal ISocketServer _Server;
/// <summary>Socket服务器。当前通讯所在的Socket服务器，其实是TcpServer/UdpServer。该属性决定本会话是客户端会话还是服务的会话</summary>
ISocketServer ISocketSession.Server => _Server;
/// <summary>自动重连次数，默认3。发生异常断开连接时，自动重连服务端。</summary>
public Int32 AutoReconnect { get; set; } = 3;
/// <summary>是否匹配空包。Http协议需要</summary>
protected Boolean MatchEmpty { get; set; }
#endregion
#region 构造
/// <summary>实例化增强TCP</summary>
public TcpSession()
{
Name = GetType().Name;
Local.Type = NetType.Tcp;
Remote.Type = NetType.Tcp;
}
/// <summary>使用监听口初始化</summary>
/// <param name="listenPort"></param>
public TcpSession(Int32 listenPort) : this() => Port = listenPort;
/// <summary>用TCP客户端初始化</summary>
/// <param name="client"></param>
public TcpSession(Socket client) : this()
{
if (client == null) return;
Client = client;
var socket = client;
if (socket.LocalEndPoint != null) Local.EndPoint = (IPEndPoint)socket.LocalEndPoint;
if (socket.RemoteEndPoint != null) Remote.EndPoint = (IPEndPoint)socket.RemoteEndPoint;
}
internal TcpSession(ISocketServer server, Socket client)
: this(client)
{
Active = true;
_Server = server;
Name = server.Name;
}
#endregion
#region 方法
internal void Start()
{
// 管道
var pp = Pipeline;
pp?.Open(pp.CreateContext(this));
ReceiveAsync();
}
/// <summary>打开</summary>
protected override Boolean OnOpen()
{
// 服务端会话没有打开
if (_Server != null) return false;
var timeout = Timeout;
var sock = Client;
if (sock == null || !sock.IsBound)
{
// 根据目标地址适配本地IPv4/IPv6
if (Remote != null && !Remote.Address.IsAny())
{
Local.Address = Local.Address.GetRightAny(Remote.Address.AddressFamily);
}
sock = Client = NetHelper.CreateTcp(Local.EndPoint.Address.IsIPv4());
//sock.SetSocketOption(SocketOptionLevel.Tcp, SocketOptionName.NoDelay, true);
if (timeout > 0)
{
sock.SendTimeout = timeout;
sock.ReceiveTimeout = timeout;
}
sock.Bind(Local.EndPoint);
CheckDynamic();
WriteLog("Open {0}", this);
}
// 打开端口前如果已设定远程地址，则自动连接
if (Remote == null || Remote.EndPoint.IsAny()) return false;
try
{
if (timeout <= 0)
sock.Connect(Remote.EndPoint);
else
{
// 采用异步来解决连接超时设置问题
var ar = sock.BeginConnect(Remote.EndPoint, null, null);
if (!ar.AsyncWaitHandle.WaitOne(timeout, false))
{
sock.Close();
throw new TimeoutException($"连接[{Remote}][{timeout}ms]超时！");
}
sock.EndConnect(ar);
}
}
catch (Exception ex)
{
if (!Disposed && !ex.IsDisposed()) OnError("Connect", ex);
/*if (ThrowException)*/
throw;
//return false;
}
_Reconnect = 0;
return true;
}
/// <summary>关闭</summary>
/// <param name="reason">关闭原因。便于日志分析</param>
protected override Boolean OnClose(String reason)
{
var client = Client;
if (client != null)
{
WriteLog("Close {0} {1}", reason, this);
// 提前关闭这个标识，否则Close时可能触发自动重连机制
Active = false;
try
{
// 温和一点关闭连接
//Client.Shutdown();
client.Close();
// 如果是服务端，这个时候就是销毁
if (_Server != null) Dispose();
}
catch (Exception ex)
{
Client = null;
if (!ex.IsDisposed()) OnError("Close", ex);
if (ThrowException) throw;
return false;
}
Client = null;
}
return true;
}
#endregion
#region 发送
private Int32 _bsize;
/// <summary>发送数据</summary>
/// <remarks>
/// 目标地址由<seealso cref="SessionBase.Remote"/>决定
/// </remarks>
/// <param name="pk">数据包</param>
/// <returns>是否成功</returns>
protected override Boolean OnSend(Packet pk)
{
var count = pk.Total;
StatSend?.Increment(count, 0);
if (Log != null && Log.Enable && LogSend) WriteLog("Send [{0}]: {1}", count, pk.ToHex());
try
{
// 修改发送缓冲区，读取SendBufferSize耗时很大
if (_bsize == 0) _bsize = Client.SendBufferSize;
if (_bsize < count) _bsize = Client.SendBufferSize = count;
if (count == 0)
Client.Send(new Byte[0]);
else if (pk.Next == null)
Client.Send(pk.Data, pk.Offset, count, SocketFlags.None);
else
Client.Send(pk.ToArray(), 0, count, SocketFlags.None);
}
catch (Exception ex)
{
if (!ex.IsDisposed())
{
OnError("Send", ex);
// 发送异常可能是连接出了问题，需要关闭
Close("发送出错");
Reconnect();
if (ThrowException) throw;
}
return false;
}
LastTime = TimerX.Now;
return true;
}
#endregion
#region 接收
internal override Boolean OnReceiveAsync(SocketAsyncEventArgs se)
{
var client = Client;
if (client == null || !Active || Disposed) throw new ObjectDisposedException(GetType().Name);
return client.ReceiveAsync(se);
}
private Int32 _empty;
/// <summary>预处理</summary>
/// <param name="pk">数据包</param>
/// <param name="remote">远程地址</param>
/// <returns>将要处理该数据包的会话</returns>
internal protected override ISocketSession OnPreReceive(Packet pk, IPEndPoint remote)
{
if (pk.Count == 0)
{
// 连续多次空数据，则断开
if ((DisconnectWhenEmptyData || _empty++ > 3))
{
Close("收到空数据");
Dispose();
return null;
}
}
else
_empty = 0;
return this;
}
/// <summary>处理收到的数据</summary>
/// <param name="e">接收事件参数</param>
protected override Boolean OnReceive(ReceivedEventArgs e)
{
var pk = e.Packet;
if (pk == null || pk.Count == 0 && !MatchEmpty) return true;
StatReceive?.Increment(pk.Count, 0);
// 分析处理
RaiseReceive(this, e);
return true;
}
///// <summary>收到异常时如何处理。默认关闭会话</summary>
///// <param name="se"></param>
///// <returns>是否当作异常处理并结束会话</returns>
//internal override Boolean OnReceiveError(SocketAsyncEventArgs se)
//{
// // 不要销毁对象，仅标记状态，便于下次需要使用时重新打开连接
// if (se.SocketError == SocketError.ConnectionReset)
// {
// Close("ReceiveAsync " + se.SocketError);
// //Active = false;
// }
// return true;
//}
#endregion
#region 自动重连
/// <summary>重连次数</summary>
private Int32 _Reconnect;
void Reconnect()
{
if (Disposed) return;
// 如果重连次数达到最大重连次数，则退出
if (Interlocked.Increment(ref _Reconnect) > AutoReconnect) return;
WriteLog("Reconnect {0}", this);
Open();
}
#endregion
#region 辅助
private String _LogPrefix;
/// <summary>日志前缀</summary>
public override String LogPrefix
{
get
{
if (_LogPrefix == null)
{
var name = _Server == null ? "" : _Server.Name;
_LogPrefix = "{0}[{1}].".F(name, ID);
}
return _LogPrefix;
}
set { _LogPrefix = value; }
}
/// <summary>已重载。</summary>
/// <returns></returns>
public override String ToString()
{
if (Remote != null && !Remote.EndPoint.IsAny())
{
if (_Server == null)
return String.Format("{0}=>{1}", Local, Remote.EndPoint);
else
return String.Format("{0}<={1}", Local, Remote.EndPoint);
}
else
return Local.ToString();
}
#endregion
}
}
|