HiPoweredDev
/
xorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add includeNil requiredField when custom struct field IsZero in Update method. Add test

This commit is contained in:
tyler 2024-03-22 03:50:20 +08:00
parent c4b5433155
commit a3687832fa
2 changed files with 23 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internal/statements/update.go
View File

@ -126,7 +126,7 @@ func (statement *Statement) BuildUpdates(tableValue reflect.Value,
if fieldValue.CanAddr() { if fieldValue.CanAddr() {
if structConvert, ok := fieldValue.Addr().Interface().(convert.Conversion); ok { if structConvert, ok := fieldValue.Addr().Interface().(convert.Conversion); ok {
if utils.IsZero(fieldValue.Interface()) { if !includeNil && !requiredField && utils.IsZero(fieldValue.Interface()) {
continue continue
} }
data, err := structConvert.ToDB() data, err := structConvert.ToDB()

24
tests/types_test.go
View File

@ -633,6 +633,9 @@ func (d *ZDecimal) FromDB(data []byte) error {
} }
func (d ZDecimal) ToDB() ([]byte, error) { func (d ZDecimal) ToDB() ([]byte, error) {
if d.value == nil {
return []byte("0"), nil
}
return []byte(fmt.Sprintf("%d", (d.value).Int64())), nil return []byte(fmt.Sprintf("%d", (d.value).Int64())), nil
} }
@ -688,6 +691,23 @@ func TestZDecimal(t *testing.T) {
assert.NoError(t, err) assert.NoError(t, err)
assert.True(t, has) assert.True(t, has)
assert.Equal(t, "test2", "test2") assert.Equal(t, "test2", "test2")
assert.False(t, m.Amount.IsZero()) assert.False(t, m2.Amount.IsZero())
assert.Equal(t, "10000000000000000", m.Amount.String()) assert.Equal(t, "10000000000000000", m2.Amount.String())
_, err = testEngine.AllCols().Update(&ZMyMoney{
Id: 1,
Account: "test3",
})
assert.NoError(t, err)
var m3 = ZMyMoney{
Id: 1,
}
has, err = testEngine.Get(&m3)
assert.NoError(t, err)
assert.True(t, has)
assert.Equal(t, "test3", "test3")
assert.True(t, m3.Amount.IsZero())
assert.Equal(t, "0", m3.Amount.String())
} }