Sequence.php 3.65 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
<?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
Benjamin Eberlei's avatar
Benjamin Eberlei committed
16
 * and is licensed under the MIT license. For more information, see
17 18 19 20 21
 * <http://www.doctrine-project.org>.
 */

namespace Doctrine\DBAL\Schema;

22 23
use Doctrine\DBAL\Schema\Visitor\Visitor;

24
/**
Benjamin Morel's avatar
Benjamin Morel committed
25
 * Sequence structure.
26
 *
Benjamin Morel's avatar
Benjamin Morel committed
27 28 29
 * @link   www.doctrine-project.org
 * @since  2.0
 * @author Benjamin Eberlei <kontakt@beberlei.de>
30 31 32 33
 */
class Sequence extends AbstractAsset
{
    /**
Benjamin Morel's avatar
Benjamin Morel committed
34
     * @var integer
35 36 37 38
     */
    protected $_allocationSize = 1;

    /**
Benjamin Morel's avatar
Benjamin Morel committed
39
     * @var integer
40 41 42 43
     */
    protected $_initialValue = 1;

    /**
Benjamin Morel's avatar
Benjamin Morel committed
44 45 46
     * @param string  $name
     * @param integer $allocationSize
     * @param integer $initialValue
47 48 49 50
     */
    public function __construct($name, $allocationSize=1, $initialValue=1)
    {
        $this->_setName($name);
51 52
        $this->_allocationSize = (is_numeric($allocationSize))?$allocationSize:1;
        $this->_initialValue = (is_numeric($initialValue))?$initialValue:1;
53 54
    }

Benjamin Morel's avatar
Benjamin Morel committed
55 56 57
    /**
     * @return integer
     */
58 59 60 61 62
    public function getAllocationSize()
    {
        return $this->_allocationSize;
    }

Benjamin Morel's avatar
Benjamin Morel committed
63 64 65
    /**
     * @return integer
     */
66 67 68 69
    public function getInitialValue()
    {
        return $this->_initialValue;
    }
70

Benjamin Morel's avatar
Benjamin Morel committed
71 72 73 74 75
    /**
     * @param integer $allocationSize
     *
     * @return void
     */
76 77 78 79 80
    public function setAllocationSize($allocationSize)
    {
        $this->_allocationSize = (is_numeric($allocationSize))?$allocationSize:1;
    }

Benjamin Morel's avatar
Benjamin Morel committed
81 82 83 84 85
    /**
     * @param integer $initialValue
     *
     * @return void
     */
86 87 88 89
    public function setInitialValue($initialValue)
    {
        $this->_initialValue = (is_numeric($initialValue))?$initialValue:1;
    }
90

91
    /**
Benjamin Morel's avatar
Benjamin Morel committed
92
     * Checks if this sequence is an autoincrement sequence for a given table.
93 94 95 96
     *
     * This is used inside the comparator to not report sequences as missing,
     * when the "from" schema implicitly creates the sequences.
     *
Benjamin Morel's avatar
Benjamin Morel committed
97
     * @param \Doctrine\DBAL\Schema\Table $table
98
     *
Benjamin Morel's avatar
Benjamin Morel committed
99
     * @return boolean
100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125
     */
    public function isAutoIncrementsFor(Table $table)
    {
        if ( ! $table->hasPrimaryKey()) {
            return false;
        }

        $pkColumns = $table->getPrimaryKey()->getColumns();

        if (count($pkColumns) != 1) {
            return false;
        }

        $column = $table->getColumn($pkColumns[0]);

        if ( ! $column->getAutoincrement()) {
            return false;
        }

        $sequenceName      = $this->getShortestName($table->getNamespaceName());
        $tableName         = $table->getShortestName($table->getNamespaceName());
        $tableSequenceName = sprintf('%s_%s_seq', $tableName, $pkColumns[0]);

        return $tableSequenceName === $sequenceName;
    }

126
    /**
Benjamin Morel's avatar
Benjamin Morel committed
127 128 129
     * @param \Doctrine\DBAL\Schema\Visitor\Visitor $visitor
     *
     * @return void
130 131 132 133 134 135
     */
    public function visit(Visitor $visitor)
    {
        $visitor->acceptSequence($this);
    }
}