web-dev-qa-db-fra.com

$ wpdb-> met à jour plusieurs lignes, comme IN dans SQL normal

Je me demandais s'il était possible d'utiliser $wpdb->update pour mettre à jour les valeurs de plusieurs lignes, comme avec IN dans un SQL "normal".

Je veux accomplir quelque chose comme cet exemple;

UPDATE [table]
SET [column_1] = [updated_value]
WHERE [column_2] IN ([comma_separated_ids])

J'ai cherché comment cela fonctionnerait, mais je n'ai pas encore trouvé de question ou de réponse/article de blog expliquant si cela peut être fait ou non.

Pour le moment, j'utilise une requête à la place, mais ce serait plus agréable avec un simple support.

6
Ludvig Sjöbeck

Comme vous pouvez le voir dans code source le signe = est codé en dur dans la méthode wpdb::update(), de sorte que, par défaut, il est not possible d'utiliser IN pour la méthode update.

Le moyen le plus simple de faire est d’utiliser la fonction wpdb::query() avec votre requête SQL, assurez-vous d’échapper correctement à toutes les valeurs

Exemple:

function wpdb_update_in( $table, $data, $where, $format = NULL, $where_format = NULL ) {

    global $wpdb;

    $table = esc_sql( $table );

    if( ! is_string( $table ) || ! isset( $wpdb->$table ) ) {
        return FALSE;
    }

    $i          = 0;
    $q          = "UPDATE " . $wpdb->$table . " SET ";
    $format     = array_values( (array) $format );
    $escaped    = array();

    foreach( (array) $data as $key => $value ) {
        $f         = isset( $format[$i] ) && in_array( $format[$i], array( '%s', '%d' ), TRUE ) ? $format[$i] : '%s';
        $escaped[] = esc_sql( $key ) . " = " . $wpdb->prepare( $f, $value );
        $i++;
    }

    $q         .= implode( $escaped, ', ' );
    $where      = (array) $where;
    $where_keys = array_keys( $where );
    $where_val  = (array) array_shift( $where );
    $q         .= " WHERE " . esc_sql( array_shift( $where_keys ) ) . ' IN (';

    if( ! in_array( $where_format, array('%s', '%d'), TRUE ) ) {
        $where_format = '%s';
    }

    $escaped = array();

    foreach( $where_val as $val ) {
        $escaped[] = $wpdb->prepare( $where_format, $val );
    }

    $q .= implode( $escaped, ', ' ) . ')';

    return $wpdb->query( $q );
}

Alors utilisez-le comme suit:

wpdb_update_in(
  'posts', // table
  array( 'post_author' => '1', 'post_status' => 'draft' ), // data
  array( 'post_author' => array( '2', '3', '4', '5' ) ), // where
  array( '%d', '%s' ), // format
  '%d' // where format
);

Le SQL exécuté sera

UPDATE wp_posts
SET post_author = 1, post_status = 'draft'
WHERE post_author IN (2, 3, 4, 5)
9
gmazzap