如何避免Gin+Gorm开发过程中的值混乱？傻傻分不清Zero-value与Not-Set？一文说清解决方法！

后端

2023-10-02 03:27:43

零值与未设置字段的概念

1. 零值

在Go语言中，每个值类型都有一个默认值，称为零值。零值是指该值类型在没有被显式初始化时所具有的值。例如，int类型的零值为0，string类型的零值为""，bool类型的零值为false。

2. 未设置字段

未设置字段是指该字段的值根本没有被设置。在Go语言中，使用指针类型可以表示一个未设置字段。例如，var p *int是一个指针类型，它指向一个int类型的变量。如果p的值为nil，则表示指向的int类型变量没有被设置。

如何区分零值与未设置字段

在Gin+Gorm开发过程中，经常会遇到需要区分请求体中的零值和未设置字段的情况。例如，在创建或更新记录时，需要判断哪些字段的值是零值，哪些字段的值是未设置。如果将零值字段当作未设置字段处理，则可能会导致数据丢失或更新失败。

以下是一些区分零值和未设置字段的方法：

1. 使用反射

可以使用反射来获取字段的值并判断其是否为零值。例如：

type User struct {
	ID        uint `gorm:"primary_key"`
	Name      string
	Age       int
	IsMarried bool
}

func main() {
	user := User{}
	reflect.ValueOf(user).Elem().FieldByName("Name").IsZero() // true
	reflect.ValueOf(user).Elem().FieldByName("Age").IsZero() // true
	reflect.ValueOf(user).Elem().FieldByName("IsMarried").IsZero() // true
}

2. 使用自定义类型

可以定义一个自定义类型来表示未设置字段。例如：

type NotSet struct{}

func (n NotSet) IsZero() bool {
	return true
}

然后可以使用这个自定义类型来表示未设置字段。例如：

type User struct {
	ID        uint `gorm:"primary_key"`
	Name      string
	Age       *int
	IsMarried *bool
}

func main() {
	user := User{
		Age:       nil,
		IsMarried: nil,
	}
	user.Age.IsZero() // true
	user.IsMarried.IsZero() // true
}

3. 使用默认值

可以为字段设置一个默认值。如果字段的值为默认值，则表示该字段没有被设置。例如：

type User struct {
	ID        uint `gorm:"primary_key"`
	Name      string `gorm:"default:''"`
	Age       int `gorm:"default:0"`
	IsMarried bool `gorm:"default:false"`
}

func main() {
	user := User{}
	user.Name == "" // true
	user.Age == 0 // true
	user.IsMarried == false // true
}