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

鬼寞
54 阅读4分钟

连接数据库是大多数应用程序的基本需求之一。在Go语言中，我们可以使用GORM库来实现对象关系映射（ORM）并连接数据库。GORM是一个功能强大且易于使用的ORM库，它提供了简洁的API和灵活的查询语言，使得数据库操作变得更加简单和高效。

在这里我将介绍如何使用GORM连接数据库，并实现常见的增删改查操作。我们将使用MySQL作为示例数据库，但GORM也支持其他数据库如PostgreSQL、SQLite等。

第一步是安装GORM库。在你的Go项目中，可以使用go get命令来获取最新版本的GORM：

go get -u gorm.io/gorm

安装完成后，我们可以开始编写代码来连接数据库和执行数据库操作。

首先，我们需要导入GORM和相应的数据库驱动。在代码文件的开头添加以下导入语句：

import (
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

接下来，我们需要定义一个数据库连接函数，用于连接到数据库。我们将使用MySQL数据库作为示例，因此需要提供MySQL数据库的连接信息，如用户名、密码、主机和数据库名称。以下是一个示例的数据库连接函数：

func ConnectDB() (*gorm.DB, error) {
	dsn := "user:password@tcp(127.0.0.1:3306)/database_name?charset=utf8mb4&parseTime=True&loc=Local"
	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		return nil, err
	}
	return db, nil
}

在这个函数中，我们使用gorm.Open函数来建立与数据库的连接。mysql.Open函数是GORM提供的MySQL数据库驱动。

在连接字符串（dsn）中，你需要替换user、password和database_name为你的实际数据库用户名、密码和数据库名称。127.0.0.1:3306表示数据库的主机和端口。

注意，我们还指定了一些参数，如charset=utf8mb4表示使用UTF-8字符集，parseTime=True表示将数据库中的时间字段解析为Go的time.Time类型，loc=Local表示使用本地时区。

现在我们已经有了一个连接到数据库的函数，接下来我们可以使用该函数来执行数据库操作。

创建模型

在执行数据库操作之前，我们需要定义模型结构。模型结构对应数据库中的表结构，使用GORM可以轻松地将模型与数据库表进行映射。

例如，我们创建一个名为User的模型来表示用户表：

type User struct {
	gorm.Model
	Name  string
	Email string
}

在这个例子中，User模型包含gorm.Model作为嵌入字段，它提供了一些通用的字段，如ID、CreatedAt、UpdatedAt和DeletedAt。Name和Email字段分别表示用户的姓名和电子邮件。

创建记录

现在我们可以使用GORM创建新的数据库记录。以下是一个示例函数来创建一个新的用户记录：

func CreateUser(name, email string) error {
	db, err := ConnectDB()
	if err != nil {
		return err
	}
	defer db.Close()

	user := User{Name: name, Email: email}
	result := db.Create(&user)
	if result.Error != nil {
		return result.Error
	}
	return nil
}

在这个函数中，我们首先通过调用ConnectDB函数连接到数据库。然后，我们定义一个User结构的实例，并设置其Name和Email字段。最后，我们使用db.Create方法将用户记录插入到数据库中。

查询记录

接下来，让我们实现一个查询用户记录的函数。以下是一个示例函数来查询名为John的用户记录：

func GetUserByName(name string) (*User, error) {
	db, err := ConnectDB()
	if err != nil {
		return nil, err
	}
	defer db.Close()

	var user User
	result:= db.Where("name = ?", name).First(&user)
	if result.Error != nil {
		return nil, result.Error
	}
	return &user, nil
}

在这个函数中，我们首先连接到数据库，然后定义一个User结构的变量user用于存储查询结果。我们使用db.Where("name = ?", name)来指定查询条件，然后使用First方法执行查询操作。

更新记录

要更新数据库中的记录，我们可以使用db.Save方法。以下是一个示例函数来更新用户的电子邮件：

func UpdateUserEmail(name, email string) error {
	db, err := ConnectDB()
	if err != nil {
		return err
	}
	defer db.Close()

	var user User
	result := db.Where("name = ?", name).First(&user)
	if result.Error != nil {
		return result.Error
	}

	user.Email = email
	result = db.Save(&user)
	if result.Error != nil {
		return result.Error
	}
	return nil
}

在这个函数中，我们首先连接到数据库，然后使用db.Where("name = ?", name).First(&user)查询要更新的用户记录。然后，我们更新user的Email字段，并使用db.Save方法保存更改。

删除记录

要删除数据库中的记录，我们可以使用db.Delete方法。以下是一个示例函数来删除名为John的用户记录：

func DeleteUserByName(name string) error {
	db, err := ConnectDB()
	if err != nil {
		return err
	}
	defer db.Close()

	var user User
	result := db.Where("name = ?", name).First(&user)
	if result.Error != nil {
		return result.Error
	}

	result = db.Delete(&user)
	if result.Error != nil {
		return result.Error
	}
	return nil
}

在这个函数中，我们首先连接到数据库，然后使用db.Where("name = ?", name).First(&user)查询要删除的用户记录。然后，我们使用db.Delete方法删除记录。

总结

通过使用GORM库，我们可以轻松地连接到数据库并执行常见的增删改查操作。在本文中，我们了解了如何使用GORM连接到MySQL数据库，并实现了创建记录、查询记录、更新记录和删除记录的函数。不仅如此，GORM还提供了更多高级功能，如事务处理、关联查询等，可以帮助我们更好地管理数据库操作。

avatar
文章
阅读
粉丝