NetworkSessionManager.php

<?php
 
/*
 *
 *  ____            _        _   __  __ _                  __  __ ____
 * |  _ \ ___   ___| | _____| |_|  \/  (_)_ __   ___      |  \/  |  _ \
 * | |_) / _ \ / __| |/ / _ \ __| |\/| | | '_ \ / _ \_____| |\/| | |_) |
 * |  __/ (_) | (__|   <  __/ |_| |  | | | | | |  __/_____| |  | |  __/
 * |_|   \___/ \___|_|\_\___|\__|_|  |_|_|_| |_|\___|     |_|  |_|_|
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * @author PocketMine Team
 * @link http://www.pocketmine.net/
 *
 *
 */
 
declare(strict_types=1);
 
namespace pocketmine\network;
 
use pocketmine\lang\Translatable;
use pocketmine\network\mcpe\NetworkSession;
use function count;
use function spl_object_id;
 
class NetworkSessionManager{
 
    private array $sessions = [];
 
    private array $pendingLoginSessions = [];
 
    public function add(NetworkSession $session) : void{
        $idx = spl_object_id($session);
        $this->sessions[$idx] = $session;
        $this->pendingLoginSessions[$idx] = $session;
    }
 
    public function markLoginReceived(NetworkSession $session) : void{
        $idx = spl_object_id($session);
        unset($this->pendingLoginSessions[$idx]);
    }
 
    public function remove(NetworkSession $session) : void{
        $idx = spl_object_id($session);
        unset($this->sessions[$idx]);
        unset($this->pendingLoginSessions[$idx]);
    }
 
    public function getSessionCount() : int{
        return count($this->sessions);
    }
 
    public function getValidSessionCount() : int{
        return count($this->sessions) - count($this->pendingLoginSessions);
    }
 
    public function getSessions() : array{ return $this->sessions; }
 
    public function tick() : void{
        foreach($this->sessions as $k => $session){
            $session->tick();
            if(!$session->isConnected()){
                unset($this->sessions[$k]);
            }
        }
    }
 
    public function close(Translatable|string $reason = "", Translatable|string|null $disconnectScreenMessage = null) : void{
        foreach($this->sessions as $session){
            $session->disconnect($reason, $disconnectScreenMessage);
        }
        $this->sessions = [];
    }
}