This repository has been archived by the owner on Feb 23, 2020. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 25
/
RngCron.php
95 lines (80 loc) · 2.53 KB
/
RngCron.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
<?php
namespace Drupal\rng;
use Drupal\Core\Queue\QueueFactory;
use Drupal\Core\Entity\EntityTypeManagerInterface;
use Drupal\rng\Entity\RuleSchedule;
/**
* RNG Cron.
*/
class RngCron {
/**
* Queue for rule scheduler.
*
* @var \Drupal\Core\Queue\QueueInterface
*/
protected $ruleSchedulerQueue;
/**
* Storage for rule scheduler entities.
*
* @var \Drupal\Core\Entity\EntityStorageInterface
*/
protected $ruleSchedulerStorage;
/**
* Constructs a new RngEntityModel object.
*
* @param \Drupal\Core\Queue\QueueFactory $queue_factory
* The queue service.
* @param \Drupal\Core\Entity\EntityTypeManagerInterface $entity_type_manager
* The entity type manager.
*/
public function __construct(QueueFactory $queue_factory, EntityTypeManagerInterface $entity_type_manager) {
$this->ruleSchedulerQueue = $queue_factory->get('rng_rule_scheduler', FALSE);
$this->ruleSchedulerStorage = $entity_type_manager->getStorage('rng_rule_scheduler');
}
/**
* React to Drupal `hook_cron` hooks.
*
* @see hook_cron();
*/
public function hook_cron() {
$this->scheduleRules();
$this->deleteScheduleRules();
}
/**
* Add scheduled rules to be executed to the queue.
*/
protected function scheduleRules() {
$ids = $this->ruleSchedulerStorage
->getQuery()
->condition('trigger_date', time(), '<=')
->condition('in_queue', 0, '=')
->condition('attempts', RuleSchedule::ATTEMPTS_MAX, '<=')
->execute();
/** @var \Drupal\rng\RuleScheduleInterface[] $rule_schedules */
$rule_schedules = $this->ruleSchedulerStorage->loadMultiple($ids);
foreach ($rule_schedules as $rule_schedule) {
$data = ['rule_scheduler_id' => $rule_schedule->id()];
if ($this->ruleSchedulerQueue->createItem($data)) {
$rule_schedule->setInQueue(TRUE)->save();
// De-activate the rule once it is queued.
if ($component = $rule_schedule->getComponent()) {
if ($rule = $component->getRule()) {
$rule->setIsActive(FALSE)->save();
}
}
}
}
}
/**
* Delete scheduled rules which have had too many attempts.
*/
protected function deleteScheduleRules() {
$ids = $this->ruleSchedulerStorage
->getQuery()
->condition('attempts', RuleSchedule::ATTEMPTS_MAX, '>')
->execute();
/** @var \Drupal\rng\RuleScheduleInterface[] $rule_schedules */
$rule_schedules = $this->ruleSchedulerStorage->loadMultiple($ids);
$this->ruleSchedulerStorage->delete($rule_schedules);
}
}