From 0e22b2dfe7a24c528fb8ac58ece089bd032f5478 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期二, 26 十二月 2023 17:49:34 +0800
Subject: [PATCH] 支持添加额外的env
---
src/main/main.go | 75 +++++++++++++++++++++++++------------
1 files changed, 50 insertions(+), 25 deletions(-)
diff --git a/src/main/main.go b/src/main/main.go
index db889ec..530cf63 100644
--- a/src/main/main.go
+++ b/src/main/main.go
@@ -27,7 +27,7 @@
)
func rancher_install_test(node rancher.Node) {
-/* nodes := []Node{
+ /* nodes := []Node{
{
Roles: []string{"etcd", "controlplane", "worker"},
IP: "192.168.20.189",
@@ -61,7 +61,7 @@
os.Exit(0)
}
-func cluster_test(clusterName string,rancherClusterConfig rancher.RancherClusterConfig) {
+func cluster_test(clusterName string, rancherClusterConfig rancher.RancherClusterConfig) {
// Create the cluster
// Rancher configuration
@@ -70,14 +70,14 @@
BearerToken: "token-nnrsc:w68zdt8s47fnpjd5xqdl5hhzpz4j2d56kt5nx49nsswcbpdzc28kh5",
}*/
- fmt.Println(rancherClusterConfig)
- fmt.Println(rancherClusterConfig.RancherURL)
- fmt.Println(rancherClusterConfig.BearerToken)
+ fmt.Println(rancherClusterConfig)
+ fmt.Println(rancherClusterConfig.RancherURL)
+ fmt.Println(rancherClusterConfig.BearerToken)
// Deploy clusterId
clusterID, err := cluster.GetClusterID(rancherClusterConfig.RancherURL, rancherClusterConfig.BearerToken, clusterName)
if err != nil {
fmt.Printf("Cluster not found: %s %v\n", clusterName, err)
- err = cluster.CreateCluster(rancherClusterConfig, clusterName)
+ _, err = cluster.CreateCluster(rancherClusterConfig, clusterName)
if err != nil {
log.Fatalf("Failed to create cluster: %v", err)
}
@@ -155,38 +155,53 @@
}
if rancherInstallCmd.Parsed() {
- // 鍒濆鍖栭厤缃В鏋愬簱
- viper.SetConfigName("config")
- viper.SetConfigType("yaml")
- viper.AddConfigPath("./config")
-
- // 璇诲彇閰嶇疆鏂囦欢
- err := viper.ReadInConfig()
- if err != nil {
- log.Fatalf("Failed to read config file: %v", err)
- }
+ //// 鍒濆鍖栭厤缃В鏋愬簱
+ //viper.SetConfigName("config")
+ //viper.SetConfigType("yaml")
+ //viper.AddConfigPath("./config")
+ //
+ //// 璇诲彇閰嶇疆鏂囦欢
+ //err := viper.ReadInConfig()
+ //if err != nil {
+ // log.Fatalf("Failed to read config file: %v", err)
+ //}
// 瑙ｆ瀽閰嶇疆鏂囦欢涓殑瀛楁
- var rancherClusterConfig rancher.RancherClusterConfig
- err = viper.Unmarshal(&rancherClusterConfig)
- if err != nil {
- log.Fatalf("Failed to unmarshal config: %v", err)
+ var rancherClusterConfig rancher.RancherClusterConfig
+ //err = viper.Unmarshal(&rancherClusterConfig)
+ //if err != nil {
+ // log.Fatalf("Failed to unmarshal config: %v", err)
+ //}
+
+ rancherClusterConfig.Nodes = make([]rancher.Node, 3)
+ rancherClusterConfig.Nodes[0] = rancher.Node{
+ Roles: []string{"worker"},
+ IP: "192.168.20.189",
+ SSHUsername: "basic",
+ SSHPassword: "123",
+ SSHPort: 22,
}
+
rancher_install_test(rancherClusterConfig.Nodes[0])
os.Exit(1)
}
// 鎵ц create 鍛戒护
if createCmd.Parsed() {
+
// 閰嶇疆 Kubernetes 闆嗙兢鐨� kubeconfig 璺緞
kubeconfig := flag.String("kubeconfig", filepath.Join(util.HomeDir(), ".kube", "config"), "kubeconfig file")
flag.Parse()
+
+ fmt.Println(&kubeconfig)
// 鍒涘缓 Kubernetes 瀹㈡埛绔�
config, err := clientcmd.BuildConfigFromFlags("", *kubeconfig)
if err != nil {
panic(err.Error())
}
+
+ fmt.Println(&config)
clientset, err := kubernetes.NewForConfig(config)
if err != nil {
@@ -199,7 +214,12 @@
os.Exit(1)
}
- err = k8s.CreateDeploymentAndService(clientset, *createNamespace, *createDeployment, *createService)
+ err = k8s.CreateDeploymentAndService(k8s.Config{
+ Client: clientset,
+ NameSpace: *createNamespace,
+ DeploymentName: *createDeployment,
+ ServiceName: *createService,
+ }, nil)
if err != nil {
panic(err)
}
@@ -267,7 +287,12 @@
for _, ns := range namespaces {
deploymentName := ns
serviceName := ns
- err := k8s.CreateDeploymentAndService(clientset, ns, deploymentName, serviceName)
+ err := k8s.CreateDeploymentAndService(k8s.Config{
+ Client: clientset,
+ NameSpace: ns,
+ DeploymentName: deploymentName,
+ ServiceName: serviceName,
+ }, nil)
if err != nil {
log.Printf("\033[97;41mFailed to create resources in namespace %s: %v\033[0m\n", ns, err)
} else {
@@ -311,13 +336,13 @@
}
// 瑙ｆ瀽閰嶇疆鏂囦欢涓殑瀛楁
- var rancherClusterConfig rancher.RancherClusterConfig
+ var rancherClusterConfig rancher.RancherClusterConfig
err = viper.Unmarshal(&rancherClusterConfig)
if err != nil {
log.Fatalf("Failed to unmarshal config: %v", err)
}
- clusterName := "kubernetus"
- fmt.Println(rancherClusterConfig)
+ clusterName := "kubernetus"
+ fmt.Println(rancherClusterConfig)
cluster_test(clusterName, rancherClusterConfig)
}
--
Gitblit v1.8.0