123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375 |
- <?php
- /*
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- *
- * This software consists of voluntary contributions made by many individuals
- * and is licensed under the MIT license. For more information, see
- * <http://www.doctrine-project.org>.
- */
- namespace Doctrine\DBAL\Connections;
- use Doctrine\DBAL\Connection;
- use Doctrine\DBAL\Driver;
- use Doctrine\DBAL\Configuration;
- use Doctrine\Common\EventManager;
- use Doctrine\DBAL\Event\ConnectionEventArgs;
- use Doctrine\DBAL\Events;
- /**
- * Master-Slave Connection
- *
- * Connection can be used with master-slave setups.
- *
- * Important for the understanding of this connection should be how and when
- * it picks the slave or master.
- *
- * 1. Slave if master was never picked before and ONLY if 'getWrappedConnection'
- * or 'executeQuery' is used.
- * 2. Master picked when 'exec', 'executeUpdate', 'insert', 'delete', 'update', 'createSavepoint',
- * 'releaseSavepoint', 'beginTransaction', 'rollback', 'commit', 'query' or
- * 'prepare' is called.
- * 3. If master was picked once during the lifetime of the connection it will always get picked afterwards.
- * 4. One slave connection is randomly picked ONCE during a request.
- *
- * ATTENTION: You can write to the slave with this connection if you execute a write query without
- * opening up a transaction. For example:
- *
- * $conn = DriverManager::getConnection(...);
- * $conn->executeQuery("DELETE FROM table");
- *
- * Be aware that Connection#executeQuery is a method specifically for READ
- * operations only.
- *
- * This connection is limited to slave operations using the
- * Connection#executeQuery operation only, because it wouldn't be compatible
- * with the ORM or SchemaManager code otherwise. Both use all the other
- * operations in a context where writes could happen to a slave, which makes
- * this restricted approach necessary.
- *
- * You can manually connect to the master at any time by calling:
- *
- * $conn->connect('master');
- *
- * Instantiation through the DriverManager looks like:
- *
- * @example
- *
- * $conn = DriverManager::getConnection(array(
- * 'wrapperClass' => 'Doctrine\DBAL\Connections\MasterSlaveConnection',
- * 'driver' => 'pdo_mysql',
- * 'master' => array('user' => '', 'password' => '', 'host' => '', 'dbname' => ''),
- * 'slaves' => array(
- * array('user' => 'slave1', 'password', 'host' => '', 'dbname' => ''),
- * array('user' => 'slave2', 'password', 'host' => '', 'dbname' => ''),
- * )
- * ));
- *
- * You can also pass 'driverOptions' and any other documented option to each of this drivers to pass additional information.
- *
- * @author Lars Strojny <lstrojny@php.net>
- * @author Benjamin Eberlei <kontakt@beberlei.de>
- */
- class MasterSlaveConnection extends Connection
- {
- /**
- * Master and slave connection (one of the randomly picked slaves).
- *
- * @var \Doctrine\DBAL\Driver\Connection[]
- */
- protected $connections = array('master' => null, 'slave' => null);
- /**
- * You can keep the slave connection and then switch back to it
- * during the request if you know what you are doing.
- *
- * @var boolean
- */
- protected $keepSlave = false;
- /**
- * Creates Master Slave Connection.
- *
- * @param array $params
- * @param \Doctrine\DBAL\Driver $driver
- * @param \Doctrine\DBAL\Configuration|null $config
- * @param \Doctrine\Common\EventManager|null $eventManager
- *
- * @throws \InvalidArgumentException
- */
- public function __construct(array $params, Driver $driver, Configuration $config = null, EventManager $eventManager = null)
- {
- if ( !isset($params['slaves']) || !isset($params['master']) ) {
- throw new \InvalidArgumentException('master or slaves configuration missing');
- }
- if ( count($params['slaves']) == 0 ) {
- throw new \InvalidArgumentException('You have to configure at least one slaves.');
- }
- $params['master']['driver'] = $params['driver'];
- foreach ($params['slaves'] as $slaveKey => $slave) {
- $params['slaves'][$slaveKey]['driver'] = $params['driver'];
- }
- $this->keepSlave = isset($params['keepSlave']) ? (bool)$params['keepSlave'] : false;
- parent::__construct($params, $driver, $config, $eventManager);
- }
- /**
- * Checks if the connection is currently towards the master or not.
- *
- * @return boolean
- */
- public function isConnectedToMaster()
- {
- return $this->_conn !== null && $this->_conn === $this->connections['master'];
- }
- /**
- * {@inheritDoc}
- */
- public function connect($connectionName = null)
- {
- $requestedConnectionChange = ($connectionName !== null);
- $connectionName = $connectionName ?: 'slave';
- if ( $connectionName !== 'slave' && $connectionName !== 'master' ) {
- throw new \InvalidArgumentException("Invalid option to connect(), only master or slave allowed.");
- }
- // If we have a connection open, and this is not an explicit connection
- // change request, then abort right here, because we are already done.
- // This prevents writes to the slave in case of "keepSlave" option enabled.
- if ($this->_conn && !$requestedConnectionChange) {
- return false;
- }
- $forceMasterAsSlave = false;
- if ($this->getTransactionNestingLevel() > 0) {
- $connectionName = 'master';
- $forceMasterAsSlave = true;
- }
- if ($this->connections[$connectionName]) {
- if ($forceMasterAsSlave) {
- $this->connections['slave'] = $this->_conn = $this->connections['master'];
- } else {
- $this->_conn = $this->connections[$connectionName];
- }
- return false;
- }
- if ($connectionName === 'master') {
- // Set slave connection to master to avoid invalid reads
- if ($this->connections['slave'] && ! $this->keepSlave) {
- unset($this->connections['slave']);
- }
- $this->connections['master'] = $this->_conn = $this->connectTo($connectionName);
- if ( ! $this->keepSlave) {
- $this->connections['slave'] = $this->connections['master'];
- }
- } else {
- $this->connections['slave'] = $this->_conn = $this->connectTo($connectionName);
- }
- if ($this->_eventManager->hasListeners(Events::postConnect)) {
- $eventArgs = new ConnectionEventArgs($this);
- $this->_eventManager->dispatchEvent(Events::postConnect, $eventArgs);
- }
- return true;
- }
- /**
- * Connects to a specific connection.
- *
- * @param string $connectionName
- *
- * @return \Doctrine\DBAL\Driver
- */
- protected function connectTo($connectionName)
- {
- $params = $this->getParams();
- $driverOptions = isset($params['driverOptions']) ? $params['driverOptions'] : array();
- $connectionParams = $this->chooseConnectionConfiguration($connectionName, $params);
- $user = isset($connectionParams['user']) ? $connectionParams['user'] : null;
- $password = isset($connectionParams['password']) ? $connectionParams['password'] : null;
- return $this->_driver->connect($connectionParams, $user, $password, $driverOptions);
- }
- /**
- * @param string $connectionName
- * @param array $params
- *
- * @return mixed
- */
- protected function chooseConnectionConfiguration($connectionName, $params)
- {
- if ($connectionName === 'master') {
- return $params['master'];
- }
- return $params['slaves'][array_rand($params['slaves'])];
- }
- /**
- * {@inheritDoc}
- */
- public function executeUpdate($query, array $params = array(), array $types = array())
- {
- $this->connect('master');
- return parent::executeUpdate($query, $params, $types);
- }
- /**
- * {@inheritDoc}
- */
- public function beginTransaction()
- {
- $this->connect('master');
- return parent::beginTransaction();
- }
- /**
- * {@inheritDoc}
- */
- public function commit()
- {
- $this->connect('master');
- return parent::commit();
- }
- /**
- * {@inheritDoc}
- */
- public function rollBack()
- {
- $this->connect('master');
- return parent::rollBack();
- }
- /**
- * {@inheritDoc}
- */
- public function delete($tableName, array $identifier, array $types = array())
- {
- $this->connect('master');
- return parent::delete($tableName, $identifier, $types);
- }
- /**
- * {@inheritDoc}
- */
- public function update($tableName, array $data, array $identifier, array $types = array())
- {
- $this->connect('master');
- return parent::update($tableName, $data, $identifier, $types);
- }
- /**
- * {@inheritDoc}
- */
- public function insert($tableName, array $data, array $types = array())
- {
- $this->connect('master');
- return parent::insert($tableName, $data, $types);
- }
- /**
- * {@inheritDoc}
- */
- public function exec($statement)
- {
- $this->connect('master');
- return parent::exec($statement);
- }
- /**
- * {@inheritDoc}
- */
- public function createSavepoint($savepoint)
- {
- $this->connect('master');
- return parent::createSavepoint($savepoint);
- }
- /**
- * {@inheritDoc}
- */
- public function releaseSavepoint($savepoint)
- {
- $this->connect('master');
- return parent::releaseSavepoint($savepoint);
- }
- /**
- * {@inheritDoc}
- */
- public function rollbackSavepoint($savepoint)
- {
- $this->connect('master');
- return parent::rollbackSavepoint($savepoint);
- }
- /**
- * {@inheritDoc}
- */
- public function query()
- {
- $this->connect('master');
- $args = func_get_args();
- $logger = $this->getConfiguration()->getSQLLogger();
- if ($logger) {
- $logger->startQuery($args[0]);
- }
- $statement = call_user_func_array(array($this->_conn, 'query'), $args);
- if ($logger) {
- $logger->stopQuery();
- }
- return $statement;
- }
- /**
- * {@inheritDoc}
- */
- public function prepare($statement)
- {
- $this->connect('master');
- return parent::prepare($statement);
- }
- }
|