使用Go語言實現(xiàn)遠程傳輸文件

更新時間：2016年08月09日 16:18:54 投稿：daisy

本文主要介紹如何利用Go語言實現(xiàn)遠程傳輸文件的功能，有需要的小伙伴們可以參考學習。下面跟著小編一起來學習學習。

前言

之前有一篇介紹如何使用Go語言通過SSH協(xié)議來執(zhí)行遠程命令：如何使用Go語言實現(xiàn)遠程執(zhí)行命令同樣，通過SSH協(xié)議也可以使用Go語言來遠程傳輸文件。

除了 SSH 的庫，為了傳輸文件，還需要用到 github.com/pkg/sftp 這個庫。

實現(xiàn)方式

廢話不多說，直接看代碼。由于是基于 SSH 協(xié)議實現(xià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ā)送文件很簡單。

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
 )

 // 這里換成實際的 SSH 連接的 用戶名，密碼，主機名或IP，SSH端口
 sftpClient, err = connect("root", "rootpass", "127.0.0.1", 22)
 if err != nil {
 log.Fatal(err)
 }
 defer sftpClient.Close()

 // 用來測試的本地文件路徑 和 遠程機器上的文件夾
 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!")
}

獲取文件

從遠程機器上獲取文件的方式略有不同，但也很簡單。

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
 )

 // 這里換成實際的 SSH 連接的 用戶名，密碼，主機名或IP，SSH端口
 sftpClient, err = connect("root", "rootpass", "127.0.0.1", 22)
 if err != nil {
 log.Fatal(err)
 }
 defer sftpClient.Close()

 // 用來測試的遠程文件路徑 和 本地文件夾
 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é)

上面的例子只是演示了文件傳輸，傳輸文件夾也很簡單，只是多了遍歷文件夾和創(chuàng)建文件夾的步驟，具體的函數(shù)可以自行查看 sftp 庫中doc。以上就是Go語言實現(xiàn)遠程傳輸文件的全部內(nèi)容，希望本文對大家學習Go語言有所幫助。

您可能感興趣的文章: