protected virtual bool OnFileTransferBegin(FileTransferMessage request, RemoteTransferRequest req)
{
//判断文件是否可读
if (request.Content == null || !request.Content.CanRead)
{
var res = req.GetResourceInfo(request.GUID);
OnMessage(String.Format("接收文件{0}失败：数据流不可读", res.FileName));
return(false);
}
return(true);
}
protected virtual void OnFileTransferFinished(FileTransferMessage request, RemoteTransferRequest req, bool succeeded, string targetFile)
{
req.CurTransState.curSize = req.CurTransState.maxSize;
req.CurTransferedSize = req.TotalFilesLength;
if (succeeded)
{
if (OnFileDownloaded != null)
{
OnFileDownloaded(request, targetFile);
}
var res = req.GetResourceInfo(request.GUID);
OnMessage(String.Format("接收文件完成：{0}({1})", res.FileName, targetFile));
}
}
public virtual void TransferFile(FileTransferMessage request)
{
var emp = OperationContext.Current.IncomingMessageProperties[RemoteEndpointMessageProperty.Name] as RemoteEndpointMessageProperty;
RemoteTransferRequest req = null;
if (!TransferRequests.TryGetValue(emp.Address, out req))
{
return;
}
var res = req.GetResourceInfo(request.GUID);
if (res == null)
{
return;
}
if (request.Index == 0)
{
OnTransmissionBegin(request, req);
}
if (!OnFileTransferBegin(request, req))
{
return;
}
req.CurTransState = new FileTransState(res.FileName, res.FileLength, request.Index);
OnMessage(String.Format("开始接收文件:{1}({0})...", res.Kind, res.FileName));
string filePath = GetFileDirectory(res.Kind).TrimEnd('\\') + "\\" + res.FileName;
CommonMethods.InheritCreateFolder(filePath, true);
bool succeeded = true;
try
{
req.CurTransState.curSize = 0;
// 文件流传输
using (var targetStream = new FileStream(filePath, FileMode.Create, FileAccess.Write, FileShare.None))
{
//定义文件缓冲区
byte[] buffer = new byte[TransmissionBufferLen];
int count = 0;
while ((count = request.Content.Read(buffer, 0, TransmissionBufferLen)) > 0)
{
targetStream.Write(buffer, 0, count);
OnFileTransferring(request, req, count);
}
request.Content.Close();
}
}
catch (Exception ex)
{
OnMessage(MessageLevel.错误, String.Format("接收文件{0}失败：{1}", res.FileName, ex.Message));
succeeded = false;
}
OnFileTransferFinished(request, req, succeeded, filePath);
if (request.Index >= req.Resources.Count - 1)
{
OnTransmissionFinished(request, req);
}
}
protected virtual void OnFileTransferring(FileTransferMessage request, RemoteTransferRequest req, int bytescount)
{
req.CurTransState.curSize += bytescount;
req.CurTransferedSize += bytescount;
}
protected virtual void OnTransmissionFinished(FileTransferMessage request, RemoteTransferRequest req)
{
req.CurTransState = null;
req.CurTransferedSize = 0;
TransferRequests.Remove(req.Address);
}
protected virtual void OnTransmissionBegin(FileTransferMessage request, RemoteTransferRequest req)
{
req.CurTransferedSize = 0;
}