notifier / libraries / action-scheduler / classes / ActionScheduler_FatalErrorMonitor.php

ActionScheduler_FatalErrorMonitor.php in WANotifier for Forms and Actions 3.1.1, at libraries/action-scheduler/classes/ActionScheduler_FatalErrorMonitor.php

99 lines 2.5 KB

Raw Download Zip

1	<?php
2
3	/**
4	* Class ActionScheduler_FatalErrorMonitor
5	*/
6	class ActionScheduler_FatalErrorMonitor {
7
8	/**
9	* ActionScheduler_ActionClaim instance.
10	*
11	* @var ActionScheduler_ActionClaim
12	*/
13	private $claim = null;
14
15	/**
16	* ActionScheduler_Store instance.
17	*
18	* @var ActionScheduler_Store
19	*/
20	private $store = null;
21
22	/**
23	* Current action's ID.
24	*
25	* @var int
26	*/
27	private $action_id = 0;
28
29	/**
30	* Construct.
31	*
32	* @param ActionScheduler_Store $store Action store.
33	*/
34	public function __construct( ActionScheduler_Store $store ) {
35	$this->store = $store;
36	}
37
38	/**
39	* Start monitoring.
40	*
41	* @param ActionScheduler_ActionClaim $claim Claimed actions.
42	*/
43	public function attach( ActionScheduler_ActionClaim $claim ) {
44	$this->claim = $claim;
45	add_action( 'shutdown', array( $this, 'handle_unexpected_shutdown' ) );
46	add_action( 'action_scheduler_before_execute', array( $this, 'track_current_action' ), 0, 1 );
47	add_action( 'action_scheduler_after_execute', array( $this, 'untrack_action' ), 0, 0 );
48	add_action( 'action_scheduler_execution_ignored', array( $this, 'untrack_action' ), 0, 0 );
49	add_action( 'action_scheduler_failed_execution', array( $this, 'untrack_action' ), 0, 0 );
50	}
51
52	/**
53	* Stop monitoring.
54	*/
55	public function detach() {
56	$this->claim = null;
57	$this->untrack_action();
58	remove_action( 'shutdown', array( $this, 'handle_unexpected_shutdown' ) );
59	remove_action( 'action_scheduler_before_execute', array( $this, 'track_current_action' ), 0 );
60	remove_action( 'action_scheduler_after_execute', array( $this, 'untrack_action' ), 0 );
61	remove_action( 'action_scheduler_execution_ignored', array( $this, 'untrack_action' ), 0 );
62	remove_action( 'action_scheduler_failed_execution', array( $this, 'untrack_action' ), 0 );
63	}
64
65	/**
66	* Track specified action.
67	*
68	* @param int $action_id Action ID to track.
69	*/
70	public function track_current_action( $action_id ) {
71	$this->action_id = $action_id;
72	}
73
74	/**
75	* Un-track action.
76	*/
77	public function untrack_action() {
78	$this->action_id = 0;
79	}
80
81	/**
82	* Handle unexpected shutdown.
83	*/
84	public function handle_unexpected_shutdown() {
85	$error = error_get_last();
86
87	if ( $error ) {
88	if ( in_array( $error['type'], array( E_ERROR, E_PARSE, E_COMPILE_ERROR, E_USER_ERROR, E_RECOVERABLE_ERROR ), true ) ) {
89	if ( ! empty( $this->action_id ) ) {
90	$this->store->mark_failure( $this->action_id );
91	do_action( 'action_scheduler_unexpected_shutdown', $this->action_id, $error );
92	}
93	}
94
95	$this->store->release_claim( $this->claim );
96	}
97	}
98	}
99