-
Notifications
You must be signed in to change notification settings - Fork 40
/
clientset.go
68 lines (59 loc) · 1.89 KB
/
clientset.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
/*
*
* * Copyright 2021 KubeClipper Authors.
* *
* * Licensed under the Apache License, Version 2.0 (the "License");
* * you may not use this file except in compliance with the License.
* * You may obtain a copy of the License at
* *
* * http://www.apache.org/licenses/LICENSE-2.0
* *
* * Unless required by applicable law or agreed to in writing, software
* * distributed under the License is distributed on an "AS IS" BASIS,
* * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* * See the License for the specific language governing permissions and
* * limitations under the License.
*
*/
package clientset
import (
"fmt"
"k8s.io/client-go/rest"
"k8s.io/client-go/util/flowcontrol"
corev1 "github.com/kubeclipper/kubeclipper/pkg/client/clientset/versioned/typed/core/v1"
iamv1 "github.com/kubeclipper/kubeclipper/pkg/client/clientset/versioned/typed/iam/v1"
)
type Interface interface {
CoreV1() corev1.CoreV1Interface
IamV1() iamv1.IamV1Interface
}
type Clientset struct {
corev1 *corev1.CoreV1Client
iamv1 *iamv1.IamV1Client
}
func (c *Clientset) CoreV1() corev1.CoreV1Interface {
return c.corev1
}
func (c *Clientset) IamV1() iamv1.IamV1Interface {
return c.iamv1
}
func NewForConfig(c *rest.Config) (*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.corev1, err = corev1.NewForConfig(&configShallowCopy)
if err != nil {
return nil, err
}
cs.iamv1, err = iamv1.NewForConfig(&configShallowCopy)
if err != nil {
return nil, err
}
return &cs, nil
}