public override async void Start(FrameBase frameBase)
{
var frame = frameBase as DataFrame;
if (frame == null)
{
return;
}
try
{
var file = FileManager.Get(frame.Guid);
if (file == null)
{
throw new Exception("未能找到已打开的文件。");
}
FileStream fs = file.Item1;
SemaphoreSlim lck = file.Item2;
using (LockGuard lg = await LockGuard.WaitAsync(lck).ConfigureAwait(false))
{
if (fs.Position != frame.DataOffset)
{
fs.Seek(frame.DataOffset, SeekOrigin.Begin);
}
await fs.WriteAsync(frame.Data, 0, frame.Data.Length).ConfigureAwait(false);
}
}
catch (Exception ex)
{
await SendError(frame, ex.Message).ConfigureAwait(false);
}
}
public override async void Start(FrameBase frameBase)
{
var frame = frameBase as CloseFrame;
if (frame == null)
{
return;
}
try
{
var file = FileManager.GetAndRemove(frame.Guid);
if (file == null)
{
throw new Exception("未能关闭文件：没有找到已打开的文件。");
}
FileStream fs = file.Item1;
SemaphoreSlim lck = file.Item2;
using (LockGuard lg = await LockGuard.WaitAsync(lck).ConfigureAwait(false))
{
fs.Close();
}
await SendFrame(new CloseAnswerFrame(frame.Guid)).ConfigureAwait(false);
}
catch (Exception ex)
{
await SendError(frame, ex.Message).ConfigureAwait(false);
}
}
/// <summary>
/// 向给定客户端发送帧。
/// </summary>
/// <exception cref="Exception"></exception>
public virtual async Task Send(ClientDataBase client, FrameBase frame)
{
using (await LockGuard.WaitAsync(client.WriteLock).ConfigureAwait(false))
{
await WriteFrame(client.TcpClient.GetStream(), frame).ConfigureAwait(false);
}
}
/// <summary>
/// 线程安全地向服务端发送帧。
/// </summary>
/// <exception cref="Exception"></exception>
public async Task SendAsync(FrameBase frame)
{
using (LockGuard lg = await LockGuard.WaitAsync(SemaphoreSend).ConfigureAwait(false))
{
if (Common.Configs.Debug)
{
Console.WriteLine("S >" + frame.GetType().ToString());
}
await WriteFrame(Tcp.GetStream(), frame).ConfigureAwait(false);
}
}
public override async void Start(FrameBase frameBase)
{
var frame = frameBase as RequestFrame;
if (frame == null)
{
return;
}
try
{
var file = FileManager.Get(frame.Guid);
if (file == null)
{
throw new Exception("未能找到已打开的文件。");
}
const int sizeMax = 1024 * 128 < Configs.MaxFrameSize ? 1024 * 128 : Configs.MaxFrameSize;
const int sizeMin = 1024 < Configs.MaxFrameSize ? 1024 : Configs.MaxFrameSize;
int size = sizeMin;
byte[] buffer = new byte[sizeMax];
long offset = 0;
using (LockGuard lg = await LockGuard.WaitAsync(file.Item2).ConfigureAwait(false))
using (FileStream fs = file.Item1)
{
SpeedRecorder recorder = new SpeedRecorder();
while (offset < fs.Length)
{
int count = await fs.ReadAsync(buffer, 0, size).ConfigureAwait(false);
DataFrame dataFrame = new DataFrame(frame.Guid)
{
FileSize = fs.Length,
DataOffset = offset,
Data = buffer.Take(count).ToArray(),
};
await SendFrame(dataFrame).ConfigureAwait(false);
offset += count;
recorder.Record(offset);
size = Math.Max(sizeMin, Math.Min(sizeMax, Common.Tools.SpeedRecorder.UpdateFrameSize((int)recorder.SpeedPerSecond, size)));
}
}
}
catch (Exception ex)
{
await SendError(frame, ex.Message).ConfigureAwait(false);
}
}