Lagdo\DbAdmin\Driver\PgSql\Db\Query

Complexity

Total Complexity

25

Size/Duplication

Total Lines	140
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

11 Methods

Rating	Name	Duplication	Size	Complexity
A	slowQuery()	0	5	1
B	insertOrUpdate()	0	22	8
A	error()	0	12	2
A	limitToOne()	0	5	3
A	explain()	0	3	1
A	user()	0	3	1
A	view()	0	12	1
A	countRows()	0	9	3
A	convertSearch()	0	5	3
A	schema()	0	3	1
A	lastAutoIncrementId()	0	3	1

<?php

namespace Lagdo\DbAdmin\Driver\PgSql\Db;

use Lagdo\DbAdmin\Driver\Entity\TableFieldEntity;
use Lagdo\DbAdmin\Driver\Entity\TableEntity;

use Lagdo\DbAdmin\Driver\Db\ConnectionInterface;

use Lagdo\DbAdmin\Driver\Db\Query as AbstractQuery;

use function strtoupper;

class Query extends AbstractQuery
{
    /**
     * @inheritDoc
     */
    protected function limitToOne(string $table, string $query, string $where, string $separator = "\n")
    {
        return (preg_match('~^INTO~', $query) ? $this->driver->limit($query, $where, 1, 0, $separator) :
            " $query" . ($this->driver->isView($this->driver->tableStatusOrName($table)) ? $where :
            " WHERE ctid = (SELECT ctid FROM " . $this->driver->table($table) . $where . $separator . "LIMIT 1)")
        );
    }

    /**
     * @inheritDoc
     */
    public function insertOrUpdate(string $table, array $rows, array $primary)
    {
        foreach ($rows as $set) {
            $update = [];
            $where = [];
            foreach ($set as $key => $val) {
                $update[] = "$key = $val";
                if (isset($primary[$this->driver->unescapeId($key)])) {
                    $where[] = "$key = $val";
                }
            }
            if (!(
                ($where && $this->driver->execute("UPDATE " . $this->driver->table($table) .
                " SET " . implode(", ", $update) . " WHERE " . implode(" AND ", $where)) &&
                $this->driver->affectedRows()) ||
                $this->driver->execute("INSERT INTO " . $this->driver->table($table) .
                " (" . implode(", ", array_keys($set)) . ") VALUES (" . implode(", ", $set) . ")")
            )) {
                return false;
            }
        }
        return true;
    }

    /**
     * @inheritDoc
     */
    public function lastAutoIncrementId()
    {
        return '0'; // there can be several sequences
    }

    /**
     * @inheritDoc
     */
    public function convertSearch(string $idf, array $val, TableFieldEntity $field)
    {
        return (preg_match('~char|text' . (!preg_match('~LIKE~', $val["op"]) ?
            '|date|time(stamp)?|boolean|uuid|' . $this->driver->numberRegex() : '') .
            '~', $field->type) ? $idf : "CAST($idf AS text)"
        );
    }

    /**
     * @inheritDoc
     */
    public function countRows(TableEntity $tableStatus, array $where)
    {
        $query = "EXPLAIN SELECT * FROM " . $this->driver->escapeId($tableStatus->name) .
            ($where ? " WHERE " . implode(" AND ", $where) : "");
        if (preg_match("~ rows=([0-9]+)~", $this->connection->result($query), $regs))
        {
            return $regs[1];
        }
        return 0;
    }

    /**
     * @inheritDoc
     */
    public function view(string $name)
    {
        $status = $this->driver->tableStatus($name);
        $type = strtoupper($status->engine);
        return [
            'name' => $name,
            'type' => $type,
            'materialized' => ($type != 'VIEW'),
            'select' => trim($this->connection->result("SELECT pg_get_viewdef(" .
                $this->connection->result("SELECT oid FROM pg_class WHERE relnamespace = " .
                "(SELECT oid FROM pg_namespace WHERE nspname = current_schema()) AND relname = " .
                $this->driver->quote($name)) . ")"))
        ];
    }

    /**
     * @inheritDoc
     */
    public function slowQuery(string $query, int $timeout)
    {
        // $this->connection->timeout = 1000 * $timeout;
        $this->connection->query("SET statement_timeout = " . (1000 * $timeout));
        return $query;
    }

    /**
     * @inheritDoc
     */
    public function explain(ConnectionInterface $connection, string $query)
    {
        return $connection->query("EXPLAIN $query");
    }

    /**
     * @inheritDoc
     */
    public function user()
    {
        return $this->connection->result("SELECT user");
    }

    /**
     * @inheritDoc
     */
    public function schema()
    {
        return $this->connection->result("SELECT current_schema()");
    }

    /**
     * @inheritDoc
     */
    public function error()
    {
        $message = parent::error();
        if (preg_match('~^(.*\n)?([^\n]*)\n( *)\^(\n.*)?$~s', $message, $match)) {
            $match1 = $match[1] ?? '';
            $match2 = $match[2] ?? '';
            $match3 = $match[3] ?? '';
            $match4 = $match[4] ?? '';
            $message = $match1 . preg_replace('~((?:[^&]|&[^;]*;){' .
                strlen($match3) . '})(.*)~', '\1<b>\2</b>', $match2) . $match4;
        }
        return $this->util->convertEolToHtml($message);
    }
}