如何使用Go语言实现远程执行命令

前言

远程执行命令有什么用？为什么要远程执行命令？ 如果你只有2，3台服务器需要管理的时候，远程执行命令确实没有没多大作用，你可以登录到每台服务器上去完成各种操作。 当你的服务器大于3台的时候，远程执行的命令的方式就可以大大提高你的生产力了。

如果你有一个可以远程执行命令的工具，那么就可以像操作单台机器那样操作多台机器，机器越多，效率提高的越多。 远程执行命令最常用的方法就是利用 ssh 协议，将命令发送到远程机器上执行，并获取返回结果。

一般命令

所谓一般命令，就是在一定时间内会执行完的命令。比如 grep, cat 等等。 执行命令的步骤是：连接，执行，获取结果

连接

连接包含了认证，可以使用 password 或者 sshkey 2种方式来认证。下面的示例为了简单，使用了密码认证的方式来完成连接。

import ( 
 "fmt"
 "time"

 "golang.org/x/crypto/ssh"
)

func connect(user, password, host string, port int) (*ssh.session, error) { 
 var (
 auth  []ssh.authmethod
 addr  string
 clientconfig *ssh.clientconfig
 client *ssh.client
 session *ssh.session
 err  error
 )
 // get auth method
 auth = make([]ssh.authmethod, 0)
 auth = append(auth, ssh.password(password))

 clientconfig = &ssh.clientconfig{
 user: user,
 auth: auth,
 timeout: 30 * time.second,
 }

 // connet to ssh
 addr = fmt.sprintf("%s:%d", host, port)

 if client, err = ssh.dial("tcp", addr, clientconfig); err != nil {
 return nil, err
 }

 // create session
 if session, err = client.newsession(); err != nil {
 return nil, err
 }

 return session, nil
}

连接的方法很简单，只要提供登录主机的 用户*， *密码*， *主机名或者ip*， *ssh端口

执行，命令获取结果

连接成功后，执行命令很简单

import ( 
 "fmt"
 "log"
 "os"
 "time"

 "golang.org/x/crypto/ssh"
)

func main() { 
 session, err := connect("root", "xxxxx", "127.0.0.1", 22)
 if err != nil {
 log.fatal(err)
 }
 defer session.close()

 session.run("ls /; ls /abc")
}

上面代码运行之后，虽然命令正常执行了，但是没有正常输出的结果，也没有异常输出的结果。 要想显示结果，需要将 session 的 stdout 和 stderr 重定向 修改 func main 为如下：

func main() { 
 session, err := connect("root", "xxxxx", "127.0.0.1", 22)
 if err != nil {
 log.fatal(err)
 }
 defer session.close()

 session.stdout = os.stdout
 session.stderr = os.stderr
 session.run("ls /; ls /abc")
}

这样就能在屏幕上显示正常，异常的信息了。

交互式命令

上面的方式无法远程执行交互式命令，比如 top ， 远程编辑一个文件，比如 vi /etc/nginx/nginx.conf 如果要支持交互式的命令，需要当前的terminal来接管远程的 pty。

func main() { 
 session, err := connect("root", "olordjesus", "dockers.iotalabs.io", 2210)
 if err != nil {
 log.fatal(err)
 }
 defer session.close()

 fd := int(os.stdin.fd())
 oldstate, err := terminal.makeraw(fd)
 if err != nil {
 panic(err)
 }
 defer terminal.restore(fd, oldstate)

 // excute command
 session.stdout = os.stdout
 session.stderr = os.stderr
 session.stdin = os.stdin

 termwidth, termheight, err := terminal.getsize(fd)
 if err != nil {
 panic(err)
 }

 // set up terminal modes
 modes := ssh.terminalmodes{
 ssh.echo:  1, // enable echoing
 ssh.tty_op_ispeed: 14400, // input speed = 14.4kbaud
 ssh.tty_op_ospeed: 14400, // output speed = 14.4kbaud
 }

 // request pseudo terminal
 if err := session.requestpty("xterm-256color", termheight, termwidth, modes); err != nil {
 log.fatal(err)
 }

 session.run("top")
}

总结

好了，这样就可以执行交互式命令了，比如上面的 top 也可以通过 vi /etc/nginx/nignx.conf 之类的命令来远程编辑文件。以上就是如何使用go语言实现远程执行命令的全部内容了，希望本文对大家学习python有所帮助。