Gleipnir/jet
4
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Faster MySQL GetTableColumnsMetaData query

Browse source
This commit is contained in:
Yosyp Buchma 2023-09-18 17:35:11 +03:00
parent e2fe799f8e
commit db808f136b
1 changed files with 29 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

47
generator/mysql/query_set.go
View file

@ -39,27 +39,36 @@ ORDER BY table_name;
func (m mySqlQuerySet) GetTableColumnsMetaData(db *sql.DB, schemaName string, tableName string) ([]metadata.Column, error) { func (m mySqlQuerySet) GetTableColumnsMetaData(db *sql.DB, schemaName string, tableName string) ([]metadata.Column, error) {
query := ` query := `
SELECT COLUMN_NAME AS "column.Name", SELECT
IS_NULLABLE = "YES" AS "column.IsNullable", col.COLUMN_NAME AS "column.Name",
columns.COLUMN_COMMENT as "column.Comment", col.IS_NULLABLE = "YES" AS "column.IsNullable",
(EXISTS( col.COLUMN_COMMENT AS "column.Comment",
SELECT 1 pk.IsPrimaryKey AS "column.IsPrimaryKey",
FROM information_schema.table_constraints t IF (col.COLUMN_TYPE = 'tinyint(1)',
JOIN information_schema.key_column_usage k USING(constraint_name,table_schema,table_name)
WHERE table_schema = ? AND table_name = ? AND t.constraint_type='PRIMARY KEY' AND k.column_name = columns.column_name
)) AS "column.IsPrimaryKey",
IF (COLUMN_TYPE = 'tinyint(1)',
'boolean', 'boolean',
IF (DATA_TYPE='enum', IF (col.DATA_TYPE = 'enum',
CONCAT(TABLE_NAME, '_', COLUMN_NAME), CONCAT(col.TABLE_NAME, '_', col.COLUMN_NAME),
DATA_TYPE) col.DATA_TYPE)
) AS "dataType.Name", ) AS "dataType.Name",
IF (DATA_TYPE = 'enum', 'enum', 'base') AS "dataType.Kind", IF (col.DATA_TYPE = 'enum', 'enum', 'base') AS "dataType.Kind",
COLUMN_TYPE LIKE '%unsigned%' AS "dataType.IsUnsigned" col.COLUMN_TYPE LIKE '%unsigned%' AS "dataType.IsUnsigned"
FROM information_schema.columns FROM
WHERE table_schema = ? AND table_name = ? information_schema.columns AS col
ORDER BY ordinal_position; LEFT JOIN (
` SELECT k.column_name, 1 AS IsPrimaryKey
FROM information_schema.table_constraints t
JOIN information_schema.key_column_usage k USING(constraint_name, table_schema, table_name)
WHERE t.table_schema = ?
AND t.table_name = ?
AND t.constraint_type = 'PRIMARY KEY'
) AS pk ON col.COLUMN_NAME = pk.column_name
WHERE
col.table_schema = ?
AND col.table_name = ?
ORDER BY
col.ordinal_position;
`
var columns []metadata.Column var columns []metadata.Column
_, err := qrm.Query(context.Background(), db, query, []interface{}{schemaName, tableName, schemaName, tableName}, &columns) _, err := qrm.Query(context.Background(), db, query, []interface{}{schemaName, tableName, schemaName, tableName}, &columns)
if err != nil { if err != nil {