gorm框架存在插入,存在则更新处理

一、gorm库

github地址:https://github.com/go-gorm/gorm
使用说明:https://gorm.io/docs/

二、探索

gorm支持clause来实现Upsert的功能,但是发现只支持根据id进行判断,如果对应id记录存在则更新,对应id记录不存在则插入。如下:

dbtable.Clauses(clause.OnConflict{
	Columns:   []clause.Column{{Name: "id"}},
	DoUpdates: clause.AssignmentColumns([]string{"remark", "is_delete"}),
}).Create(friend)

我用的数据库是mysql,需要根据userId和friendId作为条件进行更新,userId和friendId设置了联合唯一索引,用这个方式无法实现。

三、最终解决方案

先查询是否存在,不存在则插入,存在则更新,如下:

var friend FriendInfo
results := db.Table(tableName).Where("user_id = ? and friend_id = ?", uid, friend).First(&friend)
if results.Error != nil {
	if results.Error == gorm.ErrRecordNotFound {
		results := db.Table(tableName).Create(&friend)
		if results.Error != nil {
			return results.Error
		}
	} 
} else {
		friend.ID = friendRec.ID
		results := db.Table(tableName).Model(friend).Updates(map[string]interface{}{"friend_type": friend.FriendType, "remark": friend.FriendRemark, "is_delete": friend.IsDelete, "create_time": friend.CreateTime, "update_time": friend.UpdateTime})
		if results.Error != nil {
			return results.Error
		}
}
在使用 GORM 框架中,可以使用 Go 语言的协程(goroutine)来实现并发操作。下面是一个示例: ```go package main import ( "fmt" "github.com/jinzhu/gorm" _ "github.com/jinzhu/gorm/dialects/mysql" ) type User struct { ID int Name string } func main() { db, err := gorm.Open("mysql", "user:password@tcp(127.0.0.1:3306)/database?charset=utf8mb4&parseTime=True&loc=Local") if err != nil { panic("failed to connect database") } defer db.Close() // 创建一个协程来执行查询操作 go func() { var user User db.First(&user, 1) fmt.Println(user) }() // 创建另一个协程来执行插入操作 go func() { newUser := User{Name: "Alice"} db.Create(&newUser) }() // 等待一段时间,以确保协程有足够的时间执行 time.Sleep(time.Second) // 查询所有用户 var users []User db.Find(&users) fmt.Println(users) } ``` 在上面的示例中,我们使用了两个协程来执行数据库操作。第一个协程执行了查询操作,第二个协程执行了插入操作。通过使用协程,我们可以并发地执行这些操作,提高程序的性能。 需要注意的是,在使用协程时,我们需要确保数据库连接是线程安全的。在上面的示例中,我们使用了 `defer db.Close()` 来关闭数据库连接,以确保资源的释放。另外,为了确保协程有足够的时间执行,我们使用了 `time.Sleep(time.Second)` 来进行等待。 当然,具体的实现方式可能因为业务需求的同而有所差异,上述示例仅供参考。在实际使用中,你可以根据自己的需求来选择合适的并发方式。
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值