123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132 |
- <?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\Sharding;
- /**
- * Shard Manager for the Connection Pooling Shard Strategy
- *
- * @author Benjamin Eberlei <kontakt@beberlei.de>
- */
- class PoolingShardManager implements ShardManager
- {
- /**
- * @var \Doctrine\DBAL\Sharding\PoolingShardConnection
- */
- private $conn;
- /**
- * @var \Doctrine\DBAL\Sharding\ShardChoser\ShardChoser
- */
- private $choser;
- /**
- * @var string|null
- */
- private $currentDistributionValue;
- /**
- * @param \Doctrine\DBAL\Sharding\PoolingShardConnection $conn
- */
- public function __construct(PoolingShardConnection $conn)
- {
- $params = $conn->getParams();
- $this->conn = $conn;
- $this->choser = $params['shardChoser'];
- }
- /**
- * @return void
- */
- public function selectGlobal()
- {
- $this->conn->connect(0);
- $this->currentDistributionValue = null;
- }
- /**
- * @param string $distributionValue
- *
- * @return void
- */
- public function selectShard($distributionValue)
- {
- $shardId = $this->choser->pickShard($distributionValue, $this->conn);
- $this->conn->connect($shardId);
- $this->currentDistributionValue = $distributionValue;
- }
- /**
- * @return string|null
- */
- public function getCurrentDistributionValue()
- {
- return $this->currentDistributionValue;
- }
- /**
- * @return array
- */
- public function getShards()
- {
- $params = $this->conn->getParams();
- $shards = array();
- foreach ($params['shards'] as $shard) {
- $shards[] = array('id' => $shard['id']);
- }
- return $shards;
- }
- /**
- * @param string $sql
- * @param array $params
- * @param array $types
- *
- * @return array
- *
- * @throws \RuntimeException
- */
- public function queryAll($sql, array $params, array $types)
- {
- $shards = $this->getShards();
- if (!$shards) {
- throw new \RuntimeException("No shards found.");
- }
- $result = array();
- $oldDistribution = $this->getCurrentDistributionValue();
- foreach ($shards as $shard) {
- $this->selectShard($shard['id']);
- foreach ($this->conn->fetchAll($sql, $params, $types) as $row) {
- $result[] = $row;
- }
- }
- if ($oldDistribution === null) {
- $this->selectGlobal();
- } else {
- $this->selectShard($oldDistribution);
- }
- return $result;
- }
- }
|