愉快的使用cursor
This commit is contained in:
@@ -1,12 +1,16 @@
|
|||||||
---
|
---
|
||||||
|
description:
|
||||||
|
globs:
|
||||||
|
---
|
||||||
|
---
|
||||||
description: 构建agent-wdd的特定上下文的规则
|
description: 构建agent-wdd的特定上下文的规则
|
||||||
globs: *,go
|
globs: *.go
|
||||||
---
|
---
|
||||||
|
|
||||||
# 你是一个精通golang的编程大师,熟练掌握github.com/spf13/cobra框架,能够构建出非常现代的cli工具
|
# 你是一个精通golang的编程大师,熟练掌握github.com/spf13/cobra框架,能够构建出非常现代的cli工具
|
||||||
|
|
||||||
# 整个项目的架构结构如下
|
# 整个项目的架构结构如下
|
||||||
1. base 服务器基础操作 相关的功能存放于@Base.go中
|
1. base 服务器基础操作 相关的功能存放于 [Base.go](mdc:agent-wdd/cmd/Base.go)
|
||||||
1. docker docker相关的操作
|
1. docker docker相关的操作
|
||||||
1. online 使用网络安装特定版本的docker
|
1. online 使用网络安装特定版本的docker
|
||||||
2. remove 卸载docker
|
2. remove 卸载docker
|
||||||
|
|||||||
9
.cursorignore
Normal file
9
.cursorignore
Normal file
@@ -0,0 +1,9 @@
|
|||||||
|
# Add directories or file patterns to ignore during indexing (e.g. foo/ or *.csv)
|
||||||
|
|
||||||
|
./agent-deploy/
|
||||||
|
./message_pusher/
|
||||||
|
./port_forwarding/
|
||||||
|
./server/
|
||||||
|
./server-go/
|
||||||
|
./socks_txthinking/
|
||||||
|
./source/
|
||||||
@@ -1,12 +1,20 @@
|
|||||||
<component name="ProjectRunConfigurationManager">
|
<component name="ProjectRunConfigurationManager">
|
||||||
<configuration default="false" name="ServerApplication" type="SpringBootApplicationConfigurationType"
|
<configuration default="false" name="ServerApplication" type="SpringBootApplicationConfigurationType" factoryName="Spring Boot">
|
||||||
factoryName="Spring Boot">
|
<module name="server" />
|
||||||
<module name="server"/>
|
<projectPathOnTarget name="projectPathOnTarget" value="/data/wdd/ProjectOctopus" />
|
||||||
<projectPathOnTarget name="projectPathOnTarget" value="/data/wdd/ProjectOctopus"/>
|
<target name="@@@LOCAL@@@" />
|
||||||
<target name="@@@LOCAL@@@"/>
|
<option name="SPRING_BOOT_MAIN_CLASS" value="io.wdd.ServerApplication" />
|
||||||
<option name="SPRING_BOOT_MAIN_CLASS" value="io.wdd.ServerApplication"/>
|
<method v="2">
|
||||||
<method v="2">
|
<option name="Make" enabled="true" />
|
||||||
<option name="Make" enabled="true"/>
|
</method>
|
||||||
</method>
|
</configuration>
|
||||||
</configuration>
|
<configuration default="false" name="ServerApplication" type="SpringBootApplicationConfigurationType" factoryName="Spring Boot">
|
||||||
|
<module name="server" />
|
||||||
|
<projectPathOnTarget name="projectPathOnTarget" value="/data/wdd/ProjectOctopus" />
|
||||||
|
<target name="@@@LOCAL@@@" />
|
||||||
|
<option name="SPRING_BOOT_MAIN_CLASS" value="io.wdd.ServerApplication" />
|
||||||
|
<method v="2">
|
||||||
|
<option name="Make" enabled="true" />
|
||||||
|
</method>
|
||||||
|
</configuration>
|
||||||
</component>
|
</component>
|
||||||
@@ -1,11 +1,13 @@
|
|||||||
package cmd
|
package cmd
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"agent-wdd/cmd/beans"
|
||||||
"agent-wdd/config"
|
"agent-wdd/config"
|
||||||
"agent-wdd/log"
|
"agent-wdd/log"
|
||||||
"agent-wdd/op"
|
"agent-wdd/op"
|
||||||
"agent-wdd/utils"
|
"agent-wdd/utils"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"runtime"
|
||||||
|
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
@@ -18,9 +20,18 @@ var (
|
|||||||
"deltarpm", "net-tools", "iputils", "bind-utils", "lsof", "curl", "wget", "vim", "mtr", "htop",
|
"deltarpm", "net-tools", "iputils", "bind-utils", "lsof", "curl", "wget", "vim", "mtr", "htop",
|
||||||
}
|
}
|
||||||
|
|
||||||
dockerLocalInstallPath = "/root/wdd/docker-20.10.15.tgz" // 本地安装docker的文件路径
|
dockerLocalInstallPath = "/root/wdd/docker-amd64-20.10.15.tgz" // 本地安装docker的文件路径
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
switch runtime.GOARCH {
|
||||||
|
case "amd64":
|
||||||
|
dockerLocalInstallPath = "/root/wdd/docker-amd64-20.10.15.tgz" // 本地安装docker的文件路径
|
||||||
|
case "arm64":
|
||||||
|
dockerLocalInstallPath = "/root/wdd/docker-arm64-20.10.15.tgz" // 本地安装docker的文件路径
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// 添加base子命令
|
// 添加base子命令
|
||||||
func addBaseSubcommands(cmd *cobra.Command) {
|
func addBaseSubcommands(cmd *cobra.Command) {
|
||||||
// 1.1 docker
|
// 1.1 docker
|
||||||
@@ -88,7 +99,7 @@ func addDockerSubcommands(cmd *cobra.Command) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 没有传递参数,使用默认参数
|
// 没有传递参数,使用默认参数
|
||||||
version := "26.0.7"
|
version := "20.10.15"
|
||||||
log.Info("Installing Docker version: %s", version)
|
log.Info("Installing Docker version: %s", version)
|
||||||
|
|
||||||
// 安装docker
|
// 安装docker
|
||||||
@@ -113,6 +124,9 @@ func addDockerSubcommands(cmd *cobra.Command) {
|
|||||||
Short: "本地安装Docker",
|
Short: "本地安装Docker",
|
||||||
Args: cobra.ExactArgs(1),
|
Args: cobra.ExactArgs(1),
|
||||||
Run: func(cmd *cobra.Command, args []string) {
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
|
|
||||||
|
log.Info("Installing Docker from local file: %s", dockerLocalInstallPath)
|
||||||
|
|
||||||
exist := utils.FileExistAndNotNull(dockerLocalInstallPath)
|
exist := utils.FileExistAndNotNull(dockerLocalInstallPath)
|
||||||
if !exist {
|
if !exist {
|
||||||
log.Error("Docker local install file not found: %s", dockerLocalInstallPath)
|
log.Error("Docker local install file not found: %s", dockerLocalInstallPath)
|
||||||
@@ -120,7 +134,72 @@ func addDockerSubcommands(cmd *cobra.Command) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 解压文件
|
// 解压文件
|
||||||
utils.Unzip(dockerLocalInstallPath, "/root/wdd")
|
utils.UnzipFile(dockerLocalInstallPath, "/root/wdd")
|
||||||
|
|
||||||
|
// 安装docker
|
||||||
|
err := utils.MoveFolerToAnother("/root/wdd/docker", "/usr/bin")
|
||||||
|
if err != nil {
|
||||||
|
log.Error("Failed to move Docker binaries: %s", err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 设置权限
|
||||||
|
chmodCmd := []string{"chmod", "777", "-R", "/usr/bin/docker*"}
|
||||||
|
ok, resultLog := op.SingleLineCommandExecutor(chmodCmd)
|
||||||
|
if !ok {
|
||||||
|
log.Error("Failed to set permissions for Docker binaries: %s", resultLog)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 配置并启动Docker服务
|
||||||
|
// systemd daemonize docker
|
||||||
|
utils.AppendOverwriteContentToFile(beans.ContainerdDaemonService, beans.ContainerdServiceFile)
|
||||||
|
if !utils.FileExistAndNotNull(beans.ContainerdServiceFile) {
|
||||||
|
log.Error("docker deamon file not exists !")
|
||||||
|
}
|
||||||
|
utils.AppendOverwriteContentToFile(beans.DockerSocketDaemonService, beans.DockerSocketFile)
|
||||||
|
if !utils.FileExistAndNotNull(beans.DockerSocketFile) {
|
||||||
|
log.Error("docker deamon file not exists !")
|
||||||
|
}
|
||||||
|
utils.AppendOverwriteContentToFile(beans.DockerDaemonService, beans.DockerServiceFile)
|
||||||
|
if !utils.FileExistAndNotNull(beans.DockerServiceFile) {
|
||||||
|
log.Error("docker deamon file not exists !")
|
||||||
|
}
|
||||||
|
log.Info("docker dameon file append success !")
|
||||||
|
|
||||||
|
ok, resultLog = op.SystemdDaemonReload()
|
||||||
|
if !ok {
|
||||||
|
log.Error("daemon reload error ! %s", resultLog)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
op.SingleLineCommandExecutor([]string{"systemctl", "unmask", "containerd"})
|
||||||
|
op.SingleLineCommandExecutor([]string{"systemctl", "unmask", "docker.socket"})
|
||||||
|
op.SingleLineCommandExecutor([]string{"systemctl", "unmask", "docker"})
|
||||||
|
|
||||||
|
op.SingleLineCommandExecutor([]string{"groupadd", "docker"})
|
||||||
|
op.SingleLineCommandExecutor([]string{"usermod", "-aG", "docker", "root"})
|
||||||
|
op.SingleLineCommandExecutor([]string{"newgrp", "docker"})
|
||||||
|
|
||||||
|
systemdUp, resultLog := op.SystemdUp("containerd")
|
||||||
|
if !systemdUp {
|
||||||
|
log.Error("[InstallDockerBastion] - start containerd service error ! %s", resultLog)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
ok, resultLog = op.SystemdUp("docker.socket")
|
||||||
|
if !ok {
|
||||||
|
log.Error("[InstallDockerBastion] - start docker.socket error ! %s", resultLog)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
ok, resultLog = op.SystemdUp("docker")
|
||||||
|
if !ok {
|
||||||
|
log.Error("[InstallDockerBastion] - start docker service error ! %s", resultLog)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Info("Docker installed successfully from local file!")
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
121
agent-wdd/cmd/beans/DockerDaemonConfig.go
Normal file
121
agent-wdd/cmd/beans/DockerDaemonConfig.go
Normal file
@@ -0,0 +1,121 @@
|
|||||||
|
package beans
|
||||||
|
|
||||||
|
var ContainerdServiceFile = "/lib/systemd/system/containerd.service"
|
||||||
|
var DockerSocketFile = "/lib/systemd/system/docker.socket"
|
||||||
|
var DockerServiceFile = "/lib/systemd/system/docker.service"
|
||||||
|
|
||||||
|
var ContainerdDaemonService = `
|
||||||
|
# Copyright The containerd Authors.
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
|
||||||
|
[Unit]
|
||||||
|
Description=containerd container runtime
|
||||||
|
Documentation=https://containerd.io
|
||||||
|
After=network.target local-fs.target
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
ExecStartPre=-/sbin/modprobe overlay
|
||||||
|
ExecStart=/usr/bin/containerd
|
||||||
|
|
||||||
|
Type=notify
|
||||||
|
Delegate=yes
|
||||||
|
KillMode=process
|
||||||
|
Restart=always
|
||||||
|
RestartSec=5
|
||||||
|
# Having non-zero Limit*s causes performance problems due to accounting overhead
|
||||||
|
# in the kernel. We recommend using cgroups to do container-local accounting.
|
||||||
|
LimitNPROC=infinity
|
||||||
|
LimitCORE=infinity
|
||||||
|
LimitNOFILE=infinity
|
||||||
|
# Comment TasksMax if your systemd version does not supports it.
|
||||||
|
# Only systemd 226 and above support this version.
|
||||||
|
TasksMax=infinity
|
||||||
|
OOMScoreAdjust=-999
|
||||||
|
|
||||||
|
[installPrefix]
|
||||||
|
WantedBy=multi-user.target
|
||||||
|
`
|
||||||
|
|
||||||
|
var DockerSocketDaemonService = `
|
||||||
|
[Unit]
|
||||||
|
Description=Docker Socket for the API
|
||||||
|
|
||||||
|
[Socket]
|
||||||
|
ListenStream=/var/run/docker.sock
|
||||||
|
SocketMode=0660
|
||||||
|
SocketUser=root
|
||||||
|
SocketGroup=docker
|
||||||
|
|
||||||
|
[installPrefix]
|
||||||
|
WantedBy=sockets.target
|
||||||
|
`
|
||||||
|
|
||||||
|
var DockerDaemonService = `
|
||||||
|
[Unit]
|
||||||
|
Description=Docker Application Container Engine
|
||||||
|
Documentation=https://docs.docker.com
|
||||||
|
After=network-online.target docker.socket firewalld.service containerd.service
|
||||||
|
Wants=network-online.target
|
||||||
|
Requires=docker.socket containerd.service
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
Type=notify
|
||||||
|
# the default is not to use systemd for cgroups because the delegate issues still
|
||||||
|
# exists and systemd currently does not support the cgroup feature set required
|
||||||
|
# for containers run by docker
|
||||||
|
ExecStart=/usr/bin/dockerd -H fd:// --containerd=/run/containerd/containerd.sock
|
||||||
|
ExecReload=/bin/kill -s HUP $MAINPID
|
||||||
|
TimeoutSec=0
|
||||||
|
RestartSec=2
|
||||||
|
Restart=always
|
||||||
|
|
||||||
|
# Note that StartLimit* options were moved from "Service" to "Unit" in systemd 229.
|
||||||
|
# Both the old, and new location are accepted by systemd 229 and up, so using the old location
|
||||||
|
# to make them work for either version of systemd.
|
||||||
|
StartLimitBurst=3
|
||||||
|
|
||||||
|
# Note that StartLimitInterval was renamed to StartLimitIntervalSec in systemd 230.
|
||||||
|
# Both the old, and new name are accepted by systemd 230 and up, so using the old name to make
|
||||||
|
# this option work for either version of systemd.
|
||||||
|
StartLimitInterval=60s
|
||||||
|
|
||||||
|
# Having non-zero Limit*s causes performance problems due to accounting overhead
|
||||||
|
# in the kernel. We recommend using cgroups to do container-local accounting.
|
||||||
|
LimitNOFILE=infinity
|
||||||
|
LimitNPROC=infinity
|
||||||
|
LimitCORE=infinity
|
||||||
|
|
||||||
|
# Comment TasksMax if your systemd version does not support it.
|
||||||
|
# Only systemd 226 and above support this option.
|
||||||
|
TasksMax=infinity
|
||||||
|
|
||||||
|
# set delegate yes so that systemd does not reset the cgroups of docker containers
|
||||||
|
Delegate=yes
|
||||||
|
|
||||||
|
# kill only the docker process, not all processes in the cgroup
|
||||||
|
KillMode=process
|
||||||
|
OOMScoreAdjust=-500
|
||||||
|
|
||||||
|
[installPrefix]
|
||||||
|
WantedBy=multi-user.target
|
||||||
|
`
|
||||||
|
|
||||||
|
var DockerDeamonConfig = `
|
||||||
|
{
|
||||||
|
"insecure-registries": [
|
||||||
|
"DockerRegisterDomain:8033",
|
||||||
|
"harbor.wdd.io:8033"
|
||||||
|
]
|
||||||
|
}
|
||||||
|
`
|
||||||
@@ -5,6 +5,7 @@ go 1.23.5
|
|||||||
require (
|
require (
|
||||||
github.com/spf13/cobra v1.8.1
|
github.com/spf13/cobra v1.8.1
|
||||||
github.com/spf13/viper v1.19.0
|
github.com/spf13/viper v1.19.0
|
||||||
|
gopkg.in/yaml.v3 v3.0.1
|
||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
@@ -27,5 +28,5 @@ require (
|
|||||||
golang.org/x/sys v0.18.0 // indirect
|
golang.org/x/sys v0.18.0 // indirect
|
||||||
golang.org/x/text v0.14.0 // indirect
|
golang.org/x/text v0.14.0 // indirect
|
||||||
gopkg.in/ini.v1 v1.67.0 // indirect
|
gopkg.in/ini.v1 v1.67.0 // indirect
|
||||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -9,6 +9,65 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// SingleLineCommandExecutor 执行单行shell命令,返回执行结果和输出内容
|
||||||
|
// singleLineCommand: 命令及参数,如 []string{"ls", "-l"}
|
||||||
|
// 返回值:
|
||||||
|
//
|
||||||
|
// bool - 命令是否执行成功(true为成功,false为失败)
|
||||||
|
// []string - 合并后的标准输出和标准错误内容(按行分割)
|
||||||
|
func SingleLineCommandExecutor(singleLineCommand []string) (ok bool, resultLog []string) {
|
||||||
|
|
||||||
|
log.Info("[Excutor] - start => %v", singleLineCommand)
|
||||||
|
|
||||||
|
if len(singleLineCommand) == 0 {
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 创建命令实例
|
||||||
|
cmd := exec.Command(singleLineCommand[0], singleLineCommand[1:]...)
|
||||||
|
|
||||||
|
// 创建输出缓冲区
|
||||||
|
var stdoutBuf, stderrBuf bytes.Buffer
|
||||||
|
cmd.Stdout = &stdoutBuf
|
||||||
|
cmd.Stderr = &stderrBuf
|
||||||
|
|
||||||
|
// 执行命令并获取错误信息
|
||||||
|
err := cmd.Run()
|
||||||
|
|
||||||
|
// 合并输出结果
|
||||||
|
output := mergeOutput(&stdoutBuf, &stderrBuf)
|
||||||
|
|
||||||
|
// 判断执行结果
|
||||||
|
return err == nil, output
|
||||||
|
}
|
||||||
|
|
||||||
|
// 若追求极致性能,可优化为流式合并
|
||||||
|
func mergeOutput(stdout, stderr *bytes.Buffer) []string {
|
||||||
|
combined := make([]string, 0, bytes.Count(stdout.Bytes(), []byte{'\n'})+bytes.Count(stderr.Bytes(), []byte{'\n'}))
|
||||||
|
combined = append(combined, strings.Split(stdout.String(), "\n")...)
|
||||||
|
combined = append(combined, strings.Split(stderr.String(), "\n")...)
|
||||||
|
return combined
|
||||||
|
}
|
||||||
|
|
||||||
|
// mergeOutput 合并标准输出和标准错误,按行分割
|
||||||
|
//func mergeOutput(stdoutBuf, stderrBuf *bytes.Buffer) []string {
|
||||||
|
// var output []string
|
||||||
|
//
|
||||||
|
// // 处理标准输出
|
||||||
|
// scanner := bufio.NewScanner(stdoutBuf)
|
||||||
|
// for scanner.Scan() {
|
||||||
|
// output = append(output, scanner.Text())
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// // 处理标准错误
|
||||||
|
// scanner = bufio.NewScanner(stderrBuf)
|
||||||
|
// for scanner.Scan() {
|
||||||
|
// output = append(output, scanner.Text())
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// return output
|
||||||
|
//}
|
||||||
|
|
||||||
// PipeLineCommandExecutor 执行管道命令,返回执行结果和合并后的输出内容
|
// PipeLineCommandExecutor 执行管道命令,返回执行结果和合并后的输出内容
|
||||||
// pipeLineCommand: 管道命令组,如 [][]string{{"ps", "aux"}, {"grep", "nginx"}, {"wc", "-l"}}
|
// pipeLineCommand: 管道命令组,如 [][]string{{"ps", "aux"}, {"grep", "nginx"}, {"wc", "-l"}}
|
||||||
// 返回值:
|
// 返回值:
|
||||||
@@ -102,65 +161,6 @@ func PipeLineCommandExecutor(pipeLineCommand [][]string) (ok bool, resultLog []s
|
|||||||
return success, output
|
return success, output
|
||||||
}
|
}
|
||||||
|
|
||||||
// SingleLineCommandExecutor 执行单行shell命令,返回执行结果和输出内容
|
|
||||||
// singleLineCommand: 命令及参数,如 []string{"ls", "-l"}
|
|
||||||
// 返回值:
|
|
||||||
//
|
|
||||||
// bool - 命令是否执行成功(true为成功,false为失败)
|
|
||||||
// []string - 合并后的标准输出和标准错误内容(按行分割)
|
|
||||||
func SingleLineCommandExecutor(singleLineCommand []string) (ok bool, resultLog []string) {
|
|
||||||
|
|
||||||
log.Info("[Excutor] - start => %v", singleLineCommand)
|
|
||||||
|
|
||||||
if len(singleLineCommand) == 0 {
|
|
||||||
return false, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// 创建命令实例
|
|
||||||
cmd := exec.Command(singleLineCommand[0], singleLineCommand[1:]...)
|
|
||||||
|
|
||||||
// 创建输出缓冲区
|
|
||||||
var stdoutBuf, stderrBuf bytes.Buffer
|
|
||||||
cmd.Stdout = &stdoutBuf
|
|
||||||
cmd.Stderr = &stderrBuf
|
|
||||||
|
|
||||||
// 执行命令并获取错误信息
|
|
||||||
err := cmd.Run()
|
|
||||||
|
|
||||||
// 合并输出结果
|
|
||||||
output := mergeOutput(&stdoutBuf, &stderrBuf)
|
|
||||||
|
|
||||||
// 判断执行结果
|
|
||||||
return err == nil, output
|
|
||||||
}
|
|
||||||
|
|
||||||
// mergeOutput 合并标准输出和标准错误,按行分割
|
|
||||||
//func mergeOutput(stdoutBuf, stderrBuf *bytes.Buffer) []string {
|
|
||||||
// var output []string
|
|
||||||
//
|
|
||||||
// // 处理标准输出
|
|
||||||
// scanner := bufio.NewScanner(stdoutBuf)
|
|
||||||
// for scanner.Scan() {
|
|
||||||
// output = append(output, scanner.Text())
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// // 处理标准错误
|
|
||||||
// scanner = bufio.NewScanner(stderrBuf)
|
|
||||||
// for scanner.Scan() {
|
|
||||||
// output = append(output, scanner.Text())
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// return output
|
|
||||||
//}
|
|
||||||
|
|
||||||
// 若追求极致性能,可优化为流式合并
|
|
||||||
func mergeOutput(stdout, stderr *bytes.Buffer) []string {
|
|
||||||
combined := make([]string, 0, bytes.Count(stdout.Bytes(), []byte{'\n'})+bytes.Count(stderr.Bytes(), []byte{'\n'}))
|
|
||||||
combined = append(combined, strings.Split(stdout.String(), "\n")...)
|
|
||||||
combined = append(combined, strings.Split(stderr.String(), "\n")...)
|
|
||||||
return combined
|
|
||||||
}
|
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
// 成功案例
|
// 成功案例
|
||||||
success, output := SingleLineCommandExecutor([]string{"ls", "-l"})
|
success, output := SingleLineCommandExecutor([]string{"ls", "-l"})
|
||||||
|
|||||||
@@ -8,9 +8,9 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type PackageOperator struct {
|
type PackageOperator struct {
|
||||||
installPrefix []string `json:"install" yaml:"install"` // 安装前缀
|
installPrefix []string // 安装前缀
|
||||||
removePrefix []string `json:"remove" yaml:"remove"` // 移除前缀
|
removePrefix []string // 移除前缀
|
||||||
upgradePrefix []string `json:"upgrade" yaml:"upgrade"` // 升级前缀
|
upgradePrefix []string // 升级前缀
|
||||||
initCommand []string
|
initCommand []string
|
||||||
}
|
}
|
||||||
|
|
||||||
81
agent-wdd/op/SystemdExcutor.go
Normal file
81
agent-wdd/op/SystemdExcutor.go
Normal file
@@ -0,0 +1,81 @@
|
|||||||
|
package op
|
||||||
|
|
||||||
|
// SystemdUp 启动服务
|
||||||
|
func SystemdUp(serviceName string) (bool, []string) {
|
||||||
|
ok, resultLog := SingleLineCommandExecutor([]string{"systemctl", "enable", serviceName})
|
||||||
|
if !ok {
|
||||||
|
return false, resultLog
|
||||||
|
}
|
||||||
|
ok, resultLog = SingleLineCommandExecutor([]string{"systemctl", "start", serviceName})
|
||||||
|
if !ok {
|
||||||
|
return false, resultLog
|
||||||
|
}
|
||||||
|
return true, resultLog
|
||||||
|
}
|
||||||
|
|
||||||
|
// SystemdDown 停止服务
|
||||||
|
func SystemdDown(serviceName string) (bool, []string) {
|
||||||
|
ok, resultLog := SingleLineCommandExecutor([]string{"systemctl", "disable", serviceName})
|
||||||
|
if !ok {
|
||||||
|
return false, resultLog
|
||||||
|
}
|
||||||
|
ok, resultLog = SingleLineCommandExecutor([]string{"systemctl", "stop", serviceName})
|
||||||
|
if !ok {
|
||||||
|
return false, resultLog
|
||||||
|
}
|
||||||
|
return true, resultLog
|
||||||
|
}
|
||||||
|
|
||||||
|
// SystemdStatus 查看服务状态
|
||||||
|
func SystemdStatus(serviceName string) (bool, []string) {
|
||||||
|
ok, resultLog := SingleLineCommandExecutor([]string{"systemctl", "status", serviceName})
|
||||||
|
if !ok {
|
||||||
|
return false, resultLog
|
||||||
|
}
|
||||||
|
return true, resultLog
|
||||||
|
}
|
||||||
|
|
||||||
|
// SystemdRestart 重启服务
|
||||||
|
func SystemdRestart(serviceName string) (bool, []string) {
|
||||||
|
ok, resultLog := SingleLineCommandExecutor([]string{"systemctl", "restart", serviceName})
|
||||||
|
if !ok {
|
||||||
|
return false, resultLog
|
||||||
|
}
|
||||||
|
return true, resultLog
|
||||||
|
}
|
||||||
|
|
||||||
|
// SystemdReload 重新加载服务
|
||||||
|
func SystemdReload(serviceName string) (bool, []string) {
|
||||||
|
ok, resultLog := SingleLineCommandExecutor([]string{"systemctl", "reload", serviceName})
|
||||||
|
if !ok {
|
||||||
|
return false, resultLog
|
||||||
|
}
|
||||||
|
return true, resultLog
|
||||||
|
}
|
||||||
|
|
||||||
|
// SystemdEnable 启用服务
|
||||||
|
func SystemdEnable(serviceName string) (bool, []string) {
|
||||||
|
ok, resultLog := SingleLineCommandExecutor([]string{"systemctl", "enable", serviceName})
|
||||||
|
if !ok {
|
||||||
|
return false, resultLog
|
||||||
|
}
|
||||||
|
return true, resultLog
|
||||||
|
}
|
||||||
|
|
||||||
|
// SystemdDisable 禁用服务
|
||||||
|
func SystemdDisable(serviceName string) (bool, []string) {
|
||||||
|
ok, resultLog := SingleLineCommandExecutor([]string{"systemctl", "disable", serviceName})
|
||||||
|
if !ok {
|
||||||
|
return false, resultLog
|
||||||
|
}
|
||||||
|
return true, resultLog
|
||||||
|
}
|
||||||
|
|
||||||
|
// SystemdDaemonReload 重新加载服务
|
||||||
|
func SystemdDaemonReload() (bool, []string) {
|
||||||
|
ok, resultLog := SingleLineCommandExecutor([]string{"systemctl", "daemon-reload"})
|
||||||
|
if !ok {
|
||||||
|
return false, resultLog
|
||||||
|
}
|
||||||
|
return true, resultLog
|
||||||
|
}
|
||||||
@@ -283,7 +283,7 @@ func ReadAllContentFromFile(fileFullPath string) (result []string) {
|
|||||||
return result
|
return result
|
||||||
}
|
}
|
||||||
|
|
||||||
// MoveFolerToAnother 将源文件夹中除了子文件夹外的所有文件移动到目标文件夹
|
// MoveFolerToAnother 将源文件夹的所有文件递归移动到目标文件夹
|
||||||
func MoveFolerToAnother(srcDir, dstDir string) error {
|
func MoveFolerToAnother(srcDir, dstDir string) error {
|
||||||
// 读取源文件夹中的所有条目
|
// 读取源文件夹中的所有条目
|
||||||
entries, err := os.ReadDir(srcDir)
|
entries, err := os.ReadDir(srcDir)
|
||||||
@@ -291,24 +291,36 @@ func MoveFolerToAnother(srcDir, dstDir string) error {
|
|||||||
return fmt.Errorf("读取源文件夹失败: %w", err)
|
return fmt.Errorf("读取源文件夹失败: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 创建目标文件夹(如果不存在)
|
||||||
|
if err := os.MkdirAll(dstDir, os.ModePerm); err != nil {
|
||||||
|
return fmt.Errorf("创建目标文件夹失败: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
// 遍历所有条目
|
// 遍历所有条目
|
||||||
for _, entry := range entries {
|
for _, entry := range entries {
|
||||||
// 跳过子文件夹
|
|
||||||
if entry.IsDir() {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
// 构造源文件路径和目标文件路径
|
|
||||||
srcPath := filepath.Join(srcDir, entry.Name())
|
srcPath := filepath.Join(srcDir, entry.Name())
|
||||||
dstPath := filepath.Join(dstDir, entry.Name())
|
dstPath := filepath.Join(dstDir, entry.Name())
|
||||||
|
|
||||||
// 移动文件
|
// 如果是文件夹,递归处理
|
||||||
if err := os.Rename(srcPath, dstPath); err != nil {
|
if entry.IsDir() {
|
||||||
return fmt.Errorf("移动文件失败: %w", err)
|
if err := MoveFolerToAnother(srcPath, dstPath); err != nil {
|
||||||
|
return fmt.Errorf("递归移动文件夹失败: %w", err)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// 移动文件
|
||||||
|
if err := os.Rename(srcPath, dstPath); err != nil {
|
||||||
|
return fmt.Errorf("移动文件失败: %w", err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 删除源文件夹
|
||||||
|
if err := os.RemoveAll(srcDir); err != nil {
|
||||||
|
return fmt.Errorf("删除源文件夹失败: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetCurrentUserFolder 获取运行环境当前用户的根目录
|
// GetCurrentUserFolder 获取运行环境当前用户的根目录
|
||||||
|
|||||||
150
agent-wdd/utils/ZipUtils.go
Normal file
150
agent-wdd/utils/ZipUtils.go
Normal file
@@ -0,0 +1,150 @@
|
|||||||
|
package utils
|
||||||
|
|
||||||
|
import (
|
||||||
|
"agent-wdd/log"
|
||||||
|
"archive/tar"
|
||||||
|
"archive/zip"
|
||||||
|
"compress/gzip"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
)
|
||||||
|
|
||||||
|
// UnzipFile 解压文件, 支持zip和tar.gz
|
||||||
|
func UnzipFile(zipFile string, targetFolder string) {
|
||||||
|
// 检查文件扩展名以确定解压缩方法
|
||||||
|
fileExt := filepath.Ext(zipFile)
|
||||||
|
switch fileExt {
|
||||||
|
case ".zip":
|
||||||
|
// 使用标准库中的archive/zip包进行解压缩
|
||||||
|
zipReader, err := zip.OpenReader(zipFile)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("无法打开ZIP文件: %s, 错误: %s", zipFile, err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer zipReader.Close()
|
||||||
|
|
||||||
|
// 创建目标文件夹
|
||||||
|
if err := os.MkdirAll(targetFolder, 0755); err != nil {
|
||||||
|
log.Error("无法创建目标文件夹: %s, 错误: %s", targetFolder, err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 遍历ZIP文件中的所有文件
|
||||||
|
for _, file := range zipReader.File {
|
||||||
|
zippedFile, err := file.Open()
|
||||||
|
if err != nil {
|
||||||
|
log.Error("无法打开ZIP中的文件: %s, 错误: %s", file.Name, err.Error())
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
defer zippedFile.Close()
|
||||||
|
|
||||||
|
// 构建目标文件路径
|
||||||
|
targetFilePath := filepath.Join(targetFolder, file.Name)
|
||||||
|
|
||||||
|
// 如果是目录,则创建目录
|
||||||
|
if file.FileInfo().IsDir() {
|
||||||
|
if err := os.MkdirAll(targetFilePath, file.Mode()); err != nil {
|
||||||
|
log.Error("无法创建目录: %s, 错误: %s", targetFilePath, err.Error())
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// 如果是文件,则创建文件并写入内容
|
||||||
|
if err := os.MkdirAll(filepath.Dir(targetFilePath), 0755); err != nil {
|
||||||
|
log.Error("无法创建文件的父目录: %s, 错误: %s", filepath.Dir(targetFilePath), err.Error())
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
targetFile, err := os.OpenFile(targetFilePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, file.Mode())
|
||||||
|
if err != nil {
|
||||||
|
log.Error("无法创建目标文件: %s, 错误: %s", targetFilePath, err.Error())
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
defer targetFile.Close()
|
||||||
|
|
||||||
|
if _, err := io.Copy(targetFile, zippedFile); err != nil {
|
||||||
|
log.Error("无法写入文件内容: %s, 错误: %s", targetFilePath, err.Error())
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
case ".tar", ".tar.gz", ".tgz":
|
||||||
|
// 使用标准库中的archive/tar包进行解压缩
|
||||||
|
var file *os.File
|
||||||
|
var err error
|
||||||
|
if fileExt == ".tar.gz" || fileExt == ".tgz" {
|
||||||
|
file, err = os.Open(zipFile)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("无法打开TAR.GZ文件: %s, 错误: %s", zipFile, err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
gzipReader, err := gzip.NewReader(file)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("无法创建GZIP读取器: %s, 错误: %s", zipFile, err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer gzipReader.Close()
|
||||||
|
|
||||||
|
tarReader := tar.NewReader(gzipReader)
|
||||||
|
if err := extractTar(tarReader, targetFolder); err != nil {
|
||||||
|
log.Error("解压TAR.GZ文件时出错: %s, 错误: %s", zipFile, err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
file, err = os.Open(zipFile)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("无法打开TAR文件: %s, 错误: %s", zipFile, err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
tarReader := tar.NewReader(file)
|
||||||
|
if err := extractTar(tarReader, targetFolder); err != nil {
|
||||||
|
log.Error("解压TAR文件时出错: %s, 错误: %s", zipFile, err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
default:
|
||||||
|
log.Error("不支持的文件类型: %s", fileExt)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 辅助函数,用于解压TAR文件
|
||||||
|
func extractTar(tarReader *tar.Reader, targetFolder string) error {
|
||||||
|
for {
|
||||||
|
header, err := tarReader.Next()
|
||||||
|
if err == io.EOF {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
targetFilePath := filepath.Join(targetFolder, header.Name)
|
||||||
|
|
||||||
|
switch header.Typeflag {
|
||||||
|
case tar.TypeDir:
|
||||||
|
if err := os.MkdirAll(targetFilePath, 0755); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
case tar.TypeReg:
|
||||||
|
if err := os.MkdirAll(filepath.Dir(targetFilePath), 0755); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
file, err := os.OpenFile(targetFilePath, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, os.FileMode(header.Mode))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
if _, err := io.Copy(file, tarReader); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user