根據這個文檔，我們可以http.Client用httptrace這種方式追蹤 t := &transport{} req, _ := http.NewRequest("GET", "https://google.com", nil) trace := &httptrace.ClientTrace{ GotConn: t.GotConn, } req = req.WithContext(httptrace.WithClientTrace(req.Context(), trace)) client := &http.Client{Transport: t}對于google API 客戶端，這里是一個包裝代碼func NewWithClient(jsonKey []byte, cli *http.Client) (*Client, error) { if cli == nil { return nil, fmt.Errorf("client is nil") } ctx := context.WithValue(context.Background(), oauth2.HTTPClient, cli) conf, err := google.JWTConfigFromJSON(jsonKey, androidpublisher.AndroidpublisherScope) if err != nil { return nil, err } service, err := androidpublisher.NewService(ctx, option.WithHTTPClient(conf.Client(ctx))) if err != nil { return nil, err } return &Client{service}, err}我們要應用到做 HTTP 跟蹤httptrace的http.Client參數。NewWithClient我們嘗試過的type TraceTransport struct {}var traceTransport = &TraceTransport{}var trace = &httptrace.ClientTrace{ GotConn: traceTransport.GotConn,}func (t *TraceTransport) RoundTrip(req *http.Request) (*http.Response, error) { return http.DefaultTransport.RoundTrip(req)}func (t *TraceTransport) GotConn(info httptrace.GotConnInfo) { fmt.Printf("Connection reused for %v \n", info.Reused)}type ClientWrapper struct { defaultClient *http.Client}var clientWrapperTrace = &httptrace.ClientTrace{GotConn: traceTransport.GotConn}func (c *ClientWrapper) Do(req *http.Request) (*http.Response, error) { req = req.WithContext(httptrace.WithClientTrace(req.Context(), clientWrapperTrace)) return c.defaultClient.Do(req)}但是，追蹤失敗http.Client，沒有輸出GotConn。有人可以幫助我們找出上述代碼的問題嗎？http去谷歌API安卓發布者
1 回答
慕慕森
TA貢獻1856條經驗 獲得超17個贊
來自的請求無法
google/oauth2被 追蹤httptrace。您ClientWrapper傳遞的 withcontext.WithValue將在這里被忽略，并且 oauth2 有它自己的 http.Client，它只是使用from context.Value的Transport方法。*http.Client來自 androidpublisher 的請求可以通過 httptrace 跟蹤，如下所示：
ctx := httptrace.WithClientTrace(context.Background(), clientWrapperTrace)r, err := c.VerifyProduct(ctx, packageName, productID, token)
如果您只想計算請求，我認為覆蓋
http.Client.Transport是一種簡單的方法。
type TraceTransport struct {
}
func (t *TraceTransport) RoundTrip(req *http.Request) (*http.Response, error) {
fmt.Printf("RoundTrip hook %v\n", req.URL)
return http.DefaultTransport.RoundTrip(req)
}
func NewClientTrace(jsonKey []byte) (*Client, error) {
cli := &http.Client{Transport: &TraceTransport{}}
ctx := context.WithValue(context.Background(), oauth2.HTTPClient, cli)
// ...
service, err := androidpublisher.NewService(ctx, option.WithHTTPClient(conf.Client(ctx)))
// ....
}
- 1 回答
- 0 關注
- 242 瀏覽
添加回答
舉報
0/150
提交
取消