added ignore flag

codeigniter4 · Mar 25, 2019 · 8bd9ae9 · 8bd9ae9
1 parent 1e6a0dc
commit 8bd9ae9
Show file tree

Hide file tree

Showing 3 changed files with 103 additions and 1 deletion.
diff --git a/system/Database/BaseBuilder.php b/system/Database/BaseBuilder.php
@@ -208,6 +208,14 @@ class BaseBuilder
 	 */
 	protected $canLimitWhereUpdates = true;
 
+    /**
+     * Ignore errors on insert statements
+     * for example for duplicate keys.
+     *
+     * @var bool
+     */
+	protected $insertIgnore = false;
+
 	//--------------------------------------------------------------------
 
 	/**
@@ -254,6 +262,22 @@ public function getBinds(): array
 		return $this->binds;
 	}
 
+    //--------------------------------------------------------------------
+
+    /**
+     * Ignore
+     *
+     * Set ignore Flag for next insert query.
+     *
+     * @return BaseBuilder
+     */
+    public function ignore()
+    {
+        $this->insertIgnore = true;
+
+        return $this;
+    }
+
 	//--------------------------------------------------------------------
 
 	/**
@@ -1666,6 +1690,8 @@ public function insertBatch($set = null, $escape = null, $batchSize = 100, $test
 			}
 		}
 
+        $this->insertIgnore = false;
+
 		if (! $testing)
 		{
 			$this->resetWrite();
@@ -1758,6 +1784,8 @@ public function setInsertBatch($key, $value = '', $escape = null)
 	 *
 	 * @param boolean $reset TRUE: reset QB values; FALSE: leave QB values alone
 	 *
+     * @throws DatabaseException
+     *
 	 * @return string
 	 */
 	public function getCompiledInsert($reset = true)
@@ -1791,6 +1819,8 @@ public function getCompiledInsert($reset = true)
 	 * @param array   $set    An associative array of insert values
 	 * @param boolean $escape Whether to escape values and identifiers
 	 * @param boolean $test   Used when running tests
+     *
+     * @throws DatabaseException
 	 *
 	 * @return BaseResult|Query|false
 	 */
@@ -1812,6 +1842,8 @@ public function insert($set = null, $escape = null, $test = false)
 				), array_keys($this->QBSet), array_values($this->QBSet)
 		);
 
+        $this->insertIgnore = false;
+
 		if ($test === false)
 		{
 			$this->resetWrite();

diff --git a/system/Database/MySQLi/Builder.php b/system/Database/MySQLi/Builder.php
@@ -51,4 +51,40 @@ class Builder extends BaseBuilder
 	 */
 	protected $escapeChar = '`';
 
+    //--------------------------------------------------------------------
+
+    /**
+     * Insert batch statement
+     *
+     * Generates a platform-specific insert string from the supplied data.
+     *
+     * @param string $table  Table name
+     * @param array  $keys   INSERT keys
+     * @param array  $values INSERT values
+     *
+     * @return string
+     */
+    protected function _insertBatch($table, $keys, $values)
+    {
+        return 'INSERT ' . ($this->insertIgnore ? 'IGNORE' : '') . ' INTO ' . $table . ' (' . implode(', ', $keys) . ') VALUES ' . implode(', ', $values);
+    }
+
+    //--------------------------------------------------------------------
+
+    /**
+     * Insert statement
+     *
+     * Generates a platform-specific insert string from the supplied data
+     *
+     * @param string $table         The table name
+     * @param array  $keys          The insert keys
+     * @param array  $unescapedKeys The insert values
+     *
+     * @return string
+     */
+    protected function _insert($table, array $keys, array $unescapedKeys)
+    {
+        return 'INSERT ' . ($this->insertIgnore ? 'IGNORE' : '') . ' INTO ' . $table . ' (' . implode(', ', $keys) . ') VALUES (' . implode(', ', $unescapedKeys) . ')';
+    }
+
 }
diff --git a/system/Database/SQLite3/Builder.php b/system/Database/SQLite3/Builder.php
@@ -103,6 +103,40 @@ protected function _truncate($table)
 		return 'DELETE FROM ' . $table;
 	}
 
-	//--------------------------------------------------------------------
+    //--------------------------------------------------------------------
+
+    /**
+     * Insert batch statement
+     *
+     * Generates a platform-specific insert string from the supplied data.
+     *
+     * @param string $table  Table name
+     * @param array  $keys   INSERT keys
+     * @param array  $values INSERT values
+     *
+     * @return string
+     */
+    protected function _insertBatch($table, $keys, $values)
+    {
+        return 'INSERT ' . ($this->insertIgnore ? 'OR IGNORE' : '') . ' INTO ' . $table . ' (' . implode(', ', $keys) . ') VALUES ' . implode(', ', $values);
+    }
+
+    //--------------------------------------------------------------------
+
+    /**
+     * Insert statement
+     *
+     * Generates a platform-specific insert string from the supplied data
+     *
+     * @param string $table         The table name
+     * @param array  $keys          The insert keys
+     * @param array  $unescapedKeys The insert values
+     *
+     * @return string
+     */
+    protected function _insert($table, array $keys, array $unescapedKeys)
+    {
+        return 'INSERT ' . ($this->insertIgnore ? 'OR IGNORE' : '') . ' INTO ' . $table . ' (' . implode(', ', $keys) . ') VALUES (' . implode(', ', $unescapedKeys) . ')';
+    }
 
 }