56 lines
1.1 KiB
Go
56 lines
1.1 KiB
Go
package fileUtil
|
||
|
||
import (
|
||
"io"
|
||
"net/http"
|
||
"os"
|
||
"path"
|
||
)
|
||
|
||
// 下载网络文件
|
||
// netUrl：网络文件地址
|
||
// saveDir：存储位置
|
||
// saveFileName:存储的文件名
|
||
// ifTruncate:如果文件存在了，是否覆盖此文件
|
||
// 返回值：
|
||
// err:错误对象
|
||
func DownLoadNetFile(netUrl string, saveDir string, saveFileName string, ifTruncate bool) (err error) {
|
||
resp, err := http.Get(netUrl)
|
||
defer func() {
|
||
if resp != nil {
|
||
resp.Body.Close()
|
||
}
|
||
}()
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
// 创建文件夹
|
||
if IsDirExists(saveDir) == false {
|
||
os.MkdirAll(saveDir, os.ModePerm|os.ModeTemporary)
|
||
}
|
||
|
||
// 创建文件
|
||
filePath := path.Join(saveDir, saveFileName)
|
||
var fileObj *os.File
|
||
if ifTruncate {
|
||
fileObj, err = os.OpenFile(filePath, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, os.ModePerm|os.ModeTemporary)
|
||
} else {
|
||
// 如果文件已经存在，则不能打开
|
||
fileObj, err = os.OpenFile(filePath, os.O_CREATE|os.O_WRONLY|os.O_EXCL, os.ModePerm|os.ModeTemporary)
|
||
}
|
||
defer func() {
|
||
if fileObj != nil {
|
||
fileObj.Close()
|
||
}
|
||
}()
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
// 写入文件数据
|
||
_, err = io.Copy(fileObj, resp.Body)
|
||
|
||
return
|
||
}
|