mirror of
https://github.com/nextcloud/server.git
synced 2024-11-14 12:26:49 +00:00
105 lines
2.5 KiB
PHP
105 lines
2.5 KiB
PHP
<?php
|
|
|
|
declare(strict_types = 1);
|
|
|
|
/**
|
|
* SPDX-FileCopyrightText: 2021 Nextcloud GmbH and Nextcloud contributors
|
|
* SPDX-License-Identifier: AGPL-3.0-or-later
|
|
*/
|
|
|
|
namespace OC\Profiler;
|
|
|
|
use OC\AppFramework\Http\Request;
|
|
use OC\SystemConfig;
|
|
use OCP\AppFramework\Http\Response;
|
|
use OCP\DataCollector\IDataCollector;
|
|
use OCP\Profiler\IProfile;
|
|
use OCP\Profiler\IProfiler;
|
|
|
|
class Profiler implements IProfiler {
|
|
/** @var array<string, IDataCollector> */
|
|
private array $dataCollectors = [];
|
|
|
|
private ?FileProfilerStorage $storage = null;
|
|
|
|
private bool $enabled = false;
|
|
|
|
public function __construct(SystemConfig $config) {
|
|
$this->enabled = $config->getValue('profiler', false);
|
|
if ($this->enabled) {
|
|
$this->storage = new FileProfilerStorage($config->getValue('datadirectory', \OC::$SERVERROOT . '/data') . '/__profiler');
|
|
}
|
|
}
|
|
|
|
public function add(IDataCollector $dataCollector): void {
|
|
$this->dataCollectors[$dataCollector->getName()] = $dataCollector;
|
|
}
|
|
|
|
public function loadProfileFromResponse(Response $response): ?IProfile {
|
|
if (!$token = $response->getHeaders()['X-Debug-Token']) {
|
|
return null;
|
|
}
|
|
|
|
return $this->loadProfile($token);
|
|
}
|
|
|
|
public function loadProfile(string $token): ?IProfile {
|
|
if ($this->storage) {
|
|
return $this->storage->read($token);
|
|
} else {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
public function saveProfile(IProfile $profile): bool {
|
|
if ($this->storage) {
|
|
return $this->storage->write($profile);
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
public function collect(Request $request, Response $response): IProfile {
|
|
$profile = new Profile($request->getId());
|
|
$profile->setTime(time());
|
|
$profile->setUrl($request->getRequestUri());
|
|
$profile->setMethod($request->getMethod());
|
|
$profile->setStatusCode($response->getStatus());
|
|
foreach ($this->dataCollectors as $dataCollector) {
|
|
$dataCollector->collect($request, $response, null);
|
|
|
|
// We clone for subrequests
|
|
$profile->addCollector(clone $dataCollector);
|
|
}
|
|
return $profile;
|
|
}
|
|
|
|
/**
|
|
* @return array[]
|
|
*/
|
|
public function find(?string $url, ?int $limit, ?string $method, ?int $start, ?int $end,
|
|
?string $statusCode = null): array {
|
|
if ($this->storage) {
|
|
return $this->storage->find($url, $limit, $method, $start, $end, $statusCode);
|
|
} else {
|
|
return [];
|
|
}
|
|
}
|
|
|
|
public function dataProviders(): array {
|
|
return array_keys($this->dataCollectors);
|
|
}
|
|
|
|
public function isEnabled(): bool {
|
|
return $this->enabled;
|
|
}
|
|
|
|
public function setEnabled(bool $enabled): void {
|
|
$this->enabled = $enabled;
|
|
}
|
|
|
|
public function clear(): void {
|
|
$this->storage->purge();
|
|
}
|
|
}
|