139 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			139 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
| <?php
 | |
| 
 | |
| namespace Doctrine\DBAL\Schema;
 | |
| 
 | |
| use Doctrine\DBAL\Schema\Visitor\Visitor;
 | |
| use function count;
 | |
| use function is_numeric;
 | |
| use function sprintf;
 | |
| 
 | |
| /**
 | |
|  * Sequence structure.
 | |
|  */
 | |
| class Sequence extends AbstractAsset
 | |
| {
 | |
|     /** @var int */
 | |
|     protected $allocationSize = 1;
 | |
| 
 | |
|     /** @var int */
 | |
|     protected $initialValue = 1;
 | |
| 
 | |
|     /** @var int|null */
 | |
|     protected $cache = null;
 | |
| 
 | |
|     /**
 | |
|      * @param string   $name
 | |
|      * @param int      $allocationSize
 | |
|      * @param int      $initialValue
 | |
|      * @param int|null $cache
 | |
|      */
 | |
|     public function __construct($name, $allocationSize = 1, $initialValue = 1, $cache = null)
 | |
|     {
 | |
|         $this->_setName($name);
 | |
|         $this->setAllocationSize($allocationSize);
 | |
|         $this->setInitialValue($initialValue);
 | |
|         $this->cache = $cache;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @return int
 | |
|      */
 | |
|     public function getAllocationSize()
 | |
|     {
 | |
|         return $this->allocationSize;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @return int
 | |
|      */
 | |
|     public function getInitialValue()
 | |
|     {
 | |
|         return $this->initialValue;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @return int|null
 | |
|      */
 | |
|     public function getCache()
 | |
|     {
 | |
|         return $this->cache;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @param int $allocationSize
 | |
|      *
 | |
|      * @return \Doctrine\DBAL\Schema\Sequence
 | |
|      */
 | |
|     public function setAllocationSize($allocationSize)
 | |
|     {
 | |
|         $this->allocationSize = is_numeric($allocationSize) ? (int) $allocationSize : 1;
 | |
| 
 | |
|         return $this;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @param int $initialValue
 | |
|      *
 | |
|      * @return \Doctrine\DBAL\Schema\Sequence
 | |
|      */
 | |
|     public function setInitialValue($initialValue)
 | |
|     {
 | |
|         $this->initialValue = is_numeric($initialValue) ? (int) $initialValue : 1;
 | |
| 
 | |
|         return $this;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @param int $cache
 | |
|      *
 | |
|      * @return \Doctrine\DBAL\Schema\Sequence
 | |
|      */
 | |
|     public function setCache($cache)
 | |
|     {
 | |
|         $this->cache = $cache;
 | |
| 
 | |
|         return $this;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Checks if this sequence is an autoincrement sequence for a given table.
 | |
|      *
 | |
|      * This is used inside the comparator to not report sequences as missing,
 | |
|      * when the "from" schema implicitly creates the sequences.
 | |
|      *
 | |
|      * @return bool
 | |
|      */
 | |
|     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, $column->getShortestName($table->getNamespaceName()));
 | |
| 
 | |
|         return $tableSequenceName === $sequenceName;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @return void
 | |
|      */
 | |
|     public function visit(Visitor $visitor)
 | |
|     {
 | |
|         $visitor->acceptSequence($this);
 | |
|     }
 | |
| }
 | 
