mongodb

github.com/taouniverse/tao-mongodb

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

安装

go get github.com/taouniverse/tao-mongodb

使用

导入

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

配置

# 单实例配置
mongodb:
  host: localhost
  port: 27017
  user: tao
  password: 123456qwe
  db: test

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

配置字段说明

工厂模式 API

使用示例

获取客户端并执行操作

package main

import (
    "context"
    "log"
    "time"
    
    "github.com/taouniverse/tao-mongodb"
)

func main() {
    // 获取默认实例
    client, err := mongodb.Client()
    if err != nil {
        log.Fatal(err)
    }
    
    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
    defer cancel()
    
    // Ping 测试
    err = client.Ping(ctx, nil)
    if err != nil {
        log.Fatal(err)
    }
    log.Println("MongoDB 连接成功")
}

数据库操作

client, _ := mongodb.Client()
ctx := context.Background()

// 获取数据库和集合
db := client.Database("mydb")
coll := db.Collection("users")

// 插入文档
_, err := coll.InsertOne(ctx, map[string]interface{}{
    "name": "tao",
    "age": 18,
})

// 查询文档
var result map[string]interface{}
err = coll.FindOne(ctx, map[string]interface{}{"name": "tao"}).Decode(&result)

多实例使用

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

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

// 主库写入
master.Database("mydb").Collection("logs").InsertOne(ctx, logEntry)

// 从库查询
replica.Database("mydb").Collection("logs").Find(ctx, filter)

单元测试

快速测试（无需 Docker）

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

完整集成测试（需要 Docker）

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

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

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

# 生成覆盖率报告
make coverage

# 停止 MongoDB 服务
make down

手动测试

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

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

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

# 4. 停止 MongoDB
docker-compose down

开发指南