Data\MongoDataTable

Complexity

Total Complexity

26

Size/Duplication

Total Lines	115
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	1

Importance

Changes	1
Bugs	0	Features	0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	1
A	count()	0	16	3
A	getCriteriaFromFilter()	0	12	3
D	read()	0	37	9
A	create()	0	9	3
A	update()	0	14	4
A	delete()	0	10	3

<?php
namespace Data;

class MongoDataTable extends DataTable
{
    protected $collection;
    protected $name;

    public function __construct($collection, $collection_name = false)
    {
        $this->collection = $collection;
        $this->name = $collection_name;
    }

    public function count($filter = false)
    {
        $criteria = array();
        if($filter !== false)
        {
            if($filter instanceof \Data\Filter)
            {
                $criteria = $filter->to_mongo_filter();
            }
            else
            {
                $criteria = $filter;
            }
        }
        return $this->collection->count($criteria, array(), $this->name);
    }

    private function getCriteriaFromFilter($filter)
    {
        if($filter === false)
        {
            return array();
        }
        if(is_array($filter))
        {
            return $filter;
        }
        return $filter->to_mongo_filter();
    }

    public function read($filter = false, $select = false, $count = false, $skip = false, $sort = false, $params = false)
    {
        $fields   = array();
        $criteria = $this->getCriteriaFromFilter($filter);
        if($select !== false)
        {
            $fields = array_fill_keys($select, 1);
        }
        $cursor = $this->collection->find($criteria, $fields, $this->name);
        if($params !== false && isset($params['fields']))
        {
            $cursor->fields($params['fields']);
        }
        if($sort !== false)
        {
            $cursor->sort($sort);
        }
        if($skip !== false)
        {
            $cursor->skip($skip);
        }
        if($count !== false)
        {
            $cursor->limit($count);
        }
        if(method_exists($cursor, 'setTypeMap'))
        {
            $cursor->setTypeMap(['root' => 'array', 'document' => 'array', 'array' => 'array']);
            return $cursor->toArray();
        }
        $ret = array();
        foreach($cursor as $doc)
        {
            array_push($ret, $doc);
        }
        return $ret;
    }

    public function create($data)
    {
        $res = $this->collection->insert($data);
        if($res === false || $res['err'] !== null)
        {
            return false;
        }
        return $data['_id'];
    }

    public function update($filter, $data)
    {
        $criteria = $this->getCriteriaFromFilter($filter);
        if(isset($data['_id']))
        {
            unset($data['_id']);
        }
        $res = $this->collection->update($criteria, array('$set' => $data));
        if($res === false || $res['err'] !== null)
        {
            return false;
        }
        return true;
    }

    public function delete($filter)
    {
        $criteria = $this->getCriteriaFromFilter($filter);
        $res = $this->collection->remove($criteria);
        if($res === false || $res['err'] !== null)
        {
            return false;
        }
        return true;
    }
}