postgres

github.com/taouniverse/tao-postgres

Tao Universe 组件单元（Unit），基于泛型工厂模式封装 PostgreSQL 数据库。

安装

go get github.com/taouniverse/tao-postgres

使用

导入

import _ "github.com/taouniverse/tao-postgres"

配置

# 单实例配置
postgres:
  host: localhost
  port: 5432
  user: tao
  password: 123456qwe
  db: test
  ssl: disable
  time_zone: Asia/Shanghai

# 多实例配置（如主从分离）
postgres:
  default_instance: primary
  primary:
    host: localhost
    port: 5432
    user: tao
    password: 123456qwe
    db: mydb
  standby:
    host: backup.example.com
    port: 5432
    user: readonly
    password: ro_pass
    db: mydb

配置字段说明

工厂模式 API

使用示例

获取连接并执行操作

package main

import (
    "log"
    
    "github.com/taouniverse/tao-postgres"
)

func main() {
    // 获取默认实例
    db, err := postgres.DB()
    if err != nil {
        log.Fatal(err)
    }
    
    // 获取底层 sql.DB 进行 Ping 测试
    sqlDB, err := db.DB()
    if err != nil {
        log.Fatal(err)
    }
    
    err = sqlDB.Ping()
    if err != nil {
        log.Fatal(err)
    }
    log.Println("PostgreSQL 连接成功")
}

GORM 操作

db, _ := postgres.DB()

// 自动迁移
db.AutoMigrate(&User{})

// 创建记录
db.Create(&User{Name: "tao", Age: 18})

// 查询记录
var user User
db.First(&user, "name = ?", "tao")

// 更新记录
db.Model(&user).Update("age", 20)

// 删除记录
db.Delete(&user)

多实例使用

// 获取主库连接（读写）
primary, _ := postgres.GetDB("primary")

// 获取备库连接（只读）
standby, _ := postgres.GetDB("standby")

// 主库写入
primary.Create(&Order{Amount: 100})

// 备库查询
var orders []Order
standby.Find(&orders)

单元测试

快速测试（无需 Docker）

# 仅运行配置相关测试
go test -v -run "TestConfig" ./...

完整集成测试（需要 Docker）

# 启动 PostgreSQL 并运行单实例测试
make test

# 启动 PostgreSQL 并运行多实例测试
make test-multi

# 启动 PostgreSQL 并运行所有测试
make test-all

# 生成覆盖率报告
make coverage

# 停止 PostgreSQL 服务
make down

手动测试

# 1. 启动 PostgreSQL
docker-compose up -d

# 2. 运行单实例测试
go test -v ./...

# 3. 运行多实例测试
TAO_TEST_MULTI_INSTANCE=true go test -v ./...

# 4. 停止 PostgreSQL
docker-compose down

开发指南