1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
<?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();
}
}
|
