一、前言
前段时间一直在折腾基于Socket的产品在线升级模块。之前我曾写过基于.Net Remoting的、基于WCF的在线升级功能，由于并发量较小及当时代码经验的不足一直没有实际应用。这次下定决心撰写基于Socket的在线更新功能，一方面是觉得Socket的并发量较高，另一方面也是自己工作了一年多，积攒了一定的经验，应该能hold住。本文将展示的是Protype版本，Release版本已在远程测试服务器上运行，并发数过万没有什么问题，文件更新都很正常。代码的Github地址将在本文最后提供。本文将展示的在线更新功能模块涉及Devexpress WPF、Webapi、Windows Service，我会从最基础的开始说起，非常适合初入的新手，大牛或者老司机可直接略过。
二、方案
公司的产品是运行在某一BIM软件上的插件，要想做在线更新，有以下两种方案：
方案一：
插件安装后会在客户桌面上生成一个快捷方式，双击快捷方式会启动一个LaunchProduct.exe，在这里面进行更新操作，更新完之后再启动BIM软件。
方案二：
用户首先运行BIM软件，点击插件里的更新按钮，然后通过Socket下载文件。进程中Kill掉该BIM软件，执行文件替换，再自动启动该BIM软件。（不kill掉的话程序一直被占用是无法更新文件的）
三、步骤详解
无论是方案一，还是方案二，有些核心步骤是不变的。下面详细论述：
Step1： 从注册表中读取当前产品的版本、安装位置等信息。注册表是在做产品安装包时就应该要做的一件事情。产品安装完就会在客户机生成相应的注册表信息。我正好也是做产品安装包的，非常熟悉产品注册表里有哪些内容。那么这里，我封装了一个读注册表的class，可以在Github项目里找到：RegistryUtils （UpdaterClient工程中）在这里要提醒的一个地方是：有时候注册表明明有内容，C#代码调试却是null，那么解决的办法如下：
var localMachineRegistry = RegistryKey.OpenBaseKey(RegistryHive.LocalMachine, Environment.Is64BitOperatingSystem ? RegistryView.Registry64 : RegistryView.Registry32); //用这个localMachineRegistry去OpenSubKey()
//而不是直接用Registry.LocalMachine去OpenSubKey()
Step2： 从注册表里读取完本地产品的相关信息后，我把这些数据封装成一个对象，去请求产品服务器上的某个Webapi。如果有更新文件，会返回给我更新文件的大小及MD5值。更新文件的大小决定了我每个分包的大小，更新文件的MD5值用于我下载完分包进行合并后进行MD5比对，验证下载的包是否完整。Md5Utils （UpdaterShare工程中）
/// <summary>
/// Get Download File Info
/// </summary>
/// <param name="basicInfo"></param>
/// <param name="serverAddress"></param>
/// <param name="controllerName"></param>
/// <param name="actionName"></param>
/// <param name="serverResult"></param>
/// <returns></returns>
public static bool RequestDownloadFileInfo(ClientBasicInfo basicInfo,
string serverAddress,
string controllerName,
string actionName,
ref DownloadFileInfo serverResult)
{
var packageInfo = JsonConvert.SerializeObject(basicInfo); try
{
HttpClient httpClient = new HttpClient
{
BaseAddress = new Uri(serverAddress),
Timeout = TimeSpan.FromMinutes()
}; if (ConnectionTest(serverAddress))
{
StringContent strData = new StringContent(packageInfo, Encoding.UTF8, "application/json");
string postUrl = httpClient.BaseAddress + $"api/{controllerName}/{actionName}";
Uri address = new Uri(postUrl);
Task<HttpResponseMessage> task = httpClient.PostAsync(address, strData);
try
{
task.Wait();
}
catch
{
return false;
}
HttpResponseMessage response = task.Result;
if (!response.IsSuccessStatusCode)
return false; try
{
string jsonResult = response.Content.ReadAsStringAsync().Result;
serverResult = JsonConvert.DeserializeObject<DownloadFileInfo>(jsonResult);
if (serverResult != null)
{
return true;
}
}
catch(Exception ex)
{
return false;
}
}
}
catch
{
return false;
}
return false;
}
Step3： 拿到更新文件的大小及MD5值后，我们就可以开始本地通过Socket去请求服务器下载更新文件了。这里的Socket我使用的是APM写法，也就是异步编程模型。
APM是微软比较早的提供用于Socket通信的方法。其最常见的写法就是 BeginAction(byte[] buffer, int offset, int size, SocketFlags socketFlags, AsyncCallback callback, object state)，在callback回调函数里EndAction。
我在Client里是通过生成5个Task，每个Task各有一个Socket去下载1/5文件，最后合并。
var tasks = new Task[packetCount];
for (int index = ; index < packetCount; index++)
{
int packetNumber = index;
var task = new Task(() =>
{
Socket client = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
ComObject state = new ComObject { WorkSocket = client, PacketNumber = packetNumber };
client.BeginConnect(remoteEp, ConnectCallback, state);
});
tasks[packetNumber] = task;
task.Start();
}
Task.WaitAll(tasks);
那么有的线程接收文件快，有的接收文件慢。因为最后还要分别生成5个临时文件进行合并。所以需要一个线程同步，让快的或慢的都在终点线等着。这里就要用到 ManualResetEvent，其继承于EventWaitHandle，EventWaitHandle又继承于WaitHandle。ManualResetEvent是怎么使用的呢？因为代码都在Github上，这里提炼一下，总结就是： ManualResetEvent 初始为false的时候，只有在某个线程中使用ManualResetEvent.Set()方法，才能让另一线程中写在ManualResetEvent.WaitOne()之后的代码运行。假设主线程里调用了WaitOne()，那么主线程写在WaitOne之后的代码要想执行，就必须等待子线程中调用Set()方法，否则主线程会一直阻塞在WaitOne()处。
在Socket里肯定要定义一个自己产品的数据包格式，因为Socket里传的都是byte[]，你肯定要让客户端/服务器知道你发的byte[]是什么意思吧，所以要定义数据包格式：
A Packet = start_tag + version_tag + request/response_tag + length_tag + data + crc16_tag
1. 包头标识，一般用 { 0xAA, 0x55 }
2. 格式版本，暂且定为 { 0x01 }
3. 发送标识，是客户端发的呢？还是服务器发的呢？
4. 长度标识，用于记录整个数据包的长度，此标识占2个字节
5. 数据，要传输的数据
6. crc16校验码。用于判断传输的byte[]是否完整，相对于MD5，crc16校验码的字节数更短，不会占太多传输字节，非常适合用于字节数组的比较。MD5常常用于文件对比。
那么，有了长度标识与crc16校验码的双保险，我们就可以知道传输的byte[]是否完整了。
当一方发送byte[]后，另一方收到后可以拿出长度标识，判断byte[]长度是否正确；当长度正确后，使用 Crc16Utils 计算收到的byte[]的crc16码并与byte[]中的crc16码进行比对。
Step4：服务器端的Socket是写在Windows Service里的。更新文件就放在该Windows Service同路径下，因为Windows Service在启动运行之后会在注册表写下相应信息，通过注册表就能知道该Windows Service的执行路径，继而得到更新文件的路径。
/// <summary>
/// Get Latest File From Windows Service by Registry
/// </summary>
/// <param name="serviceName"></param>
/// <returns></returns>
public static string GetFilePathFromService(string serviceName)
{
try
{
ServiceController[] services = ServiceController.GetServices();
var socketService = services.FirstOrDefault(x => String.Equals(x.ServiceName, "SocketService"));
if (socketService != null)
{
var localMachineRegistry = RegistryKey.OpenBaseKey(RegistryHive.LocalMachine, Environment.Is64BitOperatingSystem ?
RegistryView.Registry64 : RegistryView.Registry32);
var key = localMachineRegistry.OpenSubKey(@"SYSTEM\CurrentControlSet\Services\" + serviceName);
if (key != null)
{
var serviceExePath = GetString(key.GetValue("ImagePath").ToString());
var folderPath = Path.GetDirectoryName(serviceExePath);
if (!String.IsNullOrEmpty(folderPath) && Directory.Exists(folderPath))
{
return folderPath;
}
}
}
}
catch(Exception ex)
{
return null;
}
return null;
}
Step5： 当下载完文件，其实已经完成了90%的工作了，剩下的无非就是简单的替换文件，更新注册表信息等等。
最后附上完整的流程图：
四、其它
我在写Socket代码的时候参考了微软的示例，还是非常有帮助的，建议先看微软的示例再看Github的代码会更方便理解，在此提供下：
Microsoft官方示例：
目前微软早已提供了更接近Socket底层的SocketAsyncEventArgs（SAEA）写法，该方法不同于APM的是：
1. APM多次Send\Receive会产生多个IAsyncResult对象，增加消耗。
2. SAEA配合BufferManager以及池化能很好的调配服务器资源，有多少坑就蹲多少人，再多了就可以考虑转移至其它服务器做均衡了。
3. SAEA的并发能力比APM略高，但是坑也不少，比如APM中，通过EndReceive是否为0我就能知道还有没有数据要接收，但是SAEA中的Available等于0时还可能有数据没接收完，这个问题的解决方法网上各种各样，各位可以自己搜搜。SAEA的服务器写法我看看之后有没有时间写写。
GitHub地址：https://github.com/airforce094/SocketUpdater
五、最后
此Github里涉及Devexpress WPF、Webapi、Windows Service，不求Star，您的阅读就是对我最大的支持。有什么问题可留言相互讨论。
《原创，转载请注明来源》
来自：airforce094