Ticket #22196: upsert.diff

wp-includes/wp-db.php

@@ -2050 +2050 @@
         }
 
         /**
+         * Insert or update (UPSERT) a row into a table.
+         *
+         *     wpdb::upsert( 'table', array( 'column' => 'foo', 'field' => 'bar' ) )
+         *     wpdb::upsert( 'table', array( 'column' => 'foo', 'field' => 1337 ), array( '%s', '%d' ), [ 'id' ] )
+         *
+         * @since 3.0.0
+         * @see wpdb::prepare()
+         * @see wpdb::$field_types
+         * @see wp_set_wpdb_vars()
+         *
+         * @param string       $table  Table name
+         * @param array        $data   Data to insert (in column => value pairs).
+         *                             Both $data columns and $data values should be "raw" (neither should be SQL escaped).
+         *                             Sending a null value will cause the column to be set to NULL - the corresponding format is ignored in this case.
+         * @param array|string $format Optional. An array of formats to be mapped to each of the value in $data.
+         *                             If string, that format will be used for all of the values in $data.
+         *                             A format is one of '%d', '%f', '%s' (integer, float, string).
+         *                             If omitted, all values in $data will be treated as strings unless otherwise specified in wpdb::$field_types.
+         * @param array|string $keys   Optional, An array of primary or unique key column names to filter out when updating.
+         *                             If string, that will be assumed to be a primary or unique key column name.
+         * @return int|false The number of rows affected, or false on error.
+         */
+        public function upsert( $table, $data, $format = null, $keys = null ) {
+                return $this->_insert_replace_helper( $table, $data, $format, 'UPSERT', $keys );
+        }
+
+        /**
          * Helper function for insert and replace.
          *
          * Runs an insert or replace query based on $type argument.
@@ -2069 +2096 @@
          *                             A format is one of '%d', '%f', '%s' (integer, float, string).
          *                             If omitted, all values in $data will be treated as strings unless otherwise specified in wpdb::$field_types.
          * @param string $type         Optional. What type of operation is this? INSERT or REPLACE. Defaults to INSERT.
+         * @param array|string $keys   Optional, An array of primary or unique key column names to filter out when updating.
+         *                             If string, that will be assumed to be a primary or unique key column name.
          * @return int|false The number of rows affected, or false on error.
          */
-        function _insert_replace_helper( $table, $data, $format = null, $type = 'INSERT' ) {
+        function _insert_replace_helper( $table, $data, $format = null, $type = 'INSERT', $keys = null ) {
                 $this->insert_id = 0;
 
-                if ( ! in_array( strtoupper( $type ), array( 'REPLACE', 'INSERT' ) ) ) {
+                if ( ! in_array( strtoupper( $type ), array( 'REPLACE', 'INSERT', 'UPSERT' ) ) ) {
                         return false;
                 }
 
@@ -2097 +2126 @@
                 $fields  = '`' . implode( '`, `', array_keys( $data ) ) . '`';
                 $formats = implode( ', ', $formats );
 
-                $sql = "$type INTO `$table` ($fields) VALUES ($formats)";
+                if ($type !== 'UPSERT') {
+                        $sql = "$type INTO `$table` ($fields) VALUES ($formats)";
+                } else {
+                        if (is_string($keys)) {
+                                $keys = array($keys);
+                        }
+                        $updates = array();
+                        foreach ( $data as $field => $value ) {
+                                if (is_array($keys) && in_array($field, $keys)) {
+                                        continue;
+                                }
+                                if ( is_null( $value['value'] ) ) {
+                                        $updates[] = "$field=NULL";
+                                        continue;
+                                }
+                                $format = $value['format'];
+                                $updates[] = "$field=$format";
+                                $values[]  = $value['value'];
+                        }
+                        $updates = implode( ', ', $updates );
+                        $sql = "INSERT INTO `$table` ($fields) VALUES ($formats) ON DUPLICATE KEY UPDATE $updates";
+                }
 
                 $this->check_current_query = false;
                 return $this->query( $this->prepare( $sql, $values ) );