Sequence.php 4.23 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
use Doctrine\DBAL\Schema\Visitor\Visitor;
23 24 25
use function count;
use function is_numeric;
use function sprintf;
26

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

    /**
42
     * @var int
43
     */
44 45 46
    protected $initialValue = 1;

    /**
47
     * @var int|null
48 49
     */
    protected $cache = null;
50 51

    /**
52 53 54 55
     * @param string   $name
     * @param int      $allocationSize
     * @param int      $initialValue
     * @param int|null $cache
56
     */
57
    public function __construct($name, $allocationSize = 1, $initialValue = 1, $cache = null)
58 59
    {
        $this->_setName($name);
60 61
        $this->setAllocationSize($allocationSize);
        $this->setInitialValue($initialValue);
62
        $this->cache          = $cache;
63 64
    }

Benjamin Morel's avatar
Benjamin Morel committed
65
    /**
66
     * @return int
Benjamin Morel's avatar
Benjamin Morel committed
67
     */
68 69
    public function getAllocationSize()
    {
70
        return $this->allocationSize;
71 72
    }

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

    /**
82
     * @return int|null
83 84 85 86
     */
    public function getCache()
    {
        return $this->cache;
87
    }
88

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

        return $this;
99 100
    }

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

        return $this;
    }

    /**
114
     * @param int $cache
115 116 117 118 119 120 121 122
     *
     * @return \Doctrine\DBAL\Schema\Sequence
     */
    public function setCache($cache)
    {
        $this->cache = $cache;

        return $this;
123
    }
124

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

        return $tableSequenceName === $sequenceName;
    }

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