mirror of https://github.com/mindoc-org/mindoc.git
46 lines
940 B
Go
46 lines
940 B
Go
package requests
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"errors"
|
|
"fmt"
|
|
)
|
|
|
|
//下载远程文件并保存到指定位置
|
|
func DownloadAndSaveFile(remoteUrl, dstFile string) (error) {
|
|
client := &http.Client{}
|
|
uri, err := url.Parse(remoteUrl)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// Create the file
|
|
out, err := os.Create(dstFile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer out.Close()
|
|
|
|
request, err := http.NewRequest("GET", uri.String(), nil)
|
|
request.Header.Add("Connection", "close")
|
|
request.Header.Add("Host", uri.Host)
|
|
request.Header.Add("Referer", uri.String())
|
|
request.Header.Add("User-Agent", "Mozilla/5.0 (Windows NT 6.1; WOW64; rv:12.0) Gecko/20100101 Firefox/12.0")
|
|
|
|
resp, err := client.Do(request)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
|
|
if resp.StatusCode == http.StatusOK {
|
|
_, err = io.Copy(out, resp.Body)
|
|
}else{
|
|
return errors.New(fmt.Sprintf("bad status: %s", resp.Status))
|
|
}
|
|
return nil
|
|
}
|