You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
2.4 KiB
71 lines
2.4 KiB
// Copyright 2016 The Go Authors. All rights reserved. |
|
// Use of this source code is governed by a BSD-style |
|
// license that can be found in the LICENSE file. |
|
|
|
package gensupport |
|
|
|
import ( |
|
"encoding/json" |
|
"errors" |
|
"net/http" |
|
|
|
"golang.org/x/net/context" |
|
"golang.org/x/net/context/ctxhttp" |
|
) |
|
|
|
// Hook is the type of a function that is called once before each HTTP request |
|
// that is sent by a generated API. It returns a function that is called after |
|
// the request returns. |
|
// Hooks are not called if the context is nil. |
|
type Hook func(ctx context.Context, req *http.Request) func(resp *http.Response) |
|
|
|
var hooks []Hook |
|
|
|
// RegisterHook registers a Hook to be called before each HTTP request by a |
|
// generated API. Hooks are called in the order they are registered. Each |
|
// hook can return a function; if it is non-nil, it is called after the HTTP |
|
// request returns. These functions are called in the reverse order. |
|
// RegisterHook should not be called concurrently with itself or SendRequest. |
|
func RegisterHook(h Hook) { |
|
hooks = append(hooks, h) |
|
} |
|
|
|
// SendRequest sends a single HTTP request using the given client. |
|
// If ctx is non-nil, it calls all hooks, then sends the request with |
|
// ctxhttp.Do, then calls any functions returned by the hooks in reverse order. |
|
func SendRequest(ctx context.Context, client *http.Client, req *http.Request) (*http.Response, error) { |
|
// Disallow Accept-Encoding because it interferes with the automatic gzip handling |
|
// done by the default http.Transport. See https://github.com/google/google-api-go-client/issues/219. |
|
if _, ok := req.Header["Accept-Encoding"]; ok { |
|
return nil, errors.New("google api: custom Accept-Encoding headers not allowed") |
|
} |
|
if ctx == nil { |
|
return client.Do(req) |
|
} |
|
// Call hooks in order of registration, store returned funcs. |
|
post := make([]func(resp *http.Response), len(hooks)) |
|
for i, h := range hooks { |
|
fn := h(ctx, req) |
|
post[i] = fn |
|
} |
|
|
|
// Send request. |
|
resp, err := ctxhttp.Do(ctx, client, req) |
|
|
|
// Call returned funcs in reverse order. |
|
for i := len(post) - 1; i >= 0; i-- { |
|
if fn := post[i]; fn != nil { |
|
fn(resp) |
|
} |
|
} |
|
return resp, err |
|
} |
|
|
|
// DecodeResponse decodes the body of res into target. If there is no body, |
|
// target is unchanged. |
|
func DecodeResponse(target interface{}, res *http.Response) error { |
|
if res.StatusCode == http.StatusNoContent { |
|
return nil |
|
} |
|
return json.NewDecoder(res.Body).Decode(target) |
|
}
|
|
|