| 1 |
<?php |
| 2 |
|
| 3 |
/** |
| 4 |
* Class ActionScheduler_FatalErrorMonitor |
| 5 |
*/ |
| 6 |
class ActionScheduler_FatalErrorMonitor { |
| 7 |
/** |
| 8 |
* @var ActionScheduler_ActionClaim |
| 9 |
*/ |
| 10 |
private $claim = NULL; |
| 11 |
/** |
| 12 |
* @var ActionScheduler_Store |
| 13 |
*/ |
| 14 |
private $store = NULL; |
| 15 |
private $action_id = 0; |
| 16 |
|
| 17 |
public function __construct( ActionScheduler_Store $store ) { |
| 18 |
$this->store = $store; |
| 19 |
} |
| 20 |
|
| 21 |
public function attach( ActionScheduler_ActionClaim $claim ) { |
| 22 |
$this->claim = $claim; |
| 23 |
add_action( 'shutdown', array( $this, 'handle_unexpected_shutdown' ) ); |
| 24 |
add_action( 'action_scheduler_before_execute', array( $this, 'track_current_action' ), 0, 1 ); |
| 25 |
add_action( 'action_scheduler_after_execute', array( $this, 'untrack_action' ), 0, 0 ); |
| 26 |
add_action( 'action_scheduler_execution_ignored', array( $this, 'untrack_action' ), 0, 0 ); |
| 27 |
add_action( 'action_scheduler_failed_execution', array( $this, 'untrack_action' ), 0, 0 ); |
| 28 |
} |
| 29 |
|
| 30 |
public function detach() { |
| 31 |
$this->claim = NULL; |
| 32 |
$this->untrack_action(); |
| 33 |
remove_action( 'shutdown', array( $this, 'handle_unexpected_shutdown' ) ); |
| 34 |
remove_action( 'action_scheduler_before_execute', array( $this, 'track_current_action' ), 0 ); |
| 35 |
remove_action( 'action_scheduler_after_execute', array( $this, 'untrack_action' ), 0 ); |
| 36 |
remove_action( 'action_scheduler_execution_ignored', array( $this, 'untrack_action' ), 0 ); |
| 37 |
remove_action( 'action_scheduler_failed_execution', array( $this, 'untrack_action' ), 0 ); |
| 38 |
} |
| 39 |
|
| 40 |
public function track_current_action( $action_id ) { |
| 41 |
$this->action_id = $action_id; |
| 42 |
} |
| 43 |
|
| 44 |
public function untrack_action() { |
| 45 |
$this->action_id = 0; |
| 46 |
} |
| 47 |
|
| 48 |
public function handle_unexpected_shutdown() { |
| 49 |
if ( $error = error_get_last() ) { |
| 50 |
if ( in_array( $error['type'], array( E_ERROR, E_PARSE, E_COMPILE_ERROR, E_USER_ERROR, E_RECOVERABLE_ERROR ) ) ) { |
| 51 |
if ( !empty($this->action_id) ) { |
| 52 |
$this->store->mark_failure( $this->action_id ); |
| 53 |
do_action( 'action_scheduler_unexpected_shutdown', $this->action_id, $error ); |
| 54 |
} |
| 55 |
} |
| 56 |
$this->store->release_claim( $this->claim ); |
| 57 |
} |
| 58 |
} |
| 59 |
} |
| 60 |
|
