mysql

github.com/taouniverse/tao-mysql

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

安装

go get github.com/taouniverse/tao-mysql

使用

导入

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

配置

# 单实例配置
mysql:
  host: localhost
  port: 3306
  user: tao
  password: 123456qwe
  db: test
  charset: utf8mb4,utf8
  parse_time: true
  local: UTC

# 多实例配置（如主从分离）
mysql:
  default_instance: master
  master:
    host: localhost
    port: 3306
    user: tao
    password: 123456qwe
    db: mydb
  replica:
    host: backup.example.com
    port: 3306
    user: readonly
    password: ro_pass
    db: mydb

配置字段说明

工厂模式 API

使用示例

获取连接并执行操作

package main

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

func main() {
    // 获取默认实例
    db, err := mysql.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("MySQL 连接成功")
}

GORM 操作

db, _ := mysql.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)

多实例使用

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

// 获取从库连接（只读）
replica, _ := mysql.GetDB("replica")

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

// 从库查询
var orders []Order
replica.Find(&orders)

单元测试

快速测试（无需 Docker）

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

完整集成测试（需要 Docker）

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

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

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

# 生成覆盖率报告
make coverage

# 停止 MySQL 服务
make down

手动测试

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

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

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

# 4. 停止 MySQL
docker-compose down

开发指南