new worker

This commit is contained in:
shimon
2022-10-20 18:07:02 +03:00
parent df8e95a0a1
commit 45e8314b49
10 changed files with 355 additions and 192 deletions
+19 -35
View File
@@ -2,15 +2,16 @@
use Ahc\Jwt\JWT;
use Ahc\Jwt\JWTException;
use Appwrite\Event\Delete;
use Appwrite\Event\SyncIn;
use Appwrite\Extend\Exception;
use Appwrite\Utopia\Request;
use Appwrite\Utopia\Response;
use Utopia\App;
use Utopia\Registry\Registry;
use Utopia\Validator\ArrayList;
use Utopia\Validator\Text;
use Utopia\Validator\WhiteList;
use Utopia\Queue\Client;
use Utopia\Queue\Connection\Redis;
App::post('/v1/edge')
->desc('Purge cache keys')
@@ -18,11 +19,12 @@ App::post('/v1/edge')
->param('keys', '', new ArrayList(new Text(100), 1000), 'Cache keys')
->inject('request')
->inject('response')
->action(function (array $keys, Request $request, Response $response) {
->inject('register')
->action(function (array $keys, Request $request, Response $response, Registry $register) {
if (empty($keys)) {
//if (empty($keys)) {
throw new Exception(Exception::KEY_NOT_FOUND);
}
//}
$token = $request->getHeader('authorization');
$token = str_replace(["Bearer"," "], "", $token);
@@ -33,40 +35,22 @@ App::post('/v1/edge')
throw new Exception(Exception::USER_JWT_INVALID, 'Failed to verify JWT. ' . $error->getMessage());
}
$syncIn = new SyncIn();
$connection = $register
->get('workerRedisConnection');
$client = new Client('syncIn', $connection);
$client->resetStats();
foreach ($keys as $key) {
$syncIn
->addKey($key)
->trigger();
$client->enqueue([
'type' => 'from endpoint',
'value' => [
'key' => $key
]
]);
}
$response
->setStatusCode(Response::STATUS_CODE_OK)
->send();
});
App::post('/v1/edge/notify')
->desc('Flush notification')
->label('scope', 'public')
->param('region', '', new WhiteList(['nyc1', 'blr1', 'fra1']), 'Cloud regions')
->inject('request')
->inject('response')
->action(function (string $region, Request $request, Response $response) {
$token = $request->getHeader('authorization');
$token = str_replace(["Bearer"," "], "", $token);
$jwt = new JWT(App::getEnv('_APP_OPENSSL_KEY_V1'), 'HS256', 600, 10);
try {
$payload = $jwt->decode($token);
} catch (JWTException $error) {
throw new Exception(Exception::USER_JWT_INVALID, 'Failed to verify JWT. ' . $error->getMessage());
}
(new Delete())
->setRegion($region)
->trigger();
$response
->setStatusCode(Response::STATUS_CODE_OK)
->send();
});