Sequence.php 4.26 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
    protected $allocationSize = 1;
37 38

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

    /**
     * @var integer|null
     */
    protected $cache = null;
47 48

    /**
jeroendedauw's avatar
jeroendedauw committed
49 50 51 52
     * @param string       $name
     * @param integer      $allocationSize
     * @param integer      $initialValue
     * @param integer|null $cache
53
     */
54
    public function __construct($name, $allocationSize = 1, $initialValue = 1, $cache = null)
55 56
    {
        $this->_setName($name);
57 58 59
        $this->allocationSize = is_numeric($allocationSize) ? $allocationSize : 1;
        $this->initialValue = is_numeric($initialValue) ? $initialValue : 1;
        $this->cache = $cache;
60 61
    }

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

Benjamin Morel's avatar
Benjamin Morel committed
70 71 72
    /**
     * @return integer
     */
73 74
    public function getInitialValue()
    {
75 76 77 78 79 80 81 82 83
        return $this->initialValue;
    }

    /**
     * @return integer|null
     */
    public function getCache()
    {
        return $this->cache;
84
    }
85

Benjamin Morel's avatar
Benjamin Morel committed
86 87 88
    /**
     * @param integer $allocationSize
     *
89
     * @return \Doctrine\DBAL\Schema\Sequence
Benjamin Morel's avatar
Benjamin Morel committed
90
     */
91 92
    public function setAllocationSize($allocationSize)
    {
93 94 95
        $this->allocationSize = is_numeric($allocationSize) ? $allocationSize : 1;

        return $this;
96 97
    }

Benjamin Morel's avatar
Benjamin Morel committed
98 99 100
    /**
     * @param integer $initialValue
     *
101
     * @return \Doctrine\DBAL\Schema\Sequence
Benjamin Morel's avatar
Benjamin Morel committed
102
     */
103 104
    public function setInitialValue($initialValue)
    {
105 106 107 108 109 110 111 112 113 114 115 116 117 118 119
        $this->initialValue = is_numeric($initialValue) ? $initialValue : 1;

        return $this;
    }

    /**
     * @param integer $cache
     *
     * @return \Doctrine\DBAL\Schema\Sequence
     */
    public function setCache($cache)
    {
        $this->cache = $cache;

        return $this;
120
    }
121

122
    /**
Benjamin Morel's avatar
Benjamin Morel committed
123
     * Checks if this sequence is an autoincrement sequence for a given table.
124 125 126 127
     *
     * 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
128
     * @param \Doctrine\DBAL\Schema\Table $table
129
     *
Benjamin Morel's avatar
Benjamin Morel committed
130
     * @return boolean
131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151
     */
    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());
152
        $tableSequenceName = sprintf('%s_%s_seq', $tableName, $column->getShortestName($table->getNamespaceName()));
153 154 155 156

        return $tableSequenceName === $sequenceName;
    }

157
    /**
Benjamin Morel's avatar
Benjamin Morel committed
158 159 160
     * @param \Doctrine\DBAL\Schema\Visitor\Visitor $visitor
     *
     * @return void
161 162 163 164 165 166
     */
    public function visit(Visitor $visitor)
    {
        $visitor->acceptSequence($this);
    }
}