pqsql-uuid
This commit is contained in:
砼砼
parent
021ad08402
commit
af328951fb
|
|
@ -44,7 +44,7 @@ func (db *postgres) SqlType(c *core.Column) string {
|
||||||
return "timestamp with time zone"
|
return "timestamp with time zone"
|
||||||
case core.Float:
|
case core.Float:
|
||||||
res = core.Real
|
res = core.Real
|
||||||
case core.TinyText, core.MediumText, core.LongText:
|
case core.TinyText, core.MediumText, core.LongText, core.Uuid:
|
||||||
res = core.Text
|
res = core.Text
|
||||||
case core.Blob, core.TinyBlob, core.MediumBlob, core.LongBlob:
|
case core.Blob, core.TinyBlob, core.MediumBlob, core.LongBlob:
|
||||||
return core.Bytea
|
return core.Bytea
|
||||||
|
|
@ -142,86 +142,96 @@ func (db *postgres) IsColumnExist(tableName string, col *core.Column) (bool, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (db *postgres) GetColumns(tableName string) ([]string, map[string]*core.Column, error) {
|
func (db *postgres) GetColumns(tableName string) ([]string, map[string]*core.Column, error) {
|
||||||
args := []interface{}{tableName}
|
args := []interface{}{tableName}
|
||||||
s := "SELECT column_name, column_default, is_nullable, data_type, character_maximum_length" +
|
s := `SELECT column_name, column_default, is_nullable, data_type, character_maximum_length, numeric_precision, numeric_precision_radix ,
|
||||||
", numeric_precision, numeric_precision_radix FROM INFORMATION_SCHEMA.COLUMNS WHERE table_name = $1"
|
CASE WHEN p.contype = 'p' THEN true ELSE false END AS primarykey,
|
||||||
|
CASE WHEN p.contype = 'u' THEN true ELSE false END AS uniquekey
|
||||||
|
FROM pg_attribute f
|
||||||
|
JOIN pg_class c ON c.oid = f.attrelid JOIN pg_type t ON t.oid = f.atttypid
|
||||||
|
LEFT JOIN pg_attrdef d ON d.adrelid = c.oid AND d.adnum = f.attnum
|
||||||
|
LEFT JOIN pg_namespace n ON n.oid = c.relnamespace
|
||||||
|
LEFT JOIN pg_constraint p ON p.conrelid = c.oid AND f.attnum = ANY (p.conkey)
|
||||||
|
LEFT JOIN pg_class AS g ON p.confrelid = g.oid
|
||||||
|
LEFT JOIN INFORMATION_SCHEMA.COLUMNS s ON s.column_name=f.attname AND c.relname=s.table_name
|
||||||
|
WHERE c.relkind = 'r'::char AND c.relname = $1 AND f.attnum > 0 ORDER BY f.attnum;`
|
||||||
|
|
||||||
rows, err := db.DB().Query(s, args...)
|
rows, err := db.DB().Query(s, args...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, nil, err
|
return nil, nil, err
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
defer rows.Close()
|
||||||
|
|
||||||
cols := make(map[string]*core.Column)
|
cols := make(map[string]*core.Column)
|
||||||
colSeq := make([]string, 0)
|
colSeq := make([]string, 0)
|
||||||
|
|
||||||
for rows.Next() {
|
for rows.Next() {
|
||||||
col := new(core.Column)
|
col := new(core.Column)
|
||||||
col.Indexes = make(map[string]bool)
|
col.Indexes = make(map[string]bool)
|
||||||
|
|
||||||
var colName, isNullable, dataType string
|
var colName, isNullable, dataType string
|
||||||
var maxLenStr, colDefault, numPrecision, numRadix *string
|
var maxLenStr, colDefault, numPrecision, numRadix *string
|
||||||
err = rows.Scan(&colName, &colDefault, &isNullable, &dataType, &maxLenStr, &numPrecision, &numRadix)
|
var isPK ,isUnique bool
|
||||||
if err != nil {
|
err = rows.Scan(&colName, &colDefault, &isNullable, &dataType, &maxLenStr, &numPrecision, &numRadix,&isPK,&isUnique)
|
||||||
return nil, nil, err
|
if err != nil {
|
||||||
}
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
//fmt.Println(args,colName, isNullable, dataType,maxLenStr, colDefault, numPrecision, numRadix,isPK ,isUnique)
|
||||||
|
var maxLen int
|
||||||
|
if maxLenStr != nil {
|
||||||
|
maxLen, err = strconv.Atoi(*maxLenStr)
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
var maxLen int
|
col.Name = strings.Trim(colName, `" `)
|
||||||
if maxLenStr != nil {
|
|
||||||
maxLen, err = strconv.Atoi(*maxLenStr)
|
|
||||||
if err != nil {
|
|
||||||
return nil, nil, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
col.Name = strings.Trim(colName, `" `)
|
if colDefault != nil || isPK {
|
||||||
|
if isPK {
|
||||||
|
col.IsPrimaryKey = true
|
||||||
|
} else {
|
||||||
|
col.Default = *colDefault
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if colDefault != nil {
|
col.Nullable = (isNullable == "YES")
|
||||||
if strings.HasPrefix(*colDefault, "nextval") {
|
|
||||||
col.IsPrimaryKey = true
|
|
||||||
} else {
|
|
||||||
col.Default = *colDefault
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
col.Nullable = (isNullable == "YES")
|
switch dataType {
|
||||||
|
case "character varying", "character":
|
||||||
|
col.SQLType = core.SQLType{core.Varchar, 0, 0}
|
||||||
|
case "timestamp without time zone":
|
||||||
|
col.SQLType = core.SQLType{core.DateTime, 0, 0}
|
||||||
|
case "timestamp with time zone":
|
||||||
|
col.SQLType = core.SQLType{core.TimeStampz, 0, 0}
|
||||||
|
case "double precision":
|
||||||
|
col.SQLType = core.SQLType{core.Double, 0, 0}
|
||||||
|
case "boolean":
|
||||||
|
col.SQLType = core.SQLType{core.Bool, 0, 0}
|
||||||
|
case "time without time zone":
|
||||||
|
col.SQLType = core.SQLType{core.Time, 0, 0}
|
||||||
|
default:
|
||||||
|
col.SQLType = core.SQLType{strings.ToUpper(dataType), 0, 0}
|
||||||
|
}
|
||||||
|
if _, ok := core.SqlTypes[col.SQLType.Name]; !ok {
|
||||||
|
return nil, nil, errors.New(fmt.Sprintf("unkonw colType %v", dataType))
|
||||||
|
}
|
||||||
|
|
||||||
switch dataType {
|
col.Length = maxLen
|
||||||
case "character varying", "character":
|
|
||||||
col.SQLType = core.SQLType{core.Varchar, 0, 0}
|
|
||||||
case "timestamp without time zone":
|
|
||||||
col.SQLType = core.SQLType{core.DateTime, 0, 0}
|
|
||||||
case "timestamp with time zone":
|
|
||||||
col.SQLType = core.SQLType{core.TimeStampz, 0, 0}
|
|
||||||
case "double precision":
|
|
||||||
col.SQLType = core.SQLType{core.Double, 0, 0}
|
|
||||||
case "boolean":
|
|
||||||
col.SQLType = core.SQLType{core.Bool, 0, 0}
|
|
||||||
case "time without time zone":
|
|
||||||
col.SQLType = core.SQLType{core.Time, 0, 0}
|
|
||||||
default:
|
|
||||||
col.SQLType = core.SQLType{strings.ToUpper(dataType), 0, 0}
|
|
||||||
}
|
|
||||||
if _, ok := core.SqlTypes[col.SQLType.Name]; !ok {
|
|
||||||
return nil, nil, errors.New(fmt.Sprintf("unkonw colType %v", dataType))
|
|
||||||
}
|
|
||||||
|
|
||||||
col.Length = maxLen
|
if col.SQLType.IsText() {
|
||||||
|
if col.Default != "" {
|
||||||
|
col.Default = "'"+col.Default+"'"
|
||||||
|
} else {
|
||||||
|
if col.DefaultIsEmpty {
|
||||||
|
col.Default = "''"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
cols[col.Name] = col
|
||||||
|
colSeq = append(colSeq, col.Name)
|
||||||
|
}
|
||||||
|
|
||||||
if col.SQLType.IsText() {
|
return colSeq, cols, nil
|
||||||
if col.Default != "" {
|
|
||||||
col.Default = "'" + col.Default + "'"
|
|
||||||
} else {
|
|
||||||
if col.DefaultIsEmpty {
|
|
||||||
col.Default = "''"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
cols[col.Name] = col
|
|
||||||
colSeq = append(colSeq, col.Name)
|
|
||||||
}
|
|
||||||
|
|
||||||
return colSeq, cols, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (db *postgres) GetTables() ([]*core.Table, error) {
|
func (db *postgres) GetTables() ([]*core.Table, error) {
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue