]> git.mxchange.org Git - friendica.git/blob - src/Protocol/ActivityPub/Queue.php
Improved workerqueue handling
[friendica.git] / src / Protocol / ActivityPub / Queue.php
1 <?php
2 /**
3  * @copyright Copyright (C) 2010-2022, the Friendica project
4  *
5  * @license GNU AGPL version 3 or any later version
6  *
7  * This program is free software: you can redistribute it and/or modify
8  * it under the terms of the GNU Affero General Public License as
9  * published by the Free Software Foundation, either version 3 of the
10  * License, or (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU Affero General Public License for more details.
16  *
17  * You should have received a copy of the GNU Affero General Public License
18  * along with this program.  If not, see <https://www.gnu.org/licenses/>.
19  *
20  */
21
22 namespace Friendica\Protocol\ActivityPub;
23
24 use Friendica\Core\Logger;
25 use Friendica\Database\Database;
26 use Friendica\Database\DBA;
27 use Friendica\Util\DateTimeFormat;
28
29 /**
30  * This class handles the processing of incoming posts
31  */
32 class Queue
33 {
34         /**
35          * Add activity to the queue
36          *
37          * @param array $activity
38          * @param string $type
39          * @param integer $uid
40          * @param string $http_signer
41          * @param boolean $push
42          * @return array
43          */
44         public static function add(array $activity, string $type, int $uid, string $http_signer, bool $push): array
45         {
46                 $fields = [
47                         'activity-id' => $activity['id'],
48                         'object-id'   => $activity['object_id'],
49                         'type'        => $type,
50                         'object-type' => $activity['object_type'],
51                         'activity'    => json_encode($activity, JSON_UNESCAPED_SLASHES | JSON_UNESCAPED_UNICODE | JSON_PRETTY_PRINT),
52                         'received'    => DateTimeFormat::utcNow(),
53                         'push'        => $push,
54                 ];
55
56                 if (!empty($activity['reply-to-id'])) {
57                         $fields['in-reply-to-id'] = $activity['reply-to-id'];
58                 }
59
60                 if (!empty($activity['object_object_type'])) {
61                         $fields['object-object-type'] = $activity['object_object_type'];
62                 }
63
64                 if (!empty($http_signer)) {
65                         $fields['signer'] = $http_signer;
66                 }
67
68                 DBA::insert('inbox-entry', $fields, Database::INSERT_IGNORE);
69
70                 $queue = DBA::selectFirst('inbox-entry', ['id'], ['activity-id' => $activity['id']]);
71                 if (!empty($queue['id'])) {
72                         $activity['entry-id'] = $queue['id'];
73                         DBA::insert('inbox-entry-receiver', ['queue-id' => $queue['id'], 'uid' => $uid], Database::INSERT_IGNORE);
74                 }
75                 return $activity;
76         }
77
78         /**
79          * Remove activity from the queue
80          *
81          * @param array $activity
82          * @return void
83          */
84         public static function remove(array $activity = [])
85         {
86                 if (empty($activity['entry-id'])) {
87                         return;
88                 }
89                 DBA::delete('inbox-entry', ['id' => $activity['entry-id']]);
90         }
91
92         /**
93          * Set the worker id for the queue entry
94          *
95          * @param array $activity
96          * @param int   $wid
97          * @return void
98          */
99         public static function setWorkerId(array $activity, int $wid)
100         {
101                 if (empty($activity['entry-id']) || empty($wid)) {
102                         return;
103                 }
104                 DBA::update('inbox-entry', ['wid' => $wid], ['id' => $activity['entry-id']]);
105         }
106
107         /**
108          * Check if there is an assigned worker task
109          *
110          * @param array $activity
111          * @return bool
112          */
113         public static function hasWorker(array $activity = []): bool
114         {
115                 if (empty($activity['worker-id'])) {
116                         return false;
117                 }
118                 return DBA::exists('workerqueue', ['id' => $activity['worker-id'], 'done' => false]);
119         }
120
121         /**
122          * Process the activity with the given id
123          *
124          * @param integer $id
125          * @return void
126          */
127         public static function process(int $id)
128         {
129                 $entry = DBA::selectFirst('inbox-entry', [], ['id' => $id]);
130                 if (empty($entry)) {
131                         return;
132                 }
133
134                 Logger::debug('Processing queue entry', ['id' => $entry['id'], 'type' => $entry['type'], 'object-type' => $entry['object-type'], 'uri' => $entry['object-id'], 'in-reply-to' => $entry['in-reply-to-id']]);
135
136                 $activity = json_decode($entry['activity'], true);
137                 $type     = $entry['type'];
138                 $push     = $entry['push'];
139
140                 $activity['entry-id']  = $entry['id'];
141                 $activity['worker-id'] = $entry['wid'];
142
143                 if (!Receiver::routeActivities($activity, $type, $push)) {
144                         self::remove($activity);
145                 }
146         }
147
148         /**
149          * Process all activities
150          *
151          * @return void
152          */
153         public static function processAll()
154         {
155                 $entries = DBA::select('inbox-entry', ['id', 'type', 'object-type'], [], ['order' => ['id' => true]]);
156                 while ($entry = DBA::fetch($entries)) {
157                         self::process($entry['id']);
158                 }
159         }
160
161         /**
162          * Clear old activities
163          *
164          * @return void
165          */
166         public static function clear()
167         {
168                 DBA::delete('inbox-entry', ["`wid` IS NULL AND `received` < ?", DateTimeFormat::utc('now - 4 hours')]);
169         }
170
171         /**
172          * Process all activities that are children of a given post url
173          *
174          * @param string $uri
175          * @return void
176          */
177         public static function processReplyByUri(string $uri)
178         {
179                 $entries = DBA::select('inbox-entry', ['id'], ['in-reply-to-id' => $uri], ['order' => ['id' => true]]);
180                 while ($entry = DBA::fetch($entries)) {
181                         self::process($entry['id']);
182                 }
183         }
184 }