public function setDatabase($database)
    {
        if (is_string($database)) {
            $database = $this->db->getDatabase($database);
        }

        $this->database = $database;

        return $this;
    }