Method monitorSessionDisconnect

internal/vnc/session_manager.go:736–782

monitorSessionDisconnect monitors a session for client disconnections and handles immediate cleanup when clients disconnect.

(session *VNCSession)

// monitorSessionDisconnect monitors a session for client disconnections
// and handles immediate cleanup when clients disconnect.
func (sm *SessionManager) monitorSessionDisconnect(session *VNCSession) {
	for {
		select {
		case <-session.disconnectChan:
			// Client disconnected, consider cleanup after a grace period
			if sm.logger != nil {
				sm.logger.Info("Client disconnected from VNC session: session_id=%s, target_type=%s, target_name=%s, active_connections=%d",
					session.ID, session.TargetType, session.TargetName, session.GetConnectionCount())
			}

			// Wait a short grace period to allow for reconnections
			time.Sleep(5 * time.Second)

			// Check if session is still disconnected and remove it
			sm.mutex.Lock()
			if existingSession, exists := sm.sessions[session.ID]; exists {
				if existingSession.GetConnectionCount() == 0 && existingSession.State == SessionStateDisconnected {
					if sm.logger != nil {
						sm.logger.Info("Removing disconnected VNC session after grace period: session_id=%s, target_type=%s, target_name=%s",
							session.ID, session.TargetType, session.TargetName)
					}

					// Shutdown the session
					if err := existingSession.Shutdown(); err != nil && sm.logger != nil {
						sm.logger.Error("Failed to shutdown disconnected session: session_id=%s, error=%v", session.ID, err)
					}

					// Remove from sessions map
					delete(sm.sessions, session.ID)

					// Free up the port
					if existingSession.Port > 0 {
						delete(sm.usedPorts, existingSession.Port)
					}

					// Notify callback of session count change
					sm.notifySessionCountChange()
				}
			}
			sm.mutex.Unlock()

		case <-sm.ctx.Done():
			// Session manager is shutting down
			return
		}
	}
}