mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
9eaf824398
* add jaeger support, link hot container & req span * adds jaeger support now with FN_JAEGER_URL, there's a simple tutorial in the operating/metrics.md file now and it's pretty easy to get up and running. * links a hot request span to a hot container span. when we change this to sample at a lower ratio we'll need to finagle the hot container span to always sample or something, otherwise we'll hide that info. at least, since we're sampling at 100% for now if this is flipped on, can see freeze/unfreeze etc. if they hit. this is useful for debugging. note that zipkin's exporter does not follow the link at all, hence jaeger... and they're backed by the Cloud Empire now (CNCF) so we'll probably use it anyway. * vendor: add thrift for jaeger
72 lines
2.4 KiB
Go
72 lines
2.4 KiB
Go
// 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)
|
|
}
|