88 lines
2.4 KiB
PHP
88 lines
2.4 KiB
PHP
<?php
|
|
|
|
namespace Driver\SQL\Query;
|
|
|
|
use Driver\SQL\SQL;
|
|
use Driver\SQL\Strategy\Strategy;
|
|
|
|
class Insert extends Query {
|
|
|
|
private string $tableName;
|
|
private array $columns;
|
|
private array $rows;
|
|
private ?Strategy $onDuplicateKey;
|
|
private ?string $returning;
|
|
|
|
public function __construct(SQL $sql, string $name, array $columns = array()) {
|
|
parent::__construct($sql);
|
|
$this->tableName = $name;
|
|
$this->columns = $columns;
|
|
$this->rows = array();
|
|
$this->onDuplicateKey = NULL;
|
|
$this->returning = NULL;
|
|
}
|
|
|
|
public function addRow(...$values): Insert {
|
|
$this->rows[] = $values;
|
|
return $this;
|
|
}
|
|
|
|
public function onDuplicateKeyStrategy(Strategy $strategy): Insert {
|
|
$this->onDuplicateKey = $strategy;
|
|
return $this;
|
|
}
|
|
|
|
public function returning(string $column): Insert {
|
|
$this->returning = $column;
|
|
return $this;
|
|
}
|
|
|
|
public function execute() {
|
|
$fetchResult = !empty($this->sql->getReturning($this->returning));
|
|
return $this->sql->executeQuery($this, $fetchResult);
|
|
}
|
|
|
|
public function getTableName(): string { return $this->tableName; }
|
|
public function getColumns(): array { return $this->columns; }
|
|
public function getRows(): array { return $this->rows; }
|
|
public function onDuplicateKey(): ?Strategy { return $this->onDuplicateKey; }
|
|
public function getReturning(): ?string { return $this->returning; }
|
|
|
|
public function build(array &$params, Query $context = NULL): ?string {
|
|
$tableName = $this->sql->tableName($this->getTableName());
|
|
$columns = $this->getColumns();
|
|
$rows = $this->getRows();
|
|
|
|
if (empty($rows)) {
|
|
$this->sql->setLastError("No rows to insert given.");
|
|
return null;
|
|
}
|
|
|
|
if (is_null($columns) || empty($columns)) {
|
|
$columnStr = "";
|
|
} else {
|
|
$columnStr = " (" . $this->sql->columnName($columns) . ")";
|
|
}
|
|
|
|
$values = array();
|
|
foreach ($rows as $row) {
|
|
$rowPlaceHolder = array();
|
|
foreach ($row as $val) {
|
|
$rowPlaceHolder[] = $this->sql->addValue($val, $params);
|
|
}
|
|
|
|
$values[] = "(" . implode(",", $rowPlaceHolder) . ")";
|
|
}
|
|
|
|
$values = implode(",", $values);
|
|
|
|
$onDuplicateKey = $this->sql->getOnDuplicateStrategy($this->onDuplicateKey(), $params);
|
|
if ($onDuplicateKey === FALSE) {
|
|
return null;
|
|
}
|
|
|
|
$returningCol = $this->getReturning();
|
|
$returning = $this->sql->getReturning($returningCol);
|
|
return "INSERT INTO $tableName$columnStr VALUES $values$onDuplicateKey$returning";
|
|
}
|
|
} |