parse dsn

Author: turboboy88

pkg/mongodb/MongodbConnectionFactory.php

@@ -12,6 +12,22 @@ class MongodbConnectionFactory implements PsrConnectionFactory
      */
     private $config;
 
+    /**
+     * The config could be an array, string DSN or null. In case of null it will attempt to connect to Mongodb localhost with default credentials.
+     *
+     * $config = [
+     *   'uri' => 'mongodb://127.0.0.1/' - Mongodb connection string. see http://docs.mongodb.org/manual/reference/connection-string/
+     *   'dbname' => 'enqueue',          - database name.
+     *   'collection_name' => 'enqueue'  - collection name
+     *   'polling_interval' => '1000',   - How often query for new messages (milliseconds)
+     * ]
+     *
+     * or
+     *
+     * mongodb://127.0.0.1:27017/dbname/collection_name?polling_interval=1000
+     *
+     * @param array|string|null $config
+     */
     public function __construct($config = 'mongodb:')
     {
         if (empty($config)) {
@@ -24,45 +40,76 @@ public function __construct($config = 'mongodb:')
         }
         $config = array_replace([
             'uri' => 'mongodb://127.0.0.1/',
-            'uriOptions' => [],
-            'driverOptions' => [],
         ], $config);
 
         $this->config = $config;
     }
 
     public function createContext()
     {
-        $client = new Client($this->config['uri'], $this->config['uriOptions'], $this->config['driverOptions']);
+        $client = new Client($this->config['uri']);
 
         return new MongodbContext($client, $this->config);
     }
 
-    private function parseDsn($dsn)
+    public static function parseDsn($dsn)
     {
-        if (false === parse_url($dsn)) {
+        $parsedUrl = parse_url($dsn);
+
+        if (false === $parsedUrl) {
             throw new \LogicException(sprintf('Failed to parse DSN "%s"', $dsn));
         }
 
-        $schema = parse_url($dsn, PHP_URL_SCHEME);
-        if (empty($schema)) {
+        if (empty($parsedUrl['scheme'])) {
             throw new \LogicException('Schema is empty');
         }
 
         $supported = [
             'mongodb' => true,
         ];
 
-        if (false == isset($supported[$schema])) {
+        if (false == isset($parsedUrl['scheme'])) {
             throw new \LogicException(sprintf(
                 'The given DSN schema "%s" is not supported. There are supported schemes: "%s".',
-                $schema,
+                $parsedUrl['scheme'],
                 implode('", "', array_keys($supported))
             ));
         }
 
-        return [
-            'uri' => $schema.':' === $dsn ? $schema.'://127.0.0.1/' : $dsn,
-        ];
+        if ($parsedUrl['scheme'].':' === $dsn) {
+            return [
+                'uri' => $parsedUrl['scheme'].'://127.0.0.1/',
+            ];
+        }
+
+        $config['uri'] = $parsedUrl['scheme'].'://'.$parsedUrl['host'];
+
+        if (isset($parsedUrl['path']) && '/' !== $parsedUrl['path']) {
+            $pathArr = explode('/', $parsedUrl['path']);
+
+            //DB name
+            if ($pathArr[1]) {
+                $config['dbname'] = $pathArr[1];
+                $config['uri'] .= '/'.$pathArr[1];
+            }
+
+            //Collection name
+            if ($pathArr[2]) {
+                $config['collection_name'] = $pathArr[2];
+            }
+        }
+
+        if (isset($parsedUrl['query'])) {
+            $config['uri'] .= '?'.$parsedUrl['query'];
+            //get polling_interval value
+            if (false !== strpos($parsedUrl['query'], 'polling_interval')) {
+                preg_match_all('/polling_interval=([0-9]+)/', $parsedUrl['query'], $matches);
+                if (isset($matches[1][0])) {
+                    $config['polling_interval'] = $matches[1][0];
+                }
+            }
+        }
+
+        return $config;
     }
 }