events.go

package common

import (
	"reflect"
	"sync"
)

type Event interface{}

type EventChan chan Event

type EventFunc func(Event)

type EventType reflect.Type

type EventManager struct {
	mu    sync.Mutex
	chans map[EventType][]EventChan
	funcs map[EventType][]*EventFunc
}

type AllEvents struct{}

var (
	Events *EventManager
)

func init() {
	Events = NewEventManager()
}

func NewEventManager() *EventManager {
	return &EventManager{
		chans: make(map[EventType][]EventChan),
		funcs: make(map[EventType][]*EventFunc),
	}
}

func (this *EventManager) NewChanReceiver(event interface{}) EventChan {
	this.mu.Lock()
	defer this.mu.Unlock()

	eventType := reflect.TypeOf(event)
	eventListener := make(EventChan)

	DebugFunc(eventType)

	if _, ok := this.chans[eventType]; ok {
		this.chans[eventType] = append(this.chans[eventType], eventListener)
	} else {
		this.chans[eventType] = []EventChan{eventListener}
	}

	return eventListener
}

func (this *EventManager) NewFuncReceiver(event interface{}, eventFunc EventFunc) *EventFunc {
	this.mu.Lock()
	defer this.mu.Unlock()

	eventType := reflect.TypeOf(event)

	DebugFunc()

	if _, ok := this.funcs[eventType]; ok {
		this.funcs[eventType] = append(this.funcs[eventType], &eventFunc)
	} else {
		this.funcs[eventType] = []*EventFunc{&eventFunc}
	}

	return &eventFunc
}

func (this *EventManager) DestroyChanReceiver(eventChan EventChan) {
	this.mu.Lock()
	defer this.mu.Unlock()

	DebugFunc()

	for _, chans := range this.chans {
		for i := range chans {
			if chans[i] == eventChan {
				close(chans[i])
				chans = append(chans[:i], chans[i+1:]...)
				break
			}
		}
	}
}

func (this *EventManager) DestroyFuncReceiver(eventFunc *EventFunc) {
	this.mu.Lock()
	defer this.mu.Unlock()

	DebugFunc()

	for eventType, funcs := range this.funcs {
		for i := range funcs {
			if funcs[i] == eventFunc {
				funcs = append(funcs[:i], funcs[i+1:]...)
				break
			}
		}

		if len(funcs) == 0 {
			delete(this.funcs, eventType)
		} else {
			this.funcs[eventType] = funcs
		}
	}
}

func (this *EventManager) Emit(event interface{}) bool {
	this.mu.Lock()
	defer this.mu.Unlock()

	b := false

	eventType := reflect.TypeOf(event)

	DebugFunc(eventType)

	if chans, ok := this.chans[eventType]; ok {
		for _, receiver := range chans {
			receiver <- event
			b = true
		}
	}

	if funcs, ok := this.funcs[eventType]; ok {
		for _, receiver := range funcs {
			(*receiver)(event)
			b = true
		}
	}

	if funcs, ok := this.funcs[reflect.TypeOf(AllEvents{})]; ok {
		for _, receiver := range funcs {
			(*receiver)(event)
			b = true
		}
	}

	return b
}