Feature model postgresql (#842)
* Support postgresql generate * Update template Var * Support to generate postgresql model * Support to generate postgresql model * Update template Co-authored-by: anqiansong <anqiansong@xiaoheiban.cn>master
anqiansong
3 years ago
committed by
GitHub
parent
476026e393
commit
089cdaa75f
@ -0,0 +1,234 @@
|
|||||||
|
package model
|
||||||
|
|
||||||
|
import (
|
||||||
|
"database/sql"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/tal-tech/go-zero/core/stores/sqlx"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
p2m = map[string]string{
|
||||||
|
"int8": "bigint",
|
||||||
|
"numeric": "bigint",
|
||||||
|
"float8": "double",
|
||||||
|
"float4": "float",
|
||||||
|
"int2": "smallint",
|
||||||
|
"int4": "integer",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
// PostgreSqlModel gets table information from information_schema、pg_catalog
|
||||||
|
type PostgreSqlModel struct {
|
||||||
|
conn sqlx.SqlConn
|
||||||
|
}
|
||||||
|
|
||||||
|
// PostgreColumn describes a column in table
|
||||||
|
type PostgreColumn struct {
|
||||||
|
Num sql.NullInt32 `db:"num"`
|
||||||
|
Field sql.NullString `db:"field"`
|
||||||
|
Type sql.NullString `db:"type"`
|
||||||
|
NotNull sql.NullBool `db:"not_null"`
|
||||||
|
Comment sql.NullString `db:"comment"`
|
||||||
|
ColumnDefault sql.NullString `db:"column_default"`
|
||||||
|
IdentityIncrement sql.NullInt32 `db:"identity_increment"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// PostgreIndex describes an index for a column
|
||||||
|
type PostgreIndex struct {
|
||||||
|
IndexName sql.NullString `db:"index_name"`
|
||||||
|
IndexId sql.NullInt32 `db:"index_id"`
|
||||||
|
IsUnique sql.NullBool `db:"is_unique"`
|
||||||
|
IsPrimary sql.NullBool `db:"is_primary"`
|
||||||
|
ColumnName sql.NullString `db:"column_name"`
|
||||||
|
IndexSort sql.NullInt32 `db:"index_sort"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewPostgreSqlModel creates an instance and return
|
||||||
|
func NewPostgreSqlModel(conn sqlx.SqlConn) *PostgreSqlModel {
|
||||||
|
return &PostgreSqlModel{
|
||||||
|
conn: conn,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetAllTables selects all tables from TABLE_SCHEMA
|
||||||
|
func (m *PostgreSqlModel) GetAllTables(schema string) ([]string, error) {
|
||||||
|
query := `select table_name from information_schema.tables where table_schema = $1`
|
||||||
|
var tables []string
|
||||||
|
err := m.conn.QueryRows(&tables, query, schema)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return tables, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// FindColumns return columns in specified database and table
|
||||||
|
func (m *PostgreSqlModel) FindColumns(schema, table string) (*ColumnData, error) {
|
||||||
|
querySql := `select t.num,t.field,t.type,t.not_null,t.comment, c.column_default, identity_increment
|
||||||
|
from (
|
||||||
|
SELECT a.attnum AS num,
|
||||||
|
c.relname,
|
||||||
|
a.attname AS field,
|
||||||
|
t.typname AS type,
|
||||||
|
a.atttypmod AS lengthvar,
|
||||||
|
a.attnotnull AS not_null,
|
||||||
|
b.description AS comment
|
||||||
|
FROM pg_class c,
|
||||||
|
pg_attribute a
|
||||||
|
LEFT OUTER JOIN pg_description b ON a.attrelid = b.objoid AND a.attnum = b.objsubid,
|
||||||
|
pg_type t
|
||||||
|
WHERE c.relname = $1
|
||||||
|
and a.attnum > 0
|
||||||
|
and a.attrelid = c.oid
|
||||||
|
and a.atttypid = t.oid
|
||||||
|
ORDER BY a.attnum) AS t
|
||||||
|
left join information_schema.columns AS c on t.relname = c.table_name
|
||||||
|
and t.field = c.column_name and c.table_schema = $2`
|
||||||
|
|
||||||
|
var reply []*PostgreColumn
|
||||||
|
err := m.conn.QueryRowsPartial(&reply, querySql, table, schema)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
list, err := m.getColumns(schema, table, reply)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var columnData ColumnData
|
||||||
|
columnData.Db = schema
|
||||||
|
columnData.Table = table
|
||||||
|
columnData.Columns = list
|
||||||
|
return &columnData, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *PostgreSqlModel) getColumns(schema, table string, in []*PostgreColumn) ([]*Column, error) {
|
||||||
|
index, err := m.getIndex(schema, table)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
var list []*Column
|
||||||
|
for _, e := range in {
|
||||||
|
var dft interface{}
|
||||||
|
if len(e.ColumnDefault.String) > 0 {
|
||||||
|
dft = e.ColumnDefault
|
||||||
|
}
|
||||||
|
|
||||||
|
isNullAble := "YES"
|
||||||
|
if e.NotNull.Bool {
|
||||||
|
isNullAble = "NO"
|
||||||
|
}
|
||||||
|
|
||||||
|
extra := "auto_increment"
|
||||||
|
if e.IdentityIncrement.Int32 != 1 {
|
||||||
|
extra = ""
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(index[e.Field.String]) > 0 {
|
||||||
|
for _, i := range index[e.Field.String] {
|
||||||
|
list = append(list, &Column{
|
||||||
|
DbColumn: &DbColumn{
|
||||||
|
Name: e.Field.String,
|
||||||
|
DataType: m.convertPostgreSqlTypeIntoMysqlType(e.Type.String),
|
||||||
|
Extra: extra,
|
||||||
|
Comment: e.Comment.String,
|
||||||
|
ColumnDefault: dft,
|
||||||
|
IsNullAble: isNullAble,
|
||||||
|
OrdinalPosition: int(e.Num.Int32),
|
||||||
|
},
|
||||||
|
Index: i,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
list = append(list, &Column{
|
||||||
|
DbColumn: &DbColumn{
|
||||||
|
Name: e.Field.String,
|
||||||
|
DataType: m.convertPostgreSqlTypeIntoMysqlType(e.Type.String),
|
||||||
|
Extra: extra,
|
||||||
|
Comment: e.Comment.String,
|
||||||
|
ColumnDefault: dft,
|
||||||
|
IsNullAble: isNullAble,
|
||||||
|
OrdinalPosition: int(e.Num.Int32),
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return list, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *PostgreSqlModel) convertPostgreSqlTypeIntoMysqlType(in string) string {
|
||||||
|
r, ok := p2m[strings.ToLower(in)]
|
||||||
|
if ok {
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
return in
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *PostgreSqlModel) getIndex(schema, table string) (map[string][]*DbIndex, error) {
|
||||||
|
indexes, err := m.FindIndex(schema, table)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
var index = make(map[string][]*DbIndex)
|
||||||
|
for _, e := range indexes {
|
||||||
|
if e.IsPrimary.Bool {
|
||||||
|
index[e.ColumnName.String] = append(index[e.ColumnName.String], &DbIndex{
|
||||||
|
IndexName: indexPri,
|
||||||
|
SeqInIndex: int(e.IndexSort.Int32),
|
||||||
|
})
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
nonUnique := 0
|
||||||
|
if !e.IsUnique.Bool {
|
||||||
|
nonUnique = 1
|
||||||
|
}
|
||||||
|
|
||||||
|
index[e.ColumnName.String] = append(index[e.ColumnName.String], &DbIndex{
|
||||||
|
IndexName: e.IndexName.String,
|
||||||
|
NonUnique: nonUnique,
|
||||||
|
SeqInIndex: int(e.IndexSort.Int32),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return index, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// FindIndex finds index with given schema, table and column.
|
||||||
|
func (m *PostgreSqlModel) FindIndex(schema, table string) ([]*PostgreIndex, error) {
|
||||||
|
querySql := `select A.INDEXNAME AS index_name,
|
||||||
|
C.INDEXRELID AS index_id,
|
||||||
|
C.INDISUNIQUE AS is_unique,
|
||||||
|
C.INDISPRIMARY AS is_primary,
|
||||||
|
G.ATTNAME AS column_name,
|
||||||
|
G.attnum AS index_sort
|
||||||
|
from PG_AM B
|
||||||
|
left join PG_CLASS F on
|
||||||
|
B.OID = F.RELAM
|
||||||
|
left join PG_STAT_ALL_INDEXES E on
|
||||||
|
F.OID = E.INDEXRELID
|
||||||
|
left join PG_INDEX C on
|
||||||
|
E.INDEXRELID = C.INDEXRELID
|
||||||
|
left outer join PG_DESCRIPTION D on
|
||||||
|
C.INDEXRELID = D.OBJOID,
|
||||||
|
PG_INDEXES A,
|
||||||
|
pg_attribute G
|
||||||
|
where A.SCHEMANAME = E.SCHEMANAME
|
||||||
|
and A.TABLENAME = E.RELNAME
|
||||||
|
and A.INDEXNAME = E.INDEXRELNAME
|
||||||
|
and F.oid = G.attrelid
|
||||||
|
and E.SCHEMANAME = $1
|
||||||
|
and E.RELNAME = $2
|
||||||
|
order by C.INDEXRELID,G.attnum`
|
||||||
|
|
||||||
|
var reply []*PostgreIndex
|
||||||
|
err := m.conn.QueryRowsPartial(&reply, querySql, schema, table)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return reply, nil
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue