第六届字节跳动青训营第8课｜青训营（实践篇02）2.然后接下来就是定义数据模型：创建一个结构体来定义学生模型，他的

使用 GORM（Go 的 ORM 库）连接数据库，并实现增删改查操作

1.那么关于使用GORM（Go 的 ORM 库）连接数据库的实践示范，我们首先就拿最常见的一个学生表进行他的一个增删改查。前提：学生表的名字为 student，然后学生数据库的字段有他的姓名name，还有年龄age，并且还有班级class和学号id。 1. 使用 GORM 连接学生数据库并执行增删改查操作需要按照以下步骤进行。假设你已经安装好了 Go 和 GORM 库。第一部分就是导入必要的库首先，你需要导入 GORM 和数据库驱动（比如 SQLite、MySQL 等）的库。在你的 Go 代码文件中，替换为你使用的数据库驱动，添加以下导入语句

package main

import (
	"fmt"
	"gorm.io/driver/sqlite" 
	"gorm.io/gorm"
)

2.然后接下来就是定义数据模型：创建一个结构体来定义学生模型，他的姓名name，年龄age，还有班级class和学号id,对应数据库中的表。在这个结构体中，你可以定义字段名、数据类型和 GORM 的标签（如：gorm:"primaryKey"、gorm:"autoIncrement" 等）来映射数据库表结构。

type Student struct {
	ID     uint   `gorm:"primaryKey;autoIncrement"`
	Name   string
	Age    int
	Class  string
}

3.初始化数据库连接：使用 GORM 连接数据库，并进行初始化设置。数据库文件名或连接字符串，根据你的数据库配置来设置，你可以像下面这样

func main() {
	dsn := "student.db" 

	db, err := gorm.Open(sqlite.Open(dsn), &gorm.Config{})
	if err != nil {
		panic("Failed to connect to database")
	}

	// 自动迁移数据库表结构
	err = db.AutoMigrate(&Student{})
	if err != nil {
		panic("Failed to migrate database")
	}

	fmt.Println("Database connected")
}

然后就是增加数据（Create）：使用 GORM 创建新的学生记录。

func createStudent(db *gorm.DB, name string, age int, class string) {
	student := Student{
		Name:  name,
		Age:   age,
		Class: class,
	}
	
	db.Create(&student)
	fmt.Println("Student created:", student)
}

接着查询数据（Read）：使用 GORM 查询学生记录

func getStudentByID(db *gorm.DB, id uint) {
	var student Student
	result := db.First(&student, id)
	if result.Error != nil {
		fmt.Println("Student not found")
		return
	}

	fmt.Println("Student found:", student)
}

更新数据（Update）：使用 GORM 更新学生记录。

func updateStudentAge(db *gorm.DB, id uint, newAge int) {
	var student Student
	result := db.First(&student, id)
	if result.Error != nil {
		fmt.Println("Student not found")
		return
	}

	student.Age = newAge
	db.Save(&student)
	fmt.Println("Student updated:", student)
}

删除数据（Delete）：使用 GORM 删除学生记录。

	var student Student
	result := db.First(&student, id)
	if result.Error != nil {
		fmt.Println("Student not found")
		return
	}

	db.Delete(&student)
	fmt.Println("Student deleted:", student)
}

调用函数执行操作：在 main 函数中，你可以调用上述定义的函数来执行对数据库的增删改查操作。

	// ... 初始化数据库连接 ...

	createStudent(db, "Alice", 18, "Class A")
	getStudentByID(db, 1)
	updateStudentAge(db, 1, 19)
	deleteStudent(db, 1)
}

请注意，上述代码是一个基本示例，你可以根据实际需求进行调整和扩展。另外，确保在使用真实数据库时提供正确的数据库连接配置信息。最后：当使用 GORM 连接数据库并执行增删改查操作时，有一些需要注意的事项：

1.错误处理：务必对数据库操作中的错误进行适当的处理。数据库操作可能因为连接问题、数据格式不正确等原因导致错误，应该适当地处理这些错误，避免程序崩溃或不稳定。

2.数据库迁移：在代码中使用 AutoMigrate 函数可以自动迁移数据库表结构，但需要注意生产环境中的数据库迁移策略。手动修改数据库结构时，确保在不影响数据完整性的前提下进行修改。

3.事务处理：在某些情况下，你可能需要使用事务来确保一系列数据库操作的原子性。GORM 支持事务处理，可以通过 db.Begin() 开始事务，然后使用 tx.Commit() 提交事务或 tx.Rollback() 回滚事务。

4.模型定义：在定义模型结构体时，需要注意字段名、数据类型以及 GORM 标签的使用。GORM 的标签用于指定主键、自增等属性，确保它们与数据库表的定义相匹配。

5.性能优化：GORM 默认会执行一些预加载（eager loading）操作，但在处理大量数据时可能导致性能问题。可以使用 .Preload() 和 .Select() 等方法来优化查询性能。

6.数据库连接管理：数据库连接是有限资源，确保在操作结束后关闭连接。如果是长时间运行的应用程序，可以考虑使用连接池来管理数据库连接。

7.数据验证：在执行插入和更新操作之前，对数据进行验证是一个良好的实践。确保输入的数据满足预期的格式和约束，以避免数据不一致或错误。

8.安全性：避免将用户输入直接拼接到 SQL 查询中，以防止 SQL 注入攻击。GORM 通过参数绑定等方式来防止 SQL 注入。

9.并发处理：如果你的应用程序需要处理并发请求，确保数据库操作是线程安全的。GORM 默认是 goroutine 安全的，但需要适当地管理数据库连接和事务。

第六届字节跳动青训营第8课 ｜ 青训营（实践篇02）

第六届字节跳动青训营第8课｜青训营（实践篇02）