Merge pull request #14 from mobility-work/feature/custom-repository-method

Add ability to set repository method on listeners

Author: makasim

DependencyInjection/Configuration.php

@@ -34,7 +34,9 @@ public function getConfigTreeBuilder()
                                     ->scalarNode('index_name')->isRequired()->cannotBeEmpty()->end()
                                     ->scalarNode('type_name')->isRequired()->cannotBeEmpty()->end()
                                     ->scalarNode('model_class')->isRequired()->cannotBeEmpty()->end()
-                                    ->scalarNode('model_id')->defaultValue('id')->cannotBeEmpty()->end()->end()
+                                    ->scalarNode('model_id')->defaultValue('id')->cannotBeEmpty()->end()
+                                    ->scalarNode('repository_method')->defaultValue('find')->cannotBeEmpty()->end()
+                    ->end()
         ;
 
         return $tb;

Doctrine/Queue/SyncIndexWithObjectChangeProcessor.php

@@ -59,24 +59,28 @@ public function process(PsrMessage $message, PsrContext $context)
             return Result::reject('The message data misses id');
         }
         if (false == isset($data['index_name'])) {
-            return Result::reject('The message data misses id');
+            return Result::reject('The message data misses index_name');
         }
         if (false == isset($data['type_name'])) {
-            return Result::reject('The message data misses id');
+            return Result::reject('The message data misses type_name');
+        }
+        if (false == isset($data['repository_method'])) {
+            return Result::reject('The message data misses repository_method');
         }
 
         $action = $data['action'];
         $modelClass = $data['model_class'];
         $id = $data['id'];
         $index = $data['index_name'];
         $type = $data['type_name'];
+        $repositoryMethod = $data['repository_method'];
 
         $repository = $this->doctrine->getManagerForClass($modelClass)->getRepository($modelClass);
         $persister = $this->persisterRegistry->getPersister($index, $type);
 
         switch ($action) {
             case self::UPDATE_ACTION:
-                if (false == $object = $repository->find($id)) {
+                if (false == $object = $repository->{$repositoryMethod}($id)) {
                     $persister->deleteById($id);
 
                     return Result::ack(sprintf('The object "%s" with id "%s" could not be found.', $modelClass, $id));
@@ -92,7 +96,7 @@ public function process(PsrMessage $message, PsrContext $context)
 
                 return self::ACK;
             case self::INSERT_ACTION:
-                if (false == $object = $repository->find($id)) {
+                if (false == $object = $repository->{$repositoryMethod}($id)) {
                     $persister->deleteById($id);
 
                     return Result::ack(sprintf('The object "%s" with id "%s" could not be found.', $modelClass, $id));

Doctrine/SyncIndexWithObjectChangeListener.php

@@ -85,6 +85,7 @@ private function sendUpdateIndexMessage($action, LifecycleEventArgs $args)
             'id' => $id,
             'index_name' => $this->config['index_name'],
             'type_name' => $this->config['type_name'],
+            'repository_method' => $this->config['repository_method'],
         ]));
 
         $this->context->createProducer()->send($queue, $message);