From 670eba9d841f30b89c40ac80c21b27d1492b3dae Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=B8=8A=E6=B5=B7=E6=98=8E=E9=A6=A8=E7=A7=91=E6=8A=80?= =?UTF-8?q?=E6=9C=89=E9=99=90=E5=85=AC=E5=8F=B8?= Date: Sun, 20 Sep 2020 07:20:18 +0800 Subject: [PATCH] Update handler.go --- pkg/filesystem/driver/onedrive/handler.go | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/pkg/filesystem/driver/onedrive/handler.go b/pkg/filesystem/driver/onedrive/handler.go index 5476f3d3..fee51228 100644 --- a/pkg/filesystem/driver/onedrive/handler.go +++ b/pkg/filesystem/driver/onedrive/handler.go @@ -156,7 +156,7 @@ func (handler Driver) Source( finalURL, err := handler.getFinalURL(cachedURL.(string)) if err != nil { return "", err - } + } return finalURL, nil } // 缓存不存在,重新获取 @@ -176,12 +176,19 @@ func (handler Driver) Source( } return "", err } -//增加国际版反向代理自定义前缀 func (handler Driver) getFinalURL(key string)(string, error){ + cdnURL, err := url.Parse(handler.Policy.BaseURL) if err != nil { return "", err } + + if cdnURL.String() == "https://login.chinacloudapi.cn/common/oauth2" { + return key, err + } + if cdnURL.String() == "https://login.microsoftonline.com/common/oauth2" { + return key, err + } if cdnURL.String() != "" { finalURL, err := url.Parse(key) if err != nil { @@ -191,6 +198,7 @@ func (handler Driver) getFinalURL(key string)(string, error){ finalURL.Scheme = cdnURL.Scheme return finalURL.String(), err } + return key, err }