Merge b47ecc8 into 82756d6

SimpleBus · Jan 3, 2017 · 2f3e527 · 2f3e527
2 parents 82756d6 + b47ecc8
commit 2f3e527
Showing 1 changed file with 41 additions and 5 deletions.
diff --git a/src/Routing/ClassBasedRoutingKeyResolver.php b/src/Routing/ClassBasedRoutingKeyResolver.php
@@ -4,12 +4,48 @@
 
 class ClassBasedRoutingKeyResolver implements RoutingKeyResolver
 {
+    /**
+     * @var string
+     */
+    private $prefix;
+
+    /**
+     * @var string
+     */
+    private $separator;
+
+    /**
+     * @var string
+     */
+    private $suffix;
+
+    /**
+     * @param string $separator
+     * @param string $prefix
+     * @param string $suffix
+     */
+    public function __construct($separator = '.', $prefix = null, $suffix = null)
+    {
+        $this->separator = $separator;
+        if ($prefix !== null) {
+            $this->prefix = $prefix;
+        }
+        if ($suffix !== null) {
+            $this->suffix = $suffix;
+        }
+    }
+
     public function resolveRoutingKeyFor($message)
     {
-        return str_replace(
-            '\\',
-            '.',
-            get_class($message)
-        );
+        $components = explode('\\', is_object($message) ? get_class($message) : $message);
+
+        if(isset($this->prefix)){
+            array_unshift($components, $this->prefix);
+        }
+        if(isset($this->suffix)){
+            $components[] = $this->suffix;
+        }
+
+        return implode($this->separator, $components);
     }
 }