登录
首页 >  Golang >  Go问答

有没有办法优化该代码? Go 中的 TCP 服务器

来源:stackoverflow

时间:2024-03-17 13:06:28 436浏览 收藏

这篇教程探讨了如何优化 Go 语言中的基本 TCP 服务器。该服务器接受连接、读取数据并写回,但其嵌套循环和对扫描器的重复声明可能导致复杂性和内存泄漏。为了解决这些问题,文章建议在 goroutine 中处理连接,以允许服务器在处理现有连接的同时接受新连接。通过这种方式,服务器可以更有效地管理连接并防止内存泄漏。

问题内容

编码员。这是基本的 tcp 服务器,它接受连接、读取传入数据并写回。

package main

    import (
        "bufio"
        "io"
        "log"
        "net"
    )

    func main() {
        li, err := net.Listen("tcp", ":8080")
        if err != nil {
            log.Fatalln(err)
        }

        defer li.Close()

        for {
            conn, err := li.Accept()
            if err != nil {
                log.Fatalln(err)
            }

            scanner := bufio.NewScanner(conn)

            for scanner.Scan() {
                ln := scanner.Text()
                io.WriteString(conn, ln+"\n")
            }

            conn.Close()
        }

    }

但是,扫描器有一个嵌套循环,并在外部循环的每次迭代中声明新的扫描器。我听说,嵌套循环会导致额外的复杂性,并且可能在无限循环的每次迭代上声明新的扫描器会导致一些内存泄漏。实际上,我不知道如何以其他方式做到这一点,我只想问两件事:

  1. 是否可以通过其他方式做同样的事情?

  2. 我们真的需要在如此低级的服务器上进行更多优化

    抽象?


解决方案


外层循环正在等待新连接,内层循环正在解析输入数据,所以从这个角度来看它没问题。并非所有嵌套循环都是邪恶的。但是,当您处理该单个连接时,服务器不再接受它们(您可以通过尝试从多个客户端连接到服务器来测试)。要解决这个问题,请在 goroutine 中处理连接:

for {
    conn, err := li.Accept()
    if err != nil {
        log.Fatalln(err)
    }

    go func() {
        defer conn.Close()
        scanner := bufio.NewScanner(conn)

        for scanner.Scan() {
            ln := scanner.Text()
            io.WriteString(conn, ln+"\n")
        }
    }()
}

终于介绍完啦!小伙伴们,这篇关于《有没有办法优化该代码? Go 中的 TCP 服务器》的介绍应该让你收获多多了吧!欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识,快来关注吧!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>