首页 > 解决方案 > 当客户端重新连接到我的 TCP 服务器时,内存不断增加

问题描述

我是 Go 新手,我正在尝试构建一个简单的 TCP 服务器。我在下面附上了我的代码。每当客户端断开连接时,它将被添加到 server.deadConns。

如果我使用一个简单的客户端并不断连接和断开与 TCP 服务器的连接(下面的代码),我的 Golang 应用程序的内存使用量就会不断上升。如果我然后停止连接并重新连接我的客户端并等待,内存使用量将慢慢开始下降。它需要很长时间才能下降,但它会下降。

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

//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
}

标签: go

解决方案


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


推荐阅读