hotgo/server/utility/simple/event.go

// Package simple
// @Link  https://github.com/bufanyun/hotgo
// @Copyright  Copyright (c) 2023 HotGo CLI
// @Author  Ms <133814250@qq.com>
// @License  https://github.com/bufanyun/hotgo/blob/master/LICENSE
package simple

import (
	"context"
	"sync"
)

type EventFunc func(ctx context.Context, args ...interface{})

type sEvent struct {
	sync.Mutex
	list map[string][]EventFunc // 所有事件的列表
}

var event *sEvent

// Event 事件实例
func Event() *sEvent {
	if event == nil {
		event = &sEvent{
			list: make(map[string][]EventFunc),
		}
	}
	return event
}

// Register 往一个分组中注册事件
func (e *sEvent) Register(group string, callback EventFunc) {
	e.Lock()
	defer e.Unlock()
	e.list[group] = append(e.list[group], callback)
}

// Call 回调一个分组的事件
func (e *sEvent) Call(group string, ctx context.Context, args ...interface{}) {
	if events, ok := e.list[group]; ok {
		for _, f := range events {
			f(ctx, args...)
		}
	}
}

// Remove 移动一个分组的事件
func (e *sEvent) Remove(group string) {
	delete(e.list, group)
}

// Clear 清空事件列表
func (e *sEvent) Clear() {
	e.list = make(map[string][]EventFunc)
}