更新 main.go

This commit is contained in:
XOF
2025-12-28 15:18:05 +08:00
parent 9baac19825
commit 6f16865eb5

17
main.go
View File

@@ -33,18 +33,8 @@ var config Config
var client = &http.Client{ var client = &http.Client{
CheckRedirect: func(req *http.Request, via []*http.Request) error { CheckRedirect: func(req *http.Request, via []*http.Request) error {
for key, val := range via[0].Header { // 不跟随重定向,让 Docker 客户端自己处理
if _, ok := req.Header[key]; !ok { return http.ErrUseLastResponse
req.Header[key] = val
}
}
// S3 重定向时添加必要的头
if strings.Contains(req.URL.Host, "amazonaws.com") {
req.Header.Set("x-amz-content-sha256", "UNSIGNED-PAYLOAD")
}
return nil
}, },
Timeout: 30 * time.Second, Timeout: 30 * time.Second,
Transport: &http.Transport{ Transport: &http.Transport{
@@ -516,9 +506,6 @@ func sendRequestWithContext(ctx context.Context, method, url string, headers htt
req.Header = headers req.Header = headers
// 为可能的 S3 重定向预先添加头
req.Header.Set("x-amz-content-sha256", "UNSIGNED-PAYLOAD")
startTime := time.Now() startTime := time.Now()
resp, err := client.Do(req) resp, err := client.Do(req)