1
|
|
|
<?php |
2
|
|
|
/** |
3
|
|
|
* Created by PhpStorm. |
4
|
|
|
* User: Petrica |
5
|
|
|
* Date: 5/28/2016 |
6
|
|
|
* Time: 23:53 |
7
|
|
|
*/ |
8
|
|
|
namespace Petrica\StatsdSystem\Gauge; |
9
|
|
|
|
10
|
|
|
use Petrica\StatsdSystem\Collection\ValuesCollection; |
11
|
|
|
use Petrica\StatsdSystem\Model\Process\Process; |
12
|
|
|
use Petrica\StatsdSystem\Model\Process\TopProcessParser; |
13
|
|
|
use Petrica\StatsdSystem\Model\TopCommand; |
14
|
|
|
|
15
|
|
|
class ProcessesGauge implements GaugeInterface |
16
|
|
|
{ |
17
|
|
|
protected $cpuAbove; |
18
|
|
|
|
19
|
|
|
protected $memoryAbove; |
20
|
|
|
|
21
|
|
|
/** |
22
|
|
|
* @var TopCommand |
23
|
|
|
*/ |
24
|
|
|
protected $command; |
25
|
|
|
|
26
|
|
|
/** |
27
|
|
|
* ProcessesGauge constructor. |
28
|
|
|
* |
29
|
|
|
* Track CPU and memory of processes |
30
|
|
|
* |
31
|
|
|
* @param float $cpuAbove |
32
|
|
|
* @param float $memoryAbove |
33
|
|
|
*/ |
34
|
1 |
|
public function __construct($cpuAbove = 5.0, $memoryAbove = 1.0) |
35
|
|
|
{ |
36
|
1 |
|
$this->cpuAbove = $cpuAbove; |
37
|
1 |
|
$this->memoryAbove = $memoryAbove; |
38
|
|
|
|
39
|
1 |
|
$this->command = new TopCommand(); |
40
|
1 |
|
} |
41
|
|
|
|
42
|
|
|
/** |
43
|
|
|
* {@inheritdoc} |
44
|
|
|
*/ |
45
|
|
|
public function getSamplingPeriod() |
46
|
|
|
{ |
47
|
|
|
return 10; |
48
|
|
|
} |
49
|
|
|
|
50
|
|
|
/** |
51
|
|
|
* {@inheritdoc} |
52
|
|
|
*/ |
53
|
1 |
|
public function getCollection() |
54
|
|
|
{ |
55
|
1 |
|
$processes = $this->getProcesses(); |
56
|
|
|
|
57
|
1 |
|
$cpu = $this->aggregateCpu($processes); |
58
|
1 |
|
$cpu = $this->filterAbove($cpu, $this->cpuAbove); |
59
|
|
|
|
60
|
1 |
|
$memory = $this->aggregateMemory($processes); |
61
|
1 |
|
$memory = $this->filterAbove($memory, $this->memoryAbove); |
62
|
|
|
|
63
|
1 |
|
$collection = new ValuesCollection(); |
64
|
|
|
|
65
|
1 |
|
foreach ($cpu as $name => $value) { |
66
|
1 |
|
$collection->add($name . '.cpu.value', $value); |
67
|
1 |
|
} |
68
|
|
|
|
69
|
1 |
|
foreach ($memory as $name => $value) { |
70
|
1 |
|
$collection->add($name . '.memory.value', $value); |
71
|
1 |
|
} |
72
|
|
|
|
73
|
1 |
|
return $collection; |
74
|
|
|
} |
75
|
|
|
|
76
|
|
|
/** |
77
|
|
|
* Return parsed data for processes |
78
|
|
|
*/ |
79
|
1 |
|
protected function getProcesses() |
80
|
|
|
{ |
81
|
1 |
|
$data = $this->getCommand()->run(); |
82
|
|
|
|
83
|
1 |
|
$parser = new TopProcessParser($data); |
84
|
1 |
|
$parser->parse(); |
85
|
|
|
|
86
|
1 |
|
return $parser->getProcesses(); |
87
|
|
|
} |
88
|
|
|
|
89
|
|
|
/** |
90
|
|
|
* Return only those processes above certail value |
91
|
|
|
* |
92
|
|
|
* @param $processes Process[] |
93
|
|
|
* @return array |
94
|
|
|
*/ |
95
|
|
|
protected function filterAbove($processes, $gate) |
96
|
|
|
{ |
97
|
1 |
|
return array_filter($processes, function ($item) use ($gate) { |
98
|
1 |
|
return $item >= $gate; |
99
|
1 |
|
}); |
100
|
|
|
} |
101
|
|
|
|
102
|
|
|
/** |
103
|
|
|
* Aggregate CPU for processes with the same name |
104
|
|
|
* |
105
|
|
|
* @param $processes Process[] |
106
|
|
|
*/ |
107
|
1 |
View Code Duplication |
protected function aggregateCpu($processes) |
|
|
|
|
108
|
|
|
{ |
109
|
1 |
|
$cpus = array(); |
110
|
|
|
|
111
|
1 |
|
foreach ($processes as $process) { |
112
|
1 |
|
if (isset($cpus[$process->getName()])) { |
113
|
1 |
|
$cpus[$process->getName()] += $process->getCpu(); |
114
|
1 |
|
} |
115
|
|
|
else { |
116
|
1 |
|
$cpus[$process->getName()] = $process->getCpu(); |
117
|
|
|
} |
118
|
1 |
|
} |
119
|
|
|
|
120
|
1 |
|
return $cpus; |
121
|
|
|
} |
122
|
|
|
|
123
|
|
|
/** |
124
|
|
|
* Aggregate memory for processes with the same name |
125
|
|
|
* |
126
|
|
|
* @param $processes Process[] |
127
|
|
|
* @return array |
128
|
|
|
*/ |
129
|
1 |
View Code Duplication |
protected function aggregateMemory($processes) |
|
|
|
|
130
|
|
|
{ |
131
|
1 |
|
$memory = array(); |
132
|
|
|
|
133
|
1 |
|
foreach ($processes as $process) { |
134
|
1 |
|
if (isset($memory[$process->getName()])) { |
135
|
1 |
|
$memory[$process->getName()] += $process->getMemory(); |
136
|
1 |
|
} |
137
|
|
|
else { |
138
|
1 |
|
$memory[$process->getName()] = $process->getMemory(); |
139
|
|
|
} |
140
|
1 |
|
} |
141
|
|
|
|
142
|
1 |
|
return $memory; |
143
|
|
|
} |
144
|
|
|
|
145
|
|
|
/** |
146
|
|
|
* Return command object for top utility |
147
|
|
|
* |
148
|
|
|
* @return TopCommand |
149
|
|
|
*/ |
150
|
|
|
protected function getCommand() |
151
|
|
|
{ |
152
|
|
|
return $this->command; |
153
|
|
|
} |
154
|
|
|
|
155
|
|
|
} |
Duplicated code is one of the most pungent code smells. If you need to duplicate the same code in three or more different places, we strongly encourage you to look into extracting the code into a single class or operation.
You can also find more detailed suggestions in the “Code” section of your repository.