Golang windows下程序自动更新
// autoupdate
package autoupdate
import (
"errors"
"fmt"
"io"
"io/ioutil"
"math/rand"
"net/http"
"os"
"os/exec"
"path/filepath"
"strings"
"time"
)
type AutoUpdate struct {
Url string //下载路径
Softname string //软件名
CurrVer string //当前版本
CurrName string //当前运行路径及文件名
}
func (au *AutoUpdate) Update() error {
if strings.HasSuffix(au.CurrName, "update.exe") { //开始更新
time.Sleep(3 * time.Second)
if au.copyFile() {
cmd := exec.Command("cmd", "/c", "start", filepath.Dir(au.CurrName)+"\\"+au.Softname+".exe")
cmd.Start()
cmd.Wait()
os.Exit(0)
}
} else { //检测更新
resp, err := http.Get(au.Url + "/" + au.Softname + ".txt?num=" + fmt.Sprintf("%d", rand.Intn(1000)))
if err != nil {
return err
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return err
}
newVer := string(body)
// fmt.Println("远端版本号:", newVer)
// fmt.Println("本地版本号:", au.CurrVer)
if newVer != au.CurrVer {
if err := au.getNewVer(); err != nil {
return err
} else {
cmd := exec.Command("cmd", "/c", "start", filepath.Dir(au.CurrName)+"\\update.exe")
cmd.Start()
cmd.Wait()
os.Exit(0)
}
} else {
os.Remove("update.exe")
}
}
return nil
}
func (au *AutoUpdate) getNewVer() error {
client := http.Client{Timeout: 900 * time.Second}
resp, err := client.Get(au.Url + "/" + au.Softname + ".exe?num=" + fmt.Sprintf("%d", rand.Intn(1000)))
if err != nil {
return err
}
defer resp.Body.Close()
if resp.Status == "200 OK" {
newFile, err := os.Create("update.exe")
if err != nil {
return err
}
defer newFile.Close()
_, err = io.Copy(newFile, resp.Body)
if err != nil {
return err
}
return nil
} else {
return errors.New(resp.Status)
}
}
func (au *AutoUpdate) copyFile() bool {
source_open, err := os.Open(au.CurrName)
if err != nil {
return false
}
defer source_open.Close()
dest_open, err := os.OpenFile(filepath.Dir(au.CurrName)+"\\"+au.Softname+".exe", os.O_CREATE|os.O_WRONLY, 644)
if err != nil {
return false
}
defer dest_open.Close()
//进行数据拷贝
_, copy_err := io.Copy(dest_open, source_open)
if copy_err != nil {
return false
} else {
return true
}
}