如何将 PostgreSQL 集成到 Go 框架中？-Golang-PHP中文网

如何将 PostgreSQL 集成到 Go 框架中？

PHPz

发布： 2024-08-10 11:03:03

原创

523人浏览过

在 go 中集成 postgresql 需使用 gorm orm。步骤如下：安装 go 和 postgresql。安装 gorm：go get -u gorm.io/gorm。配置数据库连接字符串。定义模型类。迁移数据库架构。使用 gorm 进行增删改查操作。

如何将 PostgreSQL 集成到 Go 框架中？

使用 GORM 在 Go 中集成 PostgreSQL

在 Go 应用中使用 PostgreSQL 数据库是一种常见的需求。本指南将介绍如何使用 GORM（一个广受欢迎的 ORM），将 PostgreSQL 集成到你的 Go 应用中。

先决条件

安装 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 提供了以下端点：