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

Go 是一种广泛使用的编程语言，其中之一的优点就是它的ORM（Object-Relational Mapping）库GORM。GORM可以轻松地将Go Struct对象转换为数据库表格，从而方便开发者进行数据库操作。本文将教你如何使用GORM连接数据库，并实现增、删、改、查等基本CRUD操作。

安装GORM

首先，我们需要在项目中安装GORM。可以使用以下命令在终端中安装：

go get -u github.com/go-gorm/gorm

Copied

创建数据模型

接下来，我们需要定义我们的数据模型。在GORM中，我们可以使用struct来定义数据模型。例如，我们可以创建一个名为user的struct，它包含三个属性：id、name和email。

type User struct {
    ID   int    `json:"id"`
    Name string `json:"name"`
    Email string `json:"email"`
}

Copied

连接数据库

接下来，我们需要连接到数据库。GORM支持多种数据库，包括PostgreSQL、MySQL、MongoDB等。这里我们使用的是PostgreSQL。我们可以使用以下代码来连接到数据库：

func ConnectDatabase() (*gorm.DB) {
    // 连接到 PostgreSQL 数据库
    db, err := gorm.Open(gorm.Postgres, "host=localhost port=5432 user=myuser password=mypassword dbname=mydb")
    if err != nil {
        log.Fatal("Connect Database failed", err)
    }
    return db
}

Copied

在上面的代码中，我们使用了gorm.Postgres来指定使用PostgreSQL数据库，然后提供了主机地址、端口、用户名、密码和数据库名称。如果连接成功，则返回一个*gorm.DB类型的变量，否则输出错误信息并终止程序。

创建数据表

接下来，我们需要创建数据表。GORM会根据我们的数据模型自动创建相应的数据表。我们可以使用以下代码来创建数据表：

func CreateTable() {
    // 创建数据表
    db := ConnectDatabase()
    defer db.Close()
    
    // 创建User表
    db.CreateTable(&User{})
}

Copied

在上面的代码中，我们使用了ConnectDatabase()函数来连接到数据库，然后使用CreateTable()函数来创建数据表。我们传入 &User{} 作为参数，这样GORM就知道要创建哪个表。

增加数据

接下来，我们可以使用GORM来增加数据。我们可以使用以下代码来添加一个新的用户：

func AddUser(name string, email string) {
    // 连接到数据库
    db := ConnectDatabase()
    defer db.Close()
    
    // 创建一个新的用户
    u := &User{Name: name, Email: email}
    db.Create(u)
}

Copied

在上面的代码中，我们使用了ConnectDatabase()函数来连接到数据atabase，然后使用Create()函数来创建一个新的用户。我们将Name和Email属性设置为 соответству的值，然后调用Create()函数来添加该用户到数据