From 5823670269c9eb0d8eeedd2eaddcc06310df767c Mon Sep 17 00:00:00 2001 From: Jinzhu Date: Sat, 13 Dec 2014 10:46:16 +0800 Subject: [PATCH] Refactor callbck_create --- callback_create.go | 14 +++----------- 1 file changed, 3 insertions(+), 11 deletions(-) diff --git a/callback_create.go b/callback_create.go index 4c909e1c..4734d65e 100644 --- a/callback_create.go +++ b/callback_create.go @@ -60,21 +60,13 @@ func Create(scope *Scope) { if result, err := scope.DB().Exec(scope.Sql, scope.SqlVars...); scope.Err(err) == nil { id, err = result.LastInsertId() if scope.Err(err) == nil { - if count, err := result.RowsAffected(); err == nil { - scope.db.RowsAffected = count - } + scope.db.RowsAffected, _ = result.RowsAffected() } } } else { if scope.PrimaryKey() == "" { - if rows, err := scope.DB().Query(scope.Sql, scope.SqlVars...); err != nil { - //extract column name to get fields lenght - if names, columnsErr := rows.Columns(); columnsErr != nil { - ids := make([]interface{}, len(names)) - if scope.Err(rows.Scan(ids...)) == nil { - scope.db.RowsAffected = 1 - } - } + if results, err := scope.DB().Exec(scope.Sql, scope.SqlVars...); err != nil { + scope.db.RowsAffected, _ = results.RowsAffected() } } else { if scope.Err(scope.DB().QueryRow(scope.Sql, scope.SqlVars...).Scan(&id)) == nil {