file - 如何使用go传输多个文件

Question

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

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

1. 获取要发送的文件数
2. 循环每个文件
3. 获取文件名
4. 获取文件并保存
5. 转到 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 尝试这个是个坏主意？

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

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

编辑:更新后的代码仍然无效。我认为它可能是 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 进行文件传输。当然，这意味着您必须重写您的程序。