2017-02-22 15:14:37 +08:00
|
|
|
// Copyright 2014 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 internal
|
|
|
|
|
|
|
|
import (
|
2018-11-11 07:55:36 +08:00
|
|
|
"context"
|
2017-02-22 15:14:37 +08:00
|
|
|
"net/http"
|
|
|
|
)
|
|
|
|
|
|
|
|
// HTTPClient is the context key to use with golang.org/x/net/context's
|
|
|
|
// WithValue function to associate an *http.Client value with a context.
|
|
|
|
var HTTPClient ContextKey
|
|
|
|
|
|
|
|
// ContextKey is just an empty struct. It exists so HTTPClient can be
|
|
|
|
// an immutable public variable with a unique type. It's immutable
|
|
|
|
// because nobody else can create a ContextKey, being unexported.
|
|
|
|
type ContextKey struct{}
|
|
|
|
|
2018-11-11 07:55:36 +08:00
|
|
|
var appengineClientHook func(context.Context) *http.Client
|
2017-02-22 15:14:37 +08:00
|
|
|
|
2018-11-11 07:55:36 +08:00
|
|
|
func ContextClient(ctx context.Context) *http.Client {
|
2017-02-22 15:14:37 +08:00
|
|
|
if ctx != nil {
|
|
|
|
if hc, ok := ctx.Value(HTTPClient).(*http.Client); ok {
|
2018-11-11 07:55:36 +08:00
|
|
|
return hc
|
2017-02-22 15:14:37 +08:00
|
|
|
}
|
|
|
|
}
|
2018-11-11 07:55:36 +08:00
|
|
|
if appengineClientHook != nil {
|
|
|
|
return appengineClientHook(ctx)
|
2017-02-22 15:14:37 +08:00
|
|
|
}
|
2018-11-11 07:55:36 +08:00
|
|
|
return http.DefaultClient
|
2017-02-22 15:14:37 +08:00
|
|
|
}
|