what
1322280daf
- 新增 pending 缓冲区，publish 时若无订阅者则暂存消息
- subscribe 时自动将缓冲消息投入 channel，解决服务重启后恢复任务丢失的问题
- 去除 broadcast 5ms 超时导致的消息丢失
- chan bool 改为 chan struct{}，RWMutex 改为 Mutex
- 新增 broker_test.go，12 个单元测试覆盖核心场景（含 -race）
- 为 client_test.go 中的无限循环 demo 添加 t.Skip()
98 lines
1.7 KiB
Go
98 lines
1.7 KiB
Go
package queue
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
const topic = "Golang梦工厂"
|
|
|
|
// 一个topic 测试
|
|
func TestOnceTopic(t *testing.T) {
|
|
t.Skip("infinite loop demo, not a unit test")
|
|
m := NewClient()
|
|
defer m.Close()
|
|
m.SetConditions(10)
|
|
ch, err := m.Subscribe(topic)
|
|
if err != nil {
|
|
fmt.Println("subscribe failed")
|
|
return
|
|
}
|
|
go OncePub(m)
|
|
OnceSub(ch, m)
|
|
}
|
|
|
|
// 定时推送
|
|
func OncePub(c *Client) {
|
|
t := time.NewTicker(1 * time.Second)
|
|
defer t.Stop()
|
|
for {
|
|
select {
|
|
case <-t.C:
|
|
err := c.Publish(topic, "test messs")
|
|
if err != nil {
|
|
fmt.Println("pub message failed")
|
|
}
|
|
default:
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
// 接受订阅消息
|
|
func OnceSub(m <-chan interface{}, c *Client) {
|
|
for {
|
|
val := c.GetPayload(m)
|
|
fmt.Printf("get message is %s\n", val)
|
|
}
|
|
}
|
|
|
|
// 多个topic测试
|
|
func TestManyTopic(t *testing.T) {
|
|
t.Skip("infinite loop demo, not a unit test")
|
|
m := NewClient()
|
|
defer m.Close()
|
|
m.SetConditions(10)
|
|
top := ""
|
|
for i := 0; i < 10; i++ {
|
|
top = fmt.Sprintf("Golang梦工厂_%02d", i)
|
|
go Sub(m, top)
|
|
}
|
|
ManyPub(m)
|
|
}
|
|
|
|
func ManyPub(c *Client) {
|
|
t := time.NewTicker(10 * time.Second)
|
|
defer t.Stop()
|
|
for {
|
|
select {
|
|
case <-t.C:
|
|
for i := 0; i < 10; i++ {
|
|
//多个topic 推送不同的消息
|
|
top := fmt.Sprintf("Golang梦工厂_%02d", i)
|
|
payload := fmt.Sprintf("asong真帅_%02d", i)
|
|
err := c.Publish(top, payload)
|
|
if err != nil {
|
|
fmt.Println("pub message failed")
|
|
}
|
|
}
|
|
default:
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
func Sub(c *Client, top string) {
|
|
ch, err := c.Subscribe(top)
|
|
if err != nil {
|
|
fmt.Printf("sub top:%s failed\n", top)
|
|
}
|
|
for {
|
|
val := c.GetPayload(ch)
|
|
if val != nil {
|
|
fmt.Printf("%s get message is %s\n", top, val)
|
|
}
|
|
}
|
|
}
|