Edit File by line

<?php

[0] Fix | Delete

[1] Fix | Delete

use PhpMyAdmin\SqlParser\Parser;

[2] Fix | Delete

[3] Fix | Delete

class WPML_DB_Chunk {

[4] Fix | Delete

/**

[5] Fix | Delete

* @var wpdb

[6] Fix | Delete

[7] Fix | Delete

private $wpdb;

[8] Fix | Delete

[9] Fix | Delete

/**

[10] Fix | Delete

* @var int

[11] Fix | Delete

[12] Fix | Delete

private $chunk_size;

[13] Fix | Delete

[14] Fix | Delete

/**

[15] Fix | Delete

* @param wpdb $wpdb

[16] Fix | Delete

* @param int $chunk_size

[17] Fix | Delete

[18] Fix | Delete

public function __construct( wpdb $wpdb, $chunk_size = 1000 ) {

[19] Fix | Delete

$this->wpdb = $wpdb;

[20] Fix | Delete

$this->chunk_size = $chunk_size;

[21] Fix | Delete

}

[22] Fix | Delete

[23] Fix | Delete

/**

[24] Fix | Delete

* @param string $query

[25] Fix | Delete

* @param array $args

[26] Fix | Delete

* @param int $elements_num

[27] Fix | Delete

[28] Fix | Delete

* @return array

[29] Fix | Delete

[30] Fix | Delete

* @throws \InvalidArgumentException

[31] Fix | Delete

[32] Fix | Delete

public function retrieve( $query, $args, $elements_num ) {

[33] Fix | Delete

$this->validate_query( $query );

[34] Fix | Delete

$result = array();

[35] Fix | Delete

[36] Fix | Delete

$offset = 0;

[37] Fix | Delete

while ( $offset < $elements_num ) {

[38] Fix | Delete

$new_query = $query . sprintf( ' LIMIT %d OFFSET %s', $this->chunk_size, $offset );

[39] Fix | Delete

$new_query = $this->wpdb->prepare( $new_query, $args );

[40] Fix | Delete

$rowset = $this->wpdb->get_results( $new_query, ARRAY_A );

[41] Fix | Delete

[42] Fix | Delete

if ( is_array( $rowset ) && count( $rowset ) ) {

[43] Fix | Delete

$result = array_merge( $result, $rowset );

[44] Fix | Delete

}

[45] Fix | Delete

[46] Fix | Delete

$offset += $this->chunk_size;

[47] Fix | Delete

}

[48] Fix | Delete

[49] Fix | Delete

return $result;

[50] Fix | Delete

}

[51] Fix | Delete

[52] Fix | Delete

/**

[53] Fix | Delete

* @param string $query

[54] Fix | Delete

[55] Fix | Delete

private function validate_query( $query ) {

[56] Fix | Delete

$parser = new Parser( $query );

[57] Fix | Delete

[58] Fix | Delete

if ( isset( $parser->statements ) ) {

[59] Fix | Delete

wpml_collect( $parser->statements )->each(

[60] Fix | Delete

function( $statement ) {

[61] Fix | Delete

if ( ! empty( $statement->limit ) ) {

[62] Fix | Delete

throw new InvalidArgumentException( "Query can't contain OFFSET or LIMIT keyword" );

[63] Fix | Delete

}

[64] Fix | Delete

}

[65] Fix | Delete

);

[66] Fix | Delete

}

[67] Fix | Delete

}

[68] Fix | Delete

}

[69] Fix | Delete