114 lines
2.9 KiB
Go
114 lines
2.9 KiB
Go
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", "", "内核版本")
|
|
}
|