Skip to content

energy
energy

energye / energy

通知 13
Star 2
Fork 0
energy
energy

前往新版Gitcode，体验更适合开发者的 AI 搜索 >>

提交 aedd3a09 编写于 作者: yanghye's avatar yanghye
浏览文件
操作

remove: demo lib-checkupdate

上级 2dbbca71
隐藏空白更改
内联 并排
Showing with 0 addition and 816 deletion
cmd/autoupdate/README.md 已删除 100644 → 0
浏览文件 @ 2dbbca71
### energy lib framework auto update
#### Divided into three modules
* liblcl
* enregy
* cef
#### Control whether to enable the module update based on enable
#### Field description
> first level node is the module name
>
> download: module download source
>
> url: module download source url template
>
> source: module download source, addrs
>
> sourceSelect: Download the address source selection, select the source based on the index, and replace it with the Url template
>
> enable: module update based on enable
>
> latest: module latest version
>
> versions: version list
>
>> forced: forced update, a significant update that applies to the current version
>>
>> content: update logs
---
#### update json template
```json
{
"liblcl": {
"download": {
"url": "https://{url}/energye/energy/releases/download/{version}/{OSARCH}.zip",
"source": [
"gitee.com",
"github.com"
],
"sourceSelect": 0
},
"enable": true,
"latest": "version.1",
"versions": {
"version.1": {
"energyVersion": "version.1",
"content": [
"update log 1",
"update log 2"
],
"forced": false
},
"version.0": {
"energyVersion": "",
"content": [
""
],
"forcede": false
}
}
},
"energy": {
"download": {
"url": "",
"source": [],
"sourceSelect": 0
},
"enable": false,
"latest": "",
"versions": {
"": {
"content": [
""
],
"forcede": false
}
}
},
"cef": {
"download": {
"url": "",
"source": [
""
],
"sourceSelect": 0
},
"enable": false,
"latest": "",
"versions": {
"": {
"content": [
""
],
"forcede": false
}
}
}
}
```
---
\ No newline at end of file
cmd/autoupdate/autoupdate.go 已删除 100644 → 0
浏览文件 @ 2dbbca71
//----------------------------------------
//
// Copyright © yanghy. All Rights Reserved.
//
// Licensed under Apache License Version 2.0, January 2004
//
// https://www.apache.org/licenses/LICENSE-2.0
//
//----------------------------------------
// Package autoupdate Energy lib-lcl check auto update
package autoupdate
import (
"crypto/tls"
"encoding/json"
"fmt"
"github.com/energye/energy/v2/cmd/internal"
"github.com/energye/energy/v2/cmd/internal/consts"
"github.com/energye/liblclbinres"
"io/ioutil"
"net/http"
"strconv"
"strings"
"time"
)
type LiblclCallback func(model *Model, level int, canUpdate bool)
var (
// 检查更新开关, 默认关闭
isCheckUpdate = false
CanUpdateLiblcl LiblclCallback // 参数 model: 更新模块, Level: 更新版本级别, canUpdate: 是否有更新
)
// Model 模块
type Model struct {
CurrentVersion string `json:"-"` // 当前版本
Latest string `json:"latest"` // 最新版本
Versions map[string]VersionInfo `json:"versions"` // 当前模块所有版本集合 key=版本, value=版本信息
}
// 版本信息
type VersionInfo struct {
Content string `json:"content"` // 更新内容
DownloadSource string `json:"downloadSource"` // 下载源 逗号分隔, 数组 ["gitee.com", "github.com"]
DownloadSourceSelect int `json:"downloadSourceSelect"` // 下载源 选择
DownloadUrl string `json:"downloadUrl"` // 下载地址, https://{url}/energye/energy/releases/download/{version}/{OSARCH}.zip
Module string `json:"module"` // 模块名
BuildSupportOSArch string `json:"supportOSArch"` // 已提供构建支持的系统架构
Version string `json:"version"` // 版本
}
// CheckUpdate
// Check for updates, when isCheckUpdate is true
func CheckUpdate() {
if isCheckUpdate {
check()
}
}
// IsCheckUpdate
// Set whether to check for updates
func IsCheckUpdate(v bool) {
isCheckUpdate = v
}
func check() {
client := &http.Client{Timeout: 5 * time.Second, Transport: &http.Transport{
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
}}
request, err := http.NewRequest("GET", consts.CheckUpgradeURL, nil)
if err != nil {
println("energy check update http.NewRequest error:", err.Error())
return
}
response, err := client.Do(request)
if err != nil {
println("energy check update client.Do error:", err.Error())
return
}
defer response.Body.Close()
if data, err := ioutil.ReadAll(response.Body); err == nil {
var v map[string]*Model
if err = json.Unmarshal(data, &v); err == nil {
liblcl(v["liblcl"])
} else {
println("energy check update json.Unmarshal error", err.Error())
}
} else {
println("energy check update ioutil.ReadAl(response.Body) error", err.Error())
}
}
// liblcl Model
func liblcl(model *Model) {
currentLib := liblclbinres.LibVersion()
originLib := model.Latest
if currentLib == "" {
currentLib = "0.0.0"
}
if originLib == "" {
originLib = "0.0.0"
}
model.CurrentVersion = currentLib
can, level := compare(currentLib, originLib)
if CanUpdateLiblcl != nil {
CanUpdateLiblcl(model, level, can)
}
}
func LibLCLName(version, buildSupportOSArch string) (string, bool) {
return internal.LibLCLName(version, buildSupportOSArch)
}
// 命名规则 OS+[ARCH]+BIT+[GTK2]
// ARCH: 非必需, ARM 时填写, AMD为空
// GTK2: 非必需, GTK2(Linux CEF 106) 时填写, 非Linux或GTK3时为空
func liblclOS(version, buildSupportOSArch string) (string, bool) {
archs := strings.Split(buildSupportOSArch, ",")
var goarch string
if c.Install.OS.IsWindows() && c.Install.Arch.Is386() {
goarch = "32" // windows32 = > windows386
} else {
goarch = string(c.Install.Arch)
}
noSuport := fmt.Sprintf("%v %v", c.Install.OS, goarch)
var isSupport = func(goarch string) bool {
for _, v := range archs {
if goarch == v {
return true
}
}
return false
}
if name, isOld := liblclName(c, version, cef); isOld {
if name == "" {
return noSuport, false
}
return name, true
} else {
return name, isSupport(name)
}
}
// Version comparison, returns true if the current version is smaller than the remote version
func compare(current, origin string) (bool, int) {
cmajor, cminor, crevision := versionConvert(current)
omajor, ominor, orevision := versionConvert(origin)
if omajor > cmajor {
return true, 1 // major
} else if ominor > cminor {
return true, 2 // minor
} else if orevision > crevision {
return true, 3 // revision
}
return false, 0
}
// Version number conversion=>Major version, minor version, revised version
func versionConvert(ver string) (major, minor, revision int) {
ver = strings.ToLower(ver)
lastv := strings.LastIndex(ver, "v")
if lastv != -1 {
ver = ver[lastv+1:]
}
vers := strings.Split(strings.Split(ver, "-")[0], ".")
if len(vers) >= 3 {
major, _ = strconv.Atoi(vers[0])
minor, _ = strconv.Atoi(vers[1])
revision, _ = strconv.Atoi(vers[2])
} else if len(vers) >= 2 {
major, _ = strconv.Atoi(vers[0])
minor, _ = strconv.Atoi(vers[1])
} else if len(vers) >= 1 {
major, _ = strconv.Atoi(vers[0])
}
return
}
example/lib-checkupdate/README.md 已删除 100644 → 0
浏览文件 @ 2dbbca71
### Energy liblcl auto update example
#### Demonstrating the liblcl dynamic link library module
* liblcl
#### Using the `autoupdate` tool function in the `cmd` package, check update.
#### This example demonstrates the use of UI window design
#### Reference Example Customization Update
### Check update steps
> 1. We copy a dynamic library and customize the import of dynamic link library functions to avoid conflicts with the currently updated ones
> 2. Injecting dynamic library proc by oneself `imports.SetEnergyImportDefs(version)`
> 3. initialize `inits.Init(nil, &resources)`
> 4. i18n config
> 5. open autoupdate `autoupdate.IsCheckUpdate(true)`
> 6. Set the callback function, and if there is an update, the function will be executed `autoupdate.CanUpdateLiblcl`
>> 6.1 `CanUpdateLiblcl` Obtain information on the updated module and the updated version level
>
>> 6.2. Update processing here, coding...
> 7. call `autoupdate.CheckUpdate()` Check for updates
\ No newline at end of file
example/lib-checkupdate/README.zh_CN.md 已删除 100644 → 0
浏览文件 @ 2dbbca71
### Energy liblcl 自动更新示例
#### 演示liblcl动态链接库模块
* liblcl
#### 使用`cmd`包中的`autoupdate`工具功能，检查更新。
#### 此示例使用了UI窗口设计
#### 参考该示例自定义更新, 或直接使用
### 检查更新步骤
> 1. 我们复制了一个动态库，并自定义了动态链接库函数的导入，以避免与当前更新的函数发生冲突
> 2. 自行导入动态库proc `imports.SetEnergyImportDefs(version)`
> 3. 初始化 `inits.Init(nil, &resources)`
> 4. i18n 配置
> 5. 打开自动更新 `autoupdate.IsCheckUpdate(true)`
> 6. 设置回调函数，如果有更新，函数将被执行 `autoupdate.CanUpdateLiblcl`
>> 6.1 `CanUpdateLiblcl` 回调函数返回更新模块信息和更新版本级别
>
>> 6.2. 在这里编写更新代码, coding...
> 7. 调用 `autoupdate.CheckUpdate()` 检查更新
\ No newline at end of file
example/lib-checkupdate/form/form.go 已删除 100644 → 0
浏览文件 @ 2dbbca71
//----------------------------------------
//
// Copyright © yanghy. All Rights Reserved.
//
// Licensed under Apache License Version 2.0, January 2004
//
// https://www.apache.org/licenses/LICENSE-2.0
//
//----------------------------------------
package form
import (
"github.com/energye/golcl/lcl"
"github.com/energye/golcl/lcl/types"
"github.com/energye/golcl/lcl/types/colors"
)
type callback func(m *UpdateForm)
var OnAfter func()
var OnCreate callback
type UpdateForm struct {
*lcl.TForm
TitlePanel *lcl.TPanel
UpdatePromptPanel *lcl.TPanel
UpdateContentMemo *lcl.TMemo
}
func (m *UpdateForm) OnFormCreate(sender lcl.IObject) {
var after = true
m.SetOnActivate(func(sender lcl.IObject) {
if OnAfter != nil && after {
after = false
OnAfter()
}
})
if OnCreate != nil {
OnCreate(m)
}
}
func (m *UpdateForm) NewPanel() *lcl.TPanel {
var result = lcl.NewPanel(m)
result.SetParent(m)
result.SetBevelInner(types.BvNone)
result.SetBevelOuter(types.BvNone)
result.SetBorderStyle(types.BsNone)
result.SetColor(colors.ClWhite)
return result
}
example/lib-checkupdate/main.go 已删除 100644 → 0
浏览文件 @ 2dbbca71
// auto update application
package main
import (
"archive/zip"
"embed"
"github.com/energye/energy/v2/cef/i18n"
"github.com/energye/energy/v2/cmd/autoupdate"
"github.com/energye/energy/v2/common/imports"
"github.com/energye/energy/v2/consts"
"github.com/energye/energy/v2/example/lib-checkupdate/form"
"github.com/energye/golcl/energy/inits"
"github.com/energye/golcl/energy/tools"
"github.com/energye/golcl/lcl"
"github.com/energye/golcl/lcl/api/dllimports"
"github.com/energye/golcl/lcl/types"
"github.com/energye/golcl/lcl/types/colors"
"github.com/energye/golcl/pkgs/libname"
"io"
"io/fs"
"net/http"
"os"
"path/filepath"
"runtime"
"strconv"
"strings"
)
/*
检测liblcl库的更新
下面的代码演示是未通过cef库导入的liblcl,而是单独导入,这样可以减少执行文件的大小
[]*dllimports.ImportTable
通过imports.SetEnergyImportDefs(version)将proc导入到执行文件中, 为保持和cef的proc下标一致,第0个下标为空导入
*/
var (
//go:embed resources
resources embed.FS
// form
updateForm *form.UpdateForm
version = []*dllimports.ImportTable{
dllimports.NewEnergyImport("Interface_CustomWidgetSetInitialization", 0), //linux lcl widget init
}
)
func main() {
// 针对 liblcl
energyPath := filepath.Join(os.TempDir(), "energy")
if !tools.IsExist(energyPath) {
os.Mkdir(energyPath, fs.ModePerm)
}
var (
libPath = libname.LibPath()
dstLibPath = filepath.Join(energyPath, libname.GetDLLName())
)
if tools.IsExist(dstLibPath) {
os.Remove(dstLibPath)
}
// 在初始化之前 先把 liblcl 复制一份到临时目录中
// 然后我们加载临时目录中的 liblcl
// 之后升级时使用升级新文件替换实际的 liblcl
// 可以在程度内直接替换
src, err := os.Open(libPath)
if err != nil {
panic(err)
}
dst, err := os.OpenFile(dstLibPath, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644)
if err != nil {
panic(err)
}
// 复制一份
_, err = io.Copy(dst, src)
if err != nil {
panic(err)
}
dst.Close()
src.Close()
libname.LibName = dstLibPath // 设置临时目录的 liblcl
defer os.Remove(dstLibPath)
// proc 注入到 imports
imports.SetEnergyImportDefs(version)
// 初始化golcl
inits.Init(nil, &resources)
// inits.Init 完成之后，初始化LCL控件
imports.Proc(0).Call()
// 注入资源
i18n.SetLocalFS(&resources, "resources")
//i18n.Switch(consts.LANGUAGE_en_US)
i18n.Switch(consts.LANGUAGE_zh_CN)
// 开启自动更新检查
autoupdate.IsCheckUpdate(true)
// 如果 liblcl 有更新该函数将被回调,并创建更新窗口
autoupdate.CanUpdateLiblcl = func(model *autoupdate.Model, level int, canUpdate bool) {
updateVersion := model.Versions[model.Latest]
// 这里使用窗口形式展示更新
// 运行应用后窗口创建时回调
// 通过代码设计窗口上的UI组件
form.OnCreate = func(m *form.UpdateForm) {
// 应用图标
lcl.Application.Icon().LoadFromFSFile("resources/icon.ico")
// 窗口一些属性配置
m.SetDoubleBuffered(true)
m.EnabledMinimize(false)
m.EnabledMaximize(false)
m.SetShowHint(true)
//m.SetFormStyle(types.FsSystemStayOnTop)
m.SetBorderStyle(types.BsSingle)
//m.SetBorderStyle(types.BsNone)
//m.SetShowInTaskBar(types.StNever)
m.SetColor(colors.ClWhite)
m.SetWidth(590)
m.SetHeight(390)
m.SetCaption(i18n.Resource("title")) // 自定义窗口标题
var titleHeight int32 = 0
if runtime.GOOS == "windows" {
m.SetBorderStyle(types.BsNone)
titleHeight = 32
// 自定义窗口标题栏
m.TitlePanel = m.NewPanel()
m.TitlePanel.SetColor(colors.ClTeal)
m.TitlePanel.SetHeight(titleHeight) // icon 的高
m.TitlePanel.SetWidth(m.Width())
// 模拟标题栏移动窗口
var (
isDown bool
dx, dy int32
)
m.TitlePanel.SetOnMouseDown(func(sender lcl.IObject, button types.TMouseButton, shift types.TShiftState, x, y int32) {
isDown = true
dx, dy = x, y
})
m.TitlePanel.SetOnMouseUp(func(sender lcl.IObject, button types.TMouseButton, shift types.TShiftState, x, y int32) {
isDown = false
})
m.TitlePanel.SetOnMouseMove(func(sender lcl.IObject, shift types.TShiftState, x, y int32) {
if isDown { //鼠标按下时计算移动坐标
m.SetLeft(m.Left() - (dx - x))
m.SetTop(m.Top() - (dy - y))
}
})
// title -> icon
titleIcon := lcl.NewImage(m.TitlePanel)
titleIcon.SetParent(m.TitlePanel)
titleIcon.SetWidth(32) // icon 的宽
titleIcon.SetHeight(32) // icon 的高
titleIcon.Picture().LoadFromFSFile("resources/icon.png")
// title -> text
titleText := lcl.NewLabel(m.TitlePanel)
titleText.SetParent(m.TitlePanel)
titleText.SetTop(3)
titleText.SetLeft(40)
titleText.Font().SetSize(12)
titleText.Font().SetColor(colors.ClWhite)
titleText.SetCaption(m.Caption())
// title -> close button
titleClose := lcl.NewImageButton(m.TitlePanel)
titleClose.SetParent(m.TitlePanel)
titleClose.SetImageCount(4)
titleClose.SetAutoSize(true)
titleClose.SetCursor(types.CrHandPoint)
titleClose.Picture().LoadFromFSFile("resources/btn_close.png")
titleClose.SetLeft(m.Width() - 40) //关闭按钮位置 left = 窗口宽 - 按钮图片宽
titleClose.SetHint(i18n.Resource("close"))
titleClose.SetOnClick(func(lcl.IObject) {
m.Close() // 关闭窗口
})
}
// background
bgImage := lcl.NewImage(m)
bgImage.SetParent(m)
bgImage.SetTop(titleHeight + 10)
bgImage.SetWidth(271) // 图片宽
bgImage.SetHeight(60) // 图片高
bgImage.SetLeft((m.Width() - bgImage.Width()) / 2) // 设置以窗口居中
bgImage.Picture().LoadFromFSFile("resources/bg.png")
var updatePanelHeight int32 = 200
// 更新提醒 panel
m.UpdatePromptPanel = m.NewPanel()
m.UpdatePromptPanel.SetTop(bgImage.Top() + bgImage.Height() + 20)
m.UpdatePromptPanel.SetWidth(m.Width())
m.UpdatePromptPanel.SetHeight(updatePanelHeight)
// 更新内容
m.UpdateContentMemo = lcl.NewMemo(m.UpdatePromptPanel)
m.UpdateContentMemo.SetParent(m.UpdatePromptPanel)
ucw := m.Width() / 4
m.UpdateContentMemo.SetWidth(ucw * 3)
m.UpdateContentMemo.SetLeft((m.Width() - m.UpdateContentMemo.Width()) / 2)
m.UpdateContentMemo.SetHeight(updatePanelHeight)
m.UpdateContentMemo.SetReadOnly(true)
m.UpdateContentMemo.SetColor(colors.ClWhite)
if !canUpdate {
m.UpdateContentMemo.Lines().Add(i18n.Resource("currentVersion") + ": " + model.CurrentVersion)
m.UpdateContentMemo.Lines().Add(i18n.Resource("latestVersion") + ": " + model.Latest)
m.UpdateContentMemo.SetEnabled(false)
} else { // 有更新
m.UpdateContentMemo.SetScrollBars(types.SsAutoBoth)
m.UpdateContentMemo.Lines().Add(i18n.Resource("updateContent") + " " + model.Latest)
m.UpdateContentMemo.Lines().Add(updateVersion.Content)
// liblcl 下载版本URL
liblclZipName, _ := autoupdate.LibLCLName(model.Latest, updateVersion.BuildSupportOSArch)
downSource := strings.Split(updateVersion.DownloadSource, ",")
downUrl := strings.Replace(updateVersion.DownloadUrl, "{source}", downSource[updateVersion.DownloadSourceSelect], -1) // 使用配置的下载源
downUrl = strings.Replace(downUrl, "{version}", updateVersion.Version, -1) // liblcl 所属的 enregy 版本
downUrl = strings.Replace(downUrl, "{module}", updateVersion.Module, -1) // 模块名
downUrl = strings.Replace(downUrl, "{OSARCH}", liblclZipName, -1) // 根据系统架构获取对应的文件名
m.UpdateContentMemo.Lines().Add("")
m.UpdateContentMemo.Lines().Add(i18n.Resource("downloadURL"))
m.UpdateContentMemo.Lines().Add(downUrl)
// 取消按钮
//cancelBtn := lcl.NewImageButton(m)
//cancelBtn.SetParent(m)
//cancelBtn.SetImageCount(3)
//cancelBtn.SetAutoSize(true)
//cancelBtn.SetCursor(types.CrHandPoint)
//cancelBtn.Picture().LoadFromFSFile("resources/btn-cancel.png")
//cancelBtn.SetLeft(300)
//cancelBtn.SetTop(290)
//cancelBtn.SetHint(i18n.Resource("cancel"))
//cancelBtn.SetOnClick(func(lcl.IObject) {
// //m.Close()
//})
// 下载
var isNotDownload = true
// 更新下载进度条
updateProgressBar := lcl.NewProgressBar(m)
updateProgressBar.SetParent(m)
updateProgressBar.SetTop(updatePanelHeight + m.UpdatePromptPanel.Top() + 5)
updateProgressBar.SetLeft(m.UpdateContentMemo.Left())
updateProgressBar.SetMin(0)
updateProgressBar.SetMax(100)
updateProgressBar.SetWidth(m.UpdateContentMemo.Width())
updateProgressBar.SetHeight(10)
updateProgressBar.SetVisible(false)
// 更新按钮
updateBtn := lcl.NewImageButton(m)
updateBtn.SetParent(m)
updateBtn.SetImageCount(3)
//updateBtn.SetAutoSize(true)
updateBtn.SetWidth(80)
updateBtn.SetHeight(50)
updateBtn.SetCursor(types.CrHandPoint)
updateBtn.Picture().LoadFromFSFile("resources/btn-update.png")
updateBtn.SetLeft(400)
updateBtn.SetTop(325)
updateBtn.SetHint(i18n.Resource("update"))
updateBtn.SetOnClick(func(lcl.IObject) {
if isNotDownload {
updateProgressBar.SetPosition(0)
updateProgressBar.SetVisible(isNotDownload)
isNotDownload = false
updateBtn.SetEnabled(isNotDownload)
var savePath, _ = filepath.Split(libPath)
var fileName, _ = autoupdate.LibLCLName(model.Latest, updateVersion.BuildSupportOSArch)
var saveFilePath = filepath.Join(savePath, fileName) + ".zip" // zip
m.UpdateContentMemo.Lines().Add("---------------------------------------------------------------")
m.UpdateContentMemo.Lines().Add(i18n.Resource("beginDownload"))
m.UpdateContentMemo.Lines().Add(" " + i18n.Resource("downloadURL") + ": " + downUrl)
m.UpdateContentMemo.Lines().Add(" " + i18n.Resource("savePath") + ": " + saveFilePath)
go func() {
// 下载地址, 保存目录, 回调更新进程函数
downloadFile(downUrl, saveFilePath, func(totalLength, processLength int64, err error) {
var process = int32((float64(processLength) / float64(totalLength)) * 100)
lcl.ThreadSync(func() {
if err != nil {
m.UpdateContentMemo.Lines().Add(i18n.Resource("downloadError") + ":\n " + err.Error())
m.UpdateContentMemo.Lines().Add(i18n.Resource("downloadAgain"))
} else {
updateProgressBar.SetPosition(process)
if process >= 100 {
m.UpdateContentMemo.Lines().Add(i18n.Resource("endDownload") + " -> 100")
}
}
})
})
m.UpdateContentMemo.Lines().Add("---------------------------------------------------------------")
updateProgressBar.SetPosition(0)
m.UpdateContentMemo.Lines().Add(i18n.Resource("extractUnZip") + ": " + filepath.Join(savePath, libname.GetDLLName()))
// 解压释放文件
extractUnZip(saveFilePath, savePath, func(totalLength, processLength int64, err error) {
var process = int32((float64(processLength) / float64(totalLength)) * 100)
lcl.ThreadSync(func() {
if err != nil {
m.UpdateContentMemo.Lines().Add(i18n.Resource("updateError") + ":\n " + err.Error())
m.UpdateContentMemo.Lines().Add(i18n.Resource("updateAgain"))
} else {
updateProgressBar.SetPosition(process)
if process >= 100 {
m.UpdateContentMemo.Lines().Add(i18n.Resource("extractUnZip") + ": success -> 100")
m.UpdateContentMemo.Lines().Add(i18n.Resource("updateSuccess"))
}
}
})
}, libname.GetDLLName())
isNotDownload = true
updateBtn.SetEnabled(isNotDownload)
}()
}
})
}
m.ScreenCenter()
}
// run and create update form
lcl.RunApp(&updateForm)
}
// 检查 liblcl 库是否有更新
autoupdate.CheckUpdate()
}
// 下载文件
func downloadFile(url string, localPath string, callback func(totalLength, processLength int64, err error)) error {
var (
fsize int64
buf = make([]byte, 1024*10)
written int64
)
tmpFilePath := localPath + ".download"
client := new(http.Client)
resp, err := client.Get(url)
if err != nil {
callback(0, 0, err)
return err
}
fsize, err = strconv.ParseInt(resp.Header.Get("Content-Length"), 10, 32)
if err != nil {
callback(0, 0, err)
return err
}
println("Save path: [", localPath, "] file size:", fsize)
file, err := os.Create(tmpFilePath)
if err != nil {
callback(0, 0, err)
return err
}
defer file.Close()
if resp.Body == nil {
callback(0, 0, err)
return nil
}
defer resp.Body.Close()
for {
nr, er := resp.Body.Read(buf)
if nr > 0 {
nw, err := file.Write(buf[0:nr])
if nw > 0 {
written += int64(nw)
}
callback(fsize, written, err)
if err != nil {
break
}
if nr != nw {
err = io.ErrShortWrite
break
}
}
if er != nil {
if er != io.EOF {
err = er
}
break
}
}
if err == nil {
file.Sync()
file.Close()
err = os.Rename(tmpFilePath, localPath)
if err != nil {
return err
}
}
return err
}
func writeFile(r io.Reader, w *os.File, totalLength int64, callback func(totalLength, processLength int64, err error)) {
buf := make([]byte, 1024*10)
var written int64
for {
nr, err := r.Read(buf)
if nr > 0 {
nw, err := w.Write(buf[0:nr])
if nw > 0 {
written += int64(nw)
}
callback(totalLength, written, err)
if err != nil {
break
}
if nr != nw {
err = io.ErrShortWrite
break
}
}
if err != nil {
break
}
}
}
func extractUnZip(filePath, targetPath string, callback func(totalLength, processLength int64, err error), files ...interface{}) {
if rc, err := zip.OpenReader(filePath); err == nil {
defer rc.Close()
for i := 0; i < len(files); i++ {
if f, err := rc.Open(files[i].(string)); err == nil {
defer f.Close()
st, _ := f.Stat()
targetFileName := filepath.Join(targetPath, st.Name())
if st.IsDir() {
os.MkdirAll(targetFileName, st.Mode())
continue
}
if targetFile, err := os.Create(targetFileName); err == nil {
writeFile(f, targetFile, st.Size(), callback)
targetFile.Close()
} else {
callback(0, 0, err)
}
} else {
callback(0, 0, err)
return
}
}
} else {
if err != nil {
callback(0, 0, err)
}
}
}
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或