gorm实现创建记录功能
记录学习创建一条记录,创建多条记录,指定字段的创建,关联创建,设置默认值等操作。
准备工作
建立连接
import (
"fmt"
_ "github.com/go-sql-driver/mysql"
"gorm.io/driver/mysql"
"gorm.io/gorm"
"log"
)
var db *gorm.DB
func OpenDB() {
dsn := "root:adss123@tcp(127.0.0.1:3306)/go_db?charset=utf8mb4&parseTime=True&loc=Local"
res, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
db = res
if err != nil {
log.Fatal(err)
}
fmt.Printf("成功:%v\n", db)
}
建立模型
type TestTb2 struct {
Username string
Password string
gorm.Model
}
创建一条记录
利用 db.Create() 方法实现一条记录的创建
func CreateInfo(Any any) { //根据模型创建记录
OpenDB()
create := db.Create(Any)
if create.Error != nil {
fmt.Printf("创建失败:err%v\n", create.Error)
} else {
fmt.Printf("创建成功:创建了%v条记录\n", create.RowsAffected)
}
}
其中 func (db *DB) Create(value interface{}) (tx *DB) Create()可以接收所有模型,方法返回一个DB类型的指针,DB类型是gorm定义的一个结构体,详细如下:
type DB struct {
*Config
Error error
RowsAffected int64
Statement *Statement
clone int
}
其中Error字段用来记录error,RowsAffected字段用来记录操作影响的行数。
用指定的字段创建记录
创建被选择的字段,忽略未被创建的字段
db.Select("Name", "Age", "CreatedAt").Create(&user)
// INSERT INTO `users` (`name`,`age`,`created_at`) VALUES ("jinzhu", 18, "2020-07-04 11:05:21.775")
此条命令只会创建被选的三个字段,其他字段为默认值。
样例与结果如下:
func CreateSelect(Any any) { //选择字段创建,未被选择的字段为默认值
OpenDB()
create := db.Select("username").Create(Any)//只创建主键与username
if create.Error != nil {
fmt.Printf("创建失败:err%v\n", create.Error)
} else {
fmt.Printf("创建成功:创建了%v条记录\n", create.RowsAffected)
}
}
type TestTb2 struct {
Username string
Password string
gorm.Model
}
func main() {
info := &TestTb2{Username: "1", Password: "1"}
test_gorm.CreateSelect(info)
}
结果:
第十二条为本次运行得到的新记录
忽略被选择的字段,创建未被创建的字段
func CreateOmit(Any any) { //被选择的字段忽略,创建未被选择的字段
OpenDB()
create := db.Omit("username").Create(Any)
if create.Error != nil {
fmt.Printf("创建失败:err%v\n", create.Error)
} else {
fmt.Printf("创建成功:创建了%v条记录\n", create.RowsAffected)
}
}
批量插入
要有效地插入大量记录,将一个 slice 传递给 Create 方法。 GORM 将生成单独一条SQL语句来插入所有数据,并回填主键的值,Hook方法也会被调用。
关于Hook的学习后续将会记录,本文不进行解释。
本次案例利用前文CreateInfo()方法,只是在调用时,传入参数改变为模型类型的切片。
type TestTb2 struct {
Username string
Password string
gorm.Model
}
func main() {
info := &[]TestTb2{{Username: "1", Password: "1"}, {Username: "2", Password: "2"}}
test_gorm.CreateInfo(info)
for _, v := range *info {
println(v.ID)
}
}
结果如下:
可以使用 CreateInBatches 分批创建时,你可以指定每批的数量,例如:
var users = []User{{name: "jinzhu_1"}, ...., {Name: "jinzhu_10000"}}
// 数量为 100
db.CreateInBatches(users, 100)
关联创建
在两个表产生关联时,调用之前写好的CreateInfo()方法即可完成创建。
例如:
type TestTb2 struct {
Username string
Password string
TestTb2User1ID uint
gorm.Model
}
type TestTb2User1 struct {
gorm.Model
Username string
Birthday string
TestTb2 TestTb2
}
func main() {
info := TestTb2{Username: "ylj", Password: "123"}
infoUser := &TestTb2User1{Username: "ylj", Birthday: "4.13", TestTb2: info}
test_gorm.CreateInfo(infoUser)
}
同时可以使用Select、 Omit 跳过关联保存,例如:
db.Omit("TestTb2").Create(&infoUser)
// 跳过所有关联
db.Omit(clause.Associations).Create(&infoUser)
创建关联数据时,如果关联值是非零值,这些关联会被 upsert,且它们的 Hook 方法也会被调用
具体的关联表创建以及各种关联关系详解,会在对关联的具体学习中记录。本文不过多赘述。
默认值
您可以通过标签 default 为字段定义默认值,如:
type User struct {
ID int64
Name string `gorm:"default:galeone"`
Age int64 `gorm:"default:18"`
}
Upsert 及冲突
import "gorm.io/gorm/clause"
// 在冲突时,什么都不做
db.Clauses(clause.OnConflict{DoNothing: true}).Create(&user)
// 在`id`冲突时,将列更新为默认值
db.Clauses(clause.OnConflict{
Columns: []clause.Column{{Name: "id"}},
DoUpdates: clause.Assignments(map[string]interface{}{"role": "user"}),
}).Create(&users)
// MERGE INTO "users" USING *** WHEN NOT MATCHED THEN INSERT *** WHEN MATCHED THEN UPDATE SET ***; SQL Server
// INSERT INTO `users` *** ON DUPLICATE KEY UPDATE ***; MySQL
// 使用SQL语句
db.Clauses(clause.OnConflict{
Columns: []clause.Column{{Name: "id"}},
DoUpdates: clause.Assignments(map[string]interface{}{"count": gorm.Expr("GREATEST(count, VALUES(count))")}),
}).Create(&users)
// INSERT INTO `users` *** ON DUPLICATE KEY UPDATE `count`=GREATEST(count, VALUES(count));
// 在`id`冲突时,将列更新为新值
db.Clauses(clause.OnConflict{
Columns: []clause.Column{{Name: "id"}},
DoUpdates: clause.AssignmentColumns([]string{"name", "age"}),
}).Create(&users)
// MERGE INTO "users" USING *** WHEN NOT MATCHED THEN INSERT *** WHEN MATCHED THEN UPDATE SET "name"="excluded"."name"; SQL Server
// INSERT INTO "users" *** ON CONFLICT ("id") DO UPDATE SET "name"="excluded"."name", "age"="excluded"."age"; PostgreSQL
// INSERT INTO `users` *** ON DUPLICATE KEY UPDATE `name`=VALUES(name),`age=VALUES(age); MySQL
// 在冲突时,更新除主键以外的所有列到新值。
db.Clauses(clause.OnConflict{
UpdateAll: true,
}).Create(&users)
// INSERT INTO "users" *** ON CONFLICT ("id") DO UPDATE SET "name"="excluded"."name", "age"="excluded"."age", ...;