From f13883a725e6026a22d5188cd3d4bae0e324d764 Mon Sep 17 00:00:00 2001
From: Lunny Xiao <xiaolunwen@gmail.com>
Date: Tue, 10 Mar 2020 03:01:32 +0000
Subject: [PATCH] check driver.Valuer response, and skip the column if nil
 (#1167)

update tests

Reviewed-on: https://gitea.com/xorm/xorm/pulls/1167
---
 internal/statements/statement.go |  2 +-
 internal/statements/update.go    |  3 ++
 types_null_test.go               | 61 ++++++++++++++++----------------
 3 files changed, 35 insertions(+), 31 deletions(-)

diff --git a/internal/statements/statement.go b/internal/statements/statement.go
index 4beb9a7e..d1fcaf59 100644
--- a/internal/statements/statement.go
+++ b/internal/statements/statement.go
@@ -838,7 +838,7 @@ func (statement *Statement) buildConds2(table *schemas.Table, bean interface{},
 				continue
 			} else if valNul, ok := fieldValue.Interface().(driver.Valuer); ok {
 				val, _ = valNul.Value()
-				if val == nil {
+				if val == nil && !requiredField {
 					continue
 				}
 			} else {
diff --git a/internal/statements/update.go b/internal/statements/update.go
index 10f36778..2e502243 100644
--- a/internal/statements/update.go
+++ b/internal/statements/update.go
@@ -186,6 +186,9 @@ func (statement *Statement) BuildUpdates(bean interface{},
 				val = dialects.FormatColumnTime(statement.dialect, statement.defaultTimeZone, col, t)
 			} else if nulType, ok := fieldValue.Interface().(driver.Valuer); ok {
 				val, _ = nulType.Value()
+				if val == nil && !requiredField {
+					continue
+				}
 			} else {
 				if !col.SQLType.IsJson() {
 					table, err := statement.tagParser.ParseWithCache(fieldValue)
diff --git a/types_null_test.go b/types_null_test.go
index 665849ca..f62366b7 100644
--- a/types_null_test.go
+++ b/types_null_test.go
@@ -22,6 +22,7 @@ type NullType struct {
 	Age          sql.NullInt64
 	Height       sql.NullFloat64
 	IsMan        sql.NullBool `xorm:"null"`
+	Nil          driver.Valuer
 	CustomStruct CustomStruct `xorm:"varchar(64) null"`
 }
 
@@ -72,42 +73,42 @@ func TestNullStructInsert(t *testing.T) {
 	assert.NoError(t, prepareEngine())
 	assertSync(t, new(NullType))
 
-	if true {
-		item := new(NullType)
-		_, err := testEngine.Insert(item)
-		assert.NoError(t, err)
-		assert.EqualValues(t, 1, item.Id)
-	}
+	item1 := new(NullType)
+	_, err := testEngine.Insert(item1)
+	assert.NoError(t, err)
+	assert.EqualValues(t, 1, item1.Id)
 
-	if true {
+	item := NullType{
+		Name:   sql.NullString{String: "haolei", Valid: true},
+		Age:    sql.NullInt64{Int64: 34, Valid: true},
+		Height: sql.NullFloat64{Float64: 1.72, Valid: true},
+		IsMan:  sql.NullBool{Bool: true, Valid: true},
+		Nil:    nil,
+	}
+	_, err = testEngine.Insert(&item)
+	assert.NoError(t, err)
+	assert.EqualValues(t, 2, item.Id)
+
+	items := []NullType{}
+	for i := 0; i < 5; i++ {
 		item := NullType{
-			Name:   sql.NullString{String: "haolei", Valid: true},
-			Age:    sql.NullInt64{Int64: 34, Valid: true},
-			Height: sql.NullFloat64{Float64: 1.72, Valid: true},
-			IsMan:  sql.NullBool{Bool: true, Valid: true},
+			Name:         sql.NullString{String: "haolei_" + fmt.Sprint(i+1), Valid: true},
+			Age:          sql.NullInt64{Int64: 30 + int64(i), Valid: true},
+			Height:       sql.NullFloat64{Float64: 1.5 + 1.1*float64(i), Valid: true},
+			IsMan:        sql.NullBool{Bool: true, Valid: true},
+			CustomStruct: CustomStruct{i, i + 1, i + 2},
+			Nil:          nil,
 		}
-		_, err := testEngine.Insert(&item)
-		assert.NoError(t, err)
-		assert.EqualValues(t, 2, item.Id)
+		items = append(items, item)
 	}
 
-	if true {
-		items := []NullType{}
-		for i := 0; i < 5; i++ {
-			item := NullType{
-				Name:         sql.NullString{String: "haolei_" + fmt.Sprint(i+1), Valid: true},
-				Age:          sql.NullInt64{Int64: 30 + int64(i), Valid: true},
-				Height:       sql.NullFloat64{Float64: 1.5 + 1.1*float64(i), Valid: true},
-				IsMan:        sql.NullBool{Bool: true, Valid: true},
-				CustomStruct: CustomStruct{i, i + 1, i + 2},
-			}
+	_, err = testEngine.Insert(&items)
+	assert.NoError(t, err)
 
-			items = append(items, item)
-		}
-
-		_, err := testEngine.Insert(&items)
-		assert.NoError(t, err)
-	}
+	items = make([]NullType, 0, 7)
+	err = testEngine.Find(&items)
+	assert.NoError(t, err)
+	assert.EqualValues(t, 7, len(items))
 }
 
 func TestNullStructUpdate(t *testing.T) {