mirror of
https://github.com/nextcloud/server.git
synced 2025-03-13 07:53:51 +00:00
aa5f037af7
Signed-off-by: Joas Schilling <coding@schilljs.com> Signed-off-by: Benjamin Gaussorgues <benjamin.gaussorgues@nextcloud.com>
121 lines
3.2 KiB
PHP
121 lines
3.2 KiB
PHP
<?php
|
|
|
|
declare(strict_types = 1);
|
|
|
|
/**
|
|
* @copyright 2021 Carl Schwan <carl@carlschwan.eu>
|
|
*
|
|
* @author Carl Schwan <carl@carlschwan.eu>
|
|
*
|
|
* @license GNU AGPL version 3 or any later version
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as
|
|
* published by the Free Software Foundation, either version 3 of the
|
|
* License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
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();
|
|
}
|
|
}
|