使用 Golang 进行 TCP 压缩和传输文件

从现在开始，努力学习吧！本文《使用 Golang 进行 TCP 压缩和传输文件》主要讲解了等等相关知识点，我会在golang学习网中持续更新相关的系列文章，欢迎大家关注并积极留言建议。下面就先一起来看一下本篇正文内容吧，希望能帮到你！

我编写了简单的示例代码，它有效，但接收的文件大小未压缩

我的客户端（用于连接到服务器并发送文件）：

// connect to server
    conn, err := net.Dial("tcp", serverAddr)
    CheckError(err)
    defer conn.Close()

    in, err := os.Open(srcFile)
    if err != nil {
        log.Fatal(err)
    }
    pr, pw := io.Pipe()
    gw, err := gzip.NewWriterLevel(pw, 7)
    CheckError(err)
    go func() {
        n, err := io.Copy(gw, in)
        gw.Close()
        pw.Close()
        log.Printf("copied    %v %v", n, err)
    }()
        //maybe error some next?
    _, err = io.Copy(conn, pr)

请帮忙，如何正确使用带有副本的管道

正确答案

正如我在评论中已经说过的，你的代码可以工作。我创建了一个小例子来测试或看看是否可以解决您的问题。所以我想你可以关闭这个问题。

package main

import (
    "compress/gzip"
    "io"
    "log"
    "net"
    "os"
)

func main() {
    // create a listener on a random port.
    listener, err := net.listen("tcp", "127.0.0.1:")
    if err != nil {
        log.fatal(err)
    }
    log.println("server listening on: " + listener.addr().string())
    done := make(chan struct{})
go func() {
    defer func() { done <- struct{}{} }()
    for {
        conn, err := listener.accept()
        if err != nil {
            log.println(err)
            return
        }
        go func(c net.conn) {
            defer func() {
                c.close()
                done <- struct{}{}
            }()
            buf := make([]byte, 1024)
            for {
                n, err := c.read(buf)
                if err != nil {
                    if err != io.eof {
                        log.println(err)
                    }

                    return
                }
                log.printf("received: %q", buf[:n])
                log.printf("bytes: %d", n)

            }

        }(conn)
    }
}()

    conn, err := net.dial("tcp", listener.addr().string())
    if err != nil {
        log.fatal(err)
    }
    log.println("connected to server.")

    file, err := os.open("./file.txt")
    if err != nil {
        log.fatal(err)
    }

    pr, pw := io.pipe()
    w, err := gzip.newwriterlevel(pw, 7)
    if err != nil {
        log.fatal(err)
    }
    go func() {
        n, err := io.copy(w, file)
        if err != nil {
            log.fatal(err)
        }
        w.close()
        pw.close()
        log.printf("copied to piped writer via the compressed writer: %d", n)
    }()

    n, err := io.copy(conn, pr)
    if err != nil {
        log.fatal(err)
    }
    log.printf("copied to connection: %d", n)

    conn.close()
    <-done
    listener.close()
    <-done
}

该程序的输出是一个简单的文本文件，其中包含许多重复的字符，需要压缩一些内容：该文件有 153 字节，我发送/接收了 46 字节

2022/04/04 11:23:58 Server listening on: 127.0.0.1:58250
2022/04/04 11:23:58 Connected to server.
2022/04/04 11:23:58 received: "\x1f\x8b\b\x00\x00\x00\x00\x00\x00\xff"
2022/04/04 11:23:58 bytes: 10
2022/04/04 11:23:58 copied to piped writer via the compressed writer: 153
2022/04/04 11:23:58 copied to connection: 46
2022/04/04 11:23:58 received: "*I-.I,NI,N\xc1\x01\x8aS\x8a\x13i\bx\xb9pX    \r\b\x00\x00\xff\xff\xc7\xfe\xa6c\x99\x00\x00\x00"
2022/04/04 11:23:58 bytes: 36
2022/04/04 11:23:58 accept tcp 127.0.0.1:58250: use of closed network connection

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