Skip to content

fix(locks): make cron admission atomic #522

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
sjinks wants to merge 5 commits into
base: main
Choose a base branch
from
Open

fix(locks): make cron admission atomic #522

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
a510c48
fix(locks): make cron admission atomic
sjinks Aug 27, 2026
be8b1fa
fix(locks): avoid entropy errors during generation
sjinks Aug 27, 2026
74e1448
fix(locks): call WordPress random generator globally
sjinks Aug 27, 2026
6c92881
fix(locks): support generation before WordPress bootstrap
sjinks Aug 27, 2026
1094377
fix(locks): make release failures retryable
sjinks Aug 27, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
140 changes: 140 additions & 0 deletions __tests__/unit-tests/test-lock.php
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,140 @@
<?php

namespace Automattic\WP\Cron_Control\Tests;

use Automattic\WP\Cron_Control\Lock;

class Lock_Tests extends \WP_UnitTestCase {
private $locks = array(
'concurrent-acquisition',
'rejected-admission',
'stale-lock',
'old-generation-cleanup',
'recreated-lock',
'reset-lock',
);

public function setUp(): void {
parent::setUp();

foreach ( $this->locks as $lock ) {
Lock::reset_lock( $lock );
}
}

public function tearDown(): void {
foreach ( $this->locks as $lock ) {
Lock::reset_lock( $lock );
}

parent::tearDown();
}

public function test_concurrent_acquisition_with_limit_one_admits_one_worker() {
$lock = 'concurrent-acquisition';

$this->assertTrue( Lock::check_lock( $lock, 1 ) );
$this->assertFalse( Lock::check_lock( $lock, 1 ) );
$this->assertSame( 1, Lock::get_lock_value( $lock ) );
}

public function test_rejected_admission_does_not_change_lock_value() {
$lock = 'rejected-admission';

$this->assertTrue( Lock::check_lock( $lock, 1 ) );
$this->assertFalse( Lock::check_lock( $lock, 1 ) );
$this->assertSame( 1, Lock::get_lock_value( $lock ) );
}

public function test_stale_lock_recovery_records_the_new_admission() {
global $wpdb;

$lock = 'stale-lock';
$wpdb->update(
$wpdb->options,
array( 'option_value' => '1:' . ( time() - MINUTE_IN_SECONDS ) . ':1' ),
array( 'option_name' => 'a8ccc_lock_stale-lock' )
);

$this->assertTrue( Lock::check_lock( $lock, 1, 1 ) );
$this->assertSame( 1, Lock::get_lock_value( $lock ) );
$this->assertGreaterThan( time() - 2, Lock::get_lock_timestamp( $lock ) );
}

public function test_old_generation_cleanup_does_not_release_recovered_lock() {
global $wpdb;

$lock = 'old-generation-cleanup';
$this->assertTrue( Lock::check_lock( $lock, 1 ) );
$wpdb->update(
$wpdb->options,
array( 'option_value' => '1:' . ( time() - MINUTE_IN_SECONDS ) . ':1' ),
array( 'option_name' => 'a8ccc_lock_old-generation-cleanup' )
);
$this->assertTrue( Lock::check_lock( $lock, 1, 1 ) );
$state = explode( ':', $wpdb->get_var( "SELECT `option_value` FROM `$wpdb->options` WHERE `option_name` = 'a8ccc_lock_old-generation-cleanup'" ) );

$reflection = new \ReflectionProperty( Lock::class, 'acquired_locks' );
$reflection->setAccessible( true );
$reflection->setValue(
array(
$lock => array( (int) $state[2], 1 ),
)
);

$this->assertTrue( Lock::free_lock( $lock ) );
$this->assertSame( 1, Lock::get_lock_value( $lock ) );
}

public function test_old_generation_cleanup_does_not_release_a_recreated_lock() {
global $wpdb;

$lock = 'recreated-lock';
$this->assertTrue( Lock::check_lock( $lock, 1 ) );
$state = explode( ':', $wpdb->get_var( "SELECT `option_value` FROM `$wpdb->options` WHERE `option_name` = 'a8ccc_lock_recreated-lock'" ) );
$old_generation = (int) $state[2];
$this->assertTrue( Lock::free_lock( $lock ) );
$this->assertNull( $wpdb->get_var( "SELECT `option_value` FROM `$wpdb->options` WHERE `option_name` = 'a8ccc_lock_recreated-lock'" ) );

$this->assertTrue( Lock::check_lock( $lock, 1 ) );
$state = explode( ':', $wpdb->get_var( "SELECT `option_value` FROM `$wpdb->options` WHERE `option_name` = 'a8ccc_lock_recreated-lock'" ) );
$replacement_state = implode( ':', $state );
$replacement_generation = (int) $state[2];

$reflection = new \ReflectionProperty( Lock::class, 'acquired_locks' );
$reflection->setAccessible( true );
$reflection->setValue(
array(
$lock => array( $replacement_generation, $old_generation ),
)
);

$this->assertTrue( Lock::free_lock( $lock ) );
$this->assertSame( $replacement_state, $wpdb->get_var( "SELECT `option_value` FROM `$wpdb->options` WHERE `option_name` = 'a8ccc_lock_recreated-lock'" ) );
}

public function test_old_generation_cleanup_does_not_release_a_reset_lock() {
global $wpdb;

$lock = 'reset-lock';
$this->assertTrue( Lock::check_lock( $lock, 1 ) );
$state = explode( ':', $wpdb->get_var( "SELECT `option_value` FROM `$wpdb->options` WHERE `option_name` = 'a8ccc_lock_reset-lock'" ) );
$old_generation = (int) $state[2];
$this->assertTrue( Lock::reset_lock( $lock ) );
$this->assertTrue( Lock::check_lock( $lock, 1 ) );
$state = explode( ':', $wpdb->get_var( "SELECT `option_value` FROM `$wpdb->options` WHERE `option_name` = 'a8ccc_lock_reset-lock'" ) );
$replacement_state = implode( ':', $state );
$replacement_generation = (int) $state[2];

$reflection = new \ReflectionProperty( Lock::class, 'acquired_locks' );
$reflection->setAccessible( true );
$reflection->setValue(
array(
$lock => array( $replacement_generation, $old_generation ),
)
);

$this->assertTrue( Lock::free_lock( $lock ) );
$this->assertSame( $replacement_state, $wpdb->get_var( "SELECT `option_value` FROM `$wpdb->options` WHERE `option_name` = 'a8ccc_lock_reset-lock'" ) );
}
}
6 changes: 1 addition & 5 deletions includes/class-events.php
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -348,11 +348,7 @@ private function reset_event_lock( Event $event ): bool {
$lock_key = $this->get_lock_key_for_event_action( $event );
$expires = JOB_LOCK_EXPIRY_IN_MINUTES * \MINUTE_IN_SECONDS;

if ( isset( $this->concurrent_action_whitelist[ $event->get_action() ] ) ) {
return Lock::free_lock( $lock_key, $expires );
} else {
return Lock::reset_lock( $lock_key, $expires );
}
return Lock::free_lock( $lock_key, $expires );
}

/**
Expand Down
Loading
Loading