ouyang/ghproxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix retry body

Browse Source
This commit is contained in:
wjqserver
2025-07-25 17:07:25 +08:00
parent 8689738f4f
commit 31c26b00fb
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
proxy/docker.go
View File

@@ -1,6 +1,7 @@
package proxy package proxy
import ( import (
"bytes"
"context" "context"
"github.com/go-json-experiment/json" "github.com/go-json-experiment/json"
@@ -149,12 +150,17 @@ func GhcrRequest(ctx context.Context, c *touka.Context, u string, image *imageIn
method = c.Request.Method method = c.Request.Method
ghcrclient := c.GetHTTPC() ghcrclient := c.GetHTTPC()
bodyByte, err := c.GetReqBodyFull()
if err != nil {
HandleError(c, fmt.Sprintf("Failed to read request body: %v", err))
return
}
// 构建初始请求 // 构建初始请求
rb := ghcrclient.NewRequestBuilder(method, u) rb := ghcrclient.NewRequestBuilder(method, u)
rb.NoDefaultHeaders() // 不使用默认头部, 以便完全控制 rb.NoDefaultHeaders() // 不使用默认头部, 以便完全控制
rb.SetBody(c.Request.Body) // 设置请求体 rb.SetBody(bytes.NewBuffer(bodyByte)) // 设置请求体
rb.WithContext(ctx) // 设置请求上下文 rb.WithContext(ctx) // 设置请求上下文
req, err = rb.Build() req, err = rb.Build()
if err != nil { if err != nil {
@@ -209,7 +215,7 @@ func GhcrRequest(ctx context.Context, c *touka.Context, u string, image *imageIn
// 重新构建并发送请求 // 重新构建并发送请求
rb_retry := ghcrclient.NewRequestBuilder(method, u) rb_retry := ghcrclient.NewRequestBuilder(method, u)
rb_retry.NoDefaultHeaders() rb_retry.NoDefaultHeaders()
rb_retry.SetBody(c.Request.Body) rb_retry.SetBody(bytes.NewBuffer(bodyByte))
rb_retry.WithContext(ctx) rb_retry.WithContext(ctx)
req_retry, err_retry := rb_retry.Build() req_retry, err_retry := rb_retry.Build()