ok-1

2026-02-14 05:36:00 +08:00
commit d7cd899983
37 changed files with 4169 additions and 0 deletions
--- a/cmd/node.go
+++ b/cmd/node.go
@@ -0,0 +1,113 @@
+package cmd
+
+import (
+	"fmt"
+
+	"sunhpc/internal/db"
+	"sunhpc/internal/log"
+
+	"github.com/spf13/cobra"
+)
+
+var nodeCmd = &cobra.Command{
+	Use:   "node",
+	Short: "节点管理",
+	Long:  "管理集群节点，包括添加、删除、查询等操作",
+}
+
+var nodeListCmd = &cobra.Command{
+	Use:   "list",
+	Short: "列出所有节点",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		log.Info("查询节点列表...")
+
+		// 获取数据库实例（自动使用之前配置的路径）
+		database, err := db.GetInstance()
+		if err != nil {
+			return fmt.Errorf("获取数据库连接失败: %v", err)
+		}
+		defer database.Close()
+
+		// 执行查询
+		_, err = database.Execute("SELECT id, name, rack, rank, cpus, memory, disk, os, kernel FROM nodes ORDER BY name")
+		if err != nil {
+			return fmt.Errorf("查询节点失败: %v", err)
+		}
+
+		// 获取所有结果
+		rows, err := database.FetchAll()
+		if err != nil {
+			return fmt.Errorf("获取结果失败: %v", err)
+		}
+
+		if len(rows) == 0 {
+			log.Info("暂无节点数据")
+			return nil
+		}
+
+		// 打印结果
+		fmt.Printf("%-5s %-20s %-8s %-8s %-8s %-10s %-10s %-10s\n",
+			"ID", "名称", "机架", "排名", "CPU", "内存", "磁盘", "操作系统")
+		fmt.Println("----------------------------------------------------------------------------------")
+		for _, row := range rows {
+			fmt.Printf("%-5v %-20s %-8v %-8v %-8v %-10v %-10v %-10s\n",
+				row["id"], row["name"], row["rack"], row["rank"],
+				row["cpus"], row["memory"], row["disk"], row["os"])
+		}
+
+		return nil
+	},
+}
+
+var nodeAddCmd = &cobra.Command{
+	Use:   "add <name>",
+	Short: "添加节点",
+	Args:  cobra.ExactArgs(1),
+	RunE: func(cmd *cobra.Command, args []string) error {
+		nodeName := args[0]
+		log.Infof("添加节点: %s", nodeName)
+
+		database, err := db.GetInstance()
+		if err != nil {
+			return fmt.Errorf("获取数据库连接失败: %v", err)
+		}
+		defer database.Close()
+
+		// 插入节点
+		_, err = database.Execute(
+			"INSERT INTO nodes (name, rack, rank, cpus, memory, disk, os, kernel) VALUES (?, ?, ?, ?, ?, ?, ?, ?)",
+			nodeName, rack, rank, cpus, memory, disk, nodeOS, kernel,
+		)
+		if err != nil {
+			return fmt.Errorf("添加节点失败: %v", err)
+		}
+
+		log.Infof("节点 %s 添加成功", nodeName)
+		return nil
+	},
+}
+
+var (
+	rack   int
+	rank   int
+	cpus   int
+	memory int
+	disk   int
+	nodeOS string
+	kernel string
+)
+
+func init() {
+	// 添加子命令
+	nodeCmd.AddCommand(nodeListCmd)
+	nodeCmd.AddCommand(nodeAddCmd)
+
+	// 添加参数
+	nodeAddCmd.Flags().IntVar(&rack, "rack", 0, "机架号")
+	nodeAddCmd.Flags().IntVar(&rank, "rank", 0, "排名")
+	nodeAddCmd.Flags().IntVar(&cpus, "cpus", 0, "CPU核心数")
+	nodeAddCmd.Flags().IntVar(&memory, "memory", 0, "内存大小(GB)")
+	nodeAddCmd.Flags().IntVar(&disk, "disk", 0, "磁盘大小(GB)")
+	nodeAddCmd.Flags().StringVar(&nodeOS, "os", "", "操作系统")
+	nodeAddCmd.Flags().StringVar(&kernel, "kernel", "", "内核版本")
+}