cvweiss\projectbase\MongoDoc

Complexity

Total Complexity

17

Size/Duplication

Total Lines	98
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	1

Importance

Changes

0

12 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	2
A	get()	0	4	1
A	getAll()	0	4	1
A	set()	0	6	1
A	setAll()	0	7	2
A	save()	0	7	2
A	getBulkWriter()	0	4	1
A	doCommit()	0	8	2
A	insert()	0	10	1
A	update()	0	12	2
A	delete()	0	9	1
A	__set()	0	4	1

<?php

namespace cvweiss\projectbase;

use MongoDB\Driver\BulkWrite;

class MongoDoc
{
    private $collection = null;
    private $data = null;
    private $updates = [];

    public function __construct(string $collection, array $row = null)
    {
        $this->collection = $collection;
        $this->data = $row === null ? [] : $row;
    }

    public function get(string $field)
    {
        return $this->data[$field] ?? null;
    }

    public function getAll()
    {
        return $this->data;
    }

    public function set(string $field, $value):MongoDoc
    {
        $this->data[$field] = $value;
        $this->updates[$field] = $value;
        return $this;
    }

    public function setAll(array $params):MongoDoc
    {
        foreach ($params as $key=>$value) {
            $this->set($key, $value);
        }
        return $this;
    }

    public function save(BulkWrite $bulk = null):bool
    {
        $return = isset($this->data['_id']) ? $this->update($bulk) : $this->insert($bulk);
        $this->updates = [];

        return $return;
    }

    protected function getBulkWriter(BulkWrite $bulk = null):BulkWrite
    {
        return $bulk ?? new BulkWrite(['ordered' => true]);
    }

    protected function doCommit(BulkWrite $bulk, bool $doCommit):bool
    {
        if ($doCommit !== false) {
            $return = Mongo::get()->executeBulkWrite($this->collection, $bulk);
            return  (count($return->getWriteErrors()) == 0);
        }
        return false;
    }

    protected function insert(BulkWrite $bulk = null):bool
    {
        $commit = $bulk === null;
        $bulk = $this->getBulkWriter($bulk);

        $id = $bulk->insert($this->data);
        $this->data['_id'] = $id;

        return $this->doCommit($bulk, $commit);
    }

    protected function update(BulkWrite $bulk = null):bool
    {   
        // If we have nothing to update then move along
        if (sizeof($this->updates) == 0) return true;

        $commit = $bulk === null;
        $bulk = $this->getBulkWriter($bulk);

        $bulk->update(['_id' => $this->data['_id']], ['$set' => $this->updates]);

        return $this->doCommit($bulk, $commit);
    }

    public function delete(BulkWrite $bulk = null):bool
    {
        $commit = $bulk === null;
        $bulk = $this->getBulkWriter($bulk);

        $bulk->delete(['_id' => $this->data['_id']]);

        return $this->doCommit($bulk, $commit);
    }

    public function __set($foo, $bar)
    {
        throw new \Exception("unable to set $foo to $bar; use set(\$field, \$value)");
    }
}