使用golang上传文件到SFTP主机失败

欢迎各位小伙伴来到golang学习网，相聚于此都是缘哈哈哈！今天我给大家带来《使用golang上传文件到SFTP主机失败》，这篇文章主要讲到等等知识，如果你对Golang相关的知识非常感兴趣或者正在自学，都可以关注我，我会持续更新相关文章！当然，有什么建议也欢迎在评论留言提出！一起学习！

我有以下 golang 函数来将文件上传到 sftp：

func uploadObjectToDestination(sshConfig SSHConnectionConfig, destinationPath string, srcFile io.Reader) {
    // Connect to destination host via SSH
    conn, err := ssh.Dial("tcp", sshConfig.sftpHost+sshConfig.sftpPort, sshConfig.authConfig)
    if err != nil {
        log.Fatal(err)
    }
    defer conn.Close()

    // create new SFTP client
    client, err := sftp.NewClient(conn)
    if err != nil {
        log.Fatal(err)
    }
    defer client.Close()

    log.Printf("Opening file on destination server under path %s", destinationPath)
    // create destination file
    dstFile, err := client.OpenFile(destinationPath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC)
    if err != nil {
        log.Fatal(err)
    }
    defer dstFile.Close()

    log.Printf("Copying file to %s", destinationPath)
    // copy source file to destination file
    bytes, err := io.Copy(dstFile, srcFile)
    if err != nil {
        log.Fatal(err)
    }

    log.Printf("%s - Total %d bytes copied\n", dstFile.Name(), bytes)
}

上面的代码可以在 95% 的情况下工作，但对于某些文件会失败。这些失败的文件之间的唯一关系是大小（3-4kb）。其他成功的文件较小（0.5-3kb）。在某些情况下，大小为 2-3kb 的文件也会失败。

我能够使用不同的 sftp 服务器重现相同的问题。

当使用 sftp.write 更改失败代码（io.copy）时，我可以看到相同的行为，除了该过程不返回错误，而是我看到复制了 0 个字节，这似乎与io.copy 失败。

顺便说一句，当使用 io.copy 时，我收到的错误是 context 已取消，意外的 eof。

代码从 aws lambda 运行，不存在内存或时间限制问题。

解决方案

经过几个小时的挖掘，事实证明，我的代码是问题的根源。 以下是答案，供以后参考： 还有另一个函数不在原始问题中，它从 s3 下载对象：

func getobjectfroms3(svc *s3.s3, bucket, key string) io.reader {
    var timeout = time.second * 30
    ctx := context.background()
    var cancelfn func()
    ctx, cancelfn = context.withtimeout(ctx, timeout)
    defer cancelfn()
    var input = &s3.getobjectinput{
        bucket: aws.string(bucket),
        key:    aws.string(key),
    }
    o, err := svc.getobjectwithcontext(ctx, input)
    if err != nil {
        if aerr, ok := err.(awserr.error); ok && aerr.code() == request.cancelederrorcode {
            log.fatal("download canceled due to timeout", err)
        } else {
            log.fatal("failed to download object", err)
        }
    }
    // load s3 file into memory, assuming small files
    return o.body
  }

上面的代码使用上下文，由于某种原因，对象返回的对象大小是错误的。 由于我在这里不使用上下文，因此我只是将代码转换为使用 getobject(input) 来解决问题。

func getObjectFromS3(svc *s3.S3, bucket, key string) io.Reader {
    var input = &s3.GetObjectInput{
        Bucket: aws.String(bucket),
        Key:    aws.String(key),
    }

    o, err := svc.GetObject(input)
    if err != nil {
        if aerr, ok := err.(awserr.Error); ok {
            switch aerr.Code() {
            case s3.ErrCodeNoSuchKey:
                log.Fatal(s3.ErrCodeNoSuchKey, aerr.Error())
            default:
                log.Fatal(aerr.Error())
            }
        } else {
            // Print the error, cast err to awserr.Error to get the Code and
            // Message from an error.
            log.Fatal(err.Error())
        }
    }

    // Load S3 file into memory, assuming small files
    return o.Body
}

终于介绍完啦！小伙伴们，这篇关于《使用golang上传文件到SFTP主机失败》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！