使用GORM将PostgreSQL集成到Go框架中

2024-08-10 17:49

短信预约 -IT技能 免费直播动态提醒

先决条件

安装 Go 1.16 或更高版本
安装 PostgreSQL

GORM 安装

go get -u gorm.io/gorm

连接到 PostgreSQL

首先，你需要配置一个数据库连接字符串，该字符串指定了你的 PostgreSQL 数据库的信息：

import (
    "fmt"
    "log"

    "gorm.io/driver/postgres"
    "gorm.io/gorm"
)

const (
    host     = "localhost"
    port     = 5432
    user     = "postgres"
    password = "mypassword"
    dbname   = "mydb"
)

dsn := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s", host, port, user, password, dbname)
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
if err != nil {
    log.Fatal(err)
}

创建模型

接下来，你需要定义一个模型，它代表你的数据库表：

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

迁移数据库

在向数据库中插入数据之前，你需要迁移数据库架构：

err = db.AutoMigrate(&User{})
if err != nil {
    log.Fatal(err)
}

增删改查操作

现在你可以开始对数据库进行增删改查操作了：

创建（Create）：

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

读取（Read）：
```
users := &[]User{}
db.Find(users)
```

更新（Update）：

db.Model(&user).Updates(User{Email: "new.email@example.com"})

删除（Delete）：
```
db.Delete(&user)
```

实战案例

让我们创建一个简单的 API，使用 GORM 和 PostgreSQL 来管理用户：

func main() {
    // 初始化数据库连接
    db := initDB()

    // 迁移数据库架构
    err := db.AutoMigrate(&User{})
    if err != nil {
        log.Fatal(err)
    }

    r := mux.NewRouter()
    r.HandleFunc("/users", handleUsers).Methods(http.MethodGet, http.MethodPost)
    r.HandleFunc("/users/{id}", handleUser).Methods(http.MethodGet, http.MethodPut, http.MethodDelete)

    log.Fatal(http.ListenAndServe(":8080", r))
}

func initDB() *gorm.DB {
    dsn := "user=postgres dbname=mydb password=mypassword sslmode=disable"
    db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
    if err != nil {
        log.Fatal(err)
    }
    return db
}

这个 API 提供了以下端点：