TCP服务器在客户端重新连接时，内存持续增长

从现在开始，我们要努力学习啦！今天我给大家带来《TCP服务器在客户端重新连接时，内存持续增长》，感兴趣的朋友请继续看下去吧！下文中的内容我们主要会涉及到等等知识点，如果在阅读本文过程中有遇到不清楚的地方，欢迎留言呀！我们一起讨论，一起学习！

我是 go 新手，我正在尝试构建一个简单的 tcp 服务器。我在下面附上了我的代码。每当客户端断开连接时，它就会被添加到 server.deadconns 中。

如果我使用一个简单的客户端并不断连接和断开到 tcp 服务器（下面的代码），我的 golang 应用程序的内存使用量就会不断增加。如果我停止连接并重新连接客户端并等待，内存使用量将慢慢开始下降。它需要很长时间才能下降，但它会下降。

当我在 azure 上运行 tcp 服务器时，我注意到了这个问题，并且虚拟机上的内存使用量一直在上升，直到机器崩溃。我检查了日志，发现有一些客户端一直保持连接和断开连接。然后我在自己的机器上本地尝试，发现发生了同样的事情。

//NewTCPServer Creates the TCP Server
func NewTCPServer(port string, logger *logrus.Logger) *TCPServer {
    server := new(TCPServer)
    server.socketAPI = socketapi.NewSocketAPI(logger)
    newConns := make(chan net.Conn, 128)
    server.deadConns = make(chan net.Conn, 128)
    server.sessions = make(map[net.Conn]*types.ClientSession)
    fmt.Println(time.Now().Format("2006-01-02 15:04:05.000000") + ": Listening to: " + port)
    listener, err := net.Listen("tcp", ":"+port)
    if err != nil {
        panic(err)
    }
    go func() {
        for {
            conn, err := listener.Accept()
            if err != nil {
                panic(err)
            }
            fmt.Println(time.Now().Format("2006-01-02 15:04:05.000000") + ": Connected: " + conn.RemoteAddr().String())
            newConns <- conn
        }
    }()

    go func(server *TCPServer) {
        for {
            select {
            case conn := <-newConns:
                session := new(types.ClientSession)
                tcp := createTCPSocket(&conn, server)
                session.Socket = tcp
                server.sessions[conn] = session
            case deadConn := <-server.deadConns:
                sess, ok := server.sessions[deadConn]

                if ok {
                    fmt.Println(time.Now().Format("2006-01-02 15:04:05.000000") + ": Found session")
                    if server.socketAPI.StopSubscriptionOnClose(sess) {
                        delete(server.sessions, deadConn)
                    }
                }

                fmt.Println(time.Now().Format("2006-01-02 15:04:05.000000") + ": Delete session!")
                _ = deadConn.Close()
                sess.Socket = nil
                sess = nil
            }
        }
    }(server)
    return server
}

解决方案

我刚刚发现了错误。我没有关闭 TCP 服务器代码中的通道。

终于介绍完啦！小伙伴们，这篇关于《TCP服务器在客户端重新连接时，内存持续增长》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！