如何使⽤Go语⾔实现远程执⾏命令
前⾔
远程执⾏命令有什么⽤?为什么要远程执⾏命令?如果你只有2,3台服务器需要管理的时候,远程执⾏命令确实没有没多⼤作⽤,你可以登录到每台服务器上去完成各种操作。当你的服务器⼤于3台的时候,远程执⾏的命令的⽅式就可以⼤⼤提⾼你的⽣产⼒了。
如果你有⼀个可以远程执⾏命令的⼯具,那么就可以像操作单台机器那样操作多台机器,机器越多,效率提⾼的越多。远程执⾏命令最常⽤的⽅法就是利⽤ SSH 协议,将命令发送到远程机器上执⾏,并获取返回结果。
⼀般命令
所谓⼀般命令,就是在⼀定时间内会执⾏完的命令。⽐如 grep, cat 等等。执⾏命令的步骤是:连接,执⾏,获取结果
连接
连接包含了认证,可以使⽤ password 或者 sshkey 2种⽅式来认证。下⾯的⽰例为了简单,使⽤了密码认证的⽅式来完成连接。
import (
"fmt"
"time"
"/x/crypto/ssh"
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"
"/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/f 如果要⽀持交互式的命令,需要当前的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/f 之类的命令
来远程编辑⽂件。以上就是如何使⽤Go语⾔实现远程执⾏命令的全部内容了,希望本⽂对⼤家学习python有所帮助。

版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198,我们将在24小时内删除。