不要使用gorm推送空结构
创始人
2025-01-10 21:30:33
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时,不会触发钩子函数。

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

相关内容

热门资讯

第8分钟了解!余干辅助软件哪个... 第8分钟了解!余干辅助软件哪个好!原来是真的有辅助插件(有挂总结)-哔哩哔哩1、上手简单,内置详细流...
第9分钟了解!牵手跑辅助!一贯... 第9分钟了解!牵手跑辅助!一贯真的是有辅助神器(有挂技巧)-哔哩哔哩1、牵手跑辅助辅助器安装包、牵手...
两分钟了解!浙江游戏温州熟客辅... 两分钟了解!浙江游戏温州熟客辅助!切实一直都是有辅助插件(果真有挂)-哔哩哔哩1、每一步都需要思考,...
第三分钟了解!海螺众娱脚本!真... 第三分钟了解!海螺众娱脚本!真是是真的有辅助教程(有挂解密)-哔哩哔哩1)海螺众娱脚本免费钻石:进一...
十分钟了解!决战血流辅助!一贯... 十分钟了解!决战血流辅助!一贯一直都是有辅助技巧(有挂详情)-哔哩哔哩运决战血流辅助辅助工具,进入游...
四分钟了解!开心泉州作必弊!果... 四分钟了解!开心泉州作必弊!果然存在有辅助技巧(有挂细节)-哔哩哔哩1、开心泉州作必弊透视辅助软件激...
三分钟了解!情怀麻烦将关春天辅... 三分钟了解!情怀麻烦将关春天辅助!其实一直总是有辅助神器(有挂教学)-哔哩哔哩1、完成情怀麻烦将关春...
8分钟了解!福建十三水软件开发... 8分钟了解!福建十三水软件开发!一直有辅助工具(有挂秘诀)-哔哩哔哩该软件可以轻松地帮助玩家将福建十...
九分钟了解!手游奇迹陕西辅助工... 九分钟了解!手游奇迹陕西辅助工具!果然一直都是有辅助方法(真实有挂)-哔哩哔哩该软件可以轻松地帮助玩...
第5分钟了解!微信海豚大厅辅助... 第5分钟了解!微信海豚大厅辅助!总是一直总是有辅助软件(证实有挂)-哔哩哔哩该软件可以轻松地帮助玩家...