golang sftp传输文件
之前有一篇介绍如何使用 golang 通过SSH协议来执行远程命令：golang执行远程命令 同样，通过SSH协议也可以使用 golang 来远程传输文件。除了 SSH 的库，为了传输文件，还需要用到 github.com/pkg/sftp 这个库。
实现方式
废话不多说，直接看代码。 由于是基于 SSH 协议实现的远程文件传输，所以先创建 SSH 的连接，再创建传输文件的 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 methodauth = make([]ssh.AuthMethod, 0)auth = append(auth, ssh.Password(password)) clientConfig = &ssh.ClientConfig{ User: user, Auth: auth, Timeout: 30 * time.Second,} // connet to sshaddr = fmt.Sprintf("%s:%d", host, port) if sshClient, err = ssh.Dial("tcp", addr, clientConfig); err != nil { return nil, err} // create sftp clientif sftpClient, err = sftp.NewClient(sshClient); err != nil { return nil, err} return sftpClient, nil}发送文件
使用上面的 connect 方法创建 sftpClient 后，发送文件很简单。
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!")}总结
上面的例子只是演示了文件传输，传输文件夹也很简单，只是多了遍历文件夹和创建文件夹的步骤，具体的函数可以自行查看 sftp 库中doc。
页:
[1]