当前位置：首页 > news >正文

深入学习 GORM：记录插入与数据检索

news 2025/6/30 12:57:36

引言

在使用 GORM 进行数据库操作时，掌握如何高效地插入记录和检索数据是非常重要的。本文将详细介绍通过 Create 方法插入记录、批量插入、以及各种数据检索方法，并结合实际示例进行讲解。

一、通过 Create 方法插入记录

GORM 提供了简单易用的 Create 方法来插入单条记录。以下是一个基本示例：

type User struct {ID   uintName string
}var user = User{Name: "jinzhu"}
db.Create(&user)

执行上述代码后，user.ID 将被自动设置为新插入记录的主键值。

二、批量插入和通过 map 插入记录

批量插入

当你需要插入大量数据时，可以使用 CreateInBatches 方法来指定批量插入的批次大小，以提高性能。

示例 1：批量插入用户数据

var users = []User{{Name: "jinzhu1"}, {Name: "jinzhu2"}, {Name: "jinzhu3"}}
db.Create(&users)for _, user := range users {fmt.Println(user.ID) // 输出插入后的ID
}

示例 2：大批量插入用户数据

var users = []User{{Name: "jinzhu_1"},...,{Name: "jinzhu_10000"},
}// batch size 100
db.CreateInBatches(users, 100)

根据 Map 创建记录

除了直接使用结构体插入数据外，还可以通过 map[string]interface{} 来创建记录。

示例 1：单个记录

db.Model(&User{}).Create(map[string]interface{}{"Name": "jinzhu", "Age": 18,
})

示例 2：批量记录

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

三、通过 Take、First、Last 获取数据

GORM 提供了多种方法来检索数据，包括 Take、First 和 Last 等。

自动迁移表结构

在开始查询之前，确保已经创建了相应的表结构。

db.AutoMigrate(&User{})

使用 First 查询数据

First 方法用于获取第一条记录，默认按照主键升序排序。

var user User
result := db.First(&user, 1) // 根据主键查询
fmt.Println("User ID:", user.ID)if errors.Is(result.Error, gorm.ErrRecordNotFound) {fmt.Println("record not found")
}

使用 Take 获取一条记录

Take 方法用于获取任意一条记录，没有指定排序字段。

result := db.Take(&user, 10)
if errors.Is(result.Error, gorm.ErrRecordNotFound) {fmt.Println("record not found")
}

使用 Last 获取最后一条记录

Last 方法用于获取最后一条记录，默认按照主键降序排序。

db.Last(&user)
// SELECT * FROM users ORDER BY id DESC LIMIT 1;

根据主键检索

可以通过 Find 方法根据主键检索多条记录。

db.Find(&user, []int{1, 2, 3})
// SELECT * FROM users WHERE id IN (1,2,3);

检索全部对象

Find 方法也可以用于检索所有记录。

var users []User
result := db.Find(&users)
fmt.Println("Total records:", result.RowsAffected)for _, user := range users {fmt.Println("User ID:", user.ID)
}