Nip\Records\Traits\HasDatabase\HasDatabaseRecordsTrait

Complexity

Total Complexity

9

Size/Duplication

Total Lines	62
Duplicated Lines	0 %

Test Coverage

Coverage

90.48%

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	setDB()	0	5	1
A	initDB()	0	3	1
A	hasDB()	0	3	1
A	checkDB()	0	4	2
A	newDbConnection()	0	6	2
A	getDB()	0	9	2

<?php

namespace Nip\Records\Traits\HasDatabase;

use Nip\Container\Container;
use Nip\Database\Connections\Connection;

/**
 * Trait HasDatabaseRecordsTrait
 * @package Nip\Records\Traits\HasDatabase
 */
trait HasDatabaseRecordsTrait
{

    /**
     * @var Connection
     */
    protected $connection = null;

    /**
     * @return Connection
     */
    public function getDB()
    {
        if ($this->connection == null) {
            $this->initDB();
        }

        $this->checkDB();

        return $this->connection;
    }

    protected function initDB()
    {
        $this->setDB($this->newDbConnection());
    }

    /**
     * @param Connection $connection
     * @return $this
     */
    public function setDB($connection)
    {
        $this->connection = $connection;

        return $this;
    }

    /**
     * @return Connection
     */
    protected function newDbConnection()
    {
        if (function_exists('app')) {
            return app('db.connection');
        }
        return Container::getInstance()->get('db.connection');
    }

    public function checkDB()
    {
        if (!$this->hasDB()) {
            trigger_error("Database connection missing for [" . get_class($this) . "]", E_USER_ERROR);
        }
    }

    /**
     * @return bool
     */
    public function hasDB()
    {
        return $this->connection instanceof Connection;
    }
}