Steve-Dee-Designs
/
woocommerce
mirror of https://github.com/woocommerce/woocommerce.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add partial verification support for HPOS CLI (#37446)

This commit is contained in:
Vedanshu Jain 2023-03-29 16:00:39 +05:30 committed by GitHub
parent b5aaa1f996 792bf6fc85
commit 934084fb95
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 22 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
plugins/woocommerce/changelog/hpos-end_at_support Normal file
View File

@ -0,0 +1,4 @@
Significance: patch
Type: fix
Add support for end_at ID to allow partial verification.

24
plugins/woocommerce/src/Database/Migrations/CustomOrderTable/CLIRunner.php
View File

@ -288,10 +288,16 @@ class CLIRunner {
* default: 0 * default: 0
* --- * ---
* *
* [--end-at=<order_id>]
* : Order ID to end at.
* ---
* default: -1
* ---
*
* ## EXAMPLES * ## EXAMPLES
* *
* # Verify migrated order data, 500 orders at a time. * # Verify migrated order data, 500 orders at a time.
* wp wc cot verify_cot_data --batch-size=500 --start-from=0 * wp wc cot verify_cot_data --batch-size=500 --start-from=0 --end-at=10000
* *
* @param array $args Positional arguments passed to the command. * @param array $args Positional arguments passed to the command.
* @param array $assoc_args Associative arguments (options) passed to the command. * @param array $assoc_args Associative arguments (options) passed to the command.
@ -308,6 +314,7 @@ class CLIRunner {
array( array(
'batch-size' => 500, 'batch-size' => 500,
'start-from' => 0, 'start-from' => 0,
'end-at' => -1,
) )
); );
@ -316,7 +323,9 @@ class CLIRunner {
$failed_ids = array(); $failed_ids = array();
$processed = 0; $processed = 0;
$order_id_start = (int) $assoc_args['start-from']; $order_id_start = (int) $assoc_args['start-from'];
$order_count = $this->get_verify_order_count( $order_id_start ); $order_id_end = (int) $assoc_args['end-at'];
$order_id_end = -1 === $order_id_end ? PHP_INT_MAX : $order_id_end;
$order_count = $this->get_verify_order_count( $order_id_start, $order_id_end );
$batch_size = ( (int) $assoc_args['batch-size'] ) === 0 ? 500 : (int) $assoc_args['batch-size']; $batch_size = ( (int) $assoc_args['batch-size'] ) === 0 ? 500 : (int) $assoc_args['batch-size'];
$progress = WP_CLI\Utils\make_progress_bar( 'Order Data Verification', $order_count / $batch_size ); $progress = WP_CLI\Utils\make_progress_bar( 'Order Data Verification', $order_count / $batch_size );
@ -337,8 +346,9 @@ class CLIRunner {
$order_ids = $wpdb->get_col( $order_ids = $wpdb->get_col(
$wpdb->prepare( $wpdb->prepare(
"SELECT ID FROM $wpdb->posts WHERE post_type = 'shop_order' AND ID > %d ORDER BY ID ASC LIMIT %d", "SELECT ID FROM $wpdb->posts WHERE post_type = 'shop_order' AND ID >= %d AND ID <= %d ORDER BY ID ASC LIMIT %d",
$order_id_start, $order_id_start,
$order_id_end,
$batch_size $batch_size
) )
); );
@ -424,17 +434,19 @@ class CLIRunner {
* Helper method to get count for orders needing verification. * Helper method to get count for orders needing verification.
* *
* @param int $order_id_start Order ID to start from. * @param int $order_id_start Order ID to start from.
* @param int $order_id_end Order ID to end at.
* @param bool $log Whether to also log an error message. * @param bool $log Whether to also log an error message.
* *
* @return int Order count. * @return int Order count.
*/ */
private function get_verify_order_count( int $order_id_start, $log = true ) : int { private function get_verify_order_count( int $order_id_start, int $order_id_end, $log = true ) : int {
global $wpdb; global $wpdb;
$order_count = (int) $wpdb->get_var( $order_count = (int) $wpdb->get_var(
$wpdb->prepare( $wpdb->prepare(
"SELECT COUNT(*) FROM $wpdb->posts WHERE post_type = 'shop_order' AND ID > %d", "SELECT COUNT(*) FROM $wpdb->posts WHERE post_type = 'shop_order' AND ID >= %d AND ID <= %d",
$order_id_start $order_id_start,
$order_id_end
) )
); );