Merge pull request #1133 from mexon/mat/mailstream-workerqueue

Use workerqueue for mailstream jobs instead of custom table
This commit is contained in:
Hypolite Petovan 2021-06-16 17:57:06 -04:00 committed by GitHub
commit 0d4c95b7bc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -2,7 +2,7 @@
/** /**
* Name: Mail Stream * Name: Mail Stream
* Description: Mail all items coming into your network feed to an email address * Description: Mail all items coming into your network feed to an email address
* Version: 1.1 * Version: 2.0
* Author: Matthew Exon <http://mat.exon.name> * Author: Matthew Exon <http://mat.exon.name>
*/ */
@ -10,11 +10,14 @@ use Friendica\Content\Text\BBCode;
use Friendica\Core\Hook; use Friendica\Core\Hook;
use Friendica\Core\Logger; use Friendica\Core\Logger;
use Friendica\Core\Renderer; use Friendica\Core\Renderer;
use Friendica\Core\Worker;
use Friendica\Database\DBA; use Friendica\Database\DBA;
use Friendica\DI; use Friendica\DI;
use Friendica\Model\Item; use Friendica\Model\Item;
use Friendica\Model\Post; use Friendica\Model\Post;
use Friendica\Model\User;
use Friendica\Protocol\Activity; use Friendica\Protocol\Activity;
use Friendica\Util\DateTimeFormat;
/** /**
* Sets up the addon hooks and the database table * Sets up the addon hooks and the database table
@ -26,43 +29,32 @@ function mailstream_install()
Hook::register('post_local_end', 'addon/mailstream/mailstream.php', 'mailstream_post_hook'); Hook::register('post_local_end', 'addon/mailstream/mailstream.php', 'mailstream_post_hook');
Hook::register('post_remote_end', 'addon/mailstream/mailstream.php', 'mailstream_post_hook'); Hook::register('post_remote_end', 'addon/mailstream/mailstream.php', 'mailstream_post_hook');
Hook::register('cron', 'addon/mailstream/mailstream.php', 'mailstream_cron'); Hook::register('cron', 'addon/mailstream/mailstream.php', 'mailstream_cron');
Hook::register('mailstream_send_hook', 'addon/mailstream/mailstream.php', 'mailstream_send_hook');
if (DI::config()->get('mailstream', 'dbversion') == '0.1') { Logger::info("mailstream: installed");
q('ALTER TABLE `mailstream_item` DROP INDEX `uid`'); }
q('ALTER TABLE `mailstream_item` DROP INDEX `contact-id`');
q('ALTER TABLE `mailstream_item` DROP INDEX `plink`');
q('ALTER TABLE `mailstream_item` CHANGE `plink` `uri` char(255) NOT NULL');
DI::config()->set('mailstream', 'dbversion', '0.2');
}
if (DI::config()->get('mailstream', 'dbversion') == '0.2') {
q('DELETE FROM `pconfig` WHERE `cat` = "mailstream" AND `k` = "delay"');
DI::config()->set('mailstream', 'dbversion', '0.3');
}
if (DI::config()->get('mailstream', 'dbversion') == '0.3') {
q('ALTER TABLE `mailstream_item` CHANGE `created` `created` timestamp NOT NULL DEFAULT now()');
q('ALTER TABLE `mailstream_item` CHANGE `completed` `completed` timestamp NULL DEFAULT NULL');
DI::config()->set('mailstream', 'dbversion', '0.4');
}
if (DI::config()->get('mailstream', 'dbversion') == '0.4') {
q('ALTER TABLE `mailstream_item` CONVERT TO CHARACTER SET utf8 COLLATE utf8_bin');
DI::config()->set('mailstream', 'dbversion', '0.5');
}
if (DI::config()->get('mailstream', 'dbversion') == '0.5') {
DI::config()->set('mailstream', 'dbversion', '1.0');
}
if (DI::config()->get('retriever', 'dbversion') != '1.0') { /**
$schema = file_get_contents(dirname(__file__).'/database.sql'); * Enforces that mailstream_install has set up the current version
$arr = explode(';', $schema); */
foreach ($arr as $a) { function mailstream_check_version()
$r = q($a); {
} if (!is_null(DI::config()->get('mailstream', 'dbversion'))) {
DI::config()->set('mailstream', 'dbversion', '1.0'); DI::config()->delete('mailstream', 'dbversion');
Logger::info("mailstream_check_version: old version detected, reinstalling");
mailstream_install();
Hook::loadHooks();
Hook::add(
'mailstream_convert_table_entries',
'addon/mailstream/mailstream.php',
'mailstream_convert_table_entries'
);
Hook::fork(PRIORITY_LOW, 'mailstream_convert_table_entries');
} }
} }
/** /**
* This funciton indicates a module that can be wrapped in the LegacyModule class * This function indicates a module that can be wrapped in the LegacyModule class
*/ */
function mailstream_module() function mailstream_module()
{ {
@ -114,16 +106,41 @@ function mailstream_generate_id($uri)
return $message_id; return $message_id;
} }
function mailstream_send_hook(&$a, $data)
{
$criteria = array('uid' => $data['uid'], 'contact-id' => $data['contact-id'], 'uri' => $data['uri']);
$item = Post::selectFirst([], $criteria);
if (empty($item)) {
Logger::error('mailstream_send_hook could not find item');
return;
}
$user = User::getById($item['uid']);
if (empty($user)) {
Logger::error('mailstream_send_hook could not fund user', ['uid' => $item['uid']]);
return;
}
if (!mailstream_send($data['message_id'], $item, $user)) {
Logger::debug('mailstream_send_hook send failed, will retry', $data);
if (!Worker::defer()) {
Logger::error('mailstream_send_hook failed and could not defer', $data);
}
}
}
/** /**
* Called when either a local or remote post is created. Creates a * Called when either a local or remote post is created. If
* record in the mailstream_item table to track this email, and then * mailstream is enabled and the necessary data is available, forks a
* immediately attempts to send it * workerqueue item to send the email.
* *
* @param Friendica\App $a App object (unused) * @param Friendica\App $a App object (unused)
* @param array $item content of the item (may or may not already be stored in the item table) * @param array $item content of the item (may or may not already be stored in the item table)
*/ */
function mailstream_post_hook(&$a, &$item) function mailstream_post_hook(&$a, &$item)
{ {
mailstream_check_version();
if (!DI::pConfig()->get($item['uid'], 'mailstream', 'enabled')) { if (!DI::pConfig()->get($item['uid'], 'mailstream', 'enabled')) {
Logger::debug('mailstream: not enabled for item ' . $item['id']); Logger::debug('mailstream: not enabled for item ' . $item['id']);
return; return;
@ -152,50 +169,13 @@ function mailstream_post_hook(&$a, &$item)
} }
$message_id = mailstream_generate_id($item['uri']); $message_id = mailstream_generate_id($item['uri']);
q(
"INSERT INTO `mailstream_item` (`uid`, `contact-id`, `uri`, `message-id`) " .
"VALUES (%d, '%s', '%s', '%s')",
intval($item['uid']),
intval($item['contact-id']),
DBA::escape($item['uri']),
DBA::escape($message_id)
);
$r = q(
'SELECT * FROM `mailstream_item` WHERE `uid` = %d AND `contact-id` = %d AND `uri` = "%s"',
intval($item['uid']),
intval($item['contact-id']),
DBA::escape($item['uri'])
);
if (count($r) != 1) {
Logger::info('mailstream_post_remote_hook: Unexpected number of items returned from mailstream_item');
return;
}
$ms_item = $r[0];
Logger::debug('mailstream_post_remote_hook: created mailstream_item ' . $ms_item['id'] .
' for item ' . $item['uri'] . ' ' . $item['uid'] . ' ' . $item['contact-id']);
$user = mailstream_get_user($item['uid']);
if (!$user) {
Logger::info('mailstream_post_remote_hook: no user ' . $item['uid']);
return;
}
mailstream_send($ms_item['message-id'], $item, $user);
}
/** $send_hook_data = array('uid' => $item['uid'],
* Converts a user ID into a full user record from the corresponding database table 'contact-id' => $item['contact-id'],
* 'uri' => $item['uri'],
* @param int $uid ID of the user to query 'message_id' => $message_id,
* 'tries' => 0);
* @return array results from the user table Hook::fork(PRIORITY_LOW, 'mailstream_send_hook', $send_hook_data);
*/
function mailstream_get_user($uid)
{
$r = q('SELECT * FROM `user` WHERE `uid` = %d', intval($uid));
if (count($r) != 1) {
Logger::info('mailstream_post_remote_hook: Unexpected number of users returned');
return;
}
return $r[0];
} }
/** /**
@ -326,10 +306,13 @@ function mailstream_subject($item)
intval($item['uid']) intval($item['uid'])
); );
if (!DBA::isResult($r)) { if (!DBA::isResult($r)) {
Logger::error( Logger::error(
'mailstream_subject no contact for item', 'mailstream_subject no contact for item',
['item id' => $item['id'], 'plink' => $item['plink'], 'contact id' => $item['contact-id'], 'uid' => $item['uid']] ['id' => $item['id'],
); 'plink' => $item['plink'],
'contact id' => $item['contact-id'],
'uid' => $item['uid']]
);
return DI::l10n()->t("Friendica post"); return DI::l10n()->t("Friendica post");
} }
$contact = $r[0]; $contact = $r[0];
@ -361,14 +344,24 @@ function mailstream_subject($item)
* @param string $message_id ID of the message (RFC 1036) * @param string $message_id ID of the message (RFC 1036)
* @param array $item content of the item * @param array $item content of the item
* @param array $user results from the user table * @param array $user results from the user table
*
* @return bool True if this message has been completed. False if it should be retried.
*/ */
function mailstream_send($message_id, $item, $user) function mailstream_send($message_id, $item, $user)
{ {
if (!$item['visible']) { if (!is_array($item)) {
Logger::error('mailstream_send item is empty', ['message_id' => $message_id]);
return; return;
} }
if (!$item['visible']) {
Logger::debug('mailstream_send item not yet visible', ['item uri' => $item['uri']]);
return false;
}
if (!$message_id) { if (!$message_id) {
return; Logger::error('mailstream_send no message ID supplied', ['item uri' => $item['uri'],
'user email' => $user['email']]);
return true;
} }
require_once(dirname(__file__).'/phpmailer/class.phpmailer.php'); require_once(dirname(__file__).'/phpmailer/class.phpmailer.php');
@ -418,16 +411,16 @@ function mailstream_send($message_id, $item, $user)
if (!$mail->Send()) { if (!$mail->Send()) {
throw new Exception($mail->ErrorInfo); throw new Exception($mail->ErrorInfo);
} }
Logger::debug('mailstream_send sent message ' . $mail->MessageID . ' ' . $mail->Subject); Logger::debug('mailstream_send sent message', ['message ID' => $mail->MessageID,
'subject' => $mail->Subject,
'address' => $address]);
} catch (phpmailerException $e) { } catch (phpmailerException $e) {
Logger::debug('mailstream_send PHPMailer exception sending message ' . $message_id . ': ' . $e->errorMessage()); Logger::debug('mailstream_send PHPMailer exception sending message ' . $message_id . ': ' . $e->errorMessage());
} catch (Exception $e) { } catch (Exception $e) {
Logger::debug('mailstream_send exception sending message ' . $message_id . ': ' . $e->getMessage()); Logger::debug('mailstream_send exception sending message ' . $message_id . ': ' . $e->getMessage());
} }
// In case of failure, still set the item to completed. Otherwise
// we'll just try to send it over and over again and it'll fail return true;
// every time.
q('UPDATE `mailstream_item` SET `completed` = now() WHERE `message-id` = "%s"', DBA::escape($message_id));
} }
/** /**
@ -447,60 +440,41 @@ function mailstream_html_wrap(&$text)
} }
/** /**
* Cron job for the mailstream plugin. Sends delayed messages and cleans up old successful entries from the table. * Convert v1 mailstream table entries to v2 workerqueue items
*/ */
function mailstream_cron() function mailstream_convert_table_entries()
{ {
// Only process items older than an hour in cron. This is because
// we want to give mailstream_post_remote_hook a fair chance to
// send the email itself before cron jumps in. Only if
// mailstream_post_remote_hook fails for some reason will this get
// used, and in that case it's worth holding off a bit anyway.
$query = <<< EOT $query = <<< EOT
SELECT SELECT
`mailstream_item`.`message-id`, `message-id`,
`mailstream_item`.`uri`, `uri`,
`post-user-view`.`id` `uid`,
`contact-id`
FROM FROM
`mailstream_item` `mailstream_item`
JOIN
`post-user-view`
ON (
`mailstream_item`.`uid` = `post-user-view`.`uid` AND
`mailstream_item`.`uri` = `post-user-view`.`uri` AND
`mailstream_item`.`contact-id` = `post-user-view`.`contact-id`
)
WHERE WHERE
`mailstream_item`.`completed` IS NULL AND `mailstream_item`.`completed` IS NULL
`mailstream_item`.`created` < DATE_SUB(NOW(), INTERVAL 1 HOUR) AND
`post-user-view`.`visible` = 1
ORDER BY `mailstream_item`.`created`
LIMIT 100
EOT; EOT;
$ms_item_ids = q($query); $ms_item_ids = q($query);
if (DBA::isResult($ms_item_ids)) { if (DBA::isResult($ms_item_ids)) {
Logger::debug('mailstream_cron processing ' . count($ms_item_ids) . ' items'); Logger::debug('mailstream_convert_table_entries processing ' . count($ms_item_ids) . ' items');
foreach ($ms_item_ids as $ms_item_id) { foreach ($ms_item_ids as $ms_item_id) {
$send_hook_data = array('uid' => $ms_item_id['uid'],
'contact-id' => $ms_item_id['contact-id'],
'uri' => $ms_item_id['uri'],
'message_id' => $ms_item_id['message-id'],
'tries' => 0);
if (!$ms_item_id['message-id'] || !strlen($ms_item_id['message-id'])) { if (!$ms_item_id['message-id'] || !strlen($ms_item_id['message-id'])) {
Logger::info('mailstream_cron: Item ' . $ms_item_id['id'] . Logger::info('mailstream_cron: Item ' .
' URI ' . $ms_item_id['uri'] . ' has no message-id'); $ms_item_id['id'] . ' URI ' . $ms_item_id['uri'] . ' has no message-id');
} continue;
$item = Post::selectFirst([], ['id' => $ms_item_id['id']]);
$users = q("SELECT * FROM `user` WHERE `uid` = %d", intval($item['uid']));
$user = $users[0];
if ($user && $item) {
mailstream_send($ms_item_id['message-id'], $item, $user);
} else {
Logger::info('mailstream_cron: Unable to find item ' . $ms_item_id['id']);
q(
"UPDATE `mailstream_item` SET `completed` = now() WHERE `message-id` = %d",
intval($ms_item_id['message-id'])
);
} }
Logger::info('mailstream_convert_table_entries: convert item to workerqueue', $send_hook_data);
Hook::fork(PRIORITY_LOW, 'mailstream_send_hook', $send_hook_data);
} }
} }
mailstream_tidy(); q('DROP TABLE `mailstream_item`');
} }
/** /**
@ -567,25 +541,3 @@ function mailstream_addon_settings_post()
DI::pConfig()->delete(local_user(), 'mailstream', 'attachimg'); DI::pConfig()->delete(local_user(), 'mailstream', 'attachimg');
} }
} }
/**
* Deletes records from the mailstream_item table older than one year
*/
function mailstream_tidy()
{
$query = <<< EOT
SELECT
id
FROM
mailstream_item
WHERE
completed IS NOT NULL AND
completed < DATE_SUB(NOW(), INTERVAL 1 YEAR)
EOT;
$r = q($query);
foreach ($r as $rr) {
q('DELETE FROM mailstream_item WHERE id = %d', intval($rr['id']));
}
Logger::debug('mailstream_tidy: deleted ' . count($r) . ' old items');
}