客户端重新连接到我的TCP服务器时内存不断增加

时间:2018-10-17 11:17:03

标签: go

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

如果我使用一个简单的Client并保持与TCP Server的连接和断开连接(下面的代码),则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
}

1 个答案:

答案 0 :(得分:2)

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