-
Notifications
You must be signed in to change notification settings - Fork 1
/
ServiceLoader.php
133 lines (105 loc) · 3.59 KB
/
ServiceLoader.php
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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
<?php
namespace Cmobi\MicroserviceFrameworkBundle;
use Cmobi\MicroserviceFrameworkBundle\Listener\RpcServerListener;
use Symfony\Component\DependencyInjection\ContainerAwareTrait;
use Symfony\Component\Process\Process;
class ServiceLoader
{
use ContainerAwareTrait;
private $micName;
private $processManager;
public function __construct($microserviceName, ProcessManager $processManager)
{
$this->micName = $microserviceName;
$this->processManager = $processManager;
}
public function run()
{
$jobs = [];
$this->getContainer()->get('event_dispatcher')->dispatch('microservice.start');
/** @var \SplObjectStorage $processList */
$processList = $this->getContainer()->get('cmobi_msf.process.manager')->getProcessList();
foreach ($processList as $process) {
if ($process instanceof Process) {
//$serviceName = $this->extractServiceName($process->getCommandLine());
$jobs[] = $process;
}
}
return $jobs;
}
public function stop()
{
$getProcess = new Process(sprintf("ps a | grep %s | grep -v grep | awk {'print $1'}", $this->micName));
$getProcess->run(function ($type, $buffer) {
if (Process::ERR === $type) {
throw new \Exception('Failed list process with error: ' . $buffer);
} else {
$pids = implode(' ', explode(PHP_EOL, $buffer));
$killProcess = new Process(sprintf('kill -9 %s', $pids));
$killProcess->run(function ($type, $buffer) use ($pids) {
if (Process::ERR === $type) {
throw new \Exception(sprintf(
'Failed kill process with pids [%s] and with error [%s]',
$pids,
$buffer
));
}
});
}
});
}
public function status($serviceName = null)
{
$services = [];
/** @var RpcServerListener $rpc */
$rpc = $this->getContainer()->get('cmobi_msf.rpc_server_listener');
array_merge($services, $rpc);
$workers = $this->getContainer()->get('cmobi_msf.worker_listener');
array_merge($services, $workers);
$subscribers = $this->getContainer()->get('cmobi_msf.subscriber_listener');
array_merge($services, $subscribers);
if (! is_null($serviceName)) {
if (! in_array($serviceName, $services)) {
throw new \Exception(sprintf('Service [%s] not found.', $serviceName));
}
}
}
public function listServices()
{
$listProcess = new Process(sprintf(
"ps a | grep %s | grep -v grep | awk {'print $1\" \"$8'}",
$this->micName
));
$listProcess->start();
while($listProcess->isRunning()) {
continue;
}
return $listProcess->getOutput();
}
/**
* @param $commandLine
* @return string
*/
public function extractServiceName($commandLine)
{
$command = explode(' ', $commandLine);
$job = $command[3];
$pos = strrpos($job, '_');
$serviceName = substr($job, 0, $pos);
return $serviceName;
}
/**
* @return ProcessManager
*/
public function getProcessManager()
{
return $this->processManager;
}
/**
* @return \Symfony\Component\DependencyInjection\ContainerInterface
*/
public function getContainer()
{
return $this->container;
}
}