项目需求:
局域网内有两台电脑,电脑A(Windows系统)主要是负责接收一些文件(远程桌面粘贴、FTP上传、文件夹共享等方式),希望能在A接收文件后自动传输到电脑B(Windows系统)来做一个备份,同时电脑B上有个目录,如果往这个目录里粘贴文件了,会自动传输给A来保存。
于是通过百度找到了System.IO.FileSystemWatcher这个类,通过它来监听指定的文件夹的一些消息(文件创建、文件修改、文件删除、文件重命名)来做对应的动作,目前只需求监控文件创建,其它事件不作处理。
文件传输方面,可以自己写Socket的Server和Client,但要注意粘包的问题。我这里使用了开源的NewLife.Net(https://github.com/NewLifeX/NewLife.Net),客户端和服务器都是用它的话,内置解决粘包问题的解决方案,而且管理起来很方便,自带日志输出功能强大。这里分享一下实现的代码以及一些问题。
1、创建一个Winform的工程,运行框架为.Netframework4.6
Nuget上引用NewLife.Net
界面结构如下:
本机端口,代表本机作为服务器(server)监听的端口,远程服务器IP及端口就是当本机监控到文件创建时,自动发送给哪台服务器(接收服务器同样需要运行本软件)。
本地监控自动发送文件夹:凡是在指定的这个文件夹中新建(一般是粘贴)的文件都会被自动发送走。
自动保存接收到的文件夹:凡是远程发送过来的文件,都自动保存在此文件夹下面。
2、实现代码
Program.cs中定义两个全局的变量,用来保存文件夹信息
/// <summary>
/// 要监控的接收保存文件夹
/// </summary>
public static string SaveDir = "";
/// <summary>
/// 要监控的发送文件夹
/// </summary>
public static string SendDir = "";
using的一些类
using System;
using System.Data;
using System.IO;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;
using DirectoryWatch.Class;
using NewLife.Data;
using NewLife.Log;
using NewLife.Net;
using NewLife.Net.Handlers;
在窗体加载时,定义一些共用变量以及指定窗体下方TextBox为日志输出载体
private static int remotePort = 0;//远程端口
private static int localPort = 0;//本地端口
private static string remoteIP = "";//远程IP
private FileSystemWatcher watcher;//监控文件夹
private NetServer server;//本地服务
private void MainFrm_Load(object sender, EventArgs e)
{
textBox1.UseWinFormControl();
}
本地监控文件夹选择按钮代码
private void Btn_dirbd_Click(object sender, EventArgs e)
{
using (var folderBrowser = new FolderBrowserDialog())
{
if (folderBrowser.ShowDialog() != DialogResult.OK) return;
Program.SendDir = folderBrowser.SelectedPath;
if (!Directory.Exists(Program.SendDir))
{
MessageBox.Show(@"所选路径不存在或无权访问", @"错误", MessageBoxButtons.OK, MessageBoxIcon.Error);
return;
}
if (string.Equals(Program.SaveDir.ToLower(), Program.SendDir.ToLower()))
{
MessageBox.Show(@"自动接收文件夹和自动发送文件夹不能是同一个文件夹", @"错误", MessageBoxButtons.OK, MessageBoxIcon.Error);
return;
}
txt_localPath.Text = folderBrowser.SelectedPath;
Program.SendDir = folderBrowser.SelectedPath;
}
}
本地自动保存文件夹选择按钮代码
private void Btn_saveDic_Click(object sender, EventArgs e)
{
using (var folderBrowser = new FolderBrowserDialog())
{
if (folderBrowser.ShowDialog() != DialogResult.OK) return;
Program.SaveDir = folderBrowser.SelectedPath;
if (!Directory.Exists(Program.SendDir))
{
MessageBox.Show(@"所选路径不存在或无权访问", @"错误", MessageBoxButtons.OK, MessageBoxIcon.Error);
return;
}
if (string.Equals(Program.SaveDir.ToLower(), Program.SendDir.ToLower()))
{
MessageBox.Show(@"自动接收文件夹和自动发送文件夹不能是同一个文件夹", @"错误", MessageBoxButtons.OK, MessageBoxIcon.Error);
return;
}
txt_remoteDir.Text = folderBrowser.SelectedPath;
Program.SaveDir = folderBrowser.SelectedPath;
}
}
启动代码(启动本地监控,启用本地server服务)
private void Btn_Start_Click(object sender, EventArgs e)
{
int.TryParse(txt_remotePort.Text, out remotePort);
int.TryParse(txt_localPort.Text, out localPort);
if (string.IsNullOrEmpty(txt_remoteIP.Text.Trim()))
{
MessageBox.Show(@"请填写远程服务器IP", @"错误", MessageBoxButtons.OK, MessageBoxIcon.Error);
return;
}
remoteIP = txt_remoteIP.Text.Trim();
if (remotePort == 0)
{
MessageBox.Show(@"请填写远程服务器的端口", @"错误", MessageBoxButtons.OK, MessageBoxIcon.Error);
return;
}
if (localPort == 0)
{
MessageBox.Show(@"请填写本地服务器要打开的端口", @"错误", MessageBoxButtons.OK, MessageBoxIcon.Error);
return;
}
if (string.IsNullOrEmpty(Program.SendDir))
{
MessageBox.Show(@"请选择本地自动发送文件夹路径", @"错误", MessageBoxButtons.OK, MessageBoxIcon.Error);
return;
}
if (string.IsNullOrEmpty(Program.SaveDir))
{
MessageBox.Show(@"请选择本地自动接收发送过来的文件夹路径", @"错误", MessageBoxButtons.OK, MessageBoxIcon.Error);
return;
}
if (Btn_Start.Text.Equals("停止"))
{
watcher.EnableRaisingEvents = false;
server.Stop("手动停止");
Btn_Start.Text = @"启动";
foreach (Control control in Controls)
{
if (!(control is Button) && !(control is TextBox)) continue;
if (control.Name != "Btn_Start")
{
control.Enabled = true;
}
}
return;
}
watcher = new FileSystemWatcher
{
Path = Program.SendDir,
Filter = "*.*"//监控所有文件
};
watcher.Created += OnProcess;//只监控新增文件
//watcher.Changed += OnProcess;
//watcher.Deleted += new FileSystemEventHandler(OnProcess);
//watcher.Renamed += new RenamedEventHandler(OnRenamed);
watcher.EnableRaisingEvents = true;//是否让监控事件生效
//watcher.NotifyFilter = NotifyFilters.Attributes | NotifyFilters.CreationTime | NotifyFilters.DirectoryName | NotifyFilters.FileName | NotifyFilters.LastAccess| NotifyFilters.LastWrite | NotifyFilters.Security | NotifyFilters.Size;
watcher.NotifyFilter = NotifyFilters.FileName;//这是一些通知属性,目前不用
watcher.IncludeSubdirectories = true;//包含子文件夹
server = new NetServer
{
Log = XTrace.Log,
SessionLog = XTrace.Log,
SocketLog = XTrace.Log,
Port = localPort,
ProtocolType = NetType.Tcp
};//使用NewLife.Net创建一个Server服务,只使用TCP协议
server.Received += async (x, y) =>
{
//接收文件
var session = x as NetSession;
if (!(y.Message is Packet pk)) return;
int.TryParse(Encoding.UTF8.GetString(pk.ReadBytes(0, 1)), out var fileState);//文件状态1字节
int.TryParse(Encoding.UTF8.GetString(pk.ReadBytes(1, 10)), out var headinfo);//文件总长度10字节
int.TryParse(Encoding.UTF8.GetString(pk.ReadBytes(11, 8)), out var fileNameLength);//文件名长度8字节
var fileName = Encoding.UTF8.GetString(pk.ReadBytes(19, fileNameLength));//文件名
int.TryParse(Encoding.UTF8.GetString(pk.ReadBytes(19 + fileNameLength, 10)), out var offset);//位置偏移量10字节
var data = pk.ReadBytes(29 + fileNameLength, pk.Count - (29 + fileNameLength));//数据内容
if (data.Length == 0) return;
await Task.Run(async () =>
{
var writeData = data;
using (var filestream = new FileStream($"{Program.SaveDir}\\{fileName}", FileMode.OpenOrCreate,
FileAccess.Write, FileShare.ReadWrite))
{
filestream.Seek(offset, SeekOrigin.Begin);
await filestream.WriteAsync(writeData, 0, writeData.Length);
await filestream.FlushAsync();
}//数据写入文件
});
XTrace.WriteLine($@"状态:{fileState},编号:{session.ID},文件总长度:{headinfo},文件名长度:{fileNameLength},文件名:{fileName},偏移量:{offset},内容长度:{data.Length}");
//XTrace.Log.Debug(Encoding.UTF8.GetString(pk.Data));//输出日志
};
server.Add<StandardCodec>();//解决粘包,引入StandardCodec
server.Start();
Btn_Start.Text = string.Equals("启动", Btn_Start.Text) ? "停止" : "启动";
foreach (Control control in Controls)
{
if (!(control is Button) && !(control is TextBox)) continue;
if (control.Name != "Btn_Start")
{
control.Enabled = false;
}
}
}
监控事件触发时执行的代码
private static void OnProcess(object source, FileSystemEventArgs e)
{
if (e.ChangeType == WatcherChangeTypes.Created)
{
OnCreated(source, e);
}
//else if (e.ChangeType == WatcherChangeTypes.Changed)
//{
// OnChanged(source, e);
//}
//else if (e.ChangeType == WatcherChangeTypes.Deleted)
//{
// OnDeleted(source, e);
//}
}
监控到创建文件时执行代码
/// <summary>
/// 监测文件创建事件,延时10秒后进行写入文件发送队列,防止文件尚未创建完成就执行发送(10秒内复制不完的 同样有问题)
/// 第1位 0代表新文件 1代表续传 2代表最后一次
/// 2--11位 代表文件总长度
/// 12--18 位代表文件名长度
/// 19--N 位 代表文件名信息
/// 19--(N+1)--offset位,代表此次发送文件的偏移量位置
/// 29+(N+1)--结束 代表此次发送的文件内容
/// </summary>
/// <param name="source"></param>
/// <param name="e"></param>
private static void OnCreated(object source, FileSystemEventArgs e)
{
Task.Run(async () =>
{
await Task.Delay(10000);
var TcpClient = new NetUri($"tcp://{remoteIP}:{remotePort}");//需要发送给的远程服务器
var Netclient = TcpClient.CreateRemote();
Netclient.Log = XTrace.Log;
Netclient.LogSend = true;
Netclient.LogReceive = true;
Netclient.Add<StandardCodec>();
Netclient.Received += (s, ee) =>
{
if (!(ee.Message is Packet pk1)) return;
XTrace.WriteLine("收到服务器:{0}", pk1.ToStr());
};
if (!File.Exists(e.FullPath)) return;
byte[] data;
using (var streamReader = new FileStream(e.FullPath, FileMode.Open, FileAccess.Read, FileShare.ReadWrite))
{
if (streamReader.CanRead)
{
data = new byte[streamReader.Length];
await streamReader.ReadAsync(data, 0, (int)streamReader.Length);
}
else
{
XTrace.Log.Error($"{e.FullPath}不可访问");
return;
}
}
var fileState = Encoding.UTF8.GetBytes("0");//新文件发送
var headinfo = new byte[10];//总长度
headinfo = Encoding.UTF8.GetBytes(data.Length.ToString());
var fileNameLength = new byte[8];//文件名长度
fileNameLength = Encoding.UTF8.GetBytes(Encoding.UTF8.GetBytes(e.Name).Length.ToString());
var fileNameByte = new byte[e.Name.Length];//文件名
fileNameByte = Encoding.UTF8.GetBytes(e.Name);
var offset = 0;//偏移量
var sendLength = 409600;//单次发送大小
Netclient.Open();
while (data.Length > offset)
{
if (offset > 0)
{
fileState = Encoding.UTF8.GetBytes("1");//追加文件
}
if (sendLength > data.Length - offset)
{
sendLength = data.Length - offset;
fileState = Encoding.UTF8.GetBytes("2");//最后一次发送
}
var offsetByte = new byte[10];//偏移位置byte
offsetByte = Encoding.UTF8.GetBytes(offset.ToString());
//一次发送总byte
var sendData = new byte[1 + 10 + 8 + fileNameByte.Length + 10 + sendLength];
//文件状态0第一次 1追加文件 2最后一次发送
Array.Copy(fileState, 0, sendData, 0, fileState.Length);
//文件总长度
Array.Copy(headinfo, 0, sendData, 1, headinfo.Length);
//文件名长度
Array.Copy(fileNameLength, 0, sendData, 11, fileNameLength.Length);
//文件名信息
Array.Copy(fileNameByte, 0, sendData, 19, fileNameByte.Length);
//此次内容偏移量
Array.Copy(offsetByte, 0, sendData, 19 + fileNameByte.Length, offsetByte.Length);
//一次发送的内容 offsetByte为10byte
Array.Copy(data, offset, sendData, 29 + fileNameByte.Length, sendLength);
offset += sendLength;
var pk = new Packet(sendData);
Netclient.SendMessage(pk);
}
//Netclient.Close("发送完成,关闭连接。");
});
}
效果图:
断点续传
原因:
1、在使用过程中,发现NewLife.Net不支持普通Socket编程那样的可以一直Receive来接收后续流的操作(TCP协议),每次流到达都会触发一次事件,从而不得不每次发送的时候都带上一些头部信息(文件名、偏移量、大小等),无形中增大了流量。
2、目前看的效果是NewLife.Net在服务器端接收的时候,包的顺序并不是和客户端Send的时候保持一致(如客户端发送 1 2 3 4 5),服务端可能接收的是2 1 3 5 4这样的顺序,这个问题,可能是跟我用异步有关系(我试了几次不用async 好像不一定每次都是和发送端的顺序一致),按说TCP是可以保证包的顺序的,我已经在GitHub上提问了,目前等待作者回复。