使用 GORM（Go 的 ORM 库）连接数据库，并实现增删改查操作，把实现过程整理成文章 ｜ 青训营

在 Go 中使用 GORM 连接数据库并进行增删改查操作

GORM 是 Go 语言中一个强大的 ORM（Object-Relational Mapping）库，它提供了简单易用的 API，可以帮助开发者更轻松地与数据库交互。本文将指导您如何使用 GORM 连接数据库并进行常见的增删改查操作。

步骤 1：安装 GORM

首先，您需要安装 GORM 包。在终端中运行以下命令：

go get -u gorm.io/gorm
go get -u gorm.io/driver/<YOUR_DATABASE_DRIVER>

请将 <YOUR_DATABASE_DRIVER> 替换为您要使用的数据库驱动，例如 "mysql"、"postgres" 等。

步骤 2：连接数据库

导入 GORM 包并建立数据库连接。在您的 Go 代码文件中，添加以下内容：

package main

import (
	"gorm.io/driver/<YOUR_DATABASE_DRIVER>"
	"gorm.io/gorm"
)

func main() {
	dsn := "<YOUR_DATABASE_CONNECTION_STRING>"
	db, err := gorm.Open(<YOUR_DATABASE_DRIVER>.Open(dsn), &gorm.Config{})
	if err != nil {
		panic("Failed to connect to database")
	}

	// 在这里进行增删改查操作
}

将 <YOUR_DATABASE_CONNECTION_STRING> 替换为您的数据库连接字符串。

步骤 3：定义模型

使用 GORM，您需要定义模型来映射数据库表。例如，如果您要操作一个名为 "users" 的表，可以创建一个与之对应的模型：

type User struct {
	ID       uint   `gorm:"primaryKey"`
	Name     string
	Email    string
}

步骤 4：增删改查操作

现在，您可以使用 GORM 提供的方法执行增删改查操作。

• 插入数据：

newUser := User{Name: "John Doe", Email: "john@example.com"}
db.Create(&newUser)

• 查询数据：

var user User
db.First(&user, 1) // 根据 ID 查询第一条记录
db.Where("name = ?", "John").Find(&user) // 根据条件查询多条记录

• 更新数据：

db.Model(&user).Update("Email", "new_email@example.com")

• 删除数据：

db.Delete(&user, 1) // 根据 ID 删除记录

步骤 5：关闭数据库连接

在程序结束时，务必关闭数据库连接：

sqlDB, err := db.DB()
if err != nil {
    panic("Failed to get DB instance")
}
sqlDB.Close()

完整示例

package main

import (
	"fmt"
	"gorm.io/driver/<YOUR_DATABASE_DRIVER>"
	"gorm.io/gorm"
)

type User struct {
	ID    uint   `gorm:"primaryKey"`
	Name  string
	Email string
}

func main() {
	dsn := "<YOUR_DATABASE_CONNECTION_STRING>"
	db, err := gorm.Open(<YOUR_DATABASE_DRIVER>.Open(dsn), &gorm.Config{})
	if err != nil {
		panic("Failed to connect to database")
	}

	// AutoMigrate 将自动创建表格，如果已存在则不会重新创建
	db.AutoMigrate(&User{})

	// 插入数据
	newUser := User{Name: "John Doe", Email: "john@example.com"}
	db.Create(&newUser)

	// 查询数据
	var user User
	db.First(&user, 1)
	fmt.Println(user)

	// 更新数据
	db.Model(&user).Update("Email", "new_email@example.com")

	// 删除数据
	db.Delete(&user, 1)

	// 关闭数据库连接
	sqlDB, err := db.DB()
	if err != nil {
		panic("Failed to get DB instance")
	}
	sqlDB.Close()
}

以上就是使用 GORM 在 Go 中连接数据库并进行增删改查操作的基本步骤。希望这篇文章能够帮助您快速上手使用 GORM 进行数据库操作。