-
Notifications
You must be signed in to change notification settings - Fork 39
Expand file tree
/
Copy pathElasticaPopulateProcessor.php
More file actions
86 lines (74 loc) · 2.41 KB
/
ElasticaPopulateProcessor.php
File metadata and controls
86 lines (74 loc) · 2.41 KB
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
<?php
namespace Enqueue\ElasticaBundle\Async;
use Enqueue\Client\CommandSubscriberInterface;
use Enqueue\Consumption\QueueSubscriberInterface;
use Enqueue\Consumption\Result;
use Interop\Queue\PsrContext;
use Interop\Queue\PsrMessage;
use Interop\Queue\PsrProcessor;
use Enqueue\Util\JSON;
use FOS\ElasticaBundle\Provider\ProviderRegistry;
class ElasticaPopulateProcessor implements PsrProcessor, CommandSubscriberInterface, QueueSubscriberInterface
{
/**
* @var ProviderRegistry
*/
private $providerRegistry;
/**
* @param ProviderRegistry $providerRegistry
*/
public function __construct(ProviderRegistry $providerRegistry)
{
$this->providerRegistry = $providerRegistry;
}
/**
* {@inheritdoc}
*/
public function process(PsrMessage $message, PsrContext $context)
{
if ($message->isRedelivered()) {
$this->createReplyMessage($context, $message, false);
return Result::reply($this->createReplyMessage($context, $message, false), Result::REJECT);
}
try {
$options = JSON::decode($message->getBody());
$provider = $this->providerRegistry->getProvider($options['indexName'], $options['typeName']);
$provider->populate(null, $options);
return Result::reply($this->createReplyMessage($context, $message, true));
} catch (\Exception $e) {
return Result::reply($this->createReplyMessage($context, $message, false), Result::REJECT);
}
}
/**
* @param PsrContext $context
* @param PsrMessage $message
* @param bool $successful
*
* @return PsrMessage
*/
private function createReplyMessage(PsrContext $context, PsrMessage $message, $successful)
{
$replyMessage = $context->createMessage($message->getBody(), $message->getProperties(), $message->getHeaders());
$replyMessage->setProperty('fos-populate-successful', (int) $successful);
return $replyMessage;
}
/**
* {@inheritdoc}
*/
public static function getSubscribedCommand()
{
return [
'processorName' => Commands::POPULATE,
'queueName' => Commands::POPULATE,
'queueNameHardcoded' => true,
'exclusive' => true,
];
}
/**
* {@inheritdoc}
*/
public static function getSubscribedQueues()
{
return [Commands::POPULATE];
}
}