Merge discovery package into k8s package

This commit is contained in:
TwinProduction
2020-11-11 14:46:19 -05:00
parent 799c3e9187
commit fff34fff58
6 changed files with 70 additions and 74 deletions

View File

@ -12,23 +12,19 @@ import (
)
func NewClient(clusterMode string) *kubernetes.Clientset {
var kubeConfig *rest.Config
switch clusterMode {
case "in":
kubeConfig = getInclusterConfig()
kubeConfig = getInClusterConfig()
case "out":
kubeConfig = getOutClusterConfig()
default:
panic("invalid cluster mode")
}
clientset, err := kubernetes.NewForConfig(kubeConfig)
if err != nil {
panic(err.Error())
}
return clientset
}
@ -47,20 +43,17 @@ func getOutClusterConfig() *rest.Config {
kubeconfig = flag.String("kubeconfig", "", "absolute path to the kubeconfig file")
}
flag.Parse()
config, err := clientcmd.BuildConfigFromFlags("", *kubeconfig)
if err != nil {
panic(err.Error())
}
return config
}
func getInclusterConfig() *rest.Config {
func getInClusterConfig() *rest.Config {
config, err := rest.InClusterConfig()
if err != nil {
panic(err.Error())
}
return config
}