proxy for put

upyun · Oct 31, 2024 · 0e76273 · 0e76273
1 parent de08aa9
commit 0e76273
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/upyun/rest.go b/upyun/rest.go
@@ -248,12 +248,17 @@ func (up *UpYun) put(config *PutObjectConfig) error {
 		config.Headers["X-Upyun-Append"] = "true"
 	}
 	*/
+
+	reader := config.Reader
+	if config.ProxyReader != nil {
+		reader = config.ProxyReader(0, config.Reader)
+	}
 	_, err := up.doRESTRequest(&restReqConfig{
 		method:    "PUT",
 		uri:       config.Path,
 		headers:   config.Headers,
 		closeBody: true,
-		httpBody:  config.Reader,
+		httpBody:  reader,
 		useMD5:    config.UseMD5,
 	})
 	if err != nil {