不要使用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时,不会触发钩子函数。

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

相关内容

热门资讯

保存时出现了1个错误,导致这篇... 当保存文章时出现错误时,可以通过以下步骤解决问题:查看错误信息:查看错误提示信息可以帮助我们了解具体...
汇川伺服电机位置控制模式参数配... 1. 基本控制参数设置 1)设置位置控制模式   2)绝对值位置线性模...
不能访问光猫的的管理页面 光猫是现代家庭宽带网络的重要组成部分,它可以提供高速稳定的网络连接。但是,有时候我们会遇到不能访问光...
本地主机上的图像未显示 问题描述:在本地主机上显示图像时,图像未能正常显示。解决方法:以下是一些可能的解决方法,具体取决于问...
不一致的条件格式 要解决不一致的条件格式问题,可以按照以下步骤进行:确定条件格式的规则:首先,需要明确条件格式的规则是...
表格中数据未显示 当表格中的数据未显示时,可能是由于以下几个原因导致的:HTML代码问题:检查表格的HTML代码是否正...
表格列调整大小出现问题 问题描述:表格列调整大小出现问题,无法正常调整列宽。解决方法:检查表格的布局方式是否正确。确保表格使...
Android|无法访问或保存... 这个问题可能是由于权限设置不正确导致的。您需要在应用程序清单文件中添加以下代码来请求适当的权限:此外...
【NI Multisim 14...   目录 序言 一、工具栏 🍊1.“标准”工具栏 🍊 2.视图工具...
银河麒麟V10SP1高级服务器... 银河麒麟高级服务器操作系统简介: 银河麒麟高级服务器操作系统V10是针对企业级关键业务...