apps/kubevpn
1
0
Fork 0
mirror of https://github.com/kubenetworks/kubevpn.git synced 2025-12-24 11:51:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: close temp file before rename it

Browse Source
This commit is contained in:
fengcaiwen
2023-01-12 12:37:29 +08:00
parent dfed43c6e9
commit 6f30e2361d
1 changed files with 20 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
pkg/upgrade/upgrade.go
View File

@@ -40,6 +40,10 @@ func Main(current string, client *http.Client) error {
if err != nil {
return err
}
err = temp.Close()
if err != nil {
return err
}
err = download(client, url, temp.Name())
if err != nil {
return err
@@ -53,9 +57,23 @@ func Main(current string, client *http.Client) error {
if err != nil {
return err
}
createTemp, err := os.CreateTemp("", "")
var createTemp *os.File
createTemp, err = os.CreateTemp("", "")
if err != nil {
return err
}
err = createTemp.Close()
if err != nil {
return err
}
err = os.Remove(createTemp.Name())
if err != nil {
return err
}
err = os.Rename(curFolder, createTemp.Name())
if err != nil {
return err
}
err = os.Rename(temp.Name(), curFolder)
return err
}
@@ -134,6 +152,7 @@ func download(client *http.Client, url string, filename string) error {
if err != nil {
return err
}
defer f.Close()
buf := make([]byte, 10<<(10*2)) // 10M
_, err = io.CopyBuffer(f, p, buf)
return err