245 lines
10 KiB
PHP
245 lines
10 KiB
PHP
<?php
|
|
|
|
namespace Appwrite\Platform\Workers;
|
|
|
|
use Exception;
|
|
use Utopia\CLI\Console;
|
|
use Utopia\Database\Database;
|
|
use Utopia\Database\Document;
|
|
use Utopia\Platform\Action;
|
|
use Utopia\Queue\Message;
|
|
|
|
class Usage extends Action
|
|
{
|
|
protected static array $stats = [];
|
|
protected array $periods = [
|
|
'1h' => 'Y-m-d H:00',
|
|
'1d' => 'Y-m-d 00:00',
|
|
'inf' => '0000-00-00 00:00'
|
|
];
|
|
|
|
const INFINITY_PERIOD = '_inf_';
|
|
|
|
public static function getName(): string
|
|
{
|
|
return 'usage';
|
|
}
|
|
|
|
/**
|
|
* @throws Exception
|
|
*/
|
|
public function __construct()
|
|
{
|
|
|
|
$this
|
|
->desc('Usage worker')
|
|
->inject('message')
|
|
->inject('pools')
|
|
->inject('cache')
|
|
->callback(function ($message, $pools, $cache) {
|
|
$this->action($message, $pools, $cache);
|
|
});
|
|
}
|
|
|
|
/**
|
|
* @param Message $message
|
|
* @param $pools
|
|
* @param $cache
|
|
* @return void
|
|
* @throws \Utopia\Database\Exception
|
|
*/
|
|
public function action(Message $message, $pools, $cache): void
|
|
{
|
|
$payload = $message->getPayload() ?? [];
|
|
|
|
if (empty($payload)) {
|
|
throw new Exception('Missing payload');
|
|
}
|
|
|
|
$payload = $message->getPayload() ?? [];
|
|
$project = new Document($payload['project'] ?? []);
|
|
$projectId = $project->getInternalId();
|
|
foreach ($payload['reduce'] ?? [] as $document) {
|
|
if (empty($document)) {
|
|
continue;
|
|
}
|
|
|
|
$this->reduce(
|
|
database: $project->getAttribute('database'),
|
|
projectInternalId: $project->getInternalId(),
|
|
document: new Document($document),
|
|
metrics: $payload['metrics'],
|
|
pools: $pools,
|
|
cache: $cache
|
|
);
|
|
}
|
|
|
|
self::$stats[$projectId]['database'] = $project->getAttribute('database');
|
|
foreach ($payload['metrics'] ?? [] as $metric) {
|
|
if (!isset(self::$stats[$projectId]['keys'][$metric['key']])) {
|
|
self::$stats[$projectId]['keys'][$metric['key']] = $metric['value'];
|
|
continue;
|
|
}
|
|
self::$stats[$projectId]['keys'][$metric['key']] += $metric['value'];
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* On Documents that tied by relations like functions>deployments>build || documents>collection>database || buckets>files.
|
|
* When we remove a parent document we need to deduct his children aggregation from the project scope.
|
|
|
|
* @param $database
|
|
* @param $projectInternalId
|
|
* @param Document $document
|
|
* @param array $metrics
|
|
* @param $pools
|
|
* @param $cache
|
|
* @return void
|
|
*/
|
|
private function reduce($database, $projectInternalId, Document $document, array &$metrics, $pools, $cache)
|
|
{
|
|
try {
|
|
$dbForProject = new Database(
|
|
$pools
|
|
->get($database)
|
|
->pop()
|
|
->getResource(),
|
|
$cache
|
|
);
|
|
|
|
$dbForProject->setNamespace('_' . $projectInternalId);
|
|
|
|
switch (true) {
|
|
case $document->getCollection() === 'users': // users
|
|
$sessions = count($document->getAttribute(METRIC_SESSIONS, 0));
|
|
if (!empty($sessions)) {
|
|
$metrics[] = [
|
|
'key' => METRIC_SESSIONS,
|
|
'value' => ($sessions * -1),
|
|
];
|
|
}
|
|
break;
|
|
case $document->getCollection() === 'databases': // databases
|
|
$collections = $dbForProject->getDocument('stats', md5(self::INFINITY_PERIOD . str_replace('{databaseInternalId}', $document->getInternalId(), METRIC_DATABASE_ID_COLLECTIONS)));
|
|
$documents = $dbForProject->getDocument('stats', md5(self::INFINITY_PERIOD . str_replace('{databaseInternalId}', $document->getInternalId(), METRIC_DATABASE_ID_DOCUMENTS)));
|
|
if (!empty($collections['value'])) {
|
|
$metrics[] = [
|
|
'key' => METRIC_COLLECTIONS,
|
|
'value' => ($collections['value'] * -1),
|
|
];
|
|
}
|
|
|
|
if (!empty($documents['value'])) {
|
|
$metrics[] = [
|
|
'key' => METRIC_DOCUMENTS,
|
|
'value' => ($documents['value'] * -1),
|
|
];
|
|
}
|
|
break;
|
|
case str_starts_with($document->getCollection(), 'database_') && !str_contains($document->getCollection(), 'collection'): //collections
|
|
$parts = explode('_', $document->getCollection());
|
|
$databaseInternalId = $parts[1] ?? 0;
|
|
$documents = $dbForProject->getDocument('stats', md5(self::INFINITY_PERIOD . str_replace(['{databaseInternalId}', '{collectionInternalId}'], [$databaseInternalId, $document->getInternalId()], METRIC_DATABASE_ID_COLLECTION_ID_DOCUMENTS)));
|
|
|
|
if (!empty($documents['value'])) {
|
|
$metrics[] = [
|
|
'key' => METRIC_DOCUMENTS,
|
|
'value' => ($documents['value'] * -1),
|
|
];
|
|
$metrics[] = [
|
|
'key' => str_replace('{databaseInternalId}', $databaseInternalId, METRIC_DATABASE_ID_DOCUMENTS),
|
|
'value' => ($documents['value'] * -1),
|
|
];
|
|
}
|
|
break;
|
|
|
|
case $document->getCollection() === 'buckets':
|
|
$files = $dbForProject->getDocument('stats', md5(self::INFINITY_PERIOD . str_replace('{bucketInternalId}', $document->getInternalId(), METRIC_BUCKET_ID_FILES)));
|
|
$storage = $dbForProject->getDocument('stats', md5(self::INFINITY_PERIOD . str_replace('{bucketInternalId}', $document->getInternalId(), METRIC_BUCKET_ID_FILES_STORAGE)));
|
|
|
|
if (!empty($files['value'])) {
|
|
$metrics[] = [
|
|
'key' => METRIC_FILES,
|
|
'value' => ($files['value'] * -1),
|
|
];
|
|
}
|
|
|
|
if (!empty($storage['value'])) {
|
|
$metrics[] = [
|
|
'key' => METRIC_FILES_STORAGE,
|
|
'value' => ($storage['value'] * -1),
|
|
];
|
|
}
|
|
break;
|
|
|
|
case $document->getCollection() === 'functions':
|
|
$deployments = $dbForProject->getDocument('stats', md5(self::INFINITY_PERIOD . str_replace(['{resourceType}', '{resourceInternalId}'], ['functions', $document->getInternalId()], METRIC_FUNCTION_ID_DEPLOYMENTS)));
|
|
$deploymentsStorage = $dbForProject->getDocument('stats', md5(self::INFINITY_PERIOD . str_replace(['{resourceType}', '{resourceInternalId}'], ['functions', $document->getInternalId()], METRIC_FUNCTION_ID_DEPLOYMENTS_STORAGE)));
|
|
$builds = $dbForProject->getDocument('stats', md5(self::INFINITY_PERIOD . str_replace('{functionInternalId}', $document->getInternalId(), METRIC_FUNCTION_ID_BUILDS)));
|
|
$buildsStorage = $dbForProject->getDocument('stats', md5(self::INFINITY_PERIOD . str_replace('{functionInternalId}', $document->getInternalId(), METRIC_FUNCTION_ID_BUILDS_STORAGE)));
|
|
$buildsCompute = $dbForProject->getDocument('stats', md5(self::INFINITY_PERIOD . str_replace('{functionInternalId}', $document->getInternalId(), METRIC_FUNCTION_ID_BUILDS_COMPUTE)));
|
|
$executions = $dbForProject->getDocument('stats', md5(self::INFINITY_PERIOD . str_replace('{functionInternalId}', $document->getInternalId(), METRIC_FUNCTION_ID_EXECUTIONS)));
|
|
$executionsCompute = $dbForProject->getDocument('stats', md5(self::INFINITY_PERIOD . str_replace('{functionInternalId}', $document->getInternalId(), METRIC_FUNCTION_ID_EXECUTIONS_COMPUTE)));
|
|
|
|
if (!empty($deployments['value'])) {
|
|
$metrics[] = [
|
|
'key' => METRIC_DEPLOYMENTS,
|
|
'value' => ($deployments['value'] * -1),
|
|
];
|
|
}
|
|
|
|
if (!empty($deploymentsStorage['value'])) {
|
|
$metrics[] = [
|
|
'key' => METRIC_DEPLOYMENTS_STORAGE,
|
|
'value' => ($deploymentsStorage['value'] * -1),
|
|
];
|
|
}
|
|
|
|
if (!empty($builds['value'])) {
|
|
$metrics[] = [
|
|
'key' => METRIC_BUILDS,
|
|
'value' => ($builds['value'] * -1),
|
|
];
|
|
}
|
|
|
|
if (!empty($buildsStorage['value'])) {
|
|
$metrics[] = [
|
|
'key' => METRIC_BUILDS_STORAGE,
|
|
'value' => ($buildsStorage['value'] * -1),
|
|
];
|
|
}
|
|
|
|
if (!empty($buildsCompute['value'])) {
|
|
$metrics[] = [
|
|
'key' => METRIC_BUILDS_COMPUTE,
|
|
'value' => ($buildsCompute['value'] * -1),
|
|
];
|
|
}
|
|
|
|
if (!empty($executions['value'])) {
|
|
$metrics[] = [
|
|
'key' => METRIC_EXECUTIONS,
|
|
'value' => ($executions['value'] * -1),
|
|
];
|
|
}
|
|
|
|
if (!empty($executionsCompute['value'])) {
|
|
$metrics[] = [
|
|
'key' => METRIC_EXECUTIONS_COMPUTE,
|
|
'value' => ($executionsCompute['value'] * -1),
|
|
];
|
|
}
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
} catch (\Exception $e) {
|
|
console::error("[reducer] " . " {DateTime::now()} " . " {$projectInternalId} " . " {$e->getMessage()}");
|
|
} catch (\Throwable $e) {
|
|
} finally {
|
|
$pools->reclaim();
|
|
}
|
|
}
|
|
}
|