使用Go語言實(shí)現(xiàn)遠(yuǎn)程傳輸文件
前言
之前有一篇介紹如何使用Go語言通過SSH協(xié)議來執(zhí)行遠(yuǎn)程命令:如何使用Go語言實(shí)現(xiàn)遠(yuǎn)程執(zhí)行命令 同樣,通過SSH協(xié)議也可以使用Go語言來遠(yuǎn)程傳輸文件。
除了 SSH 的庫,為了傳輸文件,還需要用到 github.com/pkg/sftp
這個(gè)庫。
實(shí)現(xiàn)方式
廢話不多說,直接看代碼。 由于是基于 SSH 協(xié)議實(shí)現(xiàn)的遠(yuǎn)程文件傳輸,所以先創(chuàng)建 SSH 的連接,再創(chuàng)建傳輸文件的 sftp 客戶端。
func connect(user, password, host string, port int) (*sftp.Client, error) { var ( auth []ssh.AuthMethod addr string clientConfig *ssh.ClientConfig sshClient *ssh.Client sftpClient *sftp.Client 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 sshClient, err = ssh.Dial("tcp", addr, clientConfig); err != nil { return nil, err } // create sftp client if sftpClient, err = sftp.NewClient(sshClient); err != nil { return nil, err } return sftpClient, nil }
發(fā)送文件
使用上面的 connect 方法創(chuàng)建 sftpClient 后,發(fā)送文件很簡(jiǎn)單。
package main import ( "fmt" "log" "os" "path" "time" "github.com/pkg/sftp" "golang.org/x/crypto/ssh" ) func main() { var ( err error sftpClient *sftp.Client ) // 這里換成實(shí)際的 SSH 連接的 用戶名,密碼,主機(jī)名或IP,SSH端口 sftpClient, err = connect("root", "rootpass", "127.0.0.1", 22) if err != nil { log.Fatal(err) } defer sftpClient.Close() // 用來測(cè)試的本地文件路徑 和 遠(yuǎn)程機(jī)器上的文件夾 var localFilePath = "/path/to/local/file/test.txt" var remoteDir = "/remote/dir/" srcFile, err := os.Open(localFilePath) if err != nil { log.Fatal(err) } defer srcFile.Close() var remoteFileName = path.Base(localFilePath) dstFile, err := sftpClient.Create(path.Join(remoteDir, remoteFileName)) if err != nil { log.Fatal(err) } defer dstFile.Close() buf := make([]byte, 1024) for { n, _ := srcFile.Read(buf) if n == 0 { break } dstFile.Write(buf) } fmt.Println("copy file to remote server finished!") }
獲取文件
從遠(yuǎn)程機(jī)器上獲取文件的方式略有不同,但也很簡(jiǎn)單。
package main import ( "fmt" "log" "os" "path" "time" "github.com/pkg/sftp" "golang.org/x/crypto/ssh" ) func main() { var ( err error sftpClient *sftp.Client ) // 這里換成實(shí)際的 SSH 連接的 用戶名,密碼,主機(jī)名或IP,SSH端口 sftpClient, err = connect("root", "rootpass", "127.0.0.1", 22) if err != nil { log.Fatal(err) } defer sftpClient.Close() // 用來測(cè)試的遠(yuǎn)程文件路徑 和 本地文件夾 var remoteFilePath = "/path/to/remote/path/test.txt" var localDir = "/local/dir" srcFile, err := sftpClient.Open(remoteFilePath) if err != nil { log.Fatal(err) } defer srcFile.Close() var localFileName = path.Base(remoteFilePath) dstFile, err := os.Create(path.Join(localDir, localFileName)) if err != nil { log.Fatal(err) } defer dstFile.Close() if _, err = srcFile.WriteTo(dstFile); err != nil { log.Fatal(err) } fmt.Println("copy file from remote server finished!") }
總結(jié)
上面的例子只是演示了文件傳輸,傳輸文件夾也很簡(jiǎn)單,只是多了遍歷文件夾和創(chuàng)建文件夾的步驟,具體的函數(shù)可以自行查看 sftp 庫中doc。以上就是Go語言實(shí)現(xiàn)遠(yuǎn)程傳輸文件的全部?jī)?nèi)容,希望本文對(duì)大家學(xué)習(xí)Go語言有所幫助。
相關(guān)文章

3個(gè)Go語言中實(shí)用重構(gòu)技術(shù)分享

Go語言里的結(jié)構(gòu)體文法實(shí)例分析

基于Go?goroutine實(shí)現(xiàn)一個(gè)簡(jiǎn)單的聊天服務(wù)

golang跳轉(zhuǎn)語句goto,break,continue的使用及區(qū)別說明

Golang 基礎(chǔ)之函數(shù)使用(匿名遞歸閉包)實(shí)例詳解