目录
前言
shellGoGosshscp说明：本文中的示例，均是基于Go1.17 64位机器
连接远程服务器并执行命令（ssh）
用户名+密码/usr/bin/whoamiClientConfigAuthAuthMethodHostKeyCallbackssh.InsecureIgnoreHostKey()ssh.FixedHostKey(key PublicKey)DialDialclientNewSessionNewSessionRunimport (
"bytes"
"fmt"
"log"
"golang.org/x/crypto/ssh"
)
func main() {
var (
username = "your username"
password = "your password"
addr = "ip:22"
)
config := &ssh.ClientConfig{
User: username,
Auth: []ssh.AuthMethod{
ssh.Password(password),
},
HostKeyCallback: ssh.InsecureIgnoreHostKey(),
}
client, err := ssh.Dial("tcp", addr, config)
if err != nil {
log.Fatal("Failed to dial: ", err)
}
defer client.Close()
// 开启一个session，用于执行一个命令
session, err := client.NewSession()
if err != nil {
log.Fatal("Failed to create session: ", err)
}
defer session.Close()
// 执行命令，并将执行的结果写到 b 中
var b bytes.Buffer
session.Stdout = &b
// 也可以使用 session.CombinedOutput() 整合输出
if err := session.Run("/usr/bin/whoami"); err != nil {
log.Fatal("Failed to run: " + err.Error())
}
fmt.Println(b.String()) // root
}
Runk8sk8s podRuntest.sh/opt# 脚本文件 #!/bin/bash today=$(date +"%Y-%m-%d") # 将数据写入文件 $(df -h > $today.log)
package main
import (
"fmt"
"log"
"golang.org/x/crypto/ssh"
)
func main() {
var (
username = "your username"
password = "your password"
addr = "ip:22"
)
config := &ssh.ClientConfig{
User: username,
Auth: []ssh.AuthMethod{
ssh.Password(password),
},
HostKeyCallback: ssh.InsecureIgnoreHostKey(),
}
client, err := ssh.Dial("tcp", addr, config)
if err != nil {
log.Fatal("Failed to dial: ", err)
}
defer client.Close()
session, err := client.NewSession()
if err != nil {
log.Fatal("Failed to create session: ", err)
}
defer session.Close()
// 调用远程服务器脚本脚本
res, err := session.CombinedOutput("sh /opt/test.sh")
if err != nil {
log.Fatal("Failed to run: " + err.Error())
}
fmt.Println(string(res))
/*
Filesystem Size Used Avail Use% Mounted on
devtmpfs 909M 0 909M 0% /dev
tmpfs 919M 24K 919M 1% /dev/shm
tmpfs 919M 540K 919M 1% /run
tmpfs 919M 0 919M 0% /sys/fs/cgroup
/dev/vda1 50G 6.9G 40G 15% /
tmpfs 184M 0 184M 0% /run/user/0
*/
}
拷贝远程服务器文件到本地（scp）
拷贝文件步骤比较简单：
ssh clientssh clientsftp clientpackage main
import (
"io"
"log"
"os"
"time"
"github.com/pkg/sftp"
"golang.org/x/crypto/ssh"
)
func main() {
var (
username = "your username"
password = "your password"
addr = "ip:22"
)
// 1. 建立 ssh client
config := &ssh.ClientConfig{
User: username,
Auth: []ssh.AuthMethod{
ssh.Password(password),
},
HostKeyCallback: ssh.InsecureIgnoreHostKey(),
}
client, err := ssh.Dial("tcp", addr, config)
if err != nil {
log.Fatal("Failed to dial: ", err)
}
defer client.Close()
// 2. 基于ssh client, 创建 sftp 客户端
sftpClient, err := sftp.NewClient(client)
if err != nil {
log.Fatal("Failed to init sftp client: ", err)
}
defer sftpClient.Close()
// 3. 打开远程服务器文件
filename := time.Now().Format("2006-01-02") + ".log"
source, err := sftpClient.Open("/opt/" + filename)
if err != nil {
log.Fatal("Failed to open remote file: ", err)
}
defer source.Close()
// 4. 创建本地文件
target, err := os.OpenFile(filename, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0644)
if err != nil {
log.Fatal("Failed to open local file: ", err)
}
defer target.Close()
// 5. 数据复制
n, err := io.Copy(target, source)
if err != nil {
log.Fatal("Failed to copy file: ", err)
}
log.Println("Succeed to copy file: ", n)
}
sftp clientWalkReadDirStatMkdirReadWriteWriteToReadFrom简单封装下
package main
import (
"fmt"
"io"
"log"
"os"
"time"
"github.com/pkg/sftp"
"golang.org/x/crypto/ssh"
)
type Cli struct {
user string
pwd string
addr string
client *ssh.Client
}
func NewCli(user, pwd, addr string) Cli {
return Cli{
user: user,
pwd: pwd,
addr: addr,
}
}
// Connect 连接远程服务器
func (c *Cli) Connect() error {
config := &ssh.ClientConfig{
User: c.user,
Auth: []ssh.AuthMethod{
ssh.Password(c.pwd),
},
HostKeyCallback: ssh.InsecureIgnoreHostKey(),
}
client, err := ssh.Dial("tcp", c.addr, config)
if nil != err {
return fmt.Errorf("connect server error: %w", err)
}
c.client = client
return nil
}
// Run 运行命令
func (c Cli) Run(shell string) (string, error) {
if c.client == nil {
if err := c.Connect(); err != nil {
return "", err
}
}
session, err := c.client.NewSession()
if err != nil {
return "", fmt.Errorf("create new session error: %w", err)
}
defer session.Close()
buf, err := session.CombinedOutput(shell)
return string(buf), err
}
// Scp 复制文件
func (c Cli) Scp(srcFileName, targetFileName string) (int64, error) {
if c.client == nil {
if err := c.Connect(); err != nil {
return 0, err
}
}
sftpClient, err := sftp.NewClient(c.client)
if err != nil {
return 0, fmt.Errorf("new sftp client error: %w", err)
}
defer sftpClient.Close()
source, err := sftpClient.Open(srcFileName)
if err != nil {
return 0, fmt.Errorf("sftp client open file error: %w", err)
}
defer source.Close()
target, err := os.OpenFile(targetFileName, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0644)
if err != nil {
return 0, fmt.Errorf("open local file error: %w", err)
}
defer target.Close()
n, err := io.Copy(target, source)
if err != nil {
return 0, fmt.Errorf("copy file error: %w", err)
}
return n, nil
}
// 调用测试
func main() {
var (
username = "your username"
password = "your password"
addr = "ip:22"
)
// 初始化
client := NewCli(username, password, addr)
// ssh 并运行脚本
_, err := client.Run("sh /opt/test.sh")
if err != nil {
log.Printf("failed to run shell,err=[%v]\n", err)
return
}
// scp 文件到本地
filename := time.Now().Format("2006-01-02") + ".log"
n, err := client.Scp("/opt/"+filename, filename)
if err != nil {
log.Printf("failed to scp file,err=[%v]\n", err)
return
}
log.Printf("Succeed to scp file, size=[%d]\n", n)
// 处理文件并删除本地文件......
}
通过上面的一系列操作，就可以实现了我的需求：
1.编写程序：
- 连接客户服务器
- 执行远程服务器的脚本，生成日志文件
- 拷贝远程服务器的日志文件到本地
- 处理日志文件
- 删除本地文件
2.在服务器上启动一个定时任务运行该程序
您可能感兴趣的文章: