|
|
@@ -18,6 +18,7 @@ package versioned
|
|
|
|
|
|
import (
|
|
|
"fmt"
|
|
|
+ "net/http"
|
|
|
|
|
|
kilov1alpha1 "github.com/squat/kilo/pkg/k8s/clientset/versioned/typed/kilo/v1alpha1"
|
|
|
discovery "k8s.io/client-go/discovery"
|
|
|
@@ -53,22 +54,45 @@ func (c *Clientset) Discovery() discovery.DiscoveryInterface {
|
|
|
// NewForConfig creates a new Clientset for the given config.
|
|
|
// If config's RateLimiter is not set and QPS and Burst are acceptable,
|
|
|
// NewForConfig will generate a rate-limiter in configShallowCopy.
|
|
|
+// NewForConfig is equivalent to NewForConfigAndClient(c, httpClient),
|
|
|
+// where httpClient was generated with rest.HTTPClientFor(c).
|
|
|
func NewForConfig(c *rest.Config) (*Clientset, error) {
|
|
|
configShallowCopy := *c
|
|
|
+
|
|
|
+ if configShallowCopy.UserAgent == "" {
|
|
|
+ configShallowCopy.UserAgent = rest.DefaultKubernetesUserAgent()
|
|
|
+ }
|
|
|
+
|
|
|
+ // share the transport between all clients
|
|
|
+ httpClient, err := rest.HTTPClientFor(&configShallowCopy)
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+
|
|
|
+ return NewForConfigAndClient(&configShallowCopy, httpClient)
|
|
|
+}
|
|
|
+
|
|
|
+// NewForConfigAndClient creates a new Clientset for the given config and http client.
|
|
|
+// Note the http client provided takes precedence over the configured transport values.
|
|
|
+// If config's RateLimiter is not set and QPS and Burst are acceptable,
|
|
|
+// NewForConfigAndClient will generate a rate-limiter in configShallowCopy.
|
|
|
+func NewForConfigAndClient(c *rest.Config, httpClient *http.Client) (*Clientset, error) {
|
|
|
+ configShallowCopy := *c
|
|
|
if configShallowCopy.RateLimiter == nil && configShallowCopy.QPS > 0 {
|
|
|
if configShallowCopy.Burst <= 0 {
|
|
|
return nil, fmt.Errorf("burst is required to be greater than 0 when RateLimiter is not set and QPS is set to greater than 0")
|
|
|
}
|
|
|
configShallowCopy.RateLimiter = flowcontrol.NewTokenBucketRateLimiter(configShallowCopy.QPS, configShallowCopy.Burst)
|
|
|
}
|
|
|
+
|
|
|
var cs Clientset
|
|
|
var err error
|
|
|
- cs.kiloV1alpha1, err = kilov1alpha1.NewForConfig(&configShallowCopy)
|
|
|
+ cs.kiloV1alpha1, err = kilov1alpha1.NewForConfigAndClient(&configShallowCopy, httpClient)
|
|
|
if err != nil {
|
|
|
return nil, err
|
|
|
}
|
|
|
|
|
|
- cs.DiscoveryClient, err = discovery.NewDiscoveryClientForConfig(&configShallowCopy)
|
|
|
+ cs.DiscoveryClient, err = discovery.NewDiscoveryClientForConfigAndClient(&configShallowCopy, httpClient)
|
|
|
if err != nil {
|
|
|
return nil, err
|
|
|
}
|
|
|
@@ -78,11 +102,11 @@ func NewForConfig(c *rest.Config) (*Clientset, error) {
|
|
|
// NewForConfigOrDie creates a new Clientset for the given config and
|
|
|
// panics if there is an error in the config.
|
|
|
func NewForConfigOrDie(c *rest.Config) *Clientset {
|
|
|
- var cs Clientset
|
|
|
- cs.kiloV1alpha1 = kilov1alpha1.NewForConfigOrDie(c)
|
|
|
-
|
|
|
- cs.DiscoveryClient = discovery.NewDiscoveryClientForConfigOrDie(c)
|
|
|
- return &cs
|
|
|
+ cs, err := NewForConfig(c)
|
|
|
+ if err != nil {
|
|
|
+ panic(err)
|
|
|
+ }
|
|
|
+ return cs
|
|
|
}
|
|
|
|
|
|
// New creates a new Clientset for the given RESTClient.
|