实现多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)
}
相关推荐
G探险者31 分钟前
为何一个系统上线要经过N轮测试?带你看懂企业级发布体系
后端
lang201509282 小时前
Spring Boot 入门:5分钟搭建Hello World
java·spring boot·后端
间彧3 小时前
Windows Server,如何使用WSFC+nginx实现集群故障转移
后端
间彧3 小时前
Nginx + Keepalived 实现高可用集群(Linux下)
后端
间彧3 小时前
在Kubernetes中如何部署高可用的Nginx Ingress Controller?
后端
间彧3 小时前
Ribbon负载均衡器和Nginx负载均衡器有什么区别
后端
间彧3 小时前
Nacos详解与项目实战
后端
间彧3 小时前
nginx、网关Gateway、Nacos、多个服务实例之间的数据链路详解
后端
间彧3 小时前
Nacos与Eureka在性能上有哪些具体差异?
后端