C#技术分享【基于socket的断点续传功能】

最近开发了一个异地灾备传输工具，每周从广州服务器同步5g大小的数据库备份文件到北京服务器。

第一个版本的传输工具开发得很简单，两个window服务，一个负责发送，另一个负责接收，利用socket建立tcp链接，在测试服务器上传输10g大小的文件没有问题。

但部署到生产环境之后，每次传输了3g左右的数据就中断了，经分析是由于网络不稳定造成的，所以需要对传输工具添加断点续传功能，当传输意外中断时，可以自动连接，并完成上一次未完成的传输。

断点续传的原理很简单，就是分割需要传输的文件，每次传输一小块数据，并附带数据的位置和大小信息，服务器成功接收数据之后，则继续下一块数据的传输，否则重复上一块数据的传输，直到成功为止。

这实际上是为你的传输功能添加了暂停功能，网络中断的我时候暂停传输了，网络恢复之后继续再传。

既然搞清楚原理了我们就赶紧开始写代码吧，下面是核心代码。

public class BreakPointPost
{
    public string FileName { get; set; }
    public long FileSize { get; set; }
    public long PackageSize { get; set; }
    public int PackageCount { get; set; }
    public int Index { get; set; }
}

private static int GetFilePackageCount(long fileSize, long packageSize)
{
    int count = 0;
    if (fileSize % packageSize > 0)
        count = Convert.ToInt32(fileSize / packageSize) + 1;
    else
        count = Convert.ToInt32(fileSize / packageSize);
    return count;
}

private static byte[] FileRead(string path, int index, long size)
{
    byte[] result = null;
    long length = (long)index * (long)size + size;
    using (FileStream stream = File.OpenRead(path))
    {
        if (length > stream.Length)
            result = new byte[stream.Length - ((long)index * (long)size)];
        else
            result = new byte[size];
        stream.Seek((long)index * (long)size, SeekOrigin.Begin);
        stream.Read(result, 0, result.Length);
    }
    return result;
}

private static void FileWrite(string path, int index, long packageSize, int receiveSize, byte[] data)
{
    using (System.IO.FileStream stream = System.IO.File.OpenWrite(path))
    {
        stream.Seek((long)index * (long)packageSize, System.IO.SeekOrigin.Begin);
        stream.Write(data, 0, receiveSize);
        stream.Flush();
    }
}

当然，完整的代码示例肯定是少不了的，以下两个版本大家按需下载。

1、控制台版（下载有惊喜）

2、Windows服务版（生产版本）

3、喜欢用GitHub的童鞋们也可以点我fork

版权声明：本文为博主原创文章，未经博主允许不得转载。