快捷導(dǎo)航

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

更新時間：2024年08月10日 11:32:17 作者：PHPz

在go中集成postgresql需使用gorm orm,步驟如下：安裝go和postgresql,安裝 gorm：go get -u gorm.io/gorm,配置數(shù)據(jù)庫連接字符串,定義模型類,遷移數(shù)據(jù)庫架構(gòu),使用 gorm 進行增刪改查操作,本指南將介紹如何使用 GORM（一個廣受歡迎的 ORM）,將PostgreSQL集成到你的Go應(yīng)用中

先決條件

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

GORM 安裝

go get -u gorm.io/gorm

連接到 PostgreSQL

首先，你需要配置一個數(shù)據(jù)庫連接字符串，該字符串指定了你的 PostgreSQL 數(shù)據(jù)庫的信息：

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)
}

創(chuàng)建模型

接下來，你需要定義一個模型，它代表你的數(shù)據(jù)庫表：

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

遷移數(shù)據(jù)庫

在向數(shù)據(jù)庫中插入數(shù)據(jù)之前，你需要遷移數(shù)據(jù)庫架構(gòu)：

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

增刪改查操作

現(xiàn)在你可以開始對數(shù)據(jù)庫進行增刪改查操作了：

創(chuàng)建（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)
```

實戰(zhàn)案例

讓我們創(chuàng)建一個簡單的 API，使用 GORM 和 PostgreSQL 來管理用戶：

func main() {
    // 初始化數(shù)據(jù)庫連接
    db := initDB()

    // 遷移數(shù)據(jù)庫架構(gòu)
    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 提供了以下端點：