Soleo\UrlShortener\MongoConnection

Complexity

Total Complexity

13

Size/Duplication

Total Lines	94
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	3

Test Coverage

Coverage

75.47%

Importance

Changes	5
Bugs	1	Features	4

6 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	6	1
A	lookup()	0	18	3
A	reverseLookup()	0	23	3
A	addNewRecord()	0	8	1
B	getIncrementUid()	0	22	4
A	cleanDB()	0	5	1

<?php

namespace Soleo\UrlShortener;

class MongoConnection implements ConnectionInterface
{

    private $client;
    private $DB;

    public function __construct($config)
    {
        $options = array("connectTimeoutMS" => 30000);
        $this->client = new \MongoDB\Client($config, $options);
        $this->DB = $this->client->selectDatabase("url_shortener");
    }

    public function lookup($slug, $update = false)
    {
        $c_db = $this->DB;
        $c_table = $c_db->selectCollection("url_table");
        $c_table->createIndex(array('shorturl' => 1 ), array('unique' => true));
        $urls = $c_table->findOne(array('shorturl' => $slug), array('longurl', '_id'));
        if ($urls) {
            if ($update) {
                $c_table->createIndex(array('hits' => 1 ));
                $c_table->update(array("_id" => $urls['_id']), array('$inc' => array("hits" => 1)));
                $days = floor(time()/24/3600);
                $c_table->update(array("_id" => $urls['_id']), array('$inc' => array("hits_d.{$days}" => 1)));
            }
            return $urls['longurl'];
        } else {
            return false;
        }
    }

    public function reverseLookup($longUrl)
    {
        /*
         * mongodb
         * {
         *   shorturl
         *   longurl
         * }
         */

        $c_db = $this->DB;

        // check if url already exists
        $c_table = $c_db->selectCollection("url_table");
        $c_table->createIndex(array('longurl' => 1 ), array('unique' => true));
        $c_table->createIndex(array('shorturl' => 1 ), array('unique' => true));
        $shorturl = $c_table->findOne(array('longurl' => $longUrl), array('shorturl'));
        if ($shorturl && $shorturl['shorturl']) {
            return $shorturl['shorturl'];
        } else {
            return false;
        }
    }

    public function addNewRecord($longUrl, $slug)
    {
        $c_db = $this->DB;
        $c_table = $c_db->selectCollection("url_table");

        $c_table->insertOne(array("shorturl" => $slug, "longurl" => $longUrl));
        return $slug;
    }

    public function getIncrementUid()
    {
        // generate increment unique id
        $collection_name = 'increment_id';
        $unique_field =  'section';
        $data_field = 'url_id';

        $c_inc = $this->DB->selectCollection($collection_name);
        $c_inc->createIndex(array($unique_field => 1), array('unique' => true));
        try {
            $ret = $c_inc->findOneAndUpdate(
                [ $unique_field => $data_field ],
                [ '$inc' => [$data_field => 1] ],
                [ "upsert" => true ]
            );
            if ($ret && $ret->$data_field) {
                return $ret->$data_field;
            }
        } catch (\Exception $e) {
            return false;
        }
    }

    public function cleanDB()
    {
        $this->DB->selectCollection("url_table")->drop();
        $this->DB->selectCollection("increment_id")->drop();
    }
}