使用Go语言实现远程传输文件
2016-08-09 16:18
896 查看
之前有一篇介绍如何使用Go语言通过SSH协议来执行远程命令:如何使用Go语言实现远程执行命令 同样,通过SSH协议也可以使用Go语言来远程传输文件。
除了 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 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 }
发送文件
使用上面的 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。以上就是Go语言实现远程传输文件的全部内容,希望本文对大家学习Go语言有所帮助。
您可能感兴趣的文章:
相关文章推荐
- DOS命令行使用pscp实现远程文件和文件夹传输
- DOS命令行使用pscp实现远程文件和文件夹传输(转)
- DOS命令行使用pscp实现远程文件和文件夹传输
- 使用GoLang实现文件远程传输
- .Net网络通讯编程[利用Socket实现字串、文件、序列化对象传输]--使用封装的网络服务4[聊天室][使用IE浏览本页]
- .Net网络通讯编程[利用Socket实现字串、文件、序列化对象传输]--类设计1[使用IE浏览本页]
- Ubuntu下安装和使用lrzsz,实现串口传输文件&&minicom
- .Net网络通讯编程[利用Socket实现字串、文件、序列化对象传输]--使用封装的网络服务4[聊天室]
- Gstreamer实现摄像头的远程采集,udp传输,本地显示和保存为AVI文件 发送端
- c++ 使用socket实现C/S端文件的下载传输
- 在远程桌面中,实现文件传送/传输/拷贝的方法.
- .Net网络通讯编程[利用Socket实现字串、文件、序列化对象传输]--使用封装的网络服务1
- Ubuntu10.10下安装和使用lrzsz,实现串口传输文件&&minicom
- Ubuntu下安装和使用lrzsz,实现串口传输文件&&minicom
- Gstreamer实现摄像头的远程采集,udp传输,本地显示和保存为AVI文件 接收保存显示端
- Linux下使用socket传输文件的C语言简单实现
- .Net网络通讯编程[利用Socket实现字串、文件、序列化对象传输]--使用封装的网络服务2
- pscp实现远程文件和文件夹传输
- .Net网络通讯编程[利用Socket实现字串、文件、序列化对象传输]--使用封装的网络服务3[聊天室][使用IE浏览本页]
- Gstreamer实现摄像头的远程采集,udp传输,本地显示和保存为AVI文件 发送端