如何使用go传输多个文件

Question

我正在尝试用 go 编写一个程序，它有两个部分。一部分是尝试将多张图片上传到另一部分服务器的客户端。

服务器端应执行以下操作：

获取要发送的文件数
循环每个文件
获取文件名
获取文件并保存
转到 3

到目前为止，服务器端正在执行以下操作：

func getFileFromClient(connection net.Conn) {
    var numberOfPics int
    var err error
    var receivedBytes int64
    var fileName string
    r := bufio.NewReader(connection)
    strNumberOfPics, err := r.ReadString('\n')
    if err != nil {
        fmt.Printf("Error reading: %s\n", err)
        return
    }
    fmt.Printf("Read: %s\n", strNumberOfPics)
    strNumberOfPics = strings.Trim(strNumberOfPics, "\n")
    numberOfPics, err = strconv.Atoi(strNumberOfPics)
    if err != nil {
        fmt.Printf("Error Atoi: %s\n", err)
        panic("Atoi")
    }
    fmt.Printf("Receiving %d pics:\n", numberOfPics)
    for i := 0; i < numberOfPics; i++ {
        // Getting the file name:
        fileName, err = r.ReadString('\n')
        if err != nil {
            fmt.Printf("Error receiving: %s\n", err)
        }
        fmt.Printf("Filename: %s\n", fileName)
        fileName = strings.Trim(fileName, "\n")
        f, err := os.Create(fileName)
        defer f.Close()
        if err != nil {
            fmt.Println("Error creating file")
        }
        receivedBytes, err = io.Copy(f, connection)
        if err != nil {
            panic("Transmission error")
        }

        fmt.Printf("Transmission finished. Received: %d \n", receivedBytes)
    }
}

io.Copy 只处理一个文件，没有其他文件（因为我认为它不会清空队列）。如果我没有的话，我不想每次都为每个文件重新连接。但我不确定我能做些什么。

有没有人对现有的包或方法有任何帮助的建议？或者示例代码？还是我完全错了，用 go 尝试这个是个坏主意？

我认为如果服务器能够在每次读取后刷新连接缓冲区就足够了，这样就不会读取其他信息 and/or 复制。

非常期待帮助，在此先感谢

编辑：更新后的代码仍然无效。我认为可能是 bufio.reader

func getFileFromClient(connection net.Conn) {
    var numberOfPics int
    var err error
    var receivedBytes int64
    var fileName string
    r := bufio.NewReader(connection)
    strNumberOfPics, err := r.ReadString('\n')
    if err != nil {
        fmt.Printf("Error reading: %s\n", err)
        return
    }
    strNumberOfPics = strings.Trim(strNumberOfPics, "\n")
    numberOfPics, err = strconv.Atoi(strNumberOfPics)
    if err != nil {
        fmt.Printf("Error Atoi: %s\n", err)
        panic("Atoi")
    }
    fmt.Printf("Receiving %d pics:\n", numberOfPics)
    for i := 0; i < numberOfPics; i++ {
        // Getting the file name:
        fileName, err = r.ReadString('\n')
        if err != nil {
            fmt.Printf("Error receiving: %s\n", err)
        }
        fileName = strings.Trim(fileName, "\n")
        fmt.Printf("Filename: %s\n", fileName)
        f, err := os.Create(fileName)
        defer f.Close()
        if err != nil {
            fmt.Println("Error creating file")
        }
        // Get the file size
        strFileSize, err := r.ReadString('\n')
        if err != nil {
            fmt.Printf("Read size error %s\n", err)
            panic("Read size")
        }
        strFileSize = strings.Trim(strFileSize, "\n")
        fileSize, err := strconv.Atoi(strFileSize)
        if err != nil {
            fmt.Printf("Error size Atoi: %s\n", err)
            panic("size Atoi")
        }
        fmt.Printf("Size of pic: %d\n", fileSize)

        receivedBytes, err = io.CopyN(f, connection, int64(fileSize))
        if err != nil {
            fmt.Printf("Transmission error: %s\n", err)
            panic("Transmission error")
        }
        fmt.Printf("Transmission finished. Received: %d \n", receivedBytes)
    }
}

编辑 2：我没有得到这个解决方案。我很确定这是因为我使用了 bufio。然而，我确实通过使用 io.copy 传输单个 zip 文件来让它工作。另一个可行的解决方案是使用 http 传输 zip 文件。如果您在尝试类似的事情时遇到困难并且需要帮助，请随时给我发消息。感谢大家的帮助

Answer 1

到目前为止保持您的实施，您缺少的是 io.Copy() 从源读取直到找到 EOF，因此它将一次性读取所有剩余的图像。

此外，客户端 必须为每个图像发送其大小（以字节为单位）（您可以在发送名称后执行此操作）。在服务器中，只需读取大小，然后使用 io.CopyN() 读取确切的字节数。

编辑：事实上，你也可以像你一样做并行发送图像而不是串行发送图像，这意味着你打开一个新连接每个文件传输然后读取所有文件而无需发送图像数量或图像大小。

如果您需要替代方案，一个不错的选择是使用良好的 HTTP 和多部分请求。内置模块 mime/multipart 允许您通过 HTTP 进行文件传输。当然，这意味着您必须重写您的程序。

Answer 2

我的建议是压缩所有要传输的图像，然后将它们作为单个多部分 POST 请求发送。这样，您就有了了解所有验收标准的标准方法。

您可以使用 https://golang.org/pkg/archive/zip/

轻松压缩多个文件

如何使用go传输多个文件

How to transfer multiple files using go

file

transfer

go