go语言适合编写高效稳定的运维脚本,利用os/exec执行系统命令、os包操作文件、time.Ticker实现定时任务,并通过goroutine并发处理;结合net/http调用API,使用crypto/ssh操作远程主机,推荐用slog统一日志,避免忽略错误处理,提升可维护性。
在golang中编写自动化运维脚本,核心在于利用其高并发、跨平台编译和标准库丰富的特点,实现稳定高效的系统管理任务。相比Shell或python,Go语言更适合构建复杂、可维护的运维工具。
使用标准库执行系统命令
Go 的 os/exec 包是执行外部命令的主要方式,适合调用系统工具如 ls、systemctl、kubectl 等。
示例:执行一个 shell 命令并获取输出:
package main <p>import ( "fmt" "log" "os/exec" )</p><p>func runCommand(name string, args ...string) (string, error) { cmd := exec.Command(name, args...) output, err := cmd.Output() if err != nil { return "", err } return string(output), nil }</p><p>func main() { out, err := runCommand("df", "-h") if err != nil { log.Fatal(err) } fmt.Println(out) }
这个例子检查磁盘使用情况,可用于定时监控服务器资源。
立即学习“go语言免费学习笔记(深入)”;
文件与目录操作
使用 os 和 io/ioutil(Go 1.16 后推荐使用 os 替代)进行文件读写、创建日志、备份配置等操作。
常见操作包括:
- 检查文件是否存在:os.Stat()
- 读取配置文件:os.ReadFile()
- 写入日志:os.WriteFile()
- 遍历目录:os.ReadDir()
例如,批量重命名日志文件:
files, _ := os.ReadDir("/var/log/myapp") for _, f := range files { if f.IsDir() { continue } old := "/var/log/myapp/" + f.Name() new := "/var/log/myapp/backup_" + f.Name() os.Rename(old, new) }
定时任务与并发控制
利用 Go 的 time.Ticker 实现轮询式监控,结合 goroutine 并发处理多台主机或多个服务。
示例:每5分钟检查一次服务状态:
func monitorService() { ticker := time.NewTicker(5 * time.Minute) defer ticker.Stop() <pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">for range ticker.C { out, err := runCommand("systemctl", "is-active", "nginx") if err != nil || string(out) != "activen" { // 发送告警 sendAlert("Nginx is down!") } }
}
若需同时检查多个服务,启动多个 goroutine 即可:
go monitorService("nginx") go monitorService("redis") go monitorService("postgresql")
集成远程操作与API调用
通过 SSH 连接远程服务器执行命令,可用第三方库如 golang.org/x/crypto/ssh。
也可以调用 kubernetes、prometheus 或企业内部 API,使用 net/http 发送请求。
例如,调用 Prometheus 查询接口获取指标:
resp, _ := http.Get("http://prom:9090/api/v1/query?query=up") // 解析 jsON 响应判断服务状态
结合 CLI 工具库如 spf13/cobra,还能将脚本封装成命令行工具,支持子命令、参数解析和帮助文档。
基本上就这些。Go 写运维脚本不复杂但容易忽略错误处理和日志记录,建议统一用 log/slog 记录运行状态,提升可维护性。