1 | <?php |
||
2 | |||
3 | /** |
||
4 | * Cycle DataMapper ORM |
||
5 | * |
||
6 | * @license MIT |
||
7 | * @author Anton Titov (Wolfy-J) |
||
8 | */ |
||
9 | |||
10 | declare(strict_types=1); |
||
11 | |||
12 | namespace Cycle\ORM\Command\Database; |
||
13 | |||
14 | use Cycle\ORM\Command\DatabaseCommand; |
||
15 | use Cycle\ORM\Command\InitCarrierInterface; |
||
16 | use Cycle\ORM\Command\Traits\ContextTrait; |
||
17 | use Cycle\ORM\Command\Traits\ErrorTrait; |
||
18 | use Cycle\ORM\Context\ConsumerInterface; |
||
19 | use Cycle\ORM\Context\ProducerInterface; |
||
20 | use Cycle\ORM\Exception\CommandException; |
||
21 | use Spiral\Database\DatabaseInterface; |
||
22 | use Spiral\Database\Driver\Postgres\Query\PostgresInsertQuery; |
||
23 | |||
24 | /** |
||
25 | * Insert data into associated table and provide lastInsertID promise. |
||
26 | */ |
||
27 | final class Insert extends DatabaseCommand implements InitCarrierInterface, ProducerInterface |
||
28 | { |
||
29 | use ContextTrait; |
||
30 | use ErrorTrait; |
||
31 | |||
32 | // Special identifier to forward insert key into |
||
33 | public const INSERT_ID = '@lastInsertID'; |
||
34 | |||
35 | /** @var array */ |
||
36 | protected $data; |
||
37 | |||
38 | /** @var string|null */ |
||
39 | protected $primaryKey; |
||
40 | |||
41 | /** @var ConsumerInterface[] */ |
||
42 | protected $consumers = []; |
||
43 | |||
44 | /** |
||
45 | * @param DatabaseInterface $db |
||
46 | * @param string $table |
||
47 | * @param array $data |
||
48 | * @param string|null $primaryKey |
||
49 | */ |
||
50 | public function __construct( |
||
51 | DatabaseInterface $db, |
||
52 | string $table, |
||
53 | array $data = [], |
||
54 | string $primaryKey = null |
||
55 | ) { |
||
56 | parent::__construct($db, $table); |
||
57 | $this->data = $data; |
||
58 | $this->primaryKey = $primaryKey; |
||
59 | } |
||
60 | |||
61 | /** |
||
62 | * @inheritdoc |
||
63 | */ |
||
64 | public function isReady(): bool |
||
65 | { |
||
66 | return $this->waitContext === []; |
||
67 | } |
||
68 | |||
69 | /** |
||
70 | * @inheritdoc |
||
71 | * |
||
72 | * Triggers only after command execution! |
||
73 | */ |
||
74 | public function forward( |
||
75 | string $key, |
||
76 | ConsumerInterface $consumer, |
||
77 | string $target, |
||
78 | bool $trigger = false, |
||
79 | int $stream = ConsumerInterface::DATA |
||
80 | ): void { |
||
81 | if ($trigger) { |
||
82 | throw new CommandException('Insert command can only forward keys after the execution'); |
||
83 | } |
||
84 | |||
85 | $this->consumers[$key][] = [$consumer, $target, $stream]; |
||
86 | } |
||
87 | |||
88 | /** |
||
89 | * @inheritdoc |
||
90 | */ |
||
91 | public function register(string $key, $value, bool $fresh = false, int $stream = self::DATA): void |
||
92 | { |
||
93 | if ($fresh || $value !== null) { |
||
94 | $this->freeContext($key); |
||
95 | } |
||
96 | |||
97 | $this->setContext($key, $value); |
||
98 | } |
||
99 | |||
100 | /** |
||
101 | * Insert values, context not included. |
||
102 | * |
||
103 | * @return array |
||
104 | */ |
||
105 | public function getData(): array |
||
106 | { |
||
107 | return array_merge($this->data, $this->context); |
||
108 | } |
||
109 | |||
110 | /** |
||
111 | * Insert data into associated table. |
||
112 | */ |
||
113 | public function execute(): void |
||
114 | { |
||
115 | $data = $this->getData(); |
||
116 | |||
117 | $insert = $this->db->insert($this->table)->values($data); |
||
0 ignored issues
–
show
Bug
introduced
by
Loading history...
|
|||
118 | if ($this->primaryKey !== null && $insert instanceof PostgresInsertQuery) { |
||
119 | $insert->returning($this->primaryKey); |
||
120 | } |
||
121 | |||
122 | $insertID = $insert->run(); |
||
123 | |||
124 | foreach ($this->consumers as $key => $consumers) { |
||
125 | $fresh = true; |
||
126 | if ($key === self::INSERT_ID) { |
||
127 | $value = $insertID; |
||
128 | } else { |
||
129 | $value = $data[$key] ?? null; |
||
130 | } |
||
131 | |||
132 | foreach ($consumers as $id => $consumer) { |
||
133 | /** @var ConsumerInterface $cn */ |
||
134 | $cn = $consumer[0]; |
||
135 | |||
136 | $cn->register($consumer[1], $value, $fresh, $consumer[2]); |
||
137 | |||
138 | if ($key !== self::INSERT_ID) { |
||
139 | // primary key is always delivered as fresh |
||
140 | $fresh = false; |
||
141 | } |
||
142 | } |
||
143 | } |
||
144 | |||
145 | parent::execute(); |
||
146 | } |
||
147 | } |
||
148 |