86 lines
1.6 KiB
PHP
86 lines
1.6 KiB
PHP
|
<?php
|
||
|
|
|
||
|
|
|
||
|
|
namespace Action_Scheduler\Migration;
|
||
|
|
|
||
|
|
|
||
|
|
use ActionScheduler_Store as Store;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Class BatchFetcher
|
||
|
|
*
|
||
|
|
* @package Action_Scheduler\Migration
|
||
|
|
*
|
||
|
|
* @since 3.0.0
|
||
|
|
*
|
||
|
|
* @codeCoverageIgnore
|
||
|
|
*/
|
||
|
|
class BatchFetcher {
|
||
|
|
/** var ActionScheduler_Store */
|
||
|
|
private $store;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* BatchFetcher constructor.
|
||
|
|
*
|
||
|
|
* @param ActionScheduler_Store $source_store Source store object.
|
||
|
|
*/
|
||
|
|
public function __construct( Store $source_store ) {
|
||
|
|
$this->store = $source_store;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Retrieve a list of actions.
|
||
|
|
*
|
||
|
|
* @param int $count The number of actions to retrieve
|
||
|
|
*
|
||
|
|
* @return int[] A list of action IDs
|
||
|
|
*/
|
||
|
|
public function fetch( $count = 10 ) {
|
||
|
|
foreach ( $this->get_query_strategies( $count ) as $query ) {
|
||
|
|
$action_ids = $this->store->query_actions( $query );
|
||
|
|
if ( ! empty( $action_ids ) ) {
|
||
|
|
return $action_ids;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return [];
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Generate a list of prioritized of action search parameters.
|
||
|
|
*
|
||
|
|
* @param int $count Number of actions to find.
|
||
|
|
*
|
||
|
|
* @return array
|
||
|
|
*/
|
||
|
|
private function get_query_strategies( $count ) {
|
||
|
|
$now = as_get_datetime_object();
|
||
|
|
$args = [
|
||
|
|
'date' => $now,
|
||
|
|
'per_page' => $count,
|
||
|
|
'offset' => 0,
|
||
|
|
'orderby' => 'date',
|
||
|
|
'order' => 'ASC',
|
||
|
|
];
|
||
|
|
|
||
|
|
$priorities = [
|
||
|
|
Store::STATUS_PENDING,
|
||
|
|
Store::STATUS_FAILED,
|
||
|
|
Store::STATUS_CANCELED,
|
||
|
|
Store::STATUS_COMPLETE,
|
||
|
|
Store::STATUS_RUNNING,
|
||
|
|
'', // any other unanticipated status
|
||
|
|
];
|
||
|
|
|
||
|
|
foreach ( $priorities as $status ) {
|
||
|
|
yield wp_parse_args( [
|
||
|
|
'status' => $status,
|
||
|
|
'date_compare' => '<=',
|
||
|
|
], $args );
|
||
|
|
yield wp_parse_args( [
|
||
|
|
'status' => $status,
|
||
|
|
'date_compare' => '>=',
|
||
|
|
], $args );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|