suda\orm\DataSource

Complexity

Total Complexity

15

Size/Duplication

Total Lines	154
Duplicated Lines	0 %

Importance

Changes

0

9 Methods

Rating	Name	Duplication	Size	Complexity
A	addWrite()	0	6	2
A	new()	0	6	2
A	add()	0	5	1
A	addRead()	0	6	2
A	selectReadConnection()	0	4	1
A	read()	0	4	1
A	write()	0	4	1
A	selectWriteConnection()	0	5	2
A	__construct()	0	8	3

<?php

namespace suda\orm;

use function in_array;
use suda\orm\connection\Connection;
use suda\orm\exception\SQLException;
use suda\orm\connection\MySQLConnection;
use suda\orm\connection\SQLiteConnection;

class DataSource
{
    /**
     * 读连接
     *
     * @var Connection[]
     */
    protected $write = [];

    /**
     * 写连接
     *
     * @var Connection[]
     */
    protected $read = [];

    /**
     * 当前写数据库
     *
     * @var Connection|null
     */
    protected $master;

    /**
     * 当前读数据库
     *
     * @var Connection|null
     */
    protected $slave;

    protected static $type = [
        'mysql' => MySQLConnection::class,
        'sqlite' => SQLiteConnection::class,
    ];


    /**
     * DataSource constructor.
     * @param Connection[]|Connection $connections
     */
    public function __construct($connections)
    {
        if (is_array($connections)) {
            foreach ($connections as $connection) {
                $this->add($connection);
            }
        } else {
            $this->add($connections);
        }
    }

    /**
     * 添加连接
     *
     * @param Connection $connection
     * @return $this
     */
    public function add(Connection $connection)
    {
        $this->addRead($connection);
        $this->addWrite($connection);
        return $this;
    }

    /**
     * 添加读连接
     *
     * @param Connection $connection
     * @return $this
     */
    public function addRead(Connection $connection)
    {
        if (!in_array($connection, $this->read)) {
            $this->read[] = $connection;
        }
        return $this;
    }


    /**
     * 添加写连接
     *
     * @param Connection $connection
     * @return $this
     */
    public function addWrite(Connection $connection)
    {
        if (!in_array($connection, $this->write)) {
            $this->write[] = $connection;
        }
        return $this;
    }

    /**
     * 获取写连接
     *
     * @return Connection
     */
    public function read(): Connection
    {
        $this->selectReadConnection();
        return $this->slave;
    }

    /**
     * 获取读连接
     *
     * @return Connection
     */
    public function write(): Connection
    {
        $this->selectWriteConnection();
        return $this->master;
    }

    /**
     * 创建连接
     *
     * @param string $type
     * @param array $config
     * @param string|null $name
     * @return Connection
     * @throws SQLException
     */
    public static function new(string $type, array $config, ?string $name = null): Connection
    {
        if (array_key_exists($type, static::$type)) {
            return new static::$type[$type]($config, $name);
        } else {
            throw new SQLException(sprintf('no connection type of %s', $type));
        }
    }

    /**
     * 读数据库选择
     *
     * @return void
     */
    protected function selectReadConnection()
    {
        $postion = mt_rand(0, count($this->read) - 1);
        $this->slave = $this->read[$postion];
    }

    /**
     * 写数据库选择
     *
     * @return void
     */
    protected function selectWriteConnection()
    {
        if ($this->master === null) {
            $position = mt_rand(0, count($this->write) - 1);
            $this->master = $this->write[$position];
        }
    }
}