github_mirrors/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2025-01-23 09:00:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: support pg serial type for auto_increment (#1563)

Browse Source 
* add correct example for pg's url

* 🐞 fix: merge

* 🐞 fix: pg default port

* ✨ feat: support serial type

Co-authored-by: kurimi1 <d0n41df@gmail.com>
This commit is contained in:
toutou_o 2022-02-24 13:39:31 +08:00 committed by GitHub
parent daa98f5a27
commit 3b07ed1b97
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/goctl/goctl.go
View File

@ -690,7 +690,7 @@ var commands = []cli.Command{
Flags: []cli.Flag{
cli.StringFlag{
Name: "url",
Usage: `the data source of database,like "postgres://root:password@127.0.0.1:54332/database?sslmode=disable"`,
Usage: `the data source of database,like "postgres://root:password@127.0.0.1:5432/database?sslmode=disable"`,
},
cli.StringFlag{
Name: "table, t",

12
tools/goctl/model/sql/model/postgresqlmodel.go
View File

@ -120,9 +120,15 @@ func (m *PostgreSqlModel) getColumns(schema, table string, in []*PostgreColumn)
isNullAble = "NO"
}
extra := "auto_increment"
if e.IdentityIncrement.Int32 != 1 {
extra = ""
extra := ""
// when identity is true, the column is auto increment
if e.IdentityIncrement.Int32 == 1 {
extra = "auto_increment"
}
// when type is serial, it's auto_increment. and the default value is tablename_columnname_seq
if strings.Contains(e.ColumnDefault.String, table+"_"+e.Field.String+"_seq") {
extra = "auto_increment"
}
if len(index[e.Field.String]) > 0 {