+postgres INSERT ... ON CONFLICT

Author: codemasher

src/Dialects/Dialect.php

@@ -70,10 +70,11 @@ public function selectCount(array $from, string $where = null, bool $distinct =
 	 * @param string      $table
 	 * @param array       $fields
 	 * @param string|null $onConflict
+	 * @param string|null $conflictTarget
 	 *
 	 * @return array
 	 */
-	public function insert(string $table, array $fields, string $onConflict = null):array;
+	public function insert(string $table, array $fields, string $onConflict = null, string $conflictTarget = null):array;
 
 	/**
 	 * @param string $table

src/Dialects/DialectAbstract.php

@@ -82,7 +82,7 @@ public function select(array $cols, array $from, string $where = null, $limit =
 	}
 
 	/** @inheritdoc */
-	public function insert(string $table, array $fields, string $onConflict = null):array{
+	public function insert(string $table, array $fields, string $onConflict = null, string $conflictTarget = null):array{
 		$sql = ['INSERT INTO'];
 		$sql[] = $this->quote($table);
 		$sql[] = '('.$this->quotes[0].implode($this->quotes[1].', '.$this->quotes[0], $fields).$this->quotes[1].')';

src/Dialects/MySQL.php

@@ -18,7 +18,7 @@ class MySQL extends DialectAbstract{
 	protected $charset = 'utf8mb4_bin';
 
 	/** @inheritdoc */
-	public function insert(string $table, array $fields, string $onConflict = null):array{
+	public function insert(string $table, array $fields, string $onConflict = null, string $conflictTarget = null):array{
 		$onConflict = strtoupper($onConflict);
 
 		switch($onConflict){

src/Dialects/Postgres.php

@@ -12,6 +12,8 @@
 
 namespace chillerlan\Database\Dialects;
 
+use chillerlan\Database\Query\QueryException;
+
 class Postgres extends DialectAbstract{
 
 	protected $quotes = ['"', '"'];
@@ -53,6 +55,51 @@ public function select(array $cols, array $from, string $where = null, $limit =
 		return $sql;
 	}
 
+	/**
+	 * @link https://www.postgresql.org/docs/9.5/static/sql-insert.html#SQL-ON-CONFLICT
+	 *
+	 * @inheritdoc
+	 */
+	public function insert(string $table, array $fields, string $onConflict = null, string $conflictTarget = null):array{
+		$sql = parent::insert($table, $fields);
+
+		if(in_array($onConflict, ['IGNORE', 'REPLACE'], true)){
+
+			if(empty($conflictTarget)){
+				throw new QueryException('postgres insert on conflict: no conflict target given');
+			}
+
+			$sql[] =  'ON CONFLICT ('.$this->quote($conflictTarget).') DO';
+
+			switch($onConflict){
+				case 'IGNORE':
+					$sql[] = 'NOTHING';
+					break;
+				case 'REPLACE':
+					$sql[] = $this->onConflictUpdate($fields);
+					break;
+			}
+
+		}
+
+		return $sql;
+	}
+
+	/**
+	 * @param array $fields
+	 *
+	 * @return string
+	 */
+	protected function onConflictUpdate(array $fields):string {
+		$onConflictUpdate = [];
+
+		foreach($fields as $f){
+			$onConflictUpdate[] = $this->quote($f).' = EXCLUDED.'.$this->quote($f);
+		}
+
+		return 'UPDATE SET '.implode(', ', $onConflictUpdate);
+	}
+
 	/** @inheritdoc */
 	public function createDatabase(string $dbname, bool $ifNotExists = null, string $collate = null):array{
 		$sql = ['CREATE DATABASE'];
@@ -295,20 +342,3 @@ public function showCreateTable(string $table):array{
 	}
 
 }
-
-
-/*
-create table querytest
-(
-	id INTEGER
-		primary key,
-	hash VARCHAR(32),
-	data TEXT,
-	value DECIMAL(9,6),
-	active BOOLEAN,
-	created TIMESTAMP default 'CURRENT_TIMESTAMP'
-)
-;
-
-
-*/

src/Query/Insert.php

@@ -30,10 +30,11 @@ interface Insert extends Statement{
 	/**
 	 * @param string      $table
 	 * @param string|null $on_conflict
+	 * @param string|null $conflict_target
 	 *
 	 * @return \chillerlan\Database\Query\Insert
 	 */
-	public function into(string $table, string $on_conflict = null);
+	public function into(string $table, string $on_conflict = null, string $conflict_target = null);
 
 	/**
 	 * @param iterable $values

src/Query/OnConflictTrait.php

@@ -24,16 +24,23 @@ trait OnConflictTrait{
 	 */
 	protected $on_conflict;
 
+	/**
+	 * @var string
+	 */
+	protected $conflict_target;
+
 	/**
 	 * @param string      $name
 	 * @param string|null $on_conflict
+	 * @param string|null $conflict_target
 	 *
 	 * @return $this
 	 * @throws \chillerlan\Database\Query\QueryException
 	 */
-	public function name(string $name, string $on_conflict = null){
+	public function name(string $name, string $on_conflict = null, string $conflict_target = null){
 		$this->name  = trim($name);
 		$on_conflict = trim(strtoupper($on_conflict));
+		$on_conflict = trim(strtoupper($conflict_target));
 
 		if(empty($this->name)){
 			throw new QueryException('no name specified');
@@ -43,6 +50,10 @@ public function name(string $name, string $on_conflict = null){
 			$this->on_conflict = $on_conflict;
 		}
 
+		if(!empty($conflict_target)){
+			$this->conflict_target = $conflict_target;
+		}
+
 		return $this;
 	}
 

src/Query/QueryBuilder.php

@@ -276,7 +276,7 @@ protected function getSQL():array{
 					throw new QueryException('no values given');
 				}
 
-				return $this->dialect->insert($this->name, array_keys($this->multi ? $this->bindValues[0] : $this->bindValues), $this->on_conflict);
+				return $this->dialect->insert($this->name, array_keys($this->multi ? $this->bindValues[0] : $this->bindValues), $this->on_conflict, $this->conflict_target);
 			}
 
 			/** @inheritdoc */