asda2432es

✏️ Editing: AbstractASQueue.php

<?php
declare( strict_types=1 );

namespace WP_Rocket\Engine\Common\Queue;

use ActionScheduler_Store;
use Exception;
use WP_Rocket\Logger\Logger;

abstract class AbstractASQueue implements QueueInterface {

/**
	 * Queue shared Group.
	 *
	 * @var string
	 */
	protected $group = 'rocket';

/**
	 * Enqueue an action to run one time, as soon as possible
	 *
	 * @param string $hook The hook to trigger.
	 * @param array  $args Arguments to pass when the hook triggers.
	 * @return int The action ID.
	 */
	public function add_async( $hook, $args = [] ) {
		try {
			if ( function_exists( 'as_enqueue_async_action' ) ) {
				return as_enqueue_async_action( $hook, $args, $this->group );
			}

return $this->schedule_single( time() + MINUTE_IN_SECONDS, $hook, $args );
		} catch ( Exception $exception ) {
			Logger::error( $exception->getMessage(), [ 'Action Scheduler Queue' ] );

return 0;
		}
	}

/**
	 * Schedule an action to run once at some time in the future
	 *
	 * @param int    $timestamp When the job will run.
	 * @param string $hook The hook to trigger.
	 * @param array  $args Arguments to pass when the hook triggers.
	 * @return int The action ID.
	 */
	public function schedule_single( $timestamp, $hook, $args = [] ) {
		try {
			return as_schedule_single_action( $timestamp, $hook, $args, $this->group );
		} catch ( Exception $exception ) {
			Logger::error( $exception->getMessage(), [ 'Action Scheduler Queue' ] );

return 0;
		}
	}

/**
	 * Schedule a recurring action
	 *
	 * @param int    $timestamp When the first instance of the job will run.
	 * @param int    $interval_in_seconds How long to wait between runs.
	 * @param string $hook The hook to trigger.
	 * @param array  $args Arguments to pass when the hook triggers.
	 * @param int    $priority Lower values take precedence over higher values. Defaults to 10, with acceptable values falling in the range 0-255.
	 * @param bool   $unique Whether the action should be unique. Defaults to true to prevent duplicates.
	 * @return int The action ID.
	 */
	public function schedule_recurring( $timestamp, $interval_in_seconds, $hook, $args = [], $priority = 10, $unique = true ) {
		if ( $this->is_scheduled( $hook, $args ) ) {
			// TODO: When v3.3.0 from Action Scheduler is commonly used use the array notation for status to reduce search queries to one.
			$pending_actions = $this->search(
				[
					'hook'   => $hook,
					'status' => ActionScheduler_Store::STATUS_PENDING,
				],
				'ids'
			);

if ( 1 < count( $pending_actions ) ) {
				$this->cancel_all( $hook, $args );
				return 0;
			}

$running_actions = $this->search(
				[
					'hook'   => $hook,
					'status' => ActionScheduler_Store::STATUS_RUNNING,
				],
				'ids'
			);

if ( 1 === count( $pending_actions ) + count( $running_actions ) ) {
				return 0;
			}

$this->cancel_all( $hook, $args );
		}

try {
			return as_schedule_recurring_action( $timestamp, $interval_in_seconds, $hook, $args, $this->group, $unique, $priority );
		} catch ( Exception $exception ) {
			Logger::error( $exception->getMessage(), [ 'Action Scheduler Queue' ] );

return 0;
		}
	}

/**
	 * Checks if the hook is scheduled.
	 *
	 * @param string $hook The hook to check.
	 * @param array  $args Passed arguments.
	 *
	 * @return bool
	 */
	public function is_scheduled( $hook, $args = [] ) {
		if ( ! function_exists( 'as_has_scheduled_action' ) ) {
			return ! is_null( $this->get_next( $hook, $args ) );
		}

try {
			return as_has_scheduled_action( $hook, $args, $this->group );
		} catch ( Exception $exception ) {
			Logger::error( $exception->getMessage(), [ 'Action Scheduler Queue' ] );

return false;
		}
	}

/**
	 * Schedule an action that recurs on a cron-like schedule.
	 *
	 * @param int    $timestamp The schedule will start on or after this time.
	 * @param string $cron_schedule A cron-link schedule string.
	 * @see http://en.wikipedia.org/wiki/Cron
	 *   *    *    *    *    *    *
	 *   ┬    ┬    ┬    ┬    ┬    ┬
	 *   |    |    |    |    |    |
	 *   |    |    |    |    |    + year [optional]
	 *   |    |    |    |    +----- day of week (0 - 7) (Sunday=0 or 7)
	 *   |    |    |    +---------- month (1 - 12)
	 *   |    |    +--------------- day of month (1 - 31)
	 *   |    +-------------------- hour (0 - 23)
	 *   +------------------------- min (0 - 59)
	 * @param string $hook The hook to trigger.
	 * @param array  $args Arguments to pass when the hook triggers.
	 * @return int The action ID
	 */
	public function schedule_cron( $timestamp, $cron_schedule, $hook, $args = [] ) {
		if ( $this->is_scheduled( $hook, $args ) ) {
			return 0;
		}

try {
			return as_schedule_cron_action( $timestamp, $cron_schedule, $hook, $args, $this->group );
		} catch ( Exception $exception ) {
			Logger::error( $exception->getMessage(), [ 'Action Scheduler Queue' ] );

return 0;
		}
	}

/**
	 * Dequeue the next scheduled instance of an action with a matching hook (and optionally matching args and group).
	 *
	 * Any recurring actions with a matching hook should also be cancelled, not just the next scheduled action.
	 *
	 * While technically only the next instance of a recurring or cron action is unscheduled by this method, that will also
	 * prevent all future instances of that recurring or cron action from being run. Recurring and cron actions are scheduled
	 * in a sequence instead of all being scheduled at once. Each successive occurrence of a recurring action is scheduled
	 * only after the former action is run. As the next instance is never run, because it's unscheduled by this function,
	 * then the following instance will never be scheduled (or exist), which is effectively the same as being unscheduled
	 * by this method also.
	 *
	 * @param string $hook The hook that the job will trigger.
	 * @param array  $args Args that would have been passed to the job.
	 */
	public function cancel( $hook, $args = [] ) {
		try {
			as_unschedule_action( $hook, $args, $this->group );
		} catch ( Exception $exception ) {
			Logger::error( $exception->getMessage(), [ 'Action Scheduler Queue' ] );
		}
	}

/**
	 * Deprecate action without any args or group.
	 *
	 * @param string $hook Action hook name.
	 * @return void
	 */
	public function deprecate_action( $hook ) {
		try {
			as_unschedule_action( $hook );
		} catch ( Exception $exception ) {
			Logger::error( $exception->getMessage(), [ 'Action Scheduler Queue' ] );
		}
	}

/**
	 * Dequeue all actions with a matching hook (and optionally matching args and group) so no matching actions are ever run.
	 *
	 * @param string $hook The hook that the job will trigger.
	 * @param array  $args Args that would have been passed to the job.
	 */
	public function cancel_all( $hook, $args = [] ) {
		try {
			as_unschedule_all_actions( $hook, $args, $this->group );
		} catch ( Exception $exception ) {
			Logger::error( $exception->getMessage(), [ 'Action Scheduler Queue' ] );
		}
	}

/**
	 * Get the date and time for the next scheduled occurrence of an action with a given hook
	 * (an optionally that matches certain args and group), if any.
	 *
	 * @param string $hook The hook that the job will trigger.
	 * @param array  $args Filter to a hook with matching args that will be passed to the job when it runs.
	 * @return int|null The date and time for the next occurrence, or null if there is no pending, scheduled action for the given hook.
	 */
	public function get_next( $hook, $args = null ) {
		try {
			$next_timestamp = as_next_scheduled_action( $hook, $args, $this->group );

if ( is_numeric( $next_timestamp ) ) {
				return $next_timestamp;
			}

return null;
		} catch ( Exception $exception ) {
			Logger::error( $exception->getMessage(), [ 'Action Scheduler Queue' ] );

return null;
		}
	}

/**
	 * Find scheduled actions
	 *
	 * @param array  $args Possible arguments, with their default values:
	 *        'hook' => '' - the name of the action that will be triggered
	 *        'args' => null - the args array that will be passed with the action
	 *        'date' => null - the scheduled date of the action. Expects a DateTime object, a unix timestamp, or a string that can parsed with strtotime(). Used in UTC timezone.
	 *        'date_compare' => '<=' - operator for testing "date". accepted values are '!=', '>', '>=', '<', '<=', '='
	 *        'modified' => null - the date the action was last updated. Expects a DateTime object, a unix timestamp, or a string that can parsed with strtotime(). Used in UTC timezone.
	 *        'modified_compare' => '<=' - operator for testing "modified". accepted values are '!=', '>', '>=', '<', '<=', '='
	 *        'group' => '' - the group the action belongs to
	 *        'status' => '' - ActionScheduler_Store::STATUS_COMPLETE or ActionScheduler_Store::STATUS_PENDING
	 *        'claimed' => null - TRUE to find claimed actions, FALSE to find unclaimed actions, a string to find a specific claim ID
	 *        'per_page' => 5 - Number of results to return
	 *        'offset' => 0
	 *        'orderby' => 'date' - accepted values are 'hook', 'group', 'modified', or 'date'
	 *        'order' => 'ASC'.
	 *
	 * @param string $return_format OBJECT, ARRAY_A, or ids.
	 * @return array
	 */
	public function search( $args = [], $return_format = OBJECT ) {
		try {
			return as_get_scheduled_actions( $args, $return_format );
		} catch ( Exception $exception ) {
			Logger::error( $exception->getMessage(), [ 'Action Scheduler Queue' ] );

return [];
		}
	}
}

Ubuntu

Sid Gifari From Gifari Industries - BD Cyber Security Team