Your IP : 216.73.216.86
<?php
namespace Bitrix\Perfmon;
class PgsqlDatabase extends BaseDatabase
{
public function getTables($full = true)
{
$tables = [];
if ($full)
{
$tableList = $this->connection->query("
select
table_name
,pg_table_size(quote_ident(table_name)) data_length
,pg_indexes_size(quote_ident(table_name)) index_length
,pg_catalog.pg_class.reltuples row_count
from
information_schema.tables
left join pg_catalog.pg_class on pg_catalog.pg_class.oid = quote_ident(table_name)::regclass::oid
where
table_schema = 'public'
");
while ($table = $tableList->fetch())
{
$tables[] = [
'TABLE_NAME' => $table['TABLE_NAME'],
'ENGINE_TYPE' => '',
'NUM_ROWS' => $table['ROW_COUNT'],
'BYTES' => $table['DATA_LENGTH'],
'BYTES_INDEX' => $table['INDEX_LENGTH'],
];
}
}
else
{
$tableList = $this->connection->query("
select
table_name
from
information_schema.tables
where
table_schema = 'public'
");
while ($table = $tableList->fetch())
{
$tables[] = [
'TABLE_NAME' => $table['TABLE_NAME'],
'ENGINE_TYPE' => '',
'NUM_ROWS' => '',
'BYTES' => '',
'BYTES_INDEX' => '',
];
}
}
$result = new \CDBResult();
$result->InitFromArray($tables);
return $result;
}
public function getIndexes($tableName = false)
{
$sqlHelper = $this->connection->getSqlHelper();
$tableColumns = [];
$columnList = $this->connection->query("
SELECT a.attnum, a.attname
FROM pg_class t
LEFT JOIN pg_attribute a ON a.attrelid = t.oid
WHERE t.relname = '" . $sqlHelper->forSql($tableName) . "'
");
while ($column = $columnList->fetch())
{
if ($column['ATTNUM'] > 0)
{
$tableColumns[$column['ATTNUM']] = $column['ATTNAME'];
}
}
$indexList = $this->connection->query("
SELECT relname, indkey, pg_get_expr(pg_index.indexprs, pg_index.indrelid) full_text
FROM pg_class, pg_index
WHERE pg_class.oid = pg_index.indexrelid
AND pg_class.oid IN (
SELECT indexrelid
FROM pg_index, pg_class
WHERE pg_class.relname = '" . $sqlHelper->forSql($tableName) . "'
AND pg_class.oid = pg_index.indrelid
)
");
$result = [];
while ($indexColumn = $indexList->fetch())
{
$result[$indexColumn['RELNAME']] = [];
if ($indexColumn['FULL_TEXT'])
{
$match = [];
if (preg_match_all('/,\s*([a-z0-9_]+)/i', $indexColumn['FULL_TEXT'], $match))
{
foreach ($match[1] as $i => $colName)
{
$result[$indexColumn['RELNAME']][$i] = mb_strtoupper($colName);
}
}
}
else
{
foreach (explode(' ', $indexColumn['INDKEY']) as $i => $indkey)
{
$result[$indexColumn['RELNAME']][$i] = mb_strtoupper($tableColumns[$indkey]);
}
}
}
return $result;
}
public function getUniqueIndexes($tableName = false)
{
$sqlHelper = $this->connection->getSqlHelper();
$tableColumns = [];
$columnList = $this->connection->query("
SELECT a.attnum, a.attname
FROM pg_class t
LEFT JOIN pg_attribute a ON a.attrelid = t.oid
WHERE t.relname = '" . $sqlHelper->forSql($tableName) . "'
");
while ($column = $columnList->fetch())
{
if ($column['ATTNUM'] > 0)
{
$tableColumns[$column['ATTNUM']] = $column['ATTNAME'];
}
}
$indexList = $this->connection->query("
SELECT relname, indkey, pg_get_expr(pg_index.indexprs, pg_index.indrelid) full_text
FROM pg_class, pg_index
WHERE pg_class.oid = pg_index.indexrelid
AND pg_class.oid IN (
SELECT indexrelid
FROM pg_index, pg_class
WHERE pg_class.relname = '" . $sqlHelper->forSql($tableName) . "'
AND pg_class.oid = pg_index.indrelid
)
AND (indisprimary OR indisunique)
ORDER BY indisprimary desc
");
$result = [];
while ($indexColumn = $indexList->fetch())
{
$result[$indexColumn['RELNAME']] = [];
if ($indexColumn['FULL_TEXT'])
{
$match = [];
if (preg_match_all('/,\s*([a-z0-9_]+)/i', $indexColumn['FULL_TEXT'], $match))
{
foreach ($match[1] as $i => $colName)
{
$result[$indexColumn['RELNAME']][$i] = mb_strtoupper($colName);
}
}
}
else
{
foreach (explode(' ', $indexColumn['INDKEY']) as $i => $indkey)
{
$result[$indexColumn['RELNAME']][$i] = mb_strtoupper($tableColumns[$indkey]);
}
}
}
return $result;
}
public function getTableFields($tableName = false)
{
$sqlHelper = $this->connection->getSqlHelper();
$strSql = "
SELECT *
FROM information_schema.columns
WHERE table_schema = 'public'
AND table_name = '" . $sqlHelper->forSql($tableName) . "'
";
$columnList = $this->connection->query($strSql);
$result = [];
$resultExt = [];
while ($column = $columnList->fetch())
{
$canSort = true;
$match = [];
switch ($column['DATA_TYPE'])
{
case 'character varying':
$DATA_TYPE = 'string';
$ORM_DATA_TYPE = 'string';
break;
case 'character':
$DATA_TYPE = 'string';
if (
$column['CHARACTER_MAXIMUM_LENGTH'] == 1
&& (
substr($column['COLUMN_DEFAULT'], 0, 3) === "'N'"
|| substr($column['COLUMN_DEFAULT'], 0, 3) === "'Y'"
)
)
{
$column['COLUMN_DEFAULT'] = $column['COLUMN_DEFAULT'][1];
$ORM_DATA_TYPE = 'boolean';
}
else
{
$ORM_DATA_TYPE = 'string';
}
break;
case 'text':
case 'bytea':
$canSort = false;
$DATA_TYPE = 'string';
$ORM_DATA_TYPE = 'string';
break;
case 'bigint':
case 'bigserial':
case 'int':
case 'int2':
case 'int4':
case 'int8':
case 'integer':
case 'serial':
case 'serial2':
case 'serial4':
case 'serial8':
case 'smallint':
case 'smallserial':
$DATA_TYPE = 'int';
$ORM_DATA_TYPE = 'integer';
break;
case 'double precision':
case 'float4':
case 'float8':
case 'numeric':
case 'real':
$DATA_TYPE = 'double';
$ORM_DATA_TYPE = 'float';
break;
case 'timestamp without time zone':
$DATA_TYPE = 'datetime';
$ORM_DATA_TYPE = 'datetime';
break;
case 'date':
$DATA_TYPE = 'date';
$ORM_DATA_TYPE = 'date';
break;
default:
$canSort = false;
$DATA_TYPE = 'unknown';
$ORM_DATA_TYPE = 'UNKNOWN';
break;
}
$result[mb_strtoupper($column['COLUMN_NAME'])] = $DATA_TYPE;
$resultExt[mb_strtoupper($column['COLUMN_NAME'])] = [
'type' => $DATA_TYPE,
'length' => $column['CHARACTER_MAXIMUM_LENGTH'],
'nullable' => $column['IS_NULLABLE'] !== 'NO',
'default' => preg_match('/^\'(.*)\'::/', $column['COLUMN_DEFAULT'], $match) ? $match[1] : $column['COLUMN_DEFAULT'],
'sortable' => $canSort,
'orm_type' => $ORM_DATA_TYPE,
'increment' => strpos($column['COLUMN_DEFAULT'], 'nextval(') !== false || $column['IS_IDENTITY'] === 'YES',
'info' => $column,
];
}
return [$result, $resultExt];
}
}