nyne/nysoure
1
0
Fork 0
mirror of https://github.com/wgh136/nysoure.git synced 2025-09-27 12:17:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix testFileUrl

Browse Source
This commit is contained in:
nyne
2025-08-28 16:35:56 +08:00
parent 488a91e651
commit 634d5a348a
1 changed files with 44 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
server/service/file.go
View File

@@ -445,32 +445,51 @@ func DownloadFile(fid, cfToken string) (string, string, error) {
} }
func testFileUrl(url string) (int64, error) { func testFileUrl(url string) (int64, error) {
client := http.Client{ client := http.Client{Timeout: 10 * time.Second}
Timeout: 10 * time.Second,
} // Try HEAD request first, fallback to GET
req, err := http.NewRequest("HEAD", url, nil) for _, method := range []string{"HEAD", "GET"} {
req, err := http.NewRequest(method, url, nil)
if err != nil { if err != nil {
return 0, model.NewRequestError("failed to create HTTP request") return 0, model.NewRequestError("failed to create HTTP request")
} }
resp, err := client.Do(req) resp, err := client.Do(req)
if err != nil { if err != nil {
if method == "GET" {
return 0, model.NewRequestError("failed to send HTTP request") return 0, model.NewRequestError("failed to send HTTP request")
} }
continue // Try GET if HEAD fails
}
defer resp.Body.Close() defer resp.Body.Close()
if resp.StatusCode != http.StatusOK { if resp.StatusCode != http.StatusOK {
if method == "GET" {
return 0, model.NewRequestError("URL is not accessible, status code: " + resp.Status) return 0, model.NewRequestError("URL is not accessible, status code: " + resp.Status)
} }
if resp.Header.Get("Content-Length") == "" { continue // Try GET if HEAD fails
}
contentLengthStr := resp.Header.Get("Content-Length")
if contentLengthStr == "" {
if method == "GET" {
return 0, model.NewRequestError("URL does not provide content length") return 0, model.NewRequestError("URL does not provide content length")
} }
contentLength, err := strconv.ParseInt(resp.Header.Get("Content-Length"), 10, 64) continue // Try GET if HEAD doesn't provide Content-Length
if err != nil {
return 0, model.NewRequestError("failed to parse Content-Length header")
} }
if contentLength <= 0 {
contentLength, err := strconv.ParseInt(contentLengthStr, 10, 64)
if err != nil || contentLength <= 0 {
if method == "GET" {
return 0, model.NewRequestError("Content-Length is not valid") return 0, model.NewRequestError("Content-Length is not valid")
} }
continue // Try GET if HEAD has invalid Content-Length
}
return contentLength, nil return contentLength, nil
}
return 0, model.NewRequestError("failed to get valid content length")
} }
// downloadFile return nil if the download is successful or the context is cancelled // downloadFile return nil if the download is successful or the context is cancelled