go - 当客户端重新连接到我的 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
}
解决方案
我刚刚发现了错误。我没有在我的 TCP 服务器代码中关闭通道。