mirror of
https://github.com/OpenDiablo2/OpenDiablo2
synced 2024-06-02 22:11:10 +00:00
c27fb572bc
* Added automap.go stub * Handle errors in original AutoMap.txt file * Completed AutoMapRecord struct and comments * AutoMap loader implemented * Update from base repo * Comments added to d2netpacket and d2netpackettype. Note, the Overview for d2netpacket is in net_packet.go. It could be placed in a doc.go file but net_packet.go seemed appropriate in this case. * Comments added to d2server * client_connection.go missed from previous commit * Comments added to d2client * Doc.go added to d2networking and other corrections * Unused parameter present to satisfy interface, named it '_' * Unhandled errors and other issues resolved in d2networking. This commit adds simple error reporting and creates a lot of mess. I will be doing one more pass to tidy things up a bit before I move on. * Commented out continue on RemoteClientConnection error and implemented stringer in message type * NetPacketType.String uses map instead of switch. * Map moved into String method
95 lines
3.1 KiB
Go
95 lines
3.1 KiB
Go
package d2server
|
|
|
|
import (
|
|
"log"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/OpenDiablo2/OpenDiablo2/d2networking/d2client/d2clientconnectiontype"
|
|
"github.com/OpenDiablo2/OpenDiablo2/d2networking/d2netpacket"
|
|
)
|
|
|
|
// ConnectionManager is responsible for cleanup up connections accepted by the game server.
|
|
// As the server communicates over UDP and is stateless we need to implement some loose state
|
|
// management via a ping/pong system. ConnectionManager also handles communication for
|
|
// graceful shutdowns.
|
|
type ConnectionManager struct {
|
|
sync.RWMutex
|
|
retries int // Number of attempts before the dropping the client
|
|
interval time.Duration // How long to wait before each ping/pong test
|
|
gameServer *GameServer // The GameServer with the connections being managed
|
|
status map[string]int // Map of inflight ping/pong requests
|
|
}
|
|
|
|
// CreateConnectionManager constructs a new ConnectionManager and calls
|
|
// ConnectionManager.Run() in a goroutine before retuning a pointer to
|
|
// the new ConnectionManager.
|
|
func CreateConnectionManager(gameServer *GameServer) *ConnectionManager {
|
|
manager := &ConnectionManager{
|
|
retries: 3,
|
|
interval: time.Millisecond * 1000,
|
|
gameServer: gameServer,
|
|
status: make(map[string]int),
|
|
}
|
|
|
|
go manager.Run()
|
|
|
|
return manager
|
|
}
|
|
|
|
// Run starts up any watchers for for the connection manager
|
|
func (c *ConnectionManager) Run() {
|
|
log.Print("Starting connection manager...")
|
|
for {
|
|
c.checkPeers()
|
|
time.Sleep(c.interval)
|
|
}
|
|
}
|
|
|
|
// checkPeers manages connection validation and cleanup for all peers.
|
|
func (c *ConnectionManager) checkPeers() {
|
|
for id, connection := range c.gameServer.clientConnections {
|
|
if connection.GetConnectionType() != d2clientconnectiontype.Local {
|
|
if err := connection.SendPacketToClient(d2netpacket.CreatePingPacket()); err != nil {
|
|
log.Printf("Cannot ping client id: %s", id)
|
|
}
|
|
c.RWMutex.Lock()
|
|
c.status[id] += 1
|
|
|
|
if c.status[id] >= c.retries {
|
|
delete(c.status, id)
|
|
c.Drop(id)
|
|
}
|
|
|
|
c.RWMutex.Unlock()
|
|
}
|
|
}
|
|
}
|
|
|
|
// Recv simply resets the counter, acknowledging we have received a pong from the client.
|
|
func (c *ConnectionManager) Recv(id string) {
|
|
c.status[id] = 0
|
|
}
|
|
|
|
// Drop removes the client id from the connection pool of the game server.
|
|
func (c *ConnectionManager) Drop(id string) {
|
|
c.gameServer.RWMutex.Lock()
|
|
defer c.gameServer.RWMutex.Unlock()
|
|
delete(c.gameServer.clientConnections, id)
|
|
log.Printf("%s has been disconnected...", id)
|
|
}
|
|
|
|
// Shutdown will notify all of the clients that the server has been shutdown.
|
|
func (c *ConnectionManager) Shutdown() {
|
|
// TODO: Currently this will never actually get called as the go routines are never signaled about the application termination.
|
|
// Things can be done more cleanly once we have graceful exits however we still need to account for other OS Signals
|
|
log.Print("Notifying clients server is shutting down...")
|
|
for _, connection := range c.gameServer.clientConnections {
|
|
err := connection.SendPacketToClient(d2netpacket.CreateServerClosedPacket())
|
|
if err != nil {
|
|
log.Printf("ConnectionManager: error sending ServerClosedPacket to client ID %s: %s", connection.GetUniqueId(), err)
|
|
}
|
|
}
|
|
Stop()
|
|
}
|