如何使用Go语言实现远程执行命令
前言
远程执行命令有什么用?为什么要远程执行命令?如果你只有2,3台服务器需要管理的时候,远程执行命令确实没有没多大作用,你可以登录到每台服务器上去完成各种操作。当你的服务器大于3台的时候,远程执行的命令的方式就可以大大提高你的生产力了。
如果你有一个可以远程执行命令的工具,那么就可以像操作单台机器那样操作多台机器,机器越多,效率提高的越多。远程执行命令最常用的方法就是利用SSH协议,将命令发送到远程机器上执行,并获取返回结果。
一般命令
所谓一般命令,就是在一定时间内会执行完的命令。比如grep,cat等等。执行命令的步骤是:连接,执行,获取结果
连接
连接包含了认证,可以使用password或者sshkey2种方式来认证。下面的示例为了简单,使用了密码认证的方式来完成连接。
import(
"fmt"
"time"
"golang.org/x/crypto/ssh"
)
funcconnect(user,password,hoststring,portint)(*ssh.Session,error){
var(
auth[]ssh.AuthMethod
addrstring
clientConfig*ssh.ClientConfig
client*ssh.Client
session*ssh.Session
errerror
)
//getauthmethod
auth=make([]ssh.AuthMethod,0)
auth=append(auth,ssh.Password(password))
clientConfig=&ssh.ClientConfig{
User:user,
Auth:auth,
Timeout:30*time.Second,
}
//connettossh
addr=fmt.Sprintf("%s:%d",host,port)
ifclient,err=ssh.Dial("tcp",addr,clientConfig);err!=nil{
returnnil,err
}
//createsession
ifsession,err=client.NewSession();err!=nil{
returnnil,err
}
returnsession,nil
}
连接的方法很简单,只要提供登录主机的用户*,*密码*,*主机名或者IP*,*SSH端口
执行,命令获取结果
连接成功后,执行命令很简单
import(
"fmt"
"log"
"os"
"time"
"golang.org/x/crypto/ssh"
)
funcmain(){
session,err:=connect("root","xxxxx","127.0.0.1",22)
iferr!=nil{
log.Fatal(err)
}
defersession.Close()
session.Run("ls/;ls/abc")
}
上面代码运行之后,虽然命令正常执行了,但是没有正常输出的结果,也没有异常输出的结果。要想显示结果,需要将session的Stdout和Stderr重定向修改funcmain为如下:
funcmain(){
session,err:=connect("root","xxxxx","127.0.0.1",22)
iferr!=nil{
log.Fatal(err)
}
defersession.Close()
session.Stdout=os.Stdout
session.Stderr=os.Stderr
session.Run("ls/;ls/abc")
}
这样就能在屏幕上显示正常,异常的信息了。
交互式命令
上面的方式无法远程执行交互式命令,比如top,远程编辑一个文件,比如vi/etc/nginx/nginx.conf如果要支持交互式的命令,需要当前的terminal来接管远程的PTY。
funcmain(){
session,err:=connect("root","olordjesus","dockers.iotalabs.io",2210)
iferr!=nil{
log.Fatal(err)
}
defersession.Close()
fd:=int(os.Stdin.Fd())
oldState,err:=terminal.MakeRaw(fd)
iferr!=nil{
panic(err)
}
deferterminal.Restore(fd,oldState)
//excutecommand
session.Stdout=os.Stdout
session.Stderr=os.Stderr
session.Stdin=os.Stdin
termWidth,termHeight,err:=terminal.GetSize(fd)
iferr!=nil{
panic(err)
}
//Setupterminalmodes
modes:=ssh.TerminalModes{
ssh.ECHO:1,//enableechoing
ssh.TTY_OP_ISPEED:14400,//inputspeed=14.4kbaud
ssh.TTY_OP_OSPEED:14400,//outputspeed=14.4kbaud
}
//Requestpseudoterminal
iferr:=session.RequestPty("xterm-256color",termHeight,termWidth,modes);err!=nil{
log.Fatal(err)
}
session.Run("top")
}
总结
好了,这样就可以执行交互式命令了,比如上面的top也可以通过vi/etc/nginx/nignx.conf之类的命令来远程编辑文件。以上就是如何使用Go语言实现远程执行命令的全部内容了,希望本文对大家学习python有所帮助。