/// <summary>
/// 处理接收到的数据
/// </summary>
/// <param name="msgData">接收到的消息</param>
private byte[] DealMessage(byte[] msgData)
{
//反序列化，取出数据
TSerializer serializer = new TBinarySerializer();
serializer.FromBytes(msgData);
DynObject reqFragment = DynSerialize.ReadDynObject(serializer);
serializer.Flush();
try
{
DynObject repFragment = OnFileDataReceived(reqFragment);
//序列化并把数据返回给客户端
serializer = new TBinarySerializer();
DynSerialize.WriteDynObject(serializer, repFragment);
byte[] repData = serializer.ToBytes();
serializer.Flush();
return(repData);
}
catch (Exception ex)
{
//如果服务端需要记录日志，则由本事件把异常信息发出
if (OnException != null)
{
OnException(ex);
}
_isRunning = false;
DynObject repFragment = new DynObject("FileFragment");
repFragment["State"] = (byte)105;
repFragment["ExcepMsg"] = "FtpServer端异常，" + ex.Message;
//序列化并把异常数据返回给客户端
serializer = new TBinarySerializer();
DynSerialize.WriteDynObject(serializer, repFragment);
byte[] repData = serializer.ToBytes();
serializer.Flush();
return(repData);
}
}
/// <summary>
/// 向服务端发送下载文件的请求
/// </summary>
/// <param name="destVPort">目的端口</param>
/// <param name="downloadPath">要下载的文件在服务端的路径</param>
/// <returns>将要下载文件的基本信息</returns>
private DynObject RequestDownload(ushort destVPort, string downloadPath)
{
DynObject fileFragment = new DynObject("FileFragment");
fileFragment["Path"] = Path.GetDirectoryName(downloadPath);
fileFragment["FileName"] = Path.GetFileName(downloadPath);
fileFragment["Extension"] = Path.GetExtension(downloadPath);
fileFragment["State"] = (byte)100;
fileFragment["MsgID"] = -1;
fileFragment["DataLength"] = 0;
fileFragment["Data"] = null;
//序列化要发送的数据
TSerializer serializer = new TBinarySerializer();
DynSerialize.WriteDynObject(serializer, fileFragment);
byte[] buffer = serializer.ToBytes();
serializer.Flush();
Send(_serverPort, buffer);
ZmqMessage msg = _ftpSocket.ReceiveMessage(new TimeSpan(0, 1, 30));
if (msg.FrameCount == 0)
{
_isWorking = false;
ReStart();
throw new TimeoutException("发送超时，请检查你的网络是否有问题,稍后重新下载");
}
//反序列化，服务器返回的数据
serializer = new TBinarySerializer();
serializer.FromBytes(msg[msg.FrameCount - 1].Buffer);
DynObject respFragment = DynSerialize.ReadDynObject(serializer);
serializer.Flush();
//发生异常
if ((byte)respFragment["State"] == 105)
{
_isWorking = false;
throw new FtpException(respFragment["ExcepMsg"] as string);
}
return(respFragment);
}
/// <summary>
/// 把异常信息传递到客户端
/// </summary>
/// <param name="exceptionSource">错误源</param>
/// <param name="msg">正常的消息头</param>
/// <param name="exceptionMsg">异常信息</param>
private byte[] ReturnExceptionToClient(string exceptionSource, TMessage msg, string exceptionMsg)
{
//把异常信息传递到客户端
TSerializer serializer = new TBinarySerializer();
TMessage exceptMsg = new TMessage(exceptionSource, TMessageType.Exception, msg.MsgID, msg.LiveLife, "", "", "", "");
serializer.WriteMessageBegin(exceptMsg);
TApplicationException tAppException = new TApplicationException(TApplicationException.ExceptionType.Unknown, exceptionMsg);
tAppException.Write(serializer);
serializer.WriteMessageEnd();
//如果不是单向的，把异常返回，并跳出处理函数
byte[] respException = serializer.ToBytes();
serializer.Flush();
// 返回异常
return(respException);
}
/// <summary>
/// 远程方法调用
/// </summary>
/// <param name="recvAddress">接收节点的虚拟IP地址，如3.1/param>
/// <param name="interfaceName">接口名称</param>
/// <param name="methodName">方法名称</param>
/// <param name="paramDict">参数字段</param>
/// <param name="messageType">消息类型</param>
public void Call(string recvAddress, ushort destVPort, string interfaceName, string methodName, Dictionary <string, object> paramDict, TMessageType messageType = TMessageType.RpcCall)
{
try
{
VirtuaIP receiverVIP = new VirtuaIP(recvAddress);
string msgID = Guid.NewGuid().ToString();
if (!InterfaceImplementMap.InterfaceAndImplementMap.ContainsKey(interfaceName))
{
throw new ApplicationException("找不到接口" + interfaceName + "的具体实现！！");
}
//本地应用端口随机获取，目标端口确定
TMessage reqMsg = new TMessage(interfaceName + "_" + methodName, messageType, msgID, -1, "", "", "", "");
DynMethodInstance dynMethodInstance = new DynMethodInstance(interfaceName, methodName);
TSerializer serializerReq = new TBinarySerializer();
serializerReq.WriteMessageBegin(reqMsg);
foreach (var para in paramDict)
{
dynMethodInstance[para.Key] = para.Value;
}
DynSerialize.WriteDynMethodInstance(serializerReq, dynMethodInstance);
serializerReq.WriteMessageEnd();
byte[] reqData = serializerReq.ToBytes();
serializerReq.Flush();
Send(receiverVIP, destVPort, reqData);
}
catch (Exception ex)
{
this.RaiseDealMessageExceptionEvent(ex);
throw new ApplicationException(ex.Message);
}
}
/// <summary>
/// 远程方法调用
/// </summary>
/// <param name="recvAddress">接收节点的虚拟IP地址，如3.1/param>
/// <param name="interfaceName">接口名称</param>
/// <param name="methodName">方法名称</param>
/// <param name="paramDict">参数字段</param>
/// <param name="liveLife">生命周期,以秒为单位</param>
/// <param name="messageType">消息类型</param>
/// <returns>服务端返回的数据或者null</returns>
public object Call(string recvAddress, ushort destVPort, string interfaceName, string methodName, Dictionary <string, object> paramDict, short liveLife, TMessageType messageType = TMessageType.RpcCall)
{
object result = null;
_msgID = _msgID < int.MaxValue ? ++_msgID : 0;
try
{
VirtuaIP receiverVIP = new VirtuaIP(recvAddress);
string msgID = _msgID.ToString();
if (!InterfaceImplementMap.InterfaceAndImplementMap.ContainsKey(interfaceName))
{
throw new ApplicationException("找不到接口" + interfaceName + "的具体实现！");
}
DynMethodInstance dynMethodInstance = new DynMethodInstance(interfaceName, methodName);
TMessage reqMsg = new TMessage(interfaceName + "_" + methodName, messageType, msgID, liveLife, "", "", "", "");
TSerializer serializerReq = new TBinarySerializer();
//将消息头信息写入流
serializerReq.WriteMessageBegin(reqMsg);
foreach (var para in paramDict)
{
dynMethodInstance[para.Key] = para.Value;
}
//将方法的实例写入流
DynSerialize.WriteDynMethodInstance(serializerReq, dynMethodInstance);
//什么都不做
serializerReq.WriteMessageEnd();
//将消息流转换成二进制流
byte[] reqData = serializerReq.ToBytes();
serializerReq.Flush();
//发送消息
Send(receiverVIP, destVPort, reqData);
var timeout = new TimeSpan(0, 0, liveLife);
//创建一个 Stopwatch 实例：使用 StartNew() 会创建一个 Stopwatch 实例并马上开始计时，即等效于如下代码：
//Stopwatch sw2 = new Stopwatch();
//sw2.Start();
var timer = Stopwatch.StartNew();
bool isReceive = false;
do
{
ZmqMessage zmqMessage = _reqSocket.ReceiveMessage(timeout - timer.Elapsed);
if (zmqMessage.FrameCount == 0)
{
throw new TimeoutException("调用超时，请检查网络是否联通！");
}
else
{
//zmqMessage[zmqMessage.FrameCount - 1].Buffer 只需要处理zmqMessage中最后一帧,这一帧存的是消息中的数据部分
result = DealRepMessage(zmqMessage[zmqMessage.FrameCount - 1].Buffer, msgID, out isReceive);
//消息ID不匹配且超时时间未到时，丢掉当前接收到的数据包，继续接受数据
if (isReceive)
{
return(result);
}
}
}while (timer.Elapsed <= timeout);
throw new TimeoutException("调用超时，请检查网络是否联通！");
}
catch (Exception ex)
{
this.RaiseDealMessageExceptionEvent(ex);
throw;
}
}
/// <summary>
/// 文件下载
/// </summary>
/// <param name="recvAddress">服务端机器的地址</param>
/// <param name="savePath">文件在本地的保存路径</param>
/// <param name="downloadPath">要下载的文件在服务端的路径</param>
public void FileDownload(string recvAddress, string savePath, string downloadPath)
{
#region 前置检查
if (_isWorking)
{
throw new FtpException("一个FtpClient不能同时执行多个任务");
}
//验证本地是否已有同名文件
if (File.Exists(Path.Combine(savePath, Path.GetFileName(downloadPath))))
{
throw new FtpException(savePath + "目录中已存在同名文件");
}
#endregion 前置检查
#region 获取文件的基本信息
_remoteAddress = new VirtuaIP(recvAddress);
_serverPort = GetServerPort("FileDownload");
_isWorking = true;
//发送下载文件的请求
long fileSize = 0; //文件大小（字节）
long remaining = 0; //剩余未下载的文件大小
DynObject fileFragment = RequestDownload(_serverPort, downloadPath);
fileSize = remaining = (long)fileFragment["FileLength"];
#endregion 获取文件的基本信息
FileStream writeFileStream = null;
_isCancelDownload = false;
while (remaining >= 0 && _isWorking)
{
if (_isCancelDownload)
{
fileFragment["State"] = (byte)103;
}
switch ((byte)fileFragment["State"])
{
case 100:
string tempFile = Path.Combine(savePath, fileFragment["MD5"] + ".temp");
//如果存在该文件的临时文件，则进行断点续传,否则以MD5为名字创建临时文件开始传送数据
if (File.Exists(tempFile))
{
writeFileStream = new FileStream(tempFile, FileMode.Open);
fileFragment["MsgID"] = (int)(writeFileStream.Length / _bufferLength);
fileFragment["State"] = (byte)101;
fileFragment["Data"] = null;
}
else
{
writeFileStream = new FileStream(tempFile, FileMode.Create);
fileFragment["MsgID"] = 0;
fileFragment["State"] = (byte)101;
fileFragment["Data"] = null;
}
remaining = fileSize - (int)fileFragment["MsgID"] * _bufferLength;
break;
case 101: //文件传送中
writeFileStream.Seek((int)fileFragment["MsgID"] * _bufferLength, SeekOrigin.Begin);
writeFileStream.Write(fileFragment["Data"] as byte[], 0, (int)fileFragment["DataLength"]);
remaining = fileSize - ((int)fileFragment["MsgID"] * _bufferLength + (int)fileFragment["DataLength"]);
if (remaining <= 0)
{
//文件下载结束
fileFragment["State"] = (byte)102;
writeFileStream.Flush();
writeFileStream.Close();
writeFileStream = null;
_isWorking = false;
//校验文件传送是否正确
string path = Path.Combine(savePath, fileFragment["MD5"] as string + ".temp");
string tempFileMD5 = GetMD5HashFromFile(path);
if (tempFileMD5 == fileFragment["MD5"] as string)
{
File.Move(path, Path.Combine(savePath, fileFragment["FileName"] as string));
}
else
{
File.Delete(path);
throw new FtpException("文件未能成功下载");
}
}
fileFragment["MsgID"] = (int)fileFragment["MsgID"] + 1;
fileFragment["DataLength"] = 0;
fileFragment["Data"] = null;
break;
case 103: //中断文件传送
fileFragment["DataLength"] = 0;
fileFragment["Data"] = null;
writeFileStream.Flush();
writeFileStream.Close();
writeFileStream = null;
_isWorking = false;
break;
case 105: //服务端发生异常
writeFileStream.Flush();
writeFileStream.Close();
writeFileStream = null;
_isWorking = false;
throw new FtpException(fileFragment["ExcepMsg"] as string);
}
//向外界通知文件床送进度
if (ProgressEvent != null)
{
ProgressEvent(fileSize, fileSize - remaining);
}
//发送请求下一个包数据（发生异常告诉对方关闭文件流）
TSerializer serialize = new TBinarySerializer();
DynSerialize.WriteDynObject(serialize, fileFragment);
byte[] buffer = serialize.ToBytes();
serialize.Flush();
Send(_serverPort, buffer);
ZmqMessage message = _ftpSocket.ReceiveMessage(new TimeSpan(0, 1, 30));
if (message.FrameCount == 0)
{
writeFileStream.Close();
writeFileStream = null;
_isWorking = false;
ReStart();
throw new TimeoutException("发送超时，请检查你的网络是否有问题,稍后重新下载");
}
serialize = new TBinarySerializer();
serialize.FromBytes(message[message.FrameCount - 1].Buffer);
fileFragment = DynSerialize.ReadDynObject(serialize);
serialize.Flush();
}
}
/// <summary>
/// 文件上传
/// </summary>
/// <param name="recvAddress">服务端机器的地址</param>
/// <param name="filePath">准备上传的文件</param>
/// <param name="savePath">文件在服务端的保存路径</param>
public void FileUpload(string recvAddress, string filePath, string savePath)
{
if (_isWorking)
{
throw new FtpException("一个FtpClient不能同时执行多个任务");
}
CheckPostfix(filePath);//检查上传的文件类型是否合法
string md5 = GetMD5HashFromFile(filePath);
_remoteAddress = new VirtuaIP(recvAddress);
_serverPort = GetServerPort("FileUpload");
using (FileStream fStream = File.OpenRead(filePath))
{
long fileSize = fStream.Length; //文件数据长度
long remaining = fileSize; // 剩余没上传的数据长度
byte[] buffer = new byte[_bufferLength];
//构建文件片段
DynObject fileFragment = new DynObject("FileFragment");
fileFragment["MD5"] = md5;
fileFragment["FileName"] = Path.GetFileName(filePath);
fileFragment["Extension"] = Path.GetExtension(filePath);
fileFragment["Path"] = savePath;
fileFragment["DataLength"] = 0;
fileFragment["MsgID"] = -1;
fileFragment["State"] = (byte)0;
_isCancelUpload = false;
_isWorking = true;
int offset = 0;
while (remaining >= 0 && _isWorking)
{
if (_isCancelUpload)
{
fileFragment["State"] = (byte)3;
}
switch ((byte)fileFragment["State"])
{
case 0: //文件传送开始
//第一次发送文件,发送一些基本信息（如文件名等）
fileFragment["DataLength"] = 0;
fileFragment["Data"] = null;
break;
case 1: //文件传送中
fStream.Seek((int)fileFragment["MsgID"] * _bufferLength, SeekOrigin.Begin);
fileFragment["DataLength"] = fStream.Read(buffer, offset, buffer.Length - offset);
fileFragment["Data"] = buffer;
//重置剩余数据长度
remaining = fileSize - ((int)fileFragment["MsgID"] * _bufferLength + (int)fileFragment["DataLength"]);
if (remaining == 0)
{
_isWorking = false;
remaining = -1;
fileFragment["State"] = (byte)2;
}
break;
case 3: //中断文件传送
fileFragment["DataLength"] = 0;
fileFragment["Data"] = null;
_isWorking = false;
break;
}
//序列化
TSerializer serializer = new TBinarySerializer();
DynSerialize.WriteDynObject(serializer, fileFragment);
byte[] data = serializer.ToBytes();
serializer.Flush();
DynObject repFragment = Transfer(_serverPort, data, 0);
fileFragment["State"] = repFragment["State"];
fileFragment["MsgID"] = repFragment["MsgID"];
//向外界通知文件传送进度
if (ProgressEvent != null)
{
ProgressEvent(fileSize, fileSize - remaining);
}
}
}
}
/// <summary>
/// 重载父类方法
/// </summary>
/// <param name="msgData">接收到的消息</param>
private void DealMessage(object obj)
{
ZmqMessage zmqMessage = obj as ZmqMessage;
//记录源地址和端口
VirtuaIP originVIP = new VirtuaIP(zmqMessage[1].Buffer);
ushort originPort = (ushort)BitConverter.ToInt16(zmqMessage[3].Buffer, 0);
byte[] requestData = zmqMessage[zmqMessage.FrameCount - 1].Buffer;
if (requestData != null && requestData.Length >= 4)
{
TSerializer serializer = new TBinarySerializer();
serializer.FromBytes(requestData);
// 获取消息
TMessage msg = serializer.ReadMessageBegin();
if (msg.Type == TMessageType.RpcCall || msg.Type == TMessageType.RpcOneway)
{
//获取方法信息构造Method
string[] temp = msg.Name.Split('_');
DynMethodInstance methodInstance = null;
try
{
methodInstance = DynSerialize.ReadDynMethodInstance(serializer, temp[0], temp[1]);
serializer.ReadMessageEnd();
serializer.Flush();
Dictionary <string, object> paramValues = new Dictionary <string, object>();
//获取参数信
foreach (string paramName in methodInstance.DynMethod.GetParameterNames())
{
paramValues[paramName] = methodInstance[paramName];
}
string className = InterfaceImplementMap.InterfaceAndImplementMap[temp[0]];
string methodFullName = className + "_" + temp[1];
//方法调用
object ret = DynTypeManager.MethodHandler(null, methodFullName, paramValues);
if (methodInstance.DynMethod.Result.DynType != DynType.Void)
{
methodInstance.Result = ret;
}
// 序列化返回结果
serializer = new TBinarySerializer();
TMessage respMsg = new TMessage(msg.Name, TMessageType.RpcReply, msg.MsgID, msg.LiveLife, "", "", "", "");
serializer.WriteMessageBegin(respMsg);
DynSerialize.WriteResult(serializer, methodInstance);
serializer.WriteMessageEnd();
byte[] respData = serializer.ToBytes();
serializer.Flush();
// 返回客户端
Send(originVIP, originPort, respData);
}
catch (Exception ex)
{
serializer.ReadMessageEnd();
serializer.Flush();
//如果服务端需要记录日志，则由本事件把异常信息发出
RaiseDealMessageExceptionEvent(ex);
byte[] returnData = ReturnExceptionToClient("RpcServer.DealMessageException", msg, ex.Message);
// 返回客户端
Send(originVIP, originPort, returnData);
return;
}
}
}
}