@@ -13,6 +13,7 @@ import (
13
13
14
14
eniconfigscheme "github.com/aws/amazon-vpc-cni-k8s/pkg/apis/crd/v1alpha1"
15
15
"github.com/aws/amazon-vpc-cni-k8s/pkg/utils/logger"
16
+ "github.com/aws/amazon-vpc-cni-k8s/utils"
16
17
rcscheme "github.com/aws/amazon-vpc-resource-controller-k8s/apis/vpcresources/v1alpha1"
17
18
"k8s.io/apimachinery/pkg/runtime"
18
19
"k8s.io/apimachinery/pkg/util/wait"
@@ -86,7 +87,7 @@ func StartKubeClientCache(cache cache.Cache) {
86
87
87
88
// CreateKubeClient creates a k8s client
88
89
func CreateKubeClient (appName string ) (client.Client , error ) {
89
- restCfg , err := getRestConfig (appName )
90
+ restCfg , err := getRestConfig ()
90
91
if err != nil {
91
92
return nil , err
92
93
}
@@ -122,9 +123,9 @@ func CreateKubeClient(appName string) (client.Client, error) {
122
123
return k8sClient , nil
123
124
}
124
125
125
- func GetKubeClientSet (appName string ) (kubernetes.Interface , error ) {
126
+ func GetKubeClientSet () (kubernetes.Interface , error ) {
126
127
// creates the in-cluster config
127
- config , err := getRestConfig (appName )
128
+ config , err := getRestConfig ()
128
129
if err != nil {
129
130
return nil , err
130
131
}
@@ -137,8 +138,8 @@ func GetKubeClientSet(appName string) (kubernetes.Interface, error) {
137
138
return clientSet , nil
138
139
}
139
140
140
- func CheckAPIServerConnectivity (appName string ) error {
141
- restCfg , err := getRestConfig (appName )
141
+ func CheckAPIServerConnectivity () error {
142
+ restCfg , err := getRestConfig ()
142
143
if err != nil {
143
144
return err
144
145
}
@@ -166,12 +167,12 @@ func CheckAPIServerConnectivity(appName string) error {
166
167
})
167
168
}
168
169
169
- func getRestConfig (appName string ) (* rest.Config , error ) {
170
+ func getRestConfig () (* rest.Config , error ) {
170
171
restCfg , err := ctrl .GetConfig ()
171
172
if err != nil {
172
173
return nil , err
173
174
}
174
- restCfg .UserAgent = appName
175
+ restCfg .UserAgent = os . Args [ 0 ] + "-" + utils . GetEnv ( "VPC_CNI_VERSION" , "" )
175
176
if endpoint , ok := os .LookupEnv ("CLUSTER_ENDPOINT" ); ok {
176
177
restCfg .Host = endpoint
177
178
}
0 commit comments