• 企业400电话
  • 微网小程序
  • AI电话机器人
  • 电商代运营
  • 全 部 栏 目

    企业400电话 网络优化推广 AI电话机器人 呼叫中心 网站建设 商标✡知产 微网小程序 电商运营 彩铃•短信 增值拓展业务
    使用Go语言实现远程传输文件

    前言

    之前有一篇介绍如何使用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语言有所帮助。

    您可能感兴趣的文章:
    • mongodb 3.4下远程连接认证失败的解决方法
    • MongoDB远程访问配置步骤详解
    • 如何使用Go语言实现远程执行命令
    • go语言实现一个简单的http客户端抓取远程url的方法
    • mongodb 实现远程连接
    • 使用goland调试远程代码的操作步骤
    上一篇:如何使用Go语言实现远程执行命令
    下一篇:浅谈Go语言的空标示符
  • 相关文章
  • 

    © 2016-2020 巨人网络通讯 版权所有

    《增值电信业务经营许可证》 苏ICP备15040257号-8

    使用Go语言实现远程传输文件 使用,语言,实现,远程,传输,