c# 智能升级程序代码(2)

最主要的就是Update()这个函数了。当程序调用au.Update时，首先检查当前是否开户了自动更新：
if (!config.Enabled)
    return;

如果启用了自动更新，就需要去下载服务器配置文件了：
WebClient client = new WebClient();
string strXml = client.DownloadString(config.ServerUrl);

然后，解析服务器配置文件到一个Dictionary中：

Dictionary<string, RemoteFile> listRemotFile = ParseRemoteXml(strXml);

接下来比较服务器配置文件和本地配置文件，找出需要下载的文件和本地需要删除的文件：

List<DownloadFileInfo> downloadList = new List<DownloadFileInfo>();
//某些文件不再需要了，删除
List<LocalFile> preDeleteFile = new List<LocalFile>();

foreach (LocalFile file in config.UpdateFileList)
{
    if (listRemotFile.ContainsKey(file.Path))
    {
        RemoteFile rf = listRemotFile[file.Path];
        if (rf.LastVer != file.LastVer)
        {
            downloadList.Add(new DownloadFileInfo(rf.Url, file.Path, rf.LastVer, rf.Size));
            file.LastVer = rf.LastVer;
            file.Size = rf.Size;

            if (rf.NeedRestart)
                bNeedRestart = true;
        }

        listRemotFile.Remove(file.Path);
    }
    else
    {
        preDeleteFile.Add(file);
    }
}

foreach (RemoteFile file in listRemotFile.Values)
{
    downloadList.Add(new DownloadFileInfo(file.Url, file.Path, file.LastVer, file.Size));
    config.UpdateFileList.Add(new LocalFile(file.Path, file.LastVer, file.Size));

    if (file.NeedRestart)
        bNeedRestart = true;
}

如果发现有需要下载的文件，则向用户显示这些文件，并提示其是否马上更新。如果用户选择了马上更新，则先删除本地不再需要的文件，然后开始下载更新文件。
if (downloadList.Count > 0)
{
    DownloadConfirm dc = new DownloadConfirm(downloadList);

    if (this.OnShow != null)
        this.OnShow();

    if (DialogResult.OK == dc.ShowDialog())
    {
        foreach (LocalFile file in preDeleteFile)
        {
            string filePath = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, file.Path);
            if (File.Exists(filePath))
                File.Delete(filePath);

            config.UpdateFileList.Remove(file);
        }

        StartDownload(downloadList);
    }
}

我们再来看一下StartDownload函数

private void StartDownload(List<DownloadFileInfo> downloadList)
{
    DownloadProgress dp = new DownloadProgress(downloadList);
    if (dp.ShowDialog() == DialogResult.OK)
    {
        //更新成功
        config.SaveConfig(Path.Combine(AppDomain.CurrentDomain.BaseDirectory, FILENAME));

        if (bNeedRestart)
        {
            MessageBox.Show("程序需要重新启动才能应用更新，请点击确定重新启动程序。", "自动更新", MessageBoxButtons.OK, MessageBoxIcon.Information);
            Process.Start(Application.ExecutablePath);
            Environment.Exit(0);
        }
    }
}

在这个函数中，先调用DownloadProgress下载所有需要下载的文件，然后更新本地配置文件，最后，如果发现某些更新文件需要重新启动应用程序的话，会提示用户重新启动程序。

至此，AutoUpdater这个类的使命就完成了，其实，整个的升级过程也就完成了。（废话）。

最后，我们来看一下这个组件是如何下载更新文件的

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Text;
using System.Windows.Forms;
using System.Threading;
using System.Net;
using System.IO;
using System.Diagnostics;

namespace Iyond.Utility
{
    public partial class DownloadProgress : Form
    {
        private bool isFinished = false;
        private List<DownloadFileInfo> downloadFileList = null;
        private ManualResetEvent evtDownload = null;
        private ManualResetEvent evtPerDonwload = null;
        private WebClient clientDownload = null;

        public DownloadProgress(List<DownloadFileInfo> downloadFileList)
        {
            InitializeComponent();

            this.downloadFileList = downloadFileList;
        }

