实现多goroutine之间的发布和订阅

实现多goroutine之间的发布和订阅

一、实现思路

  • 一个发布者,三个订阅者
  • 发布者需要跟每个订阅者之间,都要建立一个chan
  • 调用发布方法后,三个订阅者都能收到发布的信息
  • 在发布和接收之间,增加暂停,使运行结果更加直观

二、实现代码

复制代码
func Test_Main(t *testing.T) {
	sl := NewSubList()

	// 增加A的订阅
	sl.Subscribe("A", make(Broadcast))
	// 增加B的订阅
	sl.Subscribe("B", make(Broadcast))
	// 增加C的订阅
	sl.Subscribe("C", make(Broadcast))
	Sleep(3)
	sl.PublishMessage("Hello World!")
	Sleep(3)
	sl.PublishMessage("Golang very good!")

	sl.Unsubscribe("B")
	Sleep(3)
	sl.PublishMessage("Golang so easy!")
}

type Broadcast chan string

type SubList struct {
	CH map[string]Broadcast
	sync.RWMutex
}

func NewSubList() *SubList {
	s := &SubList{}
	s.CH = make(map[string]Broadcast) //所有channel
	return s
}

// Subscribe 订阅
func (s *SubList) Subscribe(name string, bc Broadcast) {
	s.Lock()
	s.CH[name] = bc
	s.Unlock()
	go s.ListeningBroadcast(name, bc)
}

// Unsubscribe 取消订阅
func (s *SubList) Unsubscribe(name string) {
	s.Lock()
	delete(s.CH, name)
	s.Unlock()
}

//发布消息
func (s *SubList) PublishMessage(message string) {
	s.RLock()
	for k, v := range s.CH {
		v <- message
		fmt.Println("向 ", k, " 发送 ", message)
	}
	s.RUnlock()
}

// ListeningBroadcast 监听信息
func (s *SubList) ListeningBroadcast(name string, bc Broadcast) {
	for {
		message := <-bc
		time.Sleep(time.Duration(1) * time.Second)
		fmt.Println(name, " 收到 ", message)
	}
}

func Sleep(i int64)  {
	fmt.Println("等待3秒......")
	time.Sleep(time.Duration(i) * time.Second)
}
相关推荐
paopaokaka_luck12 分钟前
智能推荐社交分享小程序(websocket即时通讯、协同过滤算法、时间衰减因子模型、热度得分算法)
数据库·vue.js·spring boot·后端·websocket·小程序
程序员NEO28 分钟前
Spring AI 对话记忆大揭秘:服务器重启,聊天记录不再丢失!
人工智能·后端
用户214118326360228 分钟前
惊爆!国内轻松白嫖 Claude Code,编程效率狂飙
后端
iccb101334 分钟前
我是如何实现在线客服系统的极致稳定性与安全性的
前端·javascript·后端
M1A140 分钟前
Java 面试系列第一弹:基础问题大盘点
java·后端·mysql
夕颜1111 小时前
关于 Cursor 小插曲记录
后端
考虑考虑1 小时前
go中的Map
后端·程序员·go
邓不利东2 小时前
Spring中过滤器和拦截器的区别及具体实现
java·后端·spring
头发那是一根不剩了2 小时前
Spring Boot 多数据源切换:AbstractRoutingDataSource
数据库·spring boot·后端