golang 中集成 postgresql 数据库安装:安装 postgresql 驱动、postgresql 数据库和 golang。创建连接池:管理与数据库的连接。执行查询:使用 db.query 方法执行查询。执行更新:使用 db.exec 方法执行更新(插入、更新、删除)。实战案例:使用 rest api 创建资产,并将其存储到 postgresql 数据库中。
Golang 框架如何集成 PostgreSQL 数据库
PostgreSQL 是一款功能强大的开源关系型数据库管理系统,以其可靠性和可扩展性而闻名。在 Golang 中集成 PostgreSQL 可以帮助你轻松地管理数据库连接、查询和数据操作。
先决条件
- 安装和配置 PostgreSQL 数据库
- 安装 Golang 1.18 或更高版本
安装 PostgreSQL 驱动
首先,你需要安装 PostgreSQL 驱动程序以连接到数据库:
$ go get -u github.com/jackc/pgx/v4
创建连接池
接下来,创建一个连接池以管理与数据库的连接。连接池可以提高性能并减少创建和销毁连接的开销:
立即学习“go语言免费学习笔记(深入)”;
import (
"context"
"database/sql"
"fmt"
"log"
"time"
"github.com/jackc/pgx/v4"
"github.com/jackc/pgx/v4/stdlib"
)
var db *sql.DB
func initDB() {
connectionString := fmt.Sprintf("user=root password=mysecret dbname=mydatabase host=localhost port=5432")
pool, err := pgx.ParseConfig(connectionString)
if err != nil {
log.Fatalf("unable to parse connection string: %v", err)
}
pool.MaxConnections = 10 // 根据需要调整最大连接数
pool.MaxConnLifetime = 5 * time.Minute
db = stdlib.OpenDB(*pool)
if err := db.Ping(context.Background()); err != nil {
log.Fatalf("unable to connect to database: %v", err)
}
}
执行查询
要执行查询,可以使用 db.Query 方法:
func queryAssets() ([]*Asset, error) {
rows, err := db.Query("SELECT * FROM assets")
if err != nil {
return nil, fmt.Errorf("query failed: %v", err)
}
defer rows.Close()
var assets []*Asset
for rows.Next() {
var asset Asset
if err := rows.Scan(&asset.ID, &asset.Name); err != nil {
return nil, fmt.Errorf("scan failed: %v", err)
}
assets = append(assets, &asset)
}
return assets, nil
}
执行更新
要执行更新(例如插入、更新或删除),可以使用 db.Exec 方法:
func insertAsset(asset Asset) (int64, error) {
result, err := db.Exec("INSERT INTO assets (name) VALUES (?)", asset.Name)
if err != nil {
return 0, fmt.Errorf("insert failed: %v", err)
}
return result.RowsAffected(), nil
}
实战案例
考虑一个示例应用程序,其中我们要将用户提交的数据存储到 PostgreSQL 数据库中:
package main
import (
"encoding/json"
"fmt"
"log"
"net/http"
"github.com/gorilla/mux"
)
type Asset struct {
ID int `json:"id"`
Name string `json:"name"`
}
func main() {
// 初始化数据库连接池
initDB()
r := mux.NewRouter()
r.HandleFunc("/assets", createAsset).Methods("POST")
log.Fatal(http.ListenAndServe(":8080", r))
}
func createAsset(w http.ResponseWriter, r *http.Request) {
var asset Asset
if err := json.NewDecoder(r.Body).Decode(&asset); err != nil {
http.Error(w, "Invalid request body", http.StatusBadRequest)
return
}
id, err := insertAsset(asset)
if err != nil {
http.Error(w, "Failed to create asset", http.StatusInternalServerError)
return
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(map[string]int64{"id": id})
}
这个应用程序允许用户通过一个 REST API 端点创建新资产,并将数据存储在 PostgreSQL 数据库中。