        private void OnFormClosing(object sender, FormClosingEventArgs e)
        {
            if (!isFinished && DialogResult.No == MessageBox.Show("当前正在更新，是否取消？", "自动升级", MessageBoxButtons.YesNo, MessageBoxIcon.Question))
            {
                e.Cancel = true;
                return;
            }
            else
            {
                if (clientDownload != null)
                    clientDownload.CancelAsync();

                evtDownload.Set();
                evtPerDonwload.Set();
            }
        }

        private void OnFormLoad(object sender, EventArgs e)
        {
            evtDownload = new ManualResetEvent(true);
            evtDownload.Reset();
            Thread t = new Thread(new ThreadStart(ProcDownload));
            t.Name = "download";
            t.Start();
        }

        long total = 0;
        long nDownloadedTotal = 0;

        private void ProcDownload()
        {
            evtPerDonwload = new ManualResetEvent(false);

            foreach (DownloadFileInfo file in this.downloadFileList)
            {
                total += file.Size;
            }

            while (!evtDownload.WaitOne(0, false))
            {
                if (this.downloadFileList.Count == 0)
                    break;

                DownloadFileInfo file = this.downloadFileList[0];

                //Debug.WriteLine(String.Format("Start Download:{0}", file.FileName));

                this.ShowCurrentDownloadFileName(file.FileName);

                //下载
                clientDownload = new WebClient();

                clientDownload.DownloadProgressChanged += new DownloadProgressChangedEventHandler(OnDownloadProgressChanged);
                clientDownload.DownloadFileCompleted += new AsyncCompletedEventHandler(OnDownloadFileCompleted);

                evtPerDonwload.Reset();

                clientDownload.DownloadFileAsync(new Uri(file.DownloadUrl), Path.Combine(AppDomain.CurrentDomain.BaseDirectory, file.FileFullName + ".tmp"), file);

                //等待下载完成
                evtPerDonwload.WaitOne();

                clientDownload.Dispose();
                clientDownload = null;

                //移除已下载的文件
                this.downloadFileList.Remove(file);
            }

            //Debug.WriteLine("All Downloaded");

            if (this.downloadFileList.Count == 0)
                Exit(true);
            else
                Exit(false);

            evtDownload.Set();
        }

        void OnDownloadFileCompleted(object sender, AsyncCompletedEventArgs e)
        {
            DownloadFileInfo file = e.UserState as DownloadFileInfo;
            nDownloadedTotal += file.Size;
            this.SetProcessBar(0, (int)(nDownloadedTotal * 100 / total));
            //Debug.WriteLine(String.Format("Finish Download:{0}", file.FileName));
            //替换现有文件
            string filePath = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, file.FileFullName);
            if (File.Exists(filePath))
            {
                if (File.Exists(filePath + ".old"))
                    File.Delete(filePath + ".old");

                File.Move(filePath, filePath + ".old");
            }

            File.Move(filePath + ".tmp", filePath);
            //继续下载其它文件
            evtPerDonwload.Set();
        }

        void OnDownloadProgressChanged(object sender, DownloadProgressChangedEventArgs e)
        {
            this.SetProcessBar(e.ProgressPercentage, (int)((nDownloadedTotal + e.BytesReceived) * 100 / total));
        }

        delegate void ShowCurrentDownloadFileNameCallBack(string name);
        private void ShowCurrentDownloadFileName(string name)
        {
            if (this.labelCurrentItem.InvokeRequired)
            {
                ShowCurrentDownloadFileNameCallBack cb = new ShowCurrentDownloadFileNameCallBack(ShowCurrentDownloadFileName);
                this.Invoke(cb, new object[] { name });
            }
            else
            {
                this.labelCurrentItem.Text = name;
            }
        }

        delegate void SetProcessBarCallBack(int current, int total);
        private void SetProcessBar(int current, int total)
        {
            if (this.progressBarCurrent.InvokeRequired)
            {
                SetProcessBarCallBack cb = new SetProcessBarCallBack(SetProcessBar);
                this.Invoke(cb, new object[] { current, total });
            }
            else
            {
                this.progressBarCurrent.Value = current;
                this.progressBarTotal.Value = total;
            }
        }

