bytic /
database
| 1 | <?php |
||
| 2 | |||
| 3 | namespace Nip\Database\Query; |
||
| 4 | |||
| 5 | /** |
||
| 6 | * Class Insert |
||
| 7 | * @package Nip\Database\Query |
||
| 8 | */ |
||
| 9 | class Insert extends AbstractQuery |
||
| 10 | { |
||
| 11 | protected $_cols; |
||
| 12 | |||
| 13 | protected $_values; |
||
| 14 | |||
| 15 | /** |
||
| 16 | * @return string |
||
| 17 | */ |
||
| 18 | public function assemble() |
||
| 19 | 2 | { |
|
| 20 | $return = "INSERT INTO " . $this->protect($this->getTable()); |
||
| 21 | 2 | $return .= $this->parseCols(); |
|
| 22 | 2 | $return .= $this->parseValues(); |
|
| 23 | 2 | $return .= $this->parseOnDuplicate(); |
|
| 24 | 2 | ||
| 25 | return $return; |
||
|
0 ignored issues
–
show
|
|||
| 26 | 2 | } |
|
| 27 | |||
| 28 | /** |
||
| 29 | * @return string |
||
| 30 | */ |
||
| 31 | public function parseCols() |
||
| 32 | 2 | { |
|
| 33 | if (isset($this->parts['data'][0]) && is_array($this->parts['data'][0])) { |
||
| 34 | 2 | $this->setCols(array_keys($this->parts['data'][0])); |
|
| 35 | 2 | } |
|
| 36 | return $this->_cols ? ' (' . implode(',', array_map([$this, 'protect'], $this->_cols)) . ')' : ''; |
||
| 37 | 2 | } |
|
| 38 | |||
| 39 | /** |
||
| 40 | * @param array|string $cols |
||
| 41 | * @return $this |
||
| 42 | */ |
||
| 43 | public function setCols($cols = null) |
||
| 44 | 2 | { |
|
| 45 | $this->_cols = $cols; |
||
| 46 | 2 | return $this; |
|
| 47 | 2 | } |
|
| 48 | |||
| 49 | /** |
||
| 50 | * @return string|false |
||
| 51 | */ |
||
| 52 | public function parseValues() |
||
| 53 | 2 | { |
|
| 54 | if ($this->_values instanceof AbstractQuery) { |
||
| 55 | 2 | return ' ' . (string) $this->_values; |
|
| 56 | } elseif (is_array($this->parts['data'])) { |
||
|
0 ignored issues
–
show
|
|||
| 57 | 2 | return $this->parseData(); |
|
| 58 | 2 | } |
|
| 59 | return false; |
||
| 60 | } |
||
| 61 | |||
| 62 | /** |
||
| 63 | * Parses INSERT data |
||
| 64 | * |
||
| 65 | * @return string |
||
| 66 | */ |
||
| 67 | protected function parseData() |
||
| 68 | 2 | { |
|
| 69 | $values = []; |
||
| 70 | 2 | foreach ($this->parts['data'] as $key => $data) { |
|
|
0 ignored issues
–
show
|
|||
| 71 | 2 | foreach ($data as $value) { |
|
| 72 | 2 | if (!is_array($value)) { |
|
| 73 | 2 | $value = [$value]; |
|
| 74 | 2 | } |
|
| 75 | |||
| 76 | foreach ($value as $insertValue) { |
||
| 77 | 2 | ||
| 78 | 2 | if ($insertValue === null) { |
|
| 79 | $insertValue = 'NULL'; |
||
| 80 | } else { |
||
| 81 | $insertValue = $this->getManager()->getAdapter()->quote($insertValue); |
||
| 82 | 2 | } |
|
| 83 | 2 | $values[$key][] = $insertValue; |
|
| 84 | } |
||
| 85 | } |
||
| 86 | 2 | } |
|
| 87 | foreach ($values as &$value) { |
||
| 88 | $value = "(" . implode(", ", $value) . ")"; |
||
| 89 | } |
||
| 90 | |||
| 91 | return ' VALUES ' . implode(', ', $values); |
||
| 92 | 2 | } |
|
| 93 | |||
| 94 | 2 | /** |
|
| 95 | 1 | * @return string |
|
| 96 | */ |
||
| 97 | 1 | public function parseOnDuplicate() |
|
| 98 | 1 | { |
|
| 99 | 1 | if ($this->hasPart('onDuplicate')) { |
|
| 100 | 1 | $update = $this->getManager()->newUpdate(); |
|
| 101 | 1 | ||
| 102 | $onDuplicates = $this->getPart('onDuplicate'); |
||
| 103 | $data = []; |
||
| 104 | 1 | foreach ($onDuplicates as $onDuplicate) { |
|
| 105 | foreach ($onDuplicate as $key => $value) { |
||
| 106 | 1 | $data[$key] = $value; |
|
| 107 | } |
||
| 108 | 1 | } |
|
| 109 | $update->data($data); |
||
| 110 | |||
| 111 | return " ON DUPLICATE KEY UPDATE {$update->parseUpdate()}"; |
||
| 112 | } |
||
| 113 | return ''; |
||
| 114 | } |
||
| 115 | |||
| 116 | public function setValues($values) |
||
| 117 | { |
||
| 118 | 1 | $this->_values = $values; |
|
| 119 | |||
| 120 | 1 | return $this; |
|
| 121 | 1 | } |
|
| 122 | |||
| 123 | public function onDuplicate($value) |
||
| 124 | { |
||
| 125 | $this->addPart('onDuplicate', $value); |
||
| 126 | } |
||
| 127 | } |
||
| 128 |
In the issue above, the returned value is violating the contract defined by the mentioned interface.
Let's take a look at an example: