如何使用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有所帮助。
下一篇: Go语言函数学习教程