        delegate void ExitCallBack(bool success);
        private void Exit(bool success)
        {
            if (this.InvokeRequired)
            {
                ExitCallBack cb = new ExitCallBack(Exit);
                this.Invoke(cb, new object[] { success });
            }
            else
            {
                this.isFinished = success;
                this.DialogResult = success ? DialogResult.OK : DialogResult.Cancel;
                this.Close();
            }
        }

        private void OnCancel(object sender, EventArgs e)
        {
            evtDownload.Set();
            evtPerDonwload.Set();
        }
    }
}

在构造函数中，将要下载的文件列表传进来

public DownloadProgress(List<DownloadFileInfo> downloadFileList)
{
    InitializeComponent();

    this.downloadFileList = downloadFileList;
}

在Form的Load事件中，启动下载线程，开始下载。

private void OnFormLoad(object sender, EventArgs e)
{
    evtDownload = new ManualResetEvent(true);
    evtDownload.Reset();
    Thread t = new Thread(new ThreadStart(ProcDownload));
    t.Name = "download";
    t.Start();
}

下载线程没什么特殊的，使用了WebClient的异步下载文件函数DownloadFileAsync，并且注册了两个事件，分别负责下载进度显示和下载完成后的处理：
clientDownload.DownloadProgressChanged += new DownloadProgressChangedEventHandler(OnDownloadProgressChanged);
clientDownload.DownloadFileCompleted += new AsyncCompletedEventHandler(OnDownloadFileCompleted);

大家看一下就明白了。

private void ProcDownload()
{
    evtPerDonwload = new ManualResetEvent(false);

    foreach (DownloadFileInfo file in this.downloadFileList)
    {
        total += file.Size;
    }

    while (!evtDownload.WaitOne(0, false))
    {
        if (this.downloadFileList.Count == 0)
            break;

        DownloadFileInfo file = this.downloadFileList[0];

        //Debug.WriteLine(String.Format("Start Download:{0}", file.FileName));

        this.ShowCurrentDownloadFileName(file.FileName);

        //下载
        clientDownload = new WebClient();

        clientDownload.DownloadProgressChanged += new DownloadProgressChangedEventHandler(OnDownloadProgressChanged);
        clientDownload.DownloadFileCompleted += new AsyncCompletedEventHandler(OnDownloadFileCompleted);

        evtPerDonwload.Reset();

        clientDownload.DownloadFileAsync(new Uri(file.DownloadUrl), Path.Combine(AppDomain.CurrentDomain.BaseDirectory, file.FileFullName + ".tmp"), file);

        //等待下载完成
        evtPerDonwload.WaitOne();

        clientDownload.Dispose();
        clientDownload = null;

        //移除已下载的文件
        this.downloadFileList.Remove(file);
    }

    //Debug.WriteLine("All Downloaded");

    if (this.downloadFileList.Count == 0)
        Exit(true);
    else
        Exit(false);

    evtDownload.Set();
}

最后，在OnDownloadFileCompleted函数中进行最后的处理。包括备份原文件，替换现有文件等。
void OnDownloadFileCompleted(object sender, AsyncCompletedEventArgs e)
{
    DownloadFileInfo file = e.UserState as DownloadFileInfo;
    nDownloadedTotal += file.Size;
    this.SetProcessBar(0, (int)(nDownloadedTotal * 100 / total));
    //Debug.WriteLine(String.Format("Finish Download:{0}", file.FileName));
    //替换现有文件
    string filePath = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, file.FileFullName);
    if (File.Exists(filePath))
    {
        if (File.Exists(filePath + ".old"))
            File.Delete(filePath + ".old");

        File.Move(filePath, filePath + ".old");
    }

    File.Move(filePath + ".tmp", filePath);
    //继续下载其它文件
    evtPerDonwload.Set();
}

其它的函数只是一些显示进度条和下载信息的，这里就不再详细介绍了。