@@ -809,6 +809,7 @@ func (a *API) GenerateCode() ([]byte, error) {
809809 "errors" ,
810810 "fmt" ,
811811 "io" ,
812+ "log/slog" ,
812813 "net/http" ,
813814 "net/url" ,
814815 "strconv" ,
@@ -820,6 +821,7 @@ func (a *API) GenerateCode() ([]byte, error) {
820821 if a .Name == "storage" {
821822 pn (" %q" , "github.com/googleapis/gax-go/v2" )
822823 }
824+ pn (" %q" , "github.com/googleapis/gax-go/v2/internallog" )
823825 for _ , imp := range []struct {
824826 pkg string
825827 lname string
@@ -895,7 +897,10 @@ func (a *API) GenerateCode() ([]byte, error) {
895897
896898 pn ("client, endpoint, err := htransport.NewClient(ctx, opts...)" )
897899 pn ("if err != nil { return nil, err }" )
898- pn ("s, err := New(client)" )
900+ pn ("s := &%s{client: client, BasePath: basePath, logger: internaloption.GetLogger(opts)}" , service )
901+ for _ , res := range a .doc .Resources { // add top level resources.
902+ pn ("s.%s = New%s(s)" , resourceGoField (res , nil ), resourceGoType (res ))
903+ }
899904 pn ("if err != nil { return nil, err }" )
900905 pn (`if endpoint != "" { s.BasePath = endpoint }` )
901906 pn ("return s, nil" )
@@ -908,15 +913,12 @@ func (a *API) GenerateCode() ([]byte, error) {
908913 pn ("// If you are using google.golang.org/api/googleapis/transport.APIKey, use option.WithAPIKey with NewService instead." )
909914 pn ("func New(client *http.Client) (*%s, error) {" , service )
910915 pn ("if client == nil { return nil, errors.New(\" client is nil\" ) }" )
911- pn ("s := &%s{client: client, BasePath: basePath}" , service )
912- for _ , res := range a .doc .Resources { // add top level resources.
913- pn ("s.%s = New%s(s)" , resourceGoField (res , nil ), resourceGoType (res ))
914- }
915- pn ("return s, nil" )
916+ pn ("return NewService(option.WithHTTPClient(client))" )
916917 pn ("}" )
917918
918919 pn ("\n type %s struct {" , service )
919920 pn (" client *http.Client" )
921+ pn (" logger *slog.Logger" )
920922 pn (" BasePath string // API endpoint base URL" )
921923 pn (" UserAgent string // optional additional User-Agent fragment" )
922924
@@ -995,6 +997,7 @@ func splitFileHeading(w io.Writer, pkg string) {
995997 "context" ,
996998 "fmt" ,
997999 "io" ,
1000+ "log/slog" ,
9981001 "net/http" ,
9991002 } {
10001003 pn (" %q" , imp )
@@ -2280,26 +2283,32 @@ func (meth *Method) generateCode() {
22802283 pn ("}" )
22812284 }
22822285 pn ("var body io.Reader = nil" )
2286+ var hasBody bool
22832287 if meth .IsRawRequest () {
2284- pn ("body = c.body_" )
2288+ pn ("body := bytes.NewBuffer(nil)" )
2289+ pn ("_, err := body.ReadFrom(c.body_)" )
2290+ pn ("if err != nil { return nil, err }" )
2291+ hasBody = true
22852292 } else if meth .IsProtoStructRequest () {
22862293 pn ("protoBytes, err := json.Marshal(c.req)" )
22872294 pn ("if err != nil { return nil, err }" )
2288- pn ("body = bytes.NewReader(protoBytes)" )
2295+ pn ("body := bytes.NewReader(protoBytes)" )
2296+ hasBody = true
22892297 } else {
22902298 if ba := args .bodyArg (); ba != nil && httpMethod != "GET" {
22912299 if meth .m .ID == "ml.projects.predict" {
22922300 // TODO(cbro): move ML API to rawHTTP (it will be a breaking change)
2293- // Skip JSONReader for APIs that require clients to pass in JSON already.
2294- pn ("body = strings.NewReader (c.%s.HttpBody.Data)" , ba .goname )
2301+ // Skip JSONBuffer for APIs that require clients to pass in JSON already.
2302+ pn ("body := bytes.NewBufferString (c.%s.HttpBody.Data)" , ba .goname )
22952303 } else {
22962304 style := "WithoutDataWrapper"
22972305 if a .needsDataWrapper () {
22982306 style = "WithDataWrapper"
22992307 }
2300- pn ("body, err := googleapi.%s.JSONReader (c.%s)" , style , ba .goname )
2308+ pn ("body, err := googleapi.%s.JSONBuffer (c.%s)" , style , ba .goname )
23012309 pn ("if err != nil { return nil, err }" )
23022310 }
2311+ hasBody = true
23032312 }
23042313 pn (`c.urlParams_.Set("alt", alt)` )
23052314 pn (`c.urlParams_.Set("prettyPrint", "false")` )
@@ -2337,12 +2346,18 @@ func (meth *Method) generateCode() {
23372346 }
23382347 pn (`})` )
23392348 }
2349+ logBody := "nil"
2350+ if hasBody {
2351+ logBody = "body.Bytes()"
2352+ }
23402353 if meth .supportsMediaUpload () && meth .api .Name == "storage" {
2354+ pn (`c.s.logger.DebugContext(c.ctx_, "api request", "serviceName", apiName, "rpcName", %q, "request", internallog.HTTPRequest(req, %s))` , meth .Id (), logBody )
23412355 pn ("if c.retry != nil {" )
23422356 pn (" return gensupport.SendRequestWithRetry(c.ctx_, c.s.client, req, c.retry)" )
23432357 pn ("}" )
23442358 pn ("return gensupport.SendRequest(c.ctx_, c.s.client, req)" )
23452359 } else {
2360+ pn (`c.s.logger.DebugContext(c.ctx_, "api request", "serviceName", apiName, "rpcName", %q, "request", internallog.HTTPRequest(req, %s))` , meth .Id (), logBody )
23462361 pn ("return gensupport.SendRequest(c.ctx_, c.s.client, req)" )
23472362 }
23482363 pn ("}" )
@@ -2421,6 +2436,7 @@ func (meth *Method) generateCode() {
24212436 pn ("}" )
24222437 }
24232438 if retTypeComma == "" {
2439+ pn (`c.s.logger.DebugContext(c.ctx_, "api response", "serviceName", apiName, "rpcName", %q, "response", internallog.HTTPResponse(res, nil))` , meth .Id ())
24242440 pn ("return nil" )
24252441 } else {
24262442 if meth .IsProtoStructResponse () {
@@ -2449,8 +2465,11 @@ func (meth *Method) generateCode() {
24492465 pn ("if err := res.Body.Close(); err != nil { return nil, err }" )
24502466 pn ("if err := json.NewDecoder(bytes.NewReader(b.Bytes())).Decode(target); err != nil { return nil, err }" )
24512467 pn ("ret.Data = b.String()" )
2468+ pn (`c.s.logger.DebugContext(c.ctx_, "api response", "serviceName", apiName, "rpcName", %q, "response", internallog.HTTPResponse(res, b.Bytes()))` , meth .Id ())
24522469 } else {
2453- pn ("if err := gensupport.DecodeResponse(target, res); err != nil { return nil, err }" )
2470+ pn ("b, err := gensupport.DecodeResponseBytes(target, res)" )
2471+ pn ("if err != nil { return nil, err }" )
2472+ pn (`c.s.logger.DebugContext(c.ctx_, "api response", "serviceName", apiName, "rpcName", %q, "response", internallog.HTTPResponse(res, b))` , meth .Id ())
24542473 }
24552474 pn ("return ret, nil" )
24562475 }
0 commit comments