Files
appwrite/app/workers/sync-out.php
T
2022-10-26 15:10:24 +03:00

164 lines
4.2 KiB
PHP

<?php
require_once __DIR__ . '/../worker.php';
use Ahc\Jwt\JWT;
use Appwrite\Utopia\Response;
use Swoole\Runtime;
use Swoole\Timer;
use Utopia\App;
use Utopia\Config\Config;
use Utopia\Database\Database;
use Utopia\Database\Document;
use Utopia\Database\Exception\Authorization;
use Utopia\Database\Exception\Structure;
use Utopia\Queue;
use Utopia\Queue\Message;
global $register;
$keys = [];
$counter = 0;
const SUBMITION_INTERVAL = 20;
const MAX_KEY_COUNT = 10;
const MAX_CURL_SEND_ATTEMPTS = 4;
/**
* @param string $url
* @param string $token
* @param array $keys
* @return array
*/
function send(string $url, string $token, array $keys): array
{
$payload = [];
$ch = curl_init($url);
curl_setopt($ch, CURLOPT_HTTPHEADER, [
'Authorization: Bearer ' . $token,
'Content-Type: application/json'
]);
curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
curl_setopt($ch, CURLOPT_TIMEOUT, 5);
curl_setopt($ch, CURLOPT_CUSTOMREQUEST, 'POST');
curl_setopt($ch, CURLOPT_POSTFIELDS, json_encode($keys));
for ($attempts = 0; $attempts < MAX_CURL_SEND_ATTEMPTS; $attempts++) {
$response = curl_exec($ch);
$status = curl_getinfo($ch, CURLINFO_HTTP_CODE);
$payload = [
'status' => $status,
'payload' => json_decode($response, true)
];
if ($status === 200) {
return $payload;
}
sleep(2);
}
curl_close($ch);
return $payload;
}
/**
* @throws Authorization
* @throws Structure
* @throws Exception
*/
function call($database, $regions, $keys): void
{
$jwt = new JWT(App::getEnv('_APP_OPENSSL_KEY_V1'), 'HS256', 600, 10);
$token = $jwt->encode([]);
foreach ($regions as $code => $region) {
var_dump('Sending request to ' . $code . '...............');
$response = send($region['domain'] . '/v1/edge/sync', $token, ['keys' => $keys]);
var_dump([
'keys' => $keys,
'response' => $response
]);
if ($response['status'] !== Response::STATUS_CODE_OK) {
$database->createDocument('syncs', new Document([
'region' => App::getEnv('_APP_REGION', 'nyc1'),
'target' => $code,
'keys' => $keys,
'status' => $response['status'],
'payload' => $response['payload'],
]));
}
}
}
$pools = $register->get('pools');
$queue = $pools
->get('queue')
->pop()
->getResource()
;
$connection = new Queue\Connection\Redis(fn() => $queue);
$adapter = new Queue\Adapter\Swoole($connection, 2, 'syncOut');
$server = new Queue\Server($adapter);
$server->job()
->inject('message')
->inject('dbForConsole')
->action(function (Message $message, Database $dbForConsole) use (&$keys, &$counter) {
$payload = $message->getPayload()['value'];
$regions = Config::getParam('regions', true);
$regions = array_filter(
$regions,
fn ($region) => App::getEnv('_APP_REGION', 'nyc1') !== $region
&& $region !== 'default',
ARRAY_FILTER_USE_KEY
);
if (!empty($payload['region'])) {
$regions = array_filter(
$regions,
fn ($region) => $payload['region'] === $region,
ARRAY_FILTER_USE_KEY
);
}
if (!empty($payload['chunk'])) {
call($dbForConsole, $regions, $payload['chunk']);
return;
}
$keys[$payload['key']] = null;
// if (count($keys) >= MAX_KEY_COUNT || ($counter + SUBMITION_INTERVAL) < time()) {
// call($dbForConsole, $regions, array_keys($keys));
// $counter = time();
// $keys = [];
// }
});
Runtime::enableCoroutine(SWOOLE_HOOK_ALL);
$server
->error()
->inject('error')
->action(function ($error) {
echo $error->getMessage() . PHP_EOL;
echo $error->getLine() . PHP_EOL;
});
$server
->workerStart(function () {
Timer::tick(1000, function () {
var_dump(date('m/d/Y H:i:s', time()));
});
echo "Out region [" . App::getEnv('_APP_REGION', 'nyc1') . "] cache purging worker Started" . PHP_EOL;
})
->start();