cheliequan
2023-05-31 d3a63ffbec643f7a2746dfe2db2313492ef163af
src/main/main.go
@@ -26,17 +26,8 @@
   namespaces          = []string{"guangsheng", "geruimi", "tongsheng"}
)
type Node struct {
   ClusterName string   `json:"clusterName"`
   Roles       []string `json:"roles"`
   IP          string   `json:"ip"`
   SSHUsername string   `json:"sshUsername"`
   SSHPassword string   `json:"sshPassword"`
   SSHPort     int      `json:"sshPort"`
}
func rancher_install_test() {
   nodes := []Node{
func rancher_install_test(node rancher.Node) {
/*   nodes := []Node{
      {
         Roles:       []string{"etcd", "controlplane", "worker"},
         IP:          "192.168.20.189",
@@ -59,10 +50,10 @@
         SSHPort:     22,
      },
      // Add more nodes here if needed
   }
   }*/
   //install rancher on master node
   err := rancher.InstallDockerAndRancher(nodes[0].IP, nodes[0].SSHUsername, nodes[0].SSHPassword, nodes[0].SSHPort)
   err := rancher.InstallDockerAndRancher(node.IP, node.SSHUsername, node.SSHPassword, node.SSHPort)
   if err != nil {
      log.Fatalf("Failed to install Rancher: %v", err)
   }
@@ -70,7 +61,7 @@
   os.Exit(0)
}
func cluster_test(clusterName string,nodes []Node,rancherConfig rancher.RancherConfig) {
func cluster_test(clusterName string,rancherClusterConfig rancher.RancherClusterConfig) {
   // Create the cluster
   // Rancher configuration
@@ -79,16 +70,19 @@
      BearerToken: "token-nnrsc:w68zdt8s47fnpjd5xqdl5hhzpz4j2d56kt5nx49nsswcbpdzc28kh5",
   }*/
        fmt.Println(rancherClusterConfig)
        fmt.Println(rancherClusterConfig.RancherURL)
        fmt.Println(rancherClusterConfig.BearerToken)
   //   Deploy clusterId
   clusterID, err := cluster.GetClusterID(rancherConfig.RancherURL, rancherConfig.BearerToken, clusterName)
   clusterID, err := cluster.GetClusterID(rancherClusterConfig.RancherURL, rancherClusterConfig.BearerToken, clusterName)
   if err != nil {
      log.Fatal(err)
      err = cluster.CreateCluster(rancherConfig, clusterName)
      fmt.Printf("Cluster not found: %s %v\n", clusterName, err)
      err = cluster.CreateCluster(rancherClusterConfig, clusterName)
      if err != nil {
         log.Fatalf("Failed to create cluster: %v", err)
      }
      fmt.Printf("Cluster created: %s\n", clusterName)
      clusterID, err = cluster.GetClusterID(rancherConfig.RancherURL, rancherConfig.BearerToken, clusterName)
      clusterID, err = cluster.GetClusterID(rancherClusterConfig.RancherURL, rancherClusterConfig.BearerToken, clusterName)
      if err != nil {
         log.Fatal(err)
      }
@@ -96,13 +90,13 @@
   fmt.Println(clusterID)
   //   Deploy nodeCommand
   nodeCommand, err := cluster.GetNodeCommand(rancherConfig.RancherURL, rancherConfig.BearerToken, clusterID)
   nodeCommand, err := cluster.GetNodeCommand(rancherClusterConfig.RancherURL, rancherClusterConfig.BearerToken, clusterID)
   if err != nil {
      log.Fatal(err)
   }
   fmt.Println(nodeCommand)
   for _, node := range nodes {
   for _, node := range rancherClusterConfig.Nodes {
      //Deploy Docker on each node
      err = util.InstallDocker(node.IP, node.SSHUsername, node.SSHPassword, node.SSHPort)
      if err != nil {
@@ -161,7 +155,24 @@
   }
   if rancherInstallCmd.Parsed() {
      rancher_install_test()
      // 初始化配置解析库
      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)
      }
      rancher_install_test(rancherClusterConfig.Nodes[0])
      os.Exit(1)
   }
@@ -300,41 +311,14 @@
      }
      // 解析配置文件中的字段
      var rancherConfig rancher.RancherConfig
      err = viper.Unmarshal(&rancherConfig)
      var rancherClusterConfig  rancher.RancherClusterConfig
      err = viper.Unmarshal(&rancherClusterConfig)
      if err != nil {
         log.Fatalf("Failed to unmarshal config: %v", err)
      }
           clusterName := "kubernetus"
           nodes := []Node{
              {
                 ClusterName: clusterName,
                 Roles:       []string{"worker"},
                 IP:          "192.168.20.189",
                 SSHUsername: "basic",
                 SSHPassword: "123",
                 SSHPort:     22,
              },
              {
                 ClusterName: clusterName,
                 Roles:       []string{"etcd", "worker"},
                 IP:          "192.168.20.10",
                 SSHUsername: "basic",
                 SSHPassword: "123",
                 SSHPort:     22,
              },
              {
                 ClusterName: clusterName,
                 Roles:       []string{"etcd", "controlplane","worker"},
                 IP:          "192.168.20.115",
                 SSHUsername: "basic",
                 SSHPassword: "alien123",
                 SSHPort:     22,
              },
              // Add more nodes here if needed
           }
      cluster_test(clusterName, nodes, rancherConfig)
                fmt.Println(rancherClusterConfig)
      cluster_test(clusterName, rancherClusterConfig)
   }
}