Golang PostgreSQL 数据库驱动 pq 使用指南

pq 是 Go 语言中最流行的 PostgreSQL 数据库驱动之一，它实现了 database/sql 接口，提供了稳定可靠的 PostgreSQL 数据库连接功能。

安装 pq 驱动

go get github.com/lib/pq

基本使用

1. 连接数据库

package main

import (
	"database/sql"
	"fmt"
	"log"

	_ "github.com/lib/pq"
)

func main() {
	// 连接字符串格式: "postgres://用户名:密码@主机:端口/数据库名?sslmode=disable"
	connStr := "user=postgres password=yourpassword dbname=testdb sslmode=disable"
	db, err := sql.Open("postgres", connStr)
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	// 测试连接
	err = db.Ping()
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("Successfully connected to PostgreSQL!")
}

2. 执行查询

// 查询单行
func querySingleRow(db *sql.DB) {
	var id int
	var name string
	err := db.QueryRow("SELECT id, name FROM users WHERE id = $1", 1).Scan(&id, &name)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Printf("ID: %d, Name: %s\n", id, name)
}

// 查询多行
func queryMultipleRows(db *sql.DB) {
	rows, err := db.Query("SELECT id, name FROM users WHERE age > $1", 18)
	if err != nil {
		log.Fatal(err)
	}
	defer rows.Close()

	for rows.Next() {
		var id int
		var name string
		err = rows.Scan(&id, &name)
		if err != nil {
			log.Fatal(err)
		}
		fmt.Printf("ID: %d, Name: %s\n", id, name)
	}

	// 检查遍历过程中是否有错误
	err = rows.Err()
	if err != nil {
		log.Fatal(err)
	}
}

3. 执行插入、更新和删除

// 插入数据
func insertData(db *sql.DB) {
	_, err := db.Exec("INSERT INTO users(name, age) VALUES($1, $2)", "Alice", 25)
	if err != nil {
		log.Fatal(err)
	}
}

// 更新数据
func updateData(db *sql.DB) {
	res, err := db.Exec("UPDATE users SET age = $1 WHERE name = $2", 26, "Alice")
	if err != nil {
		log.Fatal(err)
	}
	
	// 获取影响的行数
	rowsAffected, err := res.RowsAffected()
	if err != nil {
		log.Fatal(err)
	}
	fmt.Printf("Updated %d rows\n", rowsAffected)
}

// 删除数据
func deleteData(db *sql.DB) {
	_, err := db.Exec("DELETE FROM users WHERE name = $1", "Alice")
	if err != nil {
		log.Fatal(err)
	}
}

高级特性

1. 事务处理

func transferMoney(db *sql.DB, from, to int, amount float64) error {
	tx, err := db.Begin()
	if err != nil {
		return err
	}
	
	// 确保事务在函数返回前被处理
	defer func() {
		if err != nil {
			tx.Rollback()
			return
		}
		err = tx.Commit()
	}()

	// 从转出账户扣款
	_, err = tx.Exec("UPDATE accounts SET balance = balance - $1 WHERE id = $2", amount, from)
	if err != nil {
		return err
	}

	// 向转入账户加款
	_, err = tx.Exec("UPDATE accounts SET balance = balance + $1 WHERE id = $2", amount, to)
	if err != nil {
		return err
	}

	return nil
}

2. 预处理语句

func preparedStatement(db *sql.DB) {
	stmt, err := db.Prepare("SELECT name FROM users WHERE age > $1")
	if err != nil {
		log.Fatal(err)
	}
	defer stmt.Close()

	rows, err := stmt.Query(18)
	if err != nil {
		log.Fatal(err)
	}
	defer rows.Close()

	for rows.Next() {
		var name string
		err = rows.Scan(&name)
		if err != nil {
			log.Fatal(err)
		}
		fmt.Println(name)
	}
}

3. 处理 JSON 数据

PostgreSQL 支持 JSON 数据类型，pq 驱动可以很好地处理：

type User struct {
	Name string `json:"name"`
	Age  int    `json:"age"`
}

func handleJSON(db *sql.DB) {
	// 插入 JSON 数据
	user := User{Name: "Bob", Age: 30}
	_, err := db.Exec("INSERT INTO users(data) VALUES($1)", user)
	if err != nil {
		log.Fatal(err)
	}

	// 查询 JSON 数据
	var data []byte
	err = db.QueryRow("SELECT data FROM users WHERE id = $1", 1).Scan(&data)
	if err != nil {
		log.Fatal(err)
	}

	var retrievedUser User
	err = json.Unmarshal(data, &retrievedUser)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Printf("Retrieved user: %+v\n", retrievedUser)
}

连接池配置

func setupConnectionPool(db *sql.DB) {
	// 设置最大空闲连接数
	db.SetMaxIdleConns(10)
	
	// 设置最大打开连接数
	db.SetMaxOpenConns(100)
	
	// 设置连接最大存活时间
	db.SetConnMaxLifetime(time.Hour)
}

注意事项

1. 总是检查错误并处理
2. 记得关闭连接和资源（rows, stmt, tx 等）
3. 使用参数化查询防止 SQL 注入
4. 连接字符串中的密码等敏感信息应该通过环境变量获取

pq 驱动提供了丰富的功能来与 PostgreSQL 交互，以上示例涵盖了大部分常见使用场景。对于更复杂的需求，可以参考 pq 的官方文档。