golang学习笔记（5）-gorm实现创建记录功能

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", ...;