106 lines
3.2 KiB
Go
106 lines
3.2 KiB
Go
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
package mysql
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"git.magicany.cc/black1552/gin-base/database"
|
|
"github.com/gogf/gf/v2/util/gutil"
|
|
)
|
|
|
|
var (
|
|
// tableFieldsSqlByMariadb is the query statement for retrieving table fields' information in MariaDB.
|
|
// Deprecated: Use package `contrib/drivers/mariadb` instead.
|
|
// TODO remove in next version.
|
|
tableFieldsSqlByMariadb = `
|
|
SELECT
|
|
c.COLUMN_NAME AS 'Field',
|
|
( CASE WHEN ch.CHECK_CLAUSE LIKE 'json_valid%%' THEN 'json' ELSE c.COLUMN_TYPE END ) AS 'Type',
|
|
c.COLLATION_NAME AS 'Collation',
|
|
c.IS_NULLABLE AS 'Null',
|
|
c.COLUMN_KEY AS 'Key',
|
|
( CASE WHEN c.COLUMN_DEFAULT = 'NULL' OR c.COLUMN_DEFAULT IS NULL THEN NULL ELSE c.COLUMN_DEFAULT END) AS 'Default',
|
|
c.EXTRA AS 'Extra',
|
|
c.PRIVILEGES AS 'Privileges',
|
|
c.COLUMN_COMMENT AS 'Comment'
|
|
FROM
|
|
information_schema.COLUMNS AS c
|
|
LEFT JOIN information_schema.CHECK_CONSTRAINTS AS ch ON c.TABLE_NAME = ch.TABLE_NAME
|
|
AND c.TABLE_SCHEMA = ch.CONSTRAINT_SCHEMA
|
|
AND c.COLUMN_NAME = ch.CONSTRAINT_NAME
|
|
WHERE
|
|
c.TABLE_SCHEMA = '%s'
|
|
AND c.TABLE_NAME = '%s'
|
|
ORDER BY c.ORDINAL_POSITION`
|
|
)
|
|
|
|
func init() {
|
|
var err error
|
|
tableFieldsSqlByMariadb, err = database.FormatMultiLineSqlToSingle(tableFieldsSqlByMariadb)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// TableFields retrieves and returns the fields' information of specified table of current
|
|
// schema.
|
|
//
|
|
// The parameter `link` is optional, if given nil it automatically retrieves a raw sql connection
|
|
// as its link to proceed necessary sql query.
|
|
//
|
|
// Note that it returns a map containing the field name and its corresponding fields.
|
|
// As a map is unsorted, the TableField struct has a "Index" field marks its sequence in
|
|
// the fields.
|
|
//
|
|
// It's using cache feature to enhance the performance, which is never expired util the
|
|
// process restarts.
|
|
func (d *Driver) TableFields(ctx context.Context, table string, schema ...string) (fields map[string]*database.TableField, err error) {
|
|
var (
|
|
result database.Result
|
|
link database.Link
|
|
usedSchema = gutil.GetOrDefaultStr(d.GetSchema(), schema...)
|
|
tableFieldsSql string
|
|
)
|
|
if link, err = d.SlaveLink(usedSchema); err != nil {
|
|
return nil, err
|
|
}
|
|
dbType := d.GetConfig().Type
|
|
switch dbType {
|
|
// Deprecated: Use package `contrib/drivers/mariadb` instead.
|
|
// TODO remove in next version.
|
|
case "mariadb":
|
|
tableFieldsSql = fmt.Sprintf(tableFieldsSqlByMariadb, usedSchema, table)
|
|
default:
|
|
tableFieldsSql = fmt.Sprintf(`SHOW FULL COLUMNS FROM %s`, d.QuoteWord(table))
|
|
}
|
|
|
|
result, err = d.DoSelect(
|
|
ctx, link,
|
|
tableFieldsSql,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
fields = make(map[string]*database.TableField)
|
|
for i, m := range result {
|
|
fields[m["Field"].String()] = &database.TableField{
|
|
Index: i,
|
|
Name: m["Field"].String(),
|
|
Type: m["Type"].String(),
|
|
Null: strings.EqualFold(m["Null"].String(), "YES"),
|
|
Key: m["Key"].String(),
|
|
Default: m["Default"].Val(),
|
|
Extra: m["Extra"].String(),
|
|
Comment: m["Comment"].String(),
|
|
}
|
|
}
|
|
return fields, nil
|
|
}
|