cloud.google.com/goのリクエストをX-Rayでトレースしたい

cloud.google.com/goを使ってGCPへのリクエストをX-Rayのトレースとして表示したい。

が、思いがけずハマったので調査・workaroundを見つけたメモ。

2020/6/7 大きいライブラリで根が深そうなのでまだissueはあげていない。

なぜ今動かないのか

トークン取得時のリクエストにcontextを渡していないから

APIリクエストに使うhttp.*Clientを組み立てる際、元となる http.*Client の Transport フィールドの実際の型が http.*Transport ではない時、 http.DefaultTransport を尊重せずに新しいhttp.*Transportを返すから

ややこしいが http.*Client の Transport フィールドの型はhttp.RoundTripperであり、http.*Transportはそれを実装した一つの構造体型である

対策

code:code.go

import (

"context"

"net/http"

"golang.org/x/oauth2"

"golang.org/x/oauth2/google"

)

func newClientWithToken(ctx context.Context, httpClient *http.Client, cred *google.Credentials) (*http.Client, error) {

if httpClient == nil {

httpClient = http.DefaultClient

}

token, err := cred.TokenSource.Token()

if err != nil {

return nil, err

}

staticSource := oauth2.StaticTokenSource(token)

client := oauth2.NewClient(ctx, staticSource)

client.CheckRedirect = httpClient.CheckRedirect

client.Jar = httpClient.Jar

client.Timeout = httpClient.Timeout

return client, nil

}

メモ