refactor: customHost only for the GET request #111

saltbo · Jul 8, 2021 · 49f00a8 · 49f00a8
1 parent 78156c0
commit 49f00a8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/internal/pkg/provider/provider.go b/internal/pkg/provider/provider.go
@@ -51,6 +51,10 @@ func New(conf Config) (Provider, error) {
 		conf.Region = "auto"
 	}
 
+	if !strings.Contains(conf.CustomHost, "://") {
+		conf.CustomHost = "http://" + conf.CustomHost
+	}
+
 	constructor, ok := supportProviders[strings.ToUpper(conf.Provider)]
 	if !ok {
 		return nil, fmt.Errorf("provider %s not found", conf.Provider)

diff --git a/internal/pkg/provider/provider_s3.go b/internal/pkg/provider/provider_s3.go
@@ -37,6 +37,10 @@ func newS3Provider(conf Config) (*S3Provider, error) {
 		}
 
 		client.Handlers.Build.PushBack(func(r *request.Request) {
+			if r.HTTPRequest.Method != http.MethodGet {
+				return
+			}
+
 			r.HTTPRequest.URL.Scheme = cURL.Scheme
 			r.HTTPRequest.URL.Host = cURL.Host
 		})