[ Executor ] exec function - 1
This commit is contained in:
zeaslity
252
agent-go/executor/BaseFunction.go
Normal file
252
agent-go/executor/BaseFunction.go
Normal file
@@ -0,0 +1,252 @@
|
||||
package executor
|
||||
|
||||
import "strings"
|
||||
|
||||
type BaseFunc interface {
|
||||
Exec(baseFuncName string, funcArgs ...string) string
|
||||
}
|
||||
|
||||
type AgentOsOperator struct {
|
||||
InstallCommandPrefix string `json:"install_command_prefix",comment:"apt-get install or yum install"`
|
||||
|
||||
RemoveCommandPrefix string `json:"remove_command_prefix",comment:"apt-get remove or yum remove"`
|
||||
|
||||
CanAccessInternet bool `json:"can_access_internet",comment:"是否可以访问公网"`
|
||||
|
||||
IsOsTypeUbuntu bool `json:"is_os_type_ubuntu",comment:"主机操作系统是否为ubuntu系列"`
|
||||
|
||||
IsAgentInnerWall bool `json:"is_agent_inner_wall", comment:"主机是否身处国内"`
|
||||
}
|
||||
|
||||
// Exec 执行基础功能函数
|
||||
func (op *AgentOsOperator) Exec(baseFuncName string, funcArgs ...string) string {
|
||||
|
||||
var multiLineCommand [][]string
|
||||
|
||||
switch baseFuncName {
|
||||
|
||||
case "shutdownFirewall":
|
||||
multiLineCommand = op.shutdownFirewall()
|
||||
break
|
||||
case "modifyHostname":
|
||||
multiLineCommand = op.modifyHostname(funcArgs)
|
||||
break
|
||||
case "disableSwap":
|
||||
multiLineCommand = op.disableSwap()
|
||||
break
|
||||
case "installDocker":
|
||||
multiLineCommand = op.installDocker(funcArgs)
|
||||
break
|
||||
case "removeDocker":
|
||||
multiLineCommand = op.removeDocker()
|
||||
break
|
||||
case "installDockerCompose":
|
||||
multiLineCommand = op.installDockerCompose()
|
||||
break
|
||||
case "modifyDockerConfig":
|
||||
multiLineCommand = op.modifyDockerConfig()
|
||||
break
|
||||
case "installHelm":
|
||||
multiLineCommand = op.installHelm()
|
||||
break
|
||||
case "installHarbor":
|
||||
multiLineCommand = op.installHarbor(funcArgs)
|
||||
break
|
||||
case "chronyToPublicNTP":
|
||||
multiLineCommand = op.chronyToPublicNTP()
|
||||
break
|
||||
case "chronyToMaster":
|
||||
multiLineCommand = op.chronyToMaster(funcArgs)
|
||||
break
|
||||
case "installZSH":
|
||||
multiLineCommand = op.installZSH()
|
||||
break
|
||||
case "modifySshPort":
|
||||
multiLineCommand = op.modifySshPort(funcArgs)
|
||||
break
|
||||
case "openBBR":
|
||||
multiLineCommand = op.openBBR()
|
||||
break
|
||||
default:
|
||||
multiLineCommand = op.ok(funcArgs)
|
||||
|
||||
}
|
||||
|
||||
// exec the command here
|
||||
result, _ := MultiLineCommandExecutor(multiLineCommand)
|
||||
|
||||
// 归一化处理
|
||||
return strings.Join(result, "")
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) shutdownFirewall() [][]string {
|
||||
|
||||
shutdownFunc := [][]string{
|
||||
{"systemctl", "stop", "firewalld"},
|
||||
{"systemctl", "disable", "firewalld"},
|
||||
}
|
||||
|
||||
if !op.IsOsTypeUbuntu {
|
||||
shutdownFunc = append(shutdownFunc,
|
||||
[]string{
|
||||
"sed",
|
||||
"-i",
|
||||
"s/SELINUX=enforcing/SELINUX=disabled/g",
|
||||
"/etc/selinux/config",
|
||||
},
|
||||
)
|
||||
}
|
||||
|
||||
return shutdownFunc
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) modifyHostname(args []string) [][]string {
|
||||
|
||||
return [][]string{}
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) disableSwap() [][]string {
|
||||
|
||||
disableSwapFunc := [][]string{
|
||||
{
|
||||
"swapoff",
|
||||
"-a",
|
||||
},
|
||||
{
|
||||
"cp",
|
||||
"-f",
|
||||
"/etc/fstab",
|
||||
"/etc/fstab_bak",
|
||||
},
|
||||
{
|
||||
"cat",
|
||||
"/etc/fstab_bak",
|
||||
"|",
|
||||
"grep",
|
||||
"-v",
|
||||
"swap",
|
||||
">",
|
||||
"/etc/fstab",
|
||||
},
|
||||
}
|
||||
|
||||
return disableSwapFunc
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) removeDocker() [][]string {
|
||||
|
||||
removeDockerFunc := [][]string{
|
||||
{
|
||||
op.RemoveCommandPrefix,
|
||||
"docker",
|
||||
"docker-client",
|
||||
"docker-client-latest",
|
||||
"docker-ce-cli",
|
||||
"docker-common",
|
||||
"docker-latest",
|
||||
"docker-latest-logrotate",
|
||||
"docker-logrotate",
|
||||
"docker-selinux",
|
||||
"docker-engine-selinux",
|
||||
"docker-engine",
|
||||
"kubelet",
|
||||
"kubeadm",
|
||||
"kubectl",
|
||||
},
|
||||
}
|
||||
|
||||
return removeDockerFunc
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) installDocker(args []string) [][]string {
|
||||
|
||||
// remove docker all staff
|
||||
installDockerFunc := op.removeDocker()
|
||||
|
||||
if op.IsOsTypeUbuntu {
|
||||
installDockerFunc = append(installDockerFunc, [][]string{
|
||||
{
|
||||
op.InstallCommandPrefix,
|
||||
"apt-transport-https ca-certificates curl gnupg-agent software-properties-common",
|
||||
},
|
||||
{
|
||||
"apt-key",
|
||||
"add",
|
||||
"-",
|
||||
"$(curl -fsSL https://mirrors.ustc.edu.cn/docker-ce/linux/ubuntu/gpg)",
|
||||
},
|
||||
}...)
|
||||
} else {
|
||||
installDockerFunc = append(installDockerFunc, [][]string{
|
||||
{
|
||||
op.InstallCommandPrefix,
|
||||
"yum-utils device-mapper-persistent-data lvm2",
|
||||
},
|
||||
{
|
||||
"yum-config-manager",
|
||||
"--add-repo",
|
||||
"https://mirrors.ustc.edu.cn/docker-ce/linux/centos/docker-ce.repo",
|
||||
},
|
||||
{
|
||||
"sed ",
|
||||
"-i ",
|
||||
"'s/download.docker.com/mirrors.ustc.edu.cn\\/docker-ce/g' ",
|
||||
"/etc/yum.repos.d/docker-ce.repo",
|
||||
},
|
||||
{},
|
||||
}...)
|
||||
}
|
||||
|
||||
return installDockerFunc
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) installDockerCompose() [][]string {
|
||||
|
||||
return [][]string{}
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) installHelm() [][]string {
|
||||
|
||||
return [][]string{}
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) modifyDockerConfig() [][]string {
|
||||
|
||||
return [][]string{}
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) installHarbor(args []string) [][]string {
|
||||
|
||||
return [][]string{}
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) chronyToPublicNTP() [][]string {
|
||||
|
||||
return [][]string{}
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) chronyToMaster(args []string) [][]string {
|
||||
|
||||
return [][]string{}
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) installZSH() [][]string {
|
||||
|
||||
return [][]string{}
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) modifySshPort(args []string) [][]string {
|
||||
|
||||
return [][]string{}
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) openBBR() [][]string {
|
||||
|
||||
return [][]string{}
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) ok(args []string) [][]string {
|
||||
return [][]string{
|
||||
{"base function is ok , args are => " + strings.Join(args, " ")},
|
||||
}
|
||||
}
|
||||
21
agent-go/executor/BaseFunction_test.go
Normal file
21
agent-go/executor/BaseFunction_test.go
Normal file
@@ -0,0 +1,21 @@
|
||||
package executor
|
||||
|
||||
import "testing"
|
||||
|
||||
var agentOP = AgentOsOperator{
|
||||
InstallCommandPrefix: "apt-get install",
|
||||
RemoveCommandPrefix: "apt-get remove",
|
||||
CanAccessInternet: true,
|
||||
IsOsTypeUbuntu: true,
|
||||
IsAgentInnerWall: true,
|
||||
}
|
||||
|
||||
func TestBaseFunc(t *testing.T) {
|
||||
|
||||
agentOP.Exec("shutdownFirewall")
|
||||
agentOP.Exec("modifyHostname")
|
||||
agentOP.Exec("disableSwap")
|
||||
agentOP.Exec("installDocker")
|
||||
agentOP.Exec("removeDocker")
|
||||
|
||||
}
|
||||
@@ -1,136 +0,0 @@
|
||||
package function
|
||||
|
||||
import "strings"
|
||||
|
||||
type BaseFunc interface {
|
||||
Exec(baseFuncName string, funcArgs ...string) string
|
||||
}
|
||||
|
||||
type AgentOsOperator struct {
|
||||
execCommandPrefix string `json:"exec_command_prefix",comment:"apt-get or yum or zapper"`
|
||||
|
||||
canAccessInternet bool `json:"can_access_internet",comment:"是否可以访问公网"`
|
||||
}
|
||||
|
||||
// Exec 执行基础功能函数
|
||||
func (op *AgentOsOperator) Exec(baseFuncName string, funcArgs ...string) string {
|
||||
|
||||
result := ""
|
||||
|
||||
switch baseFuncName {
|
||||
|
||||
case "shutdownFirewall":
|
||||
result = op.shutdownFirewall()
|
||||
break
|
||||
case "modifyHostname":
|
||||
result = op.modifyHostname(funcArgs)
|
||||
break
|
||||
case "disableSwap":
|
||||
result = op.disableSwap()
|
||||
break
|
||||
case "installDocker":
|
||||
result = op.installDocker(funcArgs)
|
||||
break
|
||||
case "installDockerCompose":
|
||||
result = op.installDockerCompose()
|
||||
break
|
||||
case "modifyDockerConfig":
|
||||
result = op.modifyDockerConfig()
|
||||
break
|
||||
case "installHelm":
|
||||
result = op.installHelm()
|
||||
break
|
||||
case "installHarbor":
|
||||
result = op.installHarbor(funcArgs)
|
||||
break
|
||||
case "chronyToPublicNTP":
|
||||
result = op.chronyToPublicNTP()
|
||||
break
|
||||
case "chronyToMaster":
|
||||
result = op.chronyToMaster(funcArgs)
|
||||
break
|
||||
case "installZSH":
|
||||
result = op.installZSH()
|
||||
break
|
||||
case "modifySshPort":
|
||||
result = op.modifySshPort(funcArgs)
|
||||
break
|
||||
case "openBBR":
|
||||
result = op.openBBR()
|
||||
break
|
||||
default:
|
||||
result = op.ok(funcArgs)
|
||||
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) shutdownFirewall() string {
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) modifyHostname(args []string) string {
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) disableSwap() string {
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) installDocker(args []string) string {
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) installDockerCompose() string {
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) installHelm() string {
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) modifyDockerConfig() string {
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) installHarbor(args []string) string {
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) chronyToPublicNTP() string {
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) chronyToMaster(args []string) string {
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) installZSH() string {
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) modifySshPort(args []string) string {
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) openBBR() string {
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func (op *AgentOsOperator) ok(args []string) string {
|
||||
return "base function is ok , args are => " + strings.Join(args, " ")
|
||||
}
|
||||
2
agent-go/executor/script/shutdownFirewall.txt
Normal file
2
agent-go/executor/script/shutdownFirewall.txt
Normal file
@@ -0,0 +1,2 @@
|
||||
systemctl stop firewalld
|
||||
systemctl disable firewalld
|
||||
Reference in New Issue
Block a user