mirror of
https://github.com/10h30/ultimatemember.git
synced 2026-07-11 10:46:11 +09:00
5356148cc4
Introduced a new batch process to handle users lacking an `account_status` meta efficiently. Refactored legacy methods, added async scheduling, and created helper functions to manage and track progress. These changes improve performance and reliability for large user bases.
143 lines
3.4 KiB
PHP
143 lines
3.4 KiB
PHP
<?php
|
|
namespace um\common\actions;
|
|
|
|
use WP_User_Query;
|
|
|
|
if ( ! defined( 'ABSPATH' ) ) {
|
|
exit;
|
|
}
|
|
|
|
if ( ! class_exists( 'um\common\actions\Users' ) ) {
|
|
|
|
/**
|
|
* Class Users
|
|
*
|
|
* @since 2.10.3
|
|
*
|
|
* @package um\common\actions
|
|
*/
|
|
class Users {
|
|
|
|
const INTERVAL = 3600;
|
|
|
|
const SCHEDULE_ACTION = 'um_schedule_empty_account_status_check';
|
|
|
|
const BATCH_SIZE = 50;
|
|
|
|
const BATCH_ACTION = 'um_set_default_account_status';
|
|
|
|
public function __construct() {
|
|
add_action( 'init', array( &$this, 'add_recurring_action' ) );
|
|
add_action( self::SCHEDULE_ACTION, array( &$this, 'status_check' ) );
|
|
add_action( self::BATCH_ACTION, array( &$this, 'batch_check' ), 10, 3 );
|
|
}
|
|
|
|
public function add_recurring_action() {
|
|
if ( UM()->maybe_action_scheduler()->next_scheduled_action( self::SCHEDULE_ACTION ) ) {
|
|
return;
|
|
}
|
|
|
|
UM()->maybe_action_scheduler()->schedule_recurring_action(
|
|
time() + 60,
|
|
self::INTERVAL,
|
|
self::SCHEDULE_ACTION
|
|
);
|
|
}
|
|
|
|
public function status_check() {
|
|
$total_users = UM()->common()->users()::get_empty_status_users();
|
|
if ( empty( $total_users ) ) {
|
|
return;
|
|
}
|
|
|
|
UM()->maybe_action_scheduler()->enqueue_async_action(
|
|
self::BATCH_ACTION,
|
|
array(
|
|
'page' => 1,
|
|
'total' => $total_users,
|
|
'pages' => ceil( $total_users / self::BATCH_SIZE ),
|
|
)
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Perform batch checking for users based on specific conditions.
|
|
* Ignore users with `um_registration_in_progress` that can be in the process of the registration.
|
|
* Get users with empty `account_status` meta.
|
|
*
|
|
* @param int $page The current page number.
|
|
* @param int $total The total number of users to process.
|
|
* @param int $pages The total number of pages to process.
|
|
*/
|
|
public function batch_check( $page, $total, $pages ) {
|
|
$users = new WP_User_Query(
|
|
array(
|
|
'number' => self::BATCH_SIZE,
|
|
'fields' => 'ids',
|
|
'meta_query' => array(
|
|
'relation' => 'AND',
|
|
array(
|
|
'relation' => 'OR',
|
|
array(
|
|
'key' => 'um_registration_in_progress',
|
|
'compare' => 'NOT EXISTS',
|
|
),
|
|
array(
|
|
'key' => 'um_registration_in_progress',
|
|
'value' => '1',
|
|
'compare' => '!=',
|
|
),
|
|
),
|
|
array(
|
|
'relation' => 'OR',
|
|
array(
|
|
'key' => 'account_status',
|
|
'compare' => 'NOT EXISTS',
|
|
),
|
|
array(
|
|
'key' => 'account_status',
|
|
'value' => '',
|
|
'compare' => '=',
|
|
),
|
|
),
|
|
),
|
|
)
|
|
);
|
|
|
|
$results = $users->get_results();
|
|
|
|
if ( ! empty( $results ) ) {
|
|
$um_empty_status_users = get_option( '_um_log_empty_status_users', array( 0, 0 ) );
|
|
if ( ! is_array( $um_empty_status_users ) ) {
|
|
$um_empty_status_users = array( 0, count( $results ) );
|
|
}
|
|
|
|
foreach ( $results as $user_id ) {
|
|
$res = UM()->common()->users()->approve( $user_id, true, true );
|
|
if ( $res ) {
|
|
++$um_empty_status_users[0];
|
|
}
|
|
}
|
|
|
|
if ( $um_empty_status_users[0] < $um_empty_status_users[1] ) {
|
|
update_option( '_um_log_empty_status_users', $um_empty_status_users );
|
|
} else {
|
|
delete_option( '_um_log_empty_status_users' );
|
|
}
|
|
|
|
$next_page = $page + 1;
|
|
if ( $next_page <= $pages ) {
|
|
UM()->maybe_action_scheduler()->enqueue_async_action(
|
|
self::BATCH_ACTION,
|
|
array(
|
|
'page' => $next_page,
|
|
'total' => $total,
|
|
'pages' => $pages,
|
|
)
|
|
);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|