快捷導(dǎo)航

如何使用Go語(yǔ)言實(shí)現(xiàn)遠(yuǎn)程執(zhí)行命令

更新時(shí)間：2016年08月09日 15:57:06 投稿：daisy

遠(yuǎn)程執(zhí)行命令最常用的方法就是利用SSH協(xié)議,將命令發(fā)送到遠(yuǎn)程機(jī)器上執(zhí)行,并獲取返回結(jié)果。本文將介紹如何使用Go語(yǔ)言實(shí)現(xiàn)遠(yuǎn)程執(zhí)行命令。下面一起來(lái)看看。

前言

遠(yuǎn)程執(zhí)行命令有什么用？為什么要遠(yuǎn)程執(zhí)行命令？如果你只有2，3臺(tái)服務(wù)器需要管理的時(shí)候，遠(yuǎn)程執(zhí)行命令確實(shí)沒(méi)有沒(méi)多大作用，你可以登錄到每臺(tái)服務(wù)器上去完成各種操作。當(dāng)你的服務(wù)器大于3臺(tái)的時(shí)候，遠(yuǎn)程執(zhí)行的命令的方式就可以大大提高你的生產(chǎn)力了。

如果你有一個(gè)可以遠(yuǎn)程執(zhí)行命令的工具，那么就可以像操作單臺(tái)機(jī)器那樣操作多臺(tái)機(jī)器，機(jī)器越多，效率提高的越多。遠(yuǎn)程執(zhí)行命令最常用的方法就是利用 SSH 協(xié)議，將命令發(fā)送到遠(yuǎn)程機(jī)器上執(zhí)行，并獲取返回結(jié)果。

一般命令

所謂一般命令，就是在一定時(shí)間內(nèi)會(huì)執(zhí)行完的命令。比如 grep, cat 等等。執(zhí)行命令的步驟是：連接，執(zhí)行，獲取結(jié)果

連接

連接包含了認(rèn)證，可以使用 password 或者 sshkey 2種方式來(lái)認(rèn)證。下面的示例為了簡(jiǎn)單，使用了密碼認(rèn)證的方式來(lái)完成連接。

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
}

連接的方法很簡(jiǎn)單，只要提供登錄主機(jī)的用戶*， *密碼*， *主機(jī)名或者IP*， *SSH端口

執(zhí)行，命令獲取結(jié)果

連接成功后，執(zhí)行命令很簡(jiǎn)單

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")
}

上面代碼運(yùn)行之后，雖然命令正常執(zhí)行了，但是沒(méi)有正常輸出的結(jié)果，也沒(méi)有異常輸出的結(jié)果。要想顯示結(jié)果，需要將 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")
}

這樣就能在屏幕上顯示正常，異常的信息了。

交互式命令

上面的方式無(wú)法遠(yuǎn)程執(zhí)行交互式命令，比如 top ，遠(yuǎn)程編輯一個(gè)文件，比如 vi /etc/nginx/nginx.conf 如果要支持交互式的命令，需要當(dāng)前的terminal來(lái)接管遠(yuǎn)程的 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")
}

總結(jié)

好了，這樣就可以執(zhí)行交互式命令了，比如上面的 top 也可以通過(guò) vi /etc/nginx/nignx.conf 之類的命令來(lái)遠(yuǎn)程編輯文件。以上就是如何使用Go語(yǔ)言實(shí)現(xiàn)遠(yuǎn)程執(zhí)行命令的全部?jī)?nèi)容了，希望本文對(duì)大家學(xué)習(xí)python有所幫助。

您可能感興趣的文章: