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
105
106
|
<?php
/* Icinga DB Web | (c) 2024 Icinga GmbH | GPLv2 */
namespace Icinga\Module\Icingadb\Data;
use DateTime;
use DateTimeZone;
use Icinga\Module\Icingadb\Model\Host;
use Icinga\Module\Icingadb\Model\Service;
use ipl\Orm\Model;
use ipl\Orm\Query;
trait CsvResultSetUtils
{
/**
* @return array<string, ?string>
*/
public function current(): array
{
return $this->extractKeysAndValues(parent::current());
}
protected function formatValue(string $key, $value): ?string
{
if (
$value
&& (
$key === 'id'
|| substr($key, -3) === '_id'
|| substr($key, -3) === '.id'
|| substr($key, -9) === '_checksum'
|| substr($key, -4) === '_bin'
)
) {
$value = bin2hex($value);
}
if (is_bool($value)) {
return $value ? 'true' : 'false';
} elseif (is_string($value)) {
return '"' . str_replace('"', '""', $value) . '"';
} elseif (is_array($value)) {
return '"' . implode(',', $value) . '"';
} elseif ($value instanceof DateTime) {
return $value->setTimezone(new DateTimeZone('UTC'))
->format('Y-m-d\TH:i:s.vP');
} else {
return $value;
}
}
protected function extractKeysAndValues(Model $model, string $path = ''): array
{
$keysAndValues = [];
foreach ($model as $key => $value) {
$keyPath = ($path ? $path . '.' : '') . $key;
if ($value instanceof Model) {
$keysAndValues += $this->extractKeysAndValues($value, $keyPath);
} else {
$keysAndValues[$keyPath] = $this->formatValue($key, $value);
}
}
return $keysAndValues;
}
public static function stream(Query $query): void
{
if ($query->getModel() instanceof Host || $query->getModel() instanceof Service) {
$query->setResultSetClass(VolatileCsvResults::class);
} else {
$query->setResultSetClass(__CLASS__);
}
if ($query->hasLimit()) {
// Custom limits should still apply
$query->peekAhead(false);
$offset = $query->getOffset();
} else {
$query->limit(1000);
$query->peekAhead();
$offset = 0;
}
do {
$query->offset($offset);
$result = $query->execute()->disableCache();
foreach ($result as $i => $keysAndValues) {
if ($i === 0) {
echo implode(',', array_keys($keysAndValues));
}
echo "\r\n";
echo implode(',', array_values($keysAndValues));
JsonResultSet::giveMeMoreTime();
}
$offset += 1000;
} while ($result->hasMore());
exit;
}
}
|
