connectionpool.php

<?php
namespace Bitrix\Main\Data;
 
use Bitrix\Main;
use Bitrix\Main\Config;
 
class ConnectionPool
{
    protected $connections = [];
 
    protected $connectionParameters = [];
 
    protected $slavePossible = true;
    protected $ignoreDml = 0;
    protected $masterOnly = 0;
    protected $slaveConnection = null;
 
    const DEFAULT_CONNECTION_NAME = "default";
 
    public function __construct()
    {
    }
 
    protected function createConnection($name, $parameters)
    {
        $className = $parameters['className'];
 
        if (!class_exists($className))
        {
            throw new Config\ConfigurationException(sprintf(
                "Class '%s' for '%s' connection was not found", $className, $name
            ));
        }
 
        $connection = new $className($parameters);
 
        $this->connections[$name] = $connection;
 
        return $connection;
    }
 
    public function getConnection($name = "")
    {
        if ($name === "")
        {
            $name = self::DEFAULT_CONNECTION_NAME;
        }
 
        if (!isset($this->connections[$name]))
        {
            $connParameters = $this->getConnectionParameters($name);
            if (!empty($connParameters) && is_array($connParameters))
            {
                $this->createConnection($name, $connParameters);
            }
        }
 
        if (isset($this->connections[$name]))
        {
            return $this->connections[$name];
        }
 
        return null;
    }
 
    protected function getConnectionParameters($name)
    {
        if (!is_string($name))
        {
            throw new Main\ArgumentTypeException("name", "string");
        }
 
        if ($name === "")
        {
            throw new Main\ArgumentNullException("name");
        }
 
        $params = null;
        if (isset($this->connectionParameters[$name]) && !empty($this->connectionParameters[$name]))
        {
            $params = $this->connectionParameters[$name];
        }
        else
        {
            $configParams = Config\Configuration::getValue('connections');
            if (isset($configParams[$name]) && !empty($configParams[$name]))
            {
                $params = $configParams[$name];
            }
        }
 
        if ($params !== null && $name === static::DEFAULT_CONNECTION_NAME && !isset($params["include_after_connected"]))
        {
            $params["include_after_connected"] = \Bitrix\Main\Loader::getPersonal("php_interface/after_connect_d7.php");
        }
 
        return $params;
    }
 
    public function setConnectionParameters($name, $parameters)
    {
        $this->connectionParameters[$name] = $parameters;
 
        if(isset($this->connections[$name]))
        {
            unset($this->connections[$name]);
        }
    }
 
    public function getSlaveConnection($sql)
    {
        if($this->masterOnly > 0)
        {
            //We requested to process all queries
            //by master connection
        }
        elseif($this->slavePossible)
        {
            $isSelect = preg_match('/^\s*(select|show)/i', $sql) && !preg_match('/get_lock/i', $sql);
            if(!$isSelect && $this->ignoreDml <= 0)
            {
                $this->slavePossible = false;
            }
 
            if($isSelect)
            {
                if($this->slaveConnection === null)
                {
                    $this->useMasterOnly(true);
 
                    $this->slaveConnection = $this->createSlaveConnection();
 
                    $this->useMasterOnly(false);
                }
 
                if(is_object($this->slaveConnection))
                {
                    return $this->slaveConnection;
                }
            }
        }
        return null;
    }
 
    public function useMasterOnly($mode)
    {
        if($mode)
        {
            $this->masterOnly++;
        }
        else
        {
            $this->masterOnly--;
        }
    }
 
    public function ignoreDml($mode)
    {
        if($mode)
        {
            $this->ignoreDml++;
        }
        else
        {
            $this->ignoreDml--;
        }
    }
 
    protected function createSlaveConnection()
    {
        if(!Main\Loader::includeModule('cluster'))
        {
            return false;
        }
 
        $found = \CClusterSlave::GetRandomNode();
 
        if($found !== false)
        {
            $node = \CClusterDBNode::GetByID($found["ID"]);
 
            if(is_array($node) && $node["ACTIVE"] == "Y" && ($node["STATUS"] == "ONLINE" || $node["STATUS"] == "READY"))
            {
                $parameters = [
                    'host' => $node["DB_HOST"],
                    'database' => $node["DB_NAME"],
                    'login' => $node["DB_LOGIN"],
                    'password' => $node["DB_PASSWORD"],
                ];
 
                $connection = $this->cloneConnection(self::DEFAULT_CONNECTION_NAME, "node".$node["ID"], $parameters);
 
                if($connection instanceof Main\DB\Connection)
                {
                    $connection->setNodeId($node["ID"]);
                }
 
                return $connection;
            }
        }
        return false;
    }
 
    public function cloneConnection($name, $newName, array $parameters=array())
    {
        $defParameters = $this->getConnectionParameters($name);
        if (empty($defParameters) || !is_array($defParameters))
        {
            throw new Config\ConfigurationException(sprintf("Database connection '%s' is not found", $name));
        }
        $parameters = array_merge($defParameters, $parameters);
 
        $connection = $this->createConnection($newName, $parameters);
 
        return $connection;
    }
 
    public function isSlavePossible()
    {
        return $this->slavePossible;
    }
 
    public function isMasterOnly()
    {
        return ($this->masterOnly > 0);
    }
 
    public function disconnect()
    {
        foreach ($this->connections as $connection)
        {
            if ($connection instanceof Main\DB\Connection)
            {
                $connection->disconnect();
            }
        }
    }
}