wood/webp_server_go
1
0
Fork 0
mirror of https://github.com/woodchen-ink/webp_server_go.git synced 2025-07-19 06:02:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(router): improve remote address construction logic

Browse Source
This commit is contained in:
wood chen 2024-10-23 03:04:32 +08:00
parent b7d9419408
commit 2a61cc9247
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
handler/router.go
View File

@ -106,12 +106,18 @@ func Convert(c *fiber.Ctx) error {
return
}
// 构建正确的远程地址
remoteAddr := strings.TrimRight(targetUrl.String(), "/")
relativePath := strings.TrimLeft(strings.TrimPrefix(reqURI, matchedPrefix), "/")
remoteAddr = remoteAddr + "/" + relativePath
targetHostName := targetUrl.Host
targetHost := targetUrl.Scheme + "://" + targetUrl.Host
rawImageAbs, isNewDownload, err = fetchRemoteImg(remoteAddr, targetUrl.Host)
// 保留查询参数
reqURIwithQuery := strings.Replace(reqURIwithQuery, matchedPrefix, targetUrl.Path, 1)
if strings.HasSuffix(targetUrl.Path, "/") {
reqURIwithQuery = strings.TrimPrefix(reqURIwithQuery, "/")
}
realRemoteAddr := targetHost + reqURIwithQuery
rawImageAbs, isNewDownload, err = fetchRemoteImg(realRemoteAddr, targetHostName)
if err != nil {
processErr = fmt.Errorf("获取远程图像失败")
log.Errorf("%s: %v", processErr, err)