ByJG\AnyDataset\Database\MongoDbDriver

Complexity

Total Complexity

21

Size/Duplication

Total Lines	200
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	2

Importance

Changes

0

8 Methods

Rating	Name	Duplication	Size	Complexity
B	__construct()	0	22	4
A	__destruct()	0	5	1
A	getCollection()	0	4	1
A	getDbConnection()	0	4	1
A	getIterator()	0	13	3
A	insert()	0	10	3
A	setCollection()	0	5	1
C	update()	0	26	7

<?php

namespace ByJG\AnyDataset\Database;

use ByJG\AnyDataset\ConnectionManagement;
use ByJG\AnyDataset\Repository\ArrayDatasetIterator;
use InvalidArgumentException;
use MongoClient;
use MongoCollection;
use MongoDate;
use MongoDB;
use stdClass;

class MongoDbDriver implements NoSqlDriverInterface
{

    /**
     * @var MongoDB
     */
    protected $dataset = null;

    /**
     *
     * @var MongoClient;
     */
    protected $mongoClient = null;

    /**
     * Enter description here...
     *
     * @var ConnectionManagement
     */
    protected $connectionManagement;

    /**
     *
     * @var MongoCollection MongoDB collection
     */
    protected $collection;

    /**
     *
     * @var string
     */
    protected $collectionName;

    /**
     * Creates a new MongoDB connection. This class is managed from NoSqlDataset
     *
     * @param ConnectionManagement $connMngt
     * @param string $collection
     */
    public function __construct($connMngt, $collection)
    {
        $this->connectionManagement = $connMngt;

        $hosts = $this->connectionManagement->getServer();
        $port = $this->connectionManagement->getPort() == '' ? 27017 : $this->connectionManagement->getPort();
        $database = $this->connectionManagement->getDatabase();
        $username = $this->connectionManagement->getUsername();
        $password = $this->connectionManagement->getPassword();

        if ($username != '' && $password != '') {
            $auth = array('username' => $username, 'password' => $password, 'connect' => 'true');
        } else {
            $auth = array('connect' => 'true');
        }

        $connectString = sprintf('mongodb://%s:%d', $hosts, $port);
        $this->mongoClient = new MongoClient($connectString, $auth);
        $this->dataset = new MongoDB($this->mongoClient, $database);

        $this->setCollection($collection);
    }

    /**
     * Closes and destruct the MongoDB connection
     */
    public function __destruct()
    {
        $this->mongoClient->close();
        $this->dataset = null;
    }

    /**
     *
     * @return string
     */
    public function getCollection()
    {
        return $this->collectionName;
    }

    /**
     * Gets the instance of MongoDB; You do not need uses this directly.
     * If you have to, probably something is missing in this class
     * @return \MongoDB
     */
    public function getDbConnection()
    {
        return $this->dataset;
    }

    /**
     * Return a IteratorInterface
     *
     * @param array $filter
     * @param array $fields
     * @return ArrayDatasetIterator
     */
    public function getIterator($filter = null, $fields = null)
    {
        if (is_null($filter)) {
            $filter = array();
        }
        if (is_null($fields)) {
            $fields = array();
        }
        $cursor = $this->collection->find($filter, $fields);
        $arrIt = iterator_to_array($cursor);

        return new ArrayDatasetIterator($arrIt);
    }

    /**
     * Insert a document in the MongoDB
     * @param mixed $document
     * @return bool
     */
    public function insert($document)
    {
        if (is_array($document)) {
            $document['created_at'] = new MongoDate();
        } elseif ($document instanceof stdClass) {
            $document->created_at = new MongoDate();
        }

        return $this->collection->insert($document);
    }

    /**
     * Defines the new Collection
     * @param string $collection
     */
    public function setCollection($collection)
    {
        $this->collection = $this->dataset->selectCollection($collection);
        $this->collectionName = $collection;
    }

    /**
     * Update a document based on your criteria
     *
     * Options for MongoDB is an array of:
     *
     * sort array   Determines which document the operation will modify if the
     *              query selects multiple documents. findAndModify will modify
     *              the first document in the sort order specified by this argument.
     * remove boolean Optional if update field exists. When TRUE, removes the
     *              selected document. The default is FALSE.
     * update array Optional if remove field exists. Performs an update of the
     *              selected document.
     * new boolean  Optional. When TRUE, returns the modified document rather than the original.
     *              The findAndModify method ignores the new option for remove operations.
     *              The default is FALSE.
     * upsert boolean ptional. Used in conjunction with the update field. When TRUE,
     *              the findAndModify command creates a new document if the query
     *              returns no documents. The default is false. In MongoDB 2.2, the findAndModify
     *              command returns NULL when upsert is TRUE.
     *
     * @param mixed $document
     * @param array $filter
     * @param array $options See:
     * @return bool
     */
    public function update($document, $filter = null, $options = null)
    {
        if (is_null($filter)) {
            throw new InvalidArgumentException('You need to set the filter for update, or pass an empty array for all fields');
        }

        $update = array();
        if (is_array($document)) {
            $document['updated_at'] = new MongoDate();
        }
        if ($document instanceof stdClass) {
            $document->updated_at = new MongoDate();
        }
        foreach ($document as $key => $value) {

The expression $document of type object|integer|double|string|null|boolean|array<string,object<MongoDate>,{"updated_at":"object<MongoDate>"}> is not guaranteed to be traversable. How about adding an additional type check?

            if ($key[0] == '$') {
                $update[$key] = $value;
            } else {
                $update['$set'][$key] = $value;
            }
        }

        if (is_null($options)) {
            $options = array('new' => true);
        }
        return $this->collection->findAndModify($filter, $update, array(), $options);
    }
    /*

58% of this comment could be valid code. Did you maybe forget this after debugging?

      public function getAttribute($name)
      {
      $this->_db->getAttribute($name);
      }

      public function setAttribute($name, $value)
      {
      $this->_db->setAttribute ( $name, $value );
      }
     */
}