<?php

[0] Fix | Delete

[1] Fix | Delete

namespace ImageOptimization\Modules\Optimization\Components;

[2] Fix | Delete

[3] Fix | Delete

use ImageOptimization\Classes\Async_Operation\{

[4] Fix | Delete

Async_Operation,

[5] Fix | Delete

Async_Operation_Hook,

[6] Fix | Delete

Async_Operation_Queue,

[7] Fix | Delete

Exceptions\Async_Operation_Exception,

[8] Fix | Delete

Queries\Image_Optimization_Operation_Query,

[9] Fix | Delete

Queries\Operation_Query

[10] Fix | Delete

};

[11] Fix | Delete

[12] Fix | Delete

use ImageOptimization\Classes\Image\{

[13] Fix | Delete

Image_Meta,

[14] Fix | Delete

Image_Optimization_Error_Type,

[15] Fix | Delete

Image_Query_Builder,

[16] Fix | Delete

Image_Status,

[17] Fix | Delete

};

[18] Fix | Delete

[19] Fix | Delete

use ImageOptimization\Classes\Logger;

[20] Fix | Delete

use Throwable;

[21] Fix | Delete

[22] Fix | Delete

if ( ! defined( 'ABSPATH' ) ) {

[23] Fix | Delete

exit; // Exit if accessed directly.

[24] Fix | Delete

}

[25] Fix | Delete

[26] Fix | Delete

class Actions_Cleanup {

[27] Fix | Delete

const FIVE_MINUTES_IN_SECONDS = 300;

[28] Fix | Delete

[29] Fix | Delete

/**

[30] Fix | Delete

* @async

[31] Fix | Delete

* @return void

[32] Fix | Delete

*/

[33] Fix | Delete

public function cleanup_stuck_operations() {

[34] Fix | Delete

global $wpdb;

[35] Fix | Delete

[36] Fix | Delete

$table_name = $wpdb->prefix . 'actionscheduler_actions';

[37] Fix | Delete

$now = time();

[38] Fix | Delete

$threshold = $now - self::FIVE_MINUTES_IN_SECONDS;

[39] Fix | Delete

[40] Fix | Delete

// phpcs:disable WordPress.DB.PreparedSQL.InterpolatedNotPrepared

[41] Fix | Delete

$results = $wpdb->get_col(

[42] Fix | Delete

$wpdb->prepare(

[43] Fix | Delete

"

[44] Fix | Delete

SELECT action_id

[45] Fix | Delete

FROM {$table_name}

[46] Fix | Delete

WHERE last_attempt_gmt IS NOT NULL

[47] Fix | Delete

AND UNIX_TIMESTAMP(last_attempt_gmt) < %d

[48] Fix | Delete

",

[49] Fix | Delete

$threshold

[50] Fix | Delete

)

[51] Fix | Delete

);

[52] Fix | Delete

// phpcs:enable WordPress.DB.PreparedSQL.InterpolatedNotPrepared

[53] Fix | Delete

[54] Fix | Delete

if ( empty( $results ) ) {

[55] Fix | Delete

Logger::debug( 'No stuck optimization operations found for cleanup.' );

[56] Fix | Delete

return;

[57] Fix | Delete

}

[58] Fix | Delete

[59] Fix | Delete

foreach ( $results as $action_id ) {

[60] Fix | Delete

$action = Async_Operation::get_by_id( (int) $action_id );

[61] Fix | Delete

[62] Fix | Delete

if (

[63] Fix | Delete

! $action ||

[64] Fix | Delete

Async_Operation_Queue::OPTIMIZE !== $action->get_queue() ||

[65] Fix | Delete

Async_Operation::OPERATION_STATUS_RUNNING !== $action->get_status()

[66] Fix | Delete

) {

[67] Fix | Delete

continue;

[68] Fix | Delete

}

[69] Fix | Delete

[70] Fix | Delete

try {

[71] Fix | Delete

do_action(

[72] Fix | Delete

'action_scheduler_failed_action',

[73] Fix | Delete

$action_id,

[74] Fix | Delete

self::FIVE_MINUTES_IN_SECONDS

[75] Fix | Delete

);

[76] Fix | Delete

[77] Fix | Delete

Logger::debug( "Triggered retry for stuck action ID {$action_id}." );

[78] Fix | Delete

} catch ( Throwable $t ) {

[79] Fix | Delete

Logger::warn( "Failed to handle stuck operation for action ID {$action_id}: " . $t->getMessage() );

[80] Fix | Delete

}

[81] Fix | Delete

}

[82] Fix | Delete

[83] Fix | Delete

try {

[84] Fix | Delete

$this->cleanup_stuck_statuses();

[85] Fix | Delete

} catch ( Throwable $t ) {

[86] Fix | Delete

Logger::warn( 'Failed to run stuck statuses clearing job: ' . $t->getMessage() );

[87] Fix | Delete

}

[88] Fix | Delete

}

[89] Fix | Delete

[90] Fix | Delete

/**

[91] Fix | Delete

* The handler checks if there are any attachments that have the in-progress status, but no jobs are currently

[92] Fix | Delete

* run or pending. Those attachment statuses will be updated to a generic error.

[93] Fix | Delete

*

[94] Fix | Delete

* @throws Async_Operation_Exception

[95] Fix | Delete

*/

[96] Fix | Delete

public function cleanup_stuck_statuses() {

[97] Fix | Delete

$operations_query = ( new Image_Optimization_Operation_Query() )

[98] Fix | Delete

->set_status( [ Async_Operation::OPERATION_STATUS_PENDING, Async_Operation::OPERATION_STATUS_RUNNING ] )

[99] Fix | Delete

->set_limit( 1 );

[100] Fix | Delete

[101] Fix | Delete

$operations = Async_Operation::get( $operations_query );

[102] Fix | Delete

[103] Fix | Delete

if ( ! empty( $operations ) ) {

[104] Fix | Delete

return;

[105] Fix | Delete

}

[106] Fix | Delete

[107] Fix | Delete

$image_query = ( new Image_Query_Builder() )

[108] Fix | Delete

->return_optimization_in_progress_images()

[109] Fix | Delete

->set_paging_size( -1 )

[110] Fix | Delete

->execute();

[111] Fix | Delete

[112] Fix | Delete

foreach ( $image_query->posts as $attachment_id ) {

[113] Fix | Delete

( new Image_Meta( $attachment_id ) )

[114] Fix | Delete

->set_status( Image_Status::OPTIMIZATION_FAILED )

[115] Fix | Delete

->set_error_type( Image_Optimization_Error_Type::GENERIC )

[116] Fix | Delete

->save();

[117] Fix | Delete

}

[118] Fix | Delete

}

[119] Fix | Delete

[120] Fix | Delete

public function schedule_cleanup() {

[121] Fix | Delete

try {

[122] Fix | Delete

$cleanup_job_query = ( new Operation_Query() )

[123] Fix | Delete

->set_queue( Async_Operation_Queue::CLEANUP )

[124] Fix | Delete

->set_hook( Async_Operation_Hook::STUCK_OPERATION_CLEANUP )

[125] Fix | Delete

->set_status( [ Async_Operation::OPERATION_STATUS_PENDING, Async_Operation::OPERATION_STATUS_RUNNING ] )

[126] Fix | Delete

->set_limit( 1 );

[127] Fix | Delete

[128] Fix | Delete

// Prevents job duplication. For some reason unique=true is not enough

[129] Fix | Delete

if ( ! empty( Async_Operation::get( $cleanup_job_query ) ) ) {

[130] Fix | Delete

return;

[131] Fix | Delete

}

[132] Fix | Delete

[133] Fix | Delete

Async_Operation::create_recurring(

[134] Fix | Delete

time(),

[135] Fix | Delete

self::FIVE_MINUTES_IN_SECONDS,

[136] Fix | Delete

Async_Operation_Hook::STUCK_OPERATION_CLEANUP,

[137] Fix | Delete

[],

[138] Fix | Delete

Async_Operation_Queue::CLEANUP,

[139] Fix | Delete

10,

[140] Fix | Delete

true

[141] Fix | Delete

);

[142] Fix | Delete

} catch ( Async_Operation_Exception $aoe ) {

[143] Fix | Delete

Logger::warn( 'Failed to schedule recurring stuck operation cleanup: ' . $aoe->getMessage() );

[144] Fix | Delete

}

[145] Fix | Delete

}

[146] Fix | Delete

[147] Fix | Delete

public function __construct() {

[148] Fix | Delete

add_action( 'action_scheduler_init', [ $this, 'schedule_cleanup' ] );

[149] Fix | Delete

add_action( Async_Operation_Hook::STUCK_OPERATION_CLEANUP, [ $this, 'cleanup_stuck_operations' ] );

[150] Fix | Delete

}

[151] Fix | Delete

}

[152] Fix | Delete

[153] Fix | Delete