不要使用gorm推送空结构
创始人
2025-01-10 21:30:50
0

在使用gorm进行数据操作时,如果想要避免推送空结构,可以使用gorm提供的Hooks钩子函数来实现。下面是一个示例代码:

package main

import (
	"fmt"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"log"
)

type User struct {
	ID   uint
	Name string
	Age  int
}

func (u *User) BeforeSave(tx *gorm.DB) (err error) {
	if u.Name == "" && u.Age == 0 {
		return gorm.ErrEmptySlice
	}
	return nil
}

func main() {
	dsn := "root:password@tcp(127.0.0.1:3306)/test?charset=utf8mb4&parseTime=True&loc=Local"
	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
		Logger: logger.Default.LogMode(logger.Info),
	})
	if err != nil {
		log.Fatal(err)
	}

	err = db.AutoMigrate(&User{})
	if err != nil {
		log.Fatal(err)
	}

	user1 := User{Name: "", Age: 0}
	user2 := User{Name: "Alice", Age: 20}

	err = db.Save(&user1).Error
	if err != nil {
		fmt.Println("Save user1 error:", err)
	}

	err = db.Save(&user2).Error
	if err != nil {
		fmt.Println("Save user2 error:", err)
	}
}

在上面的示例中,定义了一个User结构体,并在结构体上定义了一个BeforeSave钩子函数。在该钩子函数中,判断了NameAge字段是否为空,如果为空,则返回gorm.ErrEmptySlice错误。

main函数中,通过db.Save方法保存了两个用户,一个是空结构体user1,另一个是有数据的结构体user2。当保存user1时,因为NameAge都为空,会触发BeforeSave钩子函数,抛出gorm.ErrEmptySlice错误。而保存user2时,不会触发钩子函数。

通过使用钩子函数,可以实现在保存数据前进行数据校验,避免推送空结构。

相关内容

热门资讯

AWSECS:访问外部网络时出... 如果您在AWS ECS中部署了应用程序,并且该应用程序需要访问外部网络,但是无法正常访问,可能是因为...
AWSElasticBeans... 在Dockerfile中手动配置nginx反向代理。例如,在Dockerfile中添加以下代码:FR...
AWR报告解读 WORKLOAD REPOSITORY PDB report (PDB snapshots) AW...
AWS管理控制台菜单和权限 要在AWS管理控制台中创建菜单和权限,您可以使用AWS Identity and Access Ma...
北信源内网安全管理卸载 北信源内网安全管理是一款网络安全管理软件,主要用于保护内网安全。在日常使用过程中,卸载该软件是一种常...
​ToDesk 远程工具安装及... 目录 前言 ToDesk 优势 ToDesk 下载安装 ToDesk 功能展示 文件传输 设备链接 ...
Azure构建流程(Power... 这可能是由于配置错误导致的问题。请检查构建流程任务中的“发布构建制品”步骤,确保正确配置了“Arti...
群晖外网访问终极解决方法:IP... 写在前面的话 受够了群晖的quickconnet的小水管了,急需一个新的解决方法&#x...
AWSECS:哪种网络模式具有... 使用AWS ECS中的awsvpc网络模式来获得最佳性能。awsvpc网络模式允许ECS任务直接在V...
不能访问光猫的的管理页面 光猫是现代家庭宽带网络的重要组成部分,它可以提供高速稳定的网络连接。但是,有时候我们会遇到不能访问光...