go语言下载文件

package main

import (
	//"crypto/hmac"
	//"crypto/sha1"
	//"encoding/base64"
	"fmt"
	"io"
	"net/http"
	"os"
	"strings"
	"time"
)

//此函数为将用户名密码通过指定方式生成鉴权信息:
//func generateSignature(userStringToSign, userAK, userSK string) (res string) {
//	key := []byte(userSK)
//	mac := hmac.New(sha1.New, key)
//	mac.Write([]byte(userStringToSign))
//	signature := base64.StdEncoding.EncodeToString(mac.Sum(nil))
//	res = "AUTH "+ userAK + ":" + signature
//	return
//}

func DownloadFile(url, path, ak, sk string) (pathRes, downloadRes bool) {
	var res *http.Response
	tt := strings.Split(path, "/")
	pathDir := strings.TrimSuffix(path, tt[len(tt)-1])
	pathDir = strings.TrimSuffix(pathDir, "/")
	err := os.MkdirAll(pathDir, 0777)
	if err != nil {
		fmt.Printf("mkdir file failed:[%s], error:[%v], path:[%s]\n", pathDir, err, path)
		return false, false
	} else {
		fmt.Printf("mkdir file success:[%s], error:[%v], path:[%s]\n", pathDir, err, path)
	}
	buffer, err := os.Create(path)
	if err != nil {
		fmt.Printf("open file failed, error:%v, path:[%s]\n", err, path)
		return false, false
	} else {
		fmt.Printf("open file success, error:%v, path:[%s]\n", err, path)
	}
	defer buffer.Close()
	//flag 0:work; 1:fail; 2:over
	flag := 0
	for i := 0; i < 3; i++ {
		for {
			//res, err = http.Get(url)
			client := &http.Client{}
			req, err2 := http.NewRequest("GET", url, nil)
			if err2 != nil {
				fmt.Printf("make new http request [%s] error: [%v]\n", url, err2)
				return true, false
			}
			currentDate := time.Now().Format("Mon, 02 Jan 2006 15:04:05 GMT")

			 (可选)如需鉴权操作，则通过以下方式生成Authorization所需内容
			//othersInfo := "xxx"
			生成鉴权信息时所需的信息字符串拼接，此处根据具体业务需求更改:
			//stringToSign := "GET" + "\n" + "" + "\n" + "" + "\n" + currentDate + "\n" + "/" + othersInfo
			需要鉴权则用这个生成鉴权信息，然后放到请求头中:
			//auth := generateSignature(stringToSign, ak, sk)
			//fmt.Printf("Authorization:[%s]\n",auth)

			req.Header.Add("Date",currentDate)
			//req.Header.Add("Authorization", auth)
			//req.Header.Add("Content-Type","video/mp4")
			res, err2 = client.Do(req)
			if err2 != nil {
				fmt.Printf("send http req [%s] error: [%v]\n", url, err2)
				return true, false
			}

			if res.StatusCode != 200 {
				fmt.Printf("download failed:[%s], StatusCode:[%d],url:[%s]\n", path, res.StatusCode, url)
				res.Body.Close()
				if i >= 2 {
					return true, false
				} else {
					flag = 1
					break
				}
			}
			flag = 0
			break
		}
		if flag == 1 {
			time.Sleep(1 * time.Second)
			continue
		}
		buf := make([]byte, 102400)

		for {
			n, err := res.Body.Read(buf)
			if err != nil && err != io.EOF {
				fmt.Printf("Download failed:[%s], error:%v, path:[%s]\n", url, err, path)
				res.Body.Close()
				if i >= 2 {
					return true, false
				} else {
					flag = 1
					break
				}
			}
			buffer.Write(buf[:n])
			if err == io.EOF {
				res.Body.Close()
				flag = 2
				break
			}
		}
		if flag == 1 {
			time.Sleep(1 * time.Second)
			continue
		} else if flag == 2 {
			break
		}
	}
	fmt.Printf("Download success:[%s], path:[%s]\n", url, path)
	return true, true
}

func main() {
	// 下载地址url:
	url := "http://xxxxxxxxxxxxxxxxxxx/test20220623.mp4"
	// 下载路径，全路径:
	downloadPath := "/home/testdir/download_test20220623.mp4"
	userName := "用户名"
	userKey := "密钥"
	fileRes,downloadRes := DownloadFile(url, downloadPath, userName, userKey)
	fmt.Println("\nfileRes:",fileRes,";downloadRes:",downloadRes,"")
}