Merge pull request #4314 from cividesk/CRM-13227

[civicrm-core.git] / CRM / Utils / SQL / Insert.php

diff --git a/CRM/Utils/SQL/Insert.php b/CRM/Utils/SQL/Insert.php
index e67457c78759f584f70fbeb42c8dca256876c552..20c87853474754702bcd72752abcb6c2434c6658 100644 (file)
--- a/CRM/Utils/SQL/Insert.php
+++ b/CRM/Utils/SQL/Insert.php
@@ -22,6 +22,8 @@
  */
 class CRM_Utils_SQL_Insert {
 
+  private $verb = 'INSERT INTO';
+
   /**
    * @var string
    */
@@ -93,6 +95,17 @@ class CRM_Utils_SQL_Insert {
     return $this;
   }
 
+  /**
+   * Use REPLACE INTO instead of INSERT INTO
+   *
+   * @param bool $asReplace
+   * @return CRM_Utils_SQL_Insert
+   */
+  public function usingReplace($asReplace = TRUE) {
+    $this->verb = $asReplace ? 'REPLACE INTO' : 'INSERT INTO';
+    return $this;
+  }
+
   /**
    * @param string|NULL $value
    * @return string SQL expression, e.g. "it\'s great" (with-quotes) or NULL (without-quotes)
@@ -106,7 +119,7 @@ class CRM_Utils_SQL_Insert {
    */
   public function toSQL() {
     $columns = "`" . implode('`,`', $this->columns) . "`";
-    $sql = "INSERT INTO {$this->table} ({$columns}) VALUES";
+    $sql = "{$this->verb} {$this->table} ({$columns}) VALUES";
 
     $nextDelim = '';
     foreach ($this->rows as $row) {