Skip to content

Commit b0dea8f

Browse files
committed
Fix postgres AdaptTable
1 parent d246a7e commit b0dea8f

1 file changed

Lines changed: 14 additions & 8 deletions

File tree

postgres/postgres.go

Lines changed: 14 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -703,7 +703,7 @@ func (pg *PostGres) AdaptTable(name string, col any) error {
703703
buffer.WriteString(` ADD COLUMN `)
704704
dbsql.CreateTableByColumn(&buffer, pg.ByteArrayAvailable(), c)
705705
}
706-
fmt.Println(buffer.String())
706+
log.Log.Debugf(buffer.String())
707707
_, err = db.Query(buffer.String())
708708
if err != nil {
709709
log.Log.Errorf("Error returned by SQL: %v", err)
@@ -720,15 +720,21 @@ func (pg *PostGres) AdaptTable(name string, col any) error {
720720
if err != nil {
721721
return err
722722
}
723+
var buffer bytes.Buffer
724+
buffer.WriteString(`ALTER TABLE ` + name)
725+
i := 0
723726
for _, f := range strings.Split(columStruct, ",") {
724-
725-
adaptCmd := `ALTER TABLE ` + name + ` ADD ` + f
726-
log.Log.Debugf("Create cmd %s", adaptCmd)
727-
_, err = db.Query(adaptCmd)
728-
if err != nil {
729-
log.Log.Errorf("Error returned by SQL: %v", err)
730-
return err
727+
if i > 0 {
728+
buffer.WriteString(",")
731729
}
730+
i++
731+
buffer.WriteString(` ADD COLUMN ` + f)
732+
}
733+
734+
_, err = db.Query(buffer.String())
735+
if err != nil {
736+
log.Log.Errorf("Error returned by SQL: %v", err)
737+
return err
732738
}
733739
log.Log.Debugf("Table adapted")
734740
return nil

0 commit comments

Comments
 (0)