Skip to content

Commit

Permalink
Test using pointer as struct fields
Browse files Browse the repository at this point in the history
  • Loading branch information
jinzhu committed Feb 20, 2015
1 parent b7edaee commit 406a018
Show file tree
Hide file tree
Showing 4 changed files with 97 additions and 10 deletions.
12 changes: 5 additions & 7 deletions callback_create.go
Original file line number Diff line number Diff line change
Expand Up @@ -55,27 +55,25 @@ func Create(scope *Scope) {
}

// execute create sql
var id interface{}
if scope.Dialect().SupportLastInsertId() {
if result, err := scope.DB().Exec(scope.Sql, scope.SqlVars...); scope.Err(err) == nil {
id, err = result.LastInsertId()
id, err := result.LastInsertId()
if scope.Err(err) == nil {
scope.db.RowsAffected, _ = result.RowsAffected()
if primaryField != nil {
scope.SetColumn(primaryField, id)
}
}
}
} else {
if primaryField == nil {
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 {
} else if scope.Err(scope.DB().QueryRow(scope.Sql, scope.SqlVars...).Scan(primaryField.Field.Addr().Interface())) == nil {
scope.db.RowsAffected = 1
}
}

if primaryField != nil && primaryField.IsBlank && !scope.HasError() {
scope.SetColumn(primaryField, id)
}
}
}

Expand Down
6 changes: 5 additions & 1 deletion model_struct.go
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,11 @@ type Relationship struct {

func (scope *Scope) generateSqlTag(field *StructField) {
var sqlType string
reflectValue := reflect.Indirect(reflect.New(field.Struct.Type))
structType := field.Struct.Type
if structType.Kind() == reflect.Ptr {
structType = structType.Elem()
}
reflectValue := reflect.Indirect(reflect.New(structType))
sqlSettings := parseTagSetting(field.Tag.Get("sql"))

if value, ok := sqlSettings["TYPE"]; ok {
Expand Down
84 changes: 84 additions & 0 deletions pointer_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,84 @@
package gorm_test

import "testing"

type PointerStruct struct {
ID int64
Name *string
Num *int
}

type NormalStruct struct {
ID int64
Name string
Num int
}

func TestPointerFields(t *testing.T) {
DB.DropTable(&PointerStruct{})
DB.AutoMigrate(&PointerStruct{})
var name = "pointer struct 1"
var num = 100
pointerStruct := PointerStruct{Name: &name, Num: &num}
if DB.Create(&pointerStruct).Error != nil {
t.Errorf("Failed to save pointer struct")
}

var pointerStructResult PointerStruct
if err := DB.First(&pointerStructResult, "id = ?", pointerStruct.ID).Error; err != nil || *pointerStructResult.Name != name || *pointerStructResult.Num != num {
t.Errorf("Failed to query saved pointer struct")
}

var tableName = DB.NewScope(&PointerStruct{}).TableName()

var normalStruct NormalStruct
DB.Table(tableName).First(&normalStruct)
if normalStruct.Name != name || normalStruct.Num != num {
t.Errorf("Failed to query saved Normal struct")
}

var nilPointerStruct = PointerStruct{}
if err := DB.Create(&nilPointerStruct).Error; err != nil {
t.Errorf("Failed to save nil pointer struct", err)
}

var pointerStruct2 PointerStruct
if err := DB.First(&pointerStruct2, "id = ?", nilPointerStruct.ID).Error; err != nil {
t.Errorf("Failed to query saved nil pointer struct", err)
}

var normalStruct2 NormalStruct
if err := DB.Table(tableName).First(&normalStruct2, "id = ?", nilPointerStruct.ID).Error; err != nil {
t.Errorf("Failed to query saved nil pointer struct", err)
}

var partialNilPointerStruct1 = PointerStruct{Num: &num}
if err := DB.Create(&partialNilPointerStruct1).Error; err != nil {
t.Errorf("Failed to save partial nil pointer struct", err)
}

var pointerStruct3 PointerStruct
if err := DB.First(&pointerStruct3, "id = ?", partialNilPointerStruct1.ID).Error; err != nil || *pointerStruct3.Num != num {
t.Errorf("Failed to query saved partial nil pointer struct", err)
}

var normalStruct3 NormalStruct
if err := DB.Table(tableName).First(&normalStruct3, "id = ?", partialNilPointerStruct1.ID).Error; err != nil || normalStruct3.Num != num {
t.Errorf("Failed to query saved partial pointer struct", err)
}

var partialNilPointerStruct2 = PointerStruct{Name: &name}
if err := DB.Create(&partialNilPointerStruct2).Error; err != nil {
t.Errorf("Failed to save partial nil pointer struct", err)
}

var pointerStruct4 PointerStruct
if err := DB.First(&pointerStruct4, "id = ?", partialNilPointerStruct2.ID).Error; err != nil || *pointerStruct4.Name != name {
t.Errorf("Failed to query saved partial nil pointer struct", err)
}

var normalStruct4 NormalStruct
if err := DB.Table(tableName).First(&normalStruct4, "id = ?", partialNilPointerStruct2.ID).Error; err != nil || normalStruct4.Name != name {
t.Errorf("Failed to query saved partial pointer struct", err)
}
}
5 changes: 3 additions & 2 deletions scope.go
Original file line number Diff line number Diff line change
Expand Up @@ -35,8 +35,9 @@ func (scope *Scope) IndirectValue() reflect.Value {

// NewScope create scope for callbacks, including DB's search information
func (db *DB) NewScope(value interface{}) *Scope {
db.Value = value
return &Scope{db: db, Search: db.search, Value: value}
dbClone := db.clone()
dbClone.Value = value
return &Scope{db: dbClone, Search: dbClone.search, Value: value}
}

func (scope *Scope) NeedPtr() *Scope {
Expand Down

0 comments on commit 406a018

Please sign in to comment.