From e1b9c152d4179dcfc668fe88fb646dda1c2440d1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E7=A5=A5=E8=B6=85?= Date: Tue, 7 May 2024 08:39:58 +0800 Subject: [PATCH] =?UTF-8?q?=E5=AE=9E=E7=8E=B0=E8=BF=9C=E7=A8=8B=E5=8D=B8?= =?UTF-8?q?=E8=BD=BD=E8=8A=82=E7=82=B9=E5=8A=9F=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/installers/queue_node.go | 113 ++++++++++++++++++++++++++ internal/rpc/services/service_node.go | 25 ++++++ 2 files changed, 138 insertions(+) diff --git a/internal/installers/queue_node.go b/internal/installers/queue_node.go index f2312b4e..3f165078 100644 --- a/internal/installers/queue_node.go +++ b/internal/installers/queue_node.go @@ -313,6 +313,9 @@ func (this *NodeQueue) StartNode(nodeId int64) error { if err != nil { return errors.New("edge node was not installed correctly, can not find executable file") } + if len(exe) == 0 { + return errors.New("edge node was not installed correctly, can not find executable file") + } // 我们先尝试Systemd启动 _, _, _ = installer.client.Exec("systemctl start edge-node") @@ -420,6 +423,9 @@ func (this *NodeQueue) StopNode(nodeId int64) error { if err != nil { return errors.New("edge node was not installed correctly, can not find executable file") } + if len(exe) == 0 { + return errors.New("edge node was not installed correctly, can not find executable file") + } // 我们先尝试Systemd停止 _, _, _ = installer.client.Exec("/usr/bin/systemctl stop edge-node") @@ -436,6 +442,113 @@ func (this *NodeQueue) StopNode(nodeId int64) error { return nil } +// UninstallNode 卸载节点 +func (this *NodeQueue) UninstallNode(nodeId int64) error { + node, err := models.SharedNodeDAO.FindEnabledNode(nil, nodeId) + if err != nil { + return err + } + if node == nil { + return errors.New("can not find node, ID：'" + numberutils.FormatInt64(nodeId) + "'") + } + + // 登录信息 + login, err := models.SharedNodeLoginDAO.FindEnabledNodeLoginWithNodeId(nil, nodeconfigs.NodeRoleNode, nodeId) + if err != nil { + return err + } + if login == nil { + return errors.New("can not find node login information") + } + loginParams, err := login.DecodeSSHParams() + if err != nil { + return err + } + + if len(loginParams.Host) == 0 { + // 查询节点IP + ip, _, err := models.SharedNodeIPAddressDAO.FindFirstNodeAccessIPAddress(nil, nodeId, false, nodeconfigs.NodeRoleNode) + if err != nil { + return err + } + if len(ip) > 0 { + loginParams.Host = ip + } else { + return errors.New("ssh host should not be empty") + } + } + + if loginParams.Port <= 0 { + // 从集群中读取 + sshParams, err := models.SharedNodeClusterDAO.FindClusterSSHParams(nil, int64(node.ClusterId)) + if err != nil { + return err + } + if sshParams != nil && sshParams.Port > 0 { + loginParams.Port = sshParams.Port + } else { + return errors.New("ssh port is invalid") + } + } + + if loginParams.GrantId == 0 { + // 从集群中读取 + grantId, err := models.SharedNodeClusterDAO.FindClusterGrantId(nil, int64(node.ClusterId)) + if err != nil { + return err + } + if grantId == 0 { + return errors.New("can not find node grant") + } + loginParams.GrantId = grantId + } + grant, err := models.SharedNodeGrantDAO.FindEnabledNodeGrant(nil, loginParams.GrantId) + if err != nil { + return err + } + if grant == nil { + return errors.New("can not find user grant with id '" + numberutils.FormatInt64(loginParams.GrantId) + "'") + } + + var installer = &NodeInstaller{} + err = installer.Login(&Credentials{ + Host: loginParams.Host, + Port: loginParams.Port, + Username: grant.Username, + Password: grant.Password, + PrivateKey: grant.PrivateKey, + Passphrase: grant.Passphrase, + Method: grant.Method, + Sudo: grant.Su == 1, + }) + if err != nil { + return err + } + defer func() { + _ = installer.Close() + }() + + // 检查命令是否存在 + exe, err := this.lookupNodeExe(node, installer.client) + if err != nil { + return errors.New("edge node was not installed correctly, can not find executable file") + } + if len(exe) == 0 { + return errors.New("edge node was not installed correctly, can not find executable file") + } + + // 执行uninstall + _, stderr, err := installer.client.Exec(exe + " uninstall") + if err != nil { + return fmt.Errorf("uninstall failed: %w", err) + } + if len(stderr) > 0 { + return errors.New("uninstall failed: " + stderr) + } + + return nil +} + func (this *NodeQueue) lookupNodeExe(node *models.Node, client *SSHClient) (string, error) { // 安装目录 var nodeDirs = []string{} diff --git a/internal/rpc/services/service_node.go b/internal/rpc/services/service_node.go index f4e831ec..723f76b6 100644 --- a/internal/rpc/services/service_node.go +++ b/internal/rpc/services/service_node.go @@ -967,6 +967,31 @@ func (this *NodeService) StopNode(ctx context.Context, req *pb.StopNodeRequest) return &pb.StopNodeResponse{IsOk: true}, nil } +// UninstallNode 卸载节点 +func (this *NodeService) UninstallNode(ctx context.Context, req *pb.UninstallNodeRequest) (*pb.UninstallNodeResponse, error) { + _, err := this.ValidateAdmin(ctx) + if err != nil { + return nil, err + } + + err = installers.SharedNodeQueue().UninstallNode(req.NodeId) + if err != nil { + return &pb.UninstallNodeResponse{ + IsOk: false, + Error: err.Error(), + }, nil + } + + // 修改为未安装 + var tx = this.NullTx() + err = models.SharedNodeDAO.UpdateNodeIsInstalled(tx, req.NodeId, false) + if err != nil { + return nil, err + } + + return &pb.UninstallNodeResponse{IsOk: true}, nil +} + // UpdateNodeConnectedAPINodes 更改节点连接的API节点信息 func (this *NodeService) UpdateNodeConnectedAPINodes(ctx context.Context, req *pb.UpdateNodeConnectedAPINodesRequest) (*pb.RPCSuccess, error) { // 校验节点