123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104 |
- <?php
- declare(strict_types = 1);
- 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 {
-
- 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);
-
- $profile->addCollector(clone $dataCollector);
- }
- return $profile;
- }
-
- 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();
- }
- }
|