使用Go语言实现远程传输文件
程序员文章站
2022-07-05 08:51:57
前言
之前有一篇介绍如何使用go语言通过ssh协议来执行远程命令:如何使用go语言实现远程执行命令 同样,通过ssh协议也可以使用go语言来远程传输文件。
除了 s...
前言
之前有一篇介绍如何使用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语言有所帮助。
上一篇: 利用Go语言追加内容到文件末尾
下一篇: Go语言之自定义集合Set