Go 事件驱动编程:实现一个简单的事件总线
package main
import (
"fmt"
"sync"
)
// EventBus 定义了一个事件总线接口
type EventBus interface {
// Subscribe 方法用于订阅事件
Subscribe(topic string, subscriber func(data ...interface{})) int
// Publish 方法用于发布事件
Publish(topic string, data ...interface{})
}
// eventBus 实现了 EventBus 接口
type eventBus struct {
subscribers map[int]map[string][]func(data ...interface{})
subID int
subIDMutex sync.Mutex
}
// NewEventBus 创建一个新的事件总线
func NewEventBus() EventBus {
return &eventBus{
subscribers: make(map[int]map[string][]func(data ...interface{})),
}
}
// Subscribe 方法用于订阅事件
func (b *eventBus) Subscribe(topic string, subscriber func(data ...interface{})) int {
b.subIDMutex.Lock()
defer b.subIDMutex.Unlock()
b.subID++
if _, ok := b.subscribers[b.subID]; !ok {
b.subscribers[b.subID] = make(map[string][]func(data ...interface{}))
}
b.subscribers[b.subID][topic] = append(b.subscribers[b.subID][topic], subscriber)
return b.subID
}
// Publish 方法用于发布事件
func (b *eventBus) Publish(topic string, data ...interface{}) {
for _, subscribers := range b.subscribers {
if subscribersForTopic, ok := subscribers[topic]; ok {
for _, subscriber := range subscribersForTopic {
go subscriber(data...) // 在新的 goroutine 中调用订阅者函数
}
}
}
}
func main() {
bus := NewEventBus()
id := bus.Subscribe("event", func(data ...interface{}) {
fmt.Println("Event received:", data)
})
bus.Publish("event", "Hello", "World")
fmt.Println("Subscription ID:", id)
}
这段代码定义了一个简单的事件总线,包含订阅和发布事件的方法。它演示了如何创建一个事件总线,如何订阅和取消订阅事件,以及如何在总线上发布事件。这个实现使用了一个 map
来跟踪订阅者,并且为每个订阅者分配了一个唯一的ID。发布事件时,它将调用所有订阅该事件的函数。这个设计模式在处理异步事件和消息传递在Go语言编程中非常有用。
评论已关闭