lakiboy · November 28, 2015 12:53 · Nov 28, 2015
diff --git a/DelegatingProducer.php b/DelegatingProducer.php
@@ -0,0 +1,31 @@
+<?php
+
+namespace Cinemania\CoreBundle\SimpleBus;
+
+use OldSound\RabbitMqBundle\RabbitMq\Fallback;
+use OldSound\RabbitMqBundle\RabbitMq\ProducerInterface;
+use Symfony\Component\DependencyInjection\ContainerInterface;
+
+class DelegatingProducer extends Fallback
+{
+    private $container;
+
+    public function __construct(ContainerInterface $container)
+    {
+        $this->container = $container;
+    }
+
+    public function publish($msg, $routingKey = '', $props = array())
+    {
+        if (!isset($props['producer'])) {
+            throw new \RuntimeException(sprintf('No "producer" key defined in %s', __METHOD__));
+        }
+
+        /** @var ProducerInterface $producer */
+        $producer = $this->container->get(sprintf('old_sound_rabbit_mq.%s_producer', $props['producer']));
+
+        unset($props['producer']);
+
+        return $producer->publish($msg, $routingKey, $props);
+    }
+}
diff --git a/DelegatingProducerTest.php b/DelegatingProducerTest.php
@@ -0,0 +1,51 @@
+<?php
+
+namespace Cinemania\CoreBundle\Tests\SimpleBus;
+
+use Cinemania\CoreBundle\SimpleBus\DelegatingProducer;
+
+class DelegatingProducerTest extends \PHPUnit_Framework_TestCase
+{
+    /** @var \PHPUnit_Framework_MockObject_MockObject */
+    private $container;
+
+    /** @var DelegatingProducer */
+    private $producer;
+
+    public function setUp()
+    {
+        $this->container = $this->getMock('Symfony\Component\DependencyInjection\ContainerInterface');
+        $this->producer = new DelegatingProducer($this->container);
+    }
+
+    /**
+     * @expectedException \RuntimeException
+     */
+    public function testPublishFailsWithoutProducer()
+    {
+        $this->producer->publish('string', '');
+    }
+
+    public function testPublish()
+    {
+        $message = 'message';
+        $routing = '';
+
+        // Actual producer called.
+        $producer = $this->getMock('OldSound\RabbitMqBundle\RabbitMq\ProducerInterface');
+        $producer
+            ->expects($this->once())
+            ->method('publish')
+            ->with($this->equalTo($message), $this->equalTo($routing), $this->equalTo([]))
+        ;
+
+        $this->container
+            ->expects($this->once())
+            ->method('get')
+            ->with($this->equalTo('old_sound_rabbit_mq.foo_bar_producer'))
+            ->willReturn($producer)
+        ;
+
+        $this->producer->publish($message, $routing, ['producer' => 'foo_bar']);
+    }
+}
diff --git a/ProducerNameResolver.php b/ProducerNameResolver.php
@@ -0,0 +1,33 @@
+<?php
+
+namespace Cinemania\CoreBundle\SimpleBus;
+
+use SimpleBus\Asynchronous\Properties\AdditionalPropertiesResolver;
+
+class ProducerNameResolver implements AdditionalPropertiesResolver
+{
+    public function resolveAdditionalPropertiesFor($message)
+    {
+        $name = self::tableize(substr(get_class($message), strrpos(get_class($message), '\\') + 1));
+
+        if (substr($name, -8) === '_command') {
+            $name = substr($name, 0, strlen($name) - 8);
+        } elseif (substr($name, -6) === '_event') {
+            $name = substr($name, 0, strlen($name) - 6);
+        }
+
+        return ['producer' => $name];
+    }
+
+    /**
+     * Taken from Doctrine\Common\Inflector.
+     *
+     * @param string $word
+     *
+     * @return string
+     */
+    private static function tableize($word)
+    {
+        return strtolower(preg_replace('~(?<=\\w)([A-Z])~', '_$1', $word));
+    }
+}
diff --git a/ProducerNameResolverTest.php b/ProducerNameResolverTest.php
@@ -0,0 +1,47 @@
+<?php
+
+namespace Cinemania\CoreBundle\Tests\SimpleBus;
+
+use Cinemania\CoreBundle\SimpleBus\ProducerNameResolver;
+
+class ProducerNameResolverTest extends \PHPUnit_Framework_TestCase
+{
+    /**
+     * @dataProvider getData
+     *
+     * @param object $message
+     * @param string $producer
+     */
+    public function testResolveAdditionalPropertiesFor($message, $producer)
+    {
+        $props = (new ProducerNameResolver())->resolveAdditionalPropertiesFor($message);
+
+        $this->assertEquals(['producer' => $producer], $props);
+    }
+
+    public function getData()
+    {
+        return [
+            [new FooBar(), 'foo_bar'],
+            [new FooBarCommand(), 'foo_bar'],
+            [new FooBarEvent(), 'foo_bar'],
+            [new FooBarCommandEvent(), 'foo_bar_command'],
+        ];
+    }
+}
+
+class FooBar
+{
+}
+
+class FooBarCommand
+{
+}
+
+class FooBarEvent
+{
+}
+
+class FooBarCommandEvent
+{
+}