Day06:GORM快速入门02 增加| 青训营

69 阅读5分钟

导读

本套笔记是学习完Gin框架之后的下一个阶段,通过Gin框架我们可以很方便的和前端进行交互,下一步就是和数据库进行交互,这就需要使用GORM。本套笔记就是带你快速上手GORM,学习完这个之后就可以完成大项目的开发了,后续还会进行大项目的讲解开发,制作不易,喜欢的就点个关注吧。

本套课程基于GORM 指南 | GORM - The fantastic ORM library for Golang, aims to be developer friendly.进行讲解,是对这个的补充解释说明。

注意

代码详解大部分是注释的形式给出,请留意代码注释。

GORM介绍

GORM(Go Object Relational Mapping)是一个在 Go 语言中使用的优秀的 ORM(对象关系映射)库。它提供了简单且强大的 API,使开发人员能够轻松地与数据库进行交互

创建表

使用mysql建表,使用creat()

CREATE TABLE IF NOT EXISTS `users` (  
`id` INT(11) UNSIGNED AUTO_INCREMENT PRIMARY KEY,  
`name` VARCHAR(255) NOT NULL,  
`age` INT(11) NOT NULL,  
`birthday` DATETIME NOT NULL  
);

也可以使用

type User struct {  
ID uint  
Name string  
Age int  
Birthday time.Time  
}//表模型 

err = db.AutoMigrate(&User{})

成功创建 image.png

单个增加

package main  
  
import (  
"fmt"  
"gorm.io/driver/mysql"  
"gorm.io/gorm"  
"time"  
)  
  
type User struct {  
ID uint  
Name string  
Age int  
Birthday time.Time  
}//表模型 
  
func main() {  
  
db, _ := gorm.Open(mysql.New(mysql.Config{  
DSN: "root:Lycdemima1@@tcp(127.0.0.1:3306)/test?charset=utf8&parseTime=True&loc=Local", // DSN data source name  
DefaultStringSize: 256, // string 类型字段的默认长度  
DisableDatetimePrecision: true, // 禁用 datetime 精度,MySQL 5.6 之前的数据库不支持  
DontSupportRenameIndex: true, // 重命名索引时采用删除并新建的方式,MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引  
DontSupportRenameColumn: true, // 用 `change` 重命名列,MySQL 8 之前的数据库和 MariaDB 不支持重命名列  
SkipInitializeWithVersion: false, // 根据当前 MySQL 版本自动配置  
}), &gorm.Config{})  
//以上为连接数据库,需要自己更改其中的相关配置,如root。  


user := User{Name: "Jinzhu", Age: 18, Birthday: time.Now()}//实例化模型  
  
result := db.Create(&user) // 通过数据的指针来创建  
 
 
  
fmt.Println(result)  
}

运行之后可以看到数据库里面出现了user

image.png

多个增加

package main  
  
import (  
"fmt"  
"gorm.io/driver/mysql"  
"gorm.io/gorm"  
"time"  
)  
  
type User struct {  
ID uint  
Name string  
Age int  
Birthday time.Time  
}//表的模型  
  
func main() {  
  
db, _ := gorm.Open(mysql.New(mysql.Config{  
DSN: "root:Lycdemima1@@tcp(127.0.0.1:3306)/test?charset=utf8&parseTime=True&loc=Local", // DSN data source name  
DefaultStringSize: 256, // string 类型字段的默认长度  
DisableDatetimePrecision: true, // 禁用 datetime 精度,MySQL 5.6 之前的数据库不支持  
DontSupportRenameIndex: true, // 重命名索引时采用删除并新建的方式,MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引  
DontSupportRenameColumn: true, // 用 `change` 重命名列,MySQL 8 之前的数据库和 MariaDB 不支持重命名列  
SkipInitializeWithVersion: false, // 根据当前 MySQL 版本自动配置  
}), &gorm.Config{})
//以上为连接数据库,需要自己更改其中的相关配置,如root。
  
users := []User{  
User{Name: "xiaoming", Age: 18, Birthday: time.Now()},  
User{Name: "Jackson", Age: 19, Birthday: time.Now()},  
}  
  
result := db.Create(&users) // 创建表内容
  
fmt.Println(result)  
}

成功创建出数值 image.png

用指定的字段创造数据

可以确保只有这些特定的字段被插入到数据库中,而其他字段将被忽略。

package main  
  
import (  
"fmt"  
"gorm.io/driver/mysql"  
"gorm.io/gorm"  
"time"  
)  
  
type User struct {  
ID uint  
Name string  
Age int  
Birthday time.Time  
}  
  
func main() {  
  
db, _ := gorm.Open(mysql.New(mysql.Config{  
DSN: "root:Lycdemima1@@tcp(127.0.0.1:3306)/test?charset=utf8&parseTime=True&loc=Local", // DSN data source name  
DefaultStringSize: 256, // string 类型字段的默认长度  
DisableDatetimePrecision: true, // 禁用 datetime 精度,MySQL 5.6 之前的数据库不支持  
DontSupportRenameIndex: true, // 重命名索引时采用删除并新建的方式,MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引  
DontSupportRenameColumn: true, // 用 `change` 重命名列,MySQL 8 之前的数据库和 MariaDB 不支持重命名列  
SkipInitializeWithVersion: false, // 根据当前 MySQL 版本自动配置  
}), &gorm.Config{})  
  
db.AutoMigrate(&User{})//创建表  
  
// 使用指定字段创建记录  
user := User{  
Name: "Jinzhu",  
Age: 18,  
}  

result := db.Select("Name", "Age").Create(&user)  
  
fmt.Println(result)  
}

运行之后数据库会是这样的结果

image.png

创建钩子

在使用 GORM 进行数据库操作时,你可以利用钩子(Hooks)来在执行数据库操作的不同阶段插入自定义逻辑。

func (user *User) BeforeCreate(tx *gorm.DB) error {
    // 在记录被创建之前执行逻辑
    // 可以在此处设置创建时间等字段的值
    return nil
}

func (user *User) AfterCreate(tx *gorm.DB) error {
    // 在记录被创建之后执行逻辑
    // 可以在此处执行其他相关操作
    return nil
}

func (user *User) BeforeUpdate(tx *gorm.DB) error {
    // 在记录被更新之前执行逻辑
    // 可以在此处更新修改时间等字段的值
    return nil
}

func (user *User) AfterUpdate(tx *gorm.DB) error {
    // 在记录被更新之后执行逻辑
    // 可以在此处执行其他相关操作
    return nil
}

goland示例

package main  
  
import (  
"fmt"  
"gorm.io/driver/mysql"  
"gorm.io/gorm"  
"time"  
)  
  
type User struct {  
ID uint  
Name string  
Age int  
Birthday time.Time  
}  
  
func (user *User) BeforeCreate(tx *gorm.DB) error {  
fmt.Println("钩子函数运行了")  
return nil  
}//我们定义的钩子函数  
  
func main() {  
  
db, _ := gorm.Open(mysql.New(mysql.Config{  
DSN: "root:Lycdemima1@@tcp(127.0.0.1:3306)/test?charset=utf8&parseTime=True&loc=Local", // DSN data source name  
DefaultStringSize: 256, // string 类型字段的默认长度  
DisableDatetimePrecision: true, // 禁用 datetime 精度,MySQL 5.6 之前的数据库不支持  
DontSupportRenameIndex: true, // 重命名索引时采用删除并新建的方式,MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引  
DontSupportRenameColumn: true, // 用 `change` 重命名列,MySQL 8 之前的数据库和 MariaDB 不支持重命名列  
SkipInitializeWithVersion: false, // 根据当前 MySQL 版本自动配置  
}), &gorm.Config{})  
  
db.AutoMigrate(&User{})  
  
// 使用指定字段创建记录  
user := User{  
Name: "Jinzhu",  
Age: 18,  
}  
result := db.Select("Name", "Age").Create(&user)  
  
fmt.Println(result)  
}

运行之后,钩子函数先运行,之后才会创造记录

image.png

跳过钩子

如果你想在某些情况下跳过 GORM 钩子的执行,你可以使用 gorm.DB 对象的 SkipHooks 方法来实现

result := db.Session(&gorm.Session{SkipHooks: true}).Create(&user)

我们通过在 db.Session 调用中设置 SkipHooks 字段为 true,指示跳过所有钩子的执行。然后,我们调用 Create 方法来创建记录。

根据map创建记录

这个一般不常用,不过我们也需要了解一下

使用 GORM 批量插入记录可以通过传递 []map[string]interface{}[]Struct 来实现。

// 单个记录插入 
db.Model(&User{}).Create(map[string]interface{}{ "Name": "jinzhu", "Age": 18, }) 

// 批量插入 
db.Model(&User{}).Create([]map[string]interface{}{ {"Name": "jinzhu_1", "Age": 18}, {"Name": "jinzhu_2", "Age": 20}, })

默认值

使用gorm:"default:"标签来完成

type User struct {
  gorm.Model
  Name string
  Age  *int           `gorm:"default:18"`
  Active sql.NullBool `gorm:"default:true"
 }

这样如果我们不设置age和active,也会在记录里面分配值