src/Sessions/MemorySessionPool.php - main - githubmirror/ydb-php-sdk

src/Sessions/MemorySessionPool.php
115 строк2 KB
Илья
Update destructor in MemorySessionPool.php
12 июл 2023, 13:22
12 июл 2023, 13:22bdbc5a6
<?php

namespace YdbPlatform\Ydb\Sessions;

use YdbPlatform\Ydb\Retry\Retry;
use YdbPlatform\Ydb\Session;
use YdbPlatform\Ydb\Contracts\SessionPoolContract;

class MemorySessionPool implements SessionPoolContract
{
    /**
     * @var array
     */
    protected static $sessions = [];
    protected static $retry = null;

    public function __construct(Retry &$retry)
    {
        self::$retry = $retry;
    }

    /**
     * Destroy all current sessions.
     * @return void
     */
    public function __destruct()
    {
        foreach (static::$sessions as $session_id => $session) {
            try {
                static::$retry->retry(function () use ($session) {
                    $session->delete();
                }, true);
            } catch (\Exception $e) {
            }
        }
    }

    /**
     * @return Session|null
     */
    public function getIdleSession()
    {
        foreach (static::$sessions as $session_id => $session)
        {
            if ($session->isIdle())
            {
                $this->syncSession($session_id);
                return $session;
            }
        }
    }

    /**
     * @param Session $session
     * @return void
     */
    public function addSession(Session $session)
    {
        static::$sessions[$session->id()] = $session;
    }

    /**
     * @param string $session_id
     * @return void
     */
    public function dropSession($session_id)
    {
        $session = static::$sessions[$session_id] ?? null;

        if ($session)
        {
            if ($session->isAlive())
            {
                $session->delete();
            }
            else
            {
                unset(static::$sessions[$session_id]);
            }
        }
    }

    /**
     * @param string $session_id
     * @return void
     */
    public function syncSession($session_id)
    {
        $session = static::$sessions[$session_id] ?? null;

        if ($session && $session->id() !== $session_id)
        {
            unset(static::$sessions[$session_id]);
            static::$sessions[$session->id()] = $session;
        }
    }

    /**
     * @param Session $session
     * @return void
     */
    public function sessionTaken(Session $session)
    {
        // do nothing
    }

    /**
     * @param Session $session
     * @return void
     */
    public function sessionReleased(Session $session)
    {
        // do nothing
    }
}