nats

github.com/taouniverse/tao-nats

Tao Universe 组件单元（Unit），基于泛型工厂模式封装 NATS 消息系统。

安装

go get github.com/taouniverse/tao-nats

使用

导入

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

配置

# 单实例配置
nats:
  url: nats://localhost:4222

# 多实例配置（如连接多个 NATS 集群）
nats:
  default_instance: primary
  primary:
    url: nats://primary:4222
    user: admin
    password: secret
  secondary:
    url: nats://backup:4222
    token: my-token

配置字段说明

工厂模式 API

使用示例

获取连接并发布消息

package main

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

func main() {
    // 获取默认实例
    conn, err := nats.Nats()
    if err != nil {
        log.Fatal(err)
    }
    
    // 发布消息
    err = conn.Publish("orders.created", []byte(`{"id": "123"}`))
    if err != nil {
        log.Fatal(err)
    }
}

订阅消息

conn, _ := nats.Nats()

// 普通订阅（所有消费者都收到）
sub, err := conn.Subscribe("orders.created", func(msg *nats.Msg) {
    log.Printf("Received: %s", msg.Data)
})
if err != nil {
    log.Fatal(err)
}
defer sub.Unsubscribe()

// 队列订阅（负载均衡，只有一个消费者收到）
queueSub, err := conn.QueueSubscribe("orders.process", "workers", func(msg *nats.Msg) {
    log.Println("Processing order...")
})
if err != nil {
    log.Fatal(err)
}
defer queueSub.Unsubscribe()

请求-响应模式

conn, _ := nats.Nats()

// 发送请求并等待响应
msg, err := conn.Request("inventory.check", []byte(`{"sku": "abc"}`), 2*time.Second)
if err != nil {
    log.Fatal(err)
}
log.Printf("Response: %s", msg.Data)

// 响应请求
conn.Subscribe("inventory.check", func(msg *nats.Msg) {
    msg.Respond([]byte(`{"available": true}`))
})

多实例使用

// 获取不同集群的连接
primary, _ := nats.GetNats("primary")
backup, _ := nats.GetNats("backup")

// 向主集群发布
primary.Publish("events", data)

// 向备份集群发布
backup.Publish("events", data)

单元测试

快速测试（无需 Docker）

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

完整集成测试（需要 Docker）

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

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

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

# 生成覆盖率报告
make coverage

# 停止 NATS 服务
make down

手动测试

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

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

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

# 4. 停止 NATS
docker-compose down

开发指南