From a689e56433d9900b233b6fcb81ce6a30102abb40 Mon Sep 17 00:00:00 2001 From: daheige Date: Mon, 14 Jun 2021 10:29:15 +0800 Subject: [PATCH] fix return value for *schema.Check --- migrator/migrator.go | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/migrator/migrator.go b/migrator/migrator.go index da9ddd49..2af20166 100644 --- a/migrator/migrator.go +++ b/migrator/migrator.go @@ -465,7 +465,7 @@ func buildConstraint(constraint *schema.Constraint) (sql string, results []inter return } -func (m Migrator) GuessConstraintAndTable(stmt *gorm.Statement, name string) (_ *schema.Constraint, _ *schema.Check, table string) { +func (m Migrator) GuessConstraintAndTable(stmt *gorm.Statement, name string) (*schema.Constraint, *schema.Check, string) { if stmt.Schema == nil { return nil, nil, stmt.Table } @@ -492,9 +492,11 @@ func (m Migrator) GuessConstraintAndTable(stmt *gorm.Statement, name string) (_ } if field := stmt.Schema.LookUpField(name); field != nil { - for _, cc := range checkConstraints { - if cc.Field == field { - return nil, &cc, stmt.Table + // fix return value for *schema.Check + for k := range checkConstraints { + if checkConstraints[k].Field == field { + v := checkConstraints[k] + return nil, &v, stmt.Table } }