origin/event/event.go

package event

import (
	"github.com/duanhf2012/origin/log"
	"runtime"
	"sync"
)


//事件接受器
type EventCallBack func(event IEvent)

type IEvent interface {
	GetEventType() EventType
}

type Event struct {
	Type EventType
	Data interface{}
	ref bool
}

var emptyEvent Event
func (e *Event) Reset() {
	*e = emptyEvent
}

func (e *Event) IsRef() bool {
	return e.ref
}

func (e *Event) Ref() {
	e.ref = true
}

func (e *Event) UnRef() {
	e.ref = false
}

func (e *Event) GetEventType() EventType{
	return e.Type
}

type IEventHandler interface {
	Init(processor IEventProcessor)
	GetEventProcessor() IEventProcessor  //获得事件
	NotifyEvent(IEvent)
	Destroy()
	//注册了事件
	addRegInfo(eventType EventType,eventProcessor IEventProcessor)
	removeRegInfo(eventType EventType,eventProcessor IEventProcessor)
}

type IEventChannel interface {
	PushEvent(ev IEvent) error
}

type IEventProcessor interface {
	IEventChannel

	Init(eventChannel IEventChannel)
	EventHandler(ev IEvent)
	RegEventReceiverFunc(eventType EventType, receiver IEventHandler,callback EventCallBack)
	UnRegEventReceiverFun(eventType EventType, receiver IEventHandler)

	castEvent(event IEvent) //广播事件
	addBindEvent(eventType EventType, receiver IEventHandler,callback EventCallBack)
	addListen(eventType EventType, receiver IEventHandler)
	removeBindEvent(eventType EventType, receiver IEventHandler)
	removeListen(eventType EventType, receiver IEventHandler)
}

type EventHandler struct {
	//已经注册的事件类型
	eventProcessor IEventProcessor

	//已经注册的事件
	locker sync.RWMutex
	mapRegEvent map[EventType]map[IEventProcessor]interface{}  //向其他事件处理器监听的事件类型
}

type EventProcessor struct {
	IEventChannel

	locker sync.RWMutex
	mapListenerEvent map[EventType]map[IEventProcessor]int           //监听者信息
	mapBindHandlerEvent map[EventType]map[IEventHandler]EventCallBack//收到事件处理
}

func NewEventHandler() IEventHandler{
	eh := EventHandler{}
	eh.mapRegEvent = map[EventType]map[IEventProcessor]interface{}{}

	return &eh
}

func NewEventProcessor() IEventProcessor{
	ep := EventProcessor{}
	ep.mapListenerEvent  =  map[EventType]map[IEventProcessor]int{}
	ep.mapBindHandlerEvent = map[EventType]map[IEventHandler]EventCallBack{}

	return &ep
}

func (handler *EventHandler) addRegInfo(eventType EventType,eventProcessor IEventProcessor){
	handler.locker.Lock()
	defer handler.locker.Unlock()
	if handler.mapRegEvent == nil {
		handler.mapRegEvent = map[EventType]map[IEventProcessor]interface{}{}
	}

	if _,ok := handler.mapRegEvent[eventType] ;ok == false{
		handler.mapRegEvent[eventType] = map[IEventProcessor]interface{}{}
	}
	handler.mapRegEvent[eventType][eventProcessor] = nil
}

func (handler *EventHandler) removeRegInfo(eventType EventType,eventProcessor IEventProcessor){
	if _,ok := handler.mapRegEvent[eventType];ok == true {
		delete(handler.mapRegEvent[eventType],eventProcessor)
	}
}

func (handler *EventHandler) GetEventProcessor() IEventProcessor{
	return handler.eventProcessor
}

func (handler *EventHandler) NotifyEvent(ev IEvent){
	handler.GetEventProcessor().castEvent(ev)
}

func (handler *EventHandler) Init(processor IEventProcessor){
	handler.eventProcessor = processor
	handler.mapRegEvent =map[EventType]map[IEventProcessor]interface{}{}
}


func (processor *EventProcessor) Init(eventChannel IEventChannel){
	processor.IEventChannel = eventChannel
}

func (processor *EventProcessor) addBindEvent(eventType EventType, receiver IEventHandler,callback EventCallBack){
	processor.locker.Lock()
	defer processor.locker.Unlock()

	if _,ok := processor.mapBindHandlerEvent[eventType]; ok == false {
		processor.mapBindHandlerEvent[eventType] = map[IEventHandler]EventCallBack{}
	}

	processor.mapBindHandlerEvent[eventType][receiver] = callback
}

func (processor *EventProcessor) addListen(eventType EventType, receiver IEventHandler){
	processor.locker.Lock()
	defer processor.locker.Unlock()

	if _,ok := processor.mapListenerEvent[eventType];ok == false{
		processor.mapListenerEvent[eventType] = map[IEventProcessor]int{}
	}

	processor.mapListenerEvent[eventType][receiver.GetEventProcessor()] += 1
}

func (processor *EventProcessor) removeBindEvent(eventType EventType, receiver IEventHandler){
	processor.locker.Lock()
	defer processor.locker.Unlock()
	if _,ok := processor.mapBindHandlerEvent[eventType];ok == true{
		delete(processor.mapBindHandlerEvent[eventType], receiver)
	}
}

func (processor *EventProcessor) removeListen(eventType EventType, receiver IEventHandler){
	processor.locker.Lock()
	defer processor.locker.Unlock()
	if _,ok := processor.mapListenerEvent[eventType];ok == true{
		processor.mapListenerEvent[eventType][receiver.GetEventProcessor()]-=1
		if processor.mapListenerEvent[eventType][receiver.GetEventProcessor()] <= 0 {
			delete(processor.mapListenerEvent[eventType], receiver.GetEventProcessor())
		}
	}
}

func (processor *EventProcessor) RegEventReceiverFunc(eventType EventType, receiver IEventHandler,callback EventCallBack){
	//记录receiver自己注册过的事件
	receiver.addRegInfo(eventType, processor)
	//记录当前所属IEventProcessor注册的回调
	receiver.GetEventProcessor().addBindEvent(eventType, receiver,callback)
	//将注册加入到监听中
	processor.addListen(eventType, receiver)
}

func (processor *EventProcessor) UnRegEventReceiverFun(eventType EventType, receiver IEventHandler) {
	processor.removeListen(eventType, receiver)
	receiver.GetEventProcessor().removeBindEvent(eventType, receiver)
	receiver.removeRegInfo(eventType, processor)
}

func (handler *EventHandler) Destroy(){
	handler.locker.Lock()
	defer handler.locker.Unlock()
	for eventTyp,mapEventProcess := range handler.mapRegEvent {
		if mapEventProcess == nil {
			continue
		}

		for eventProcess := range mapEventProcess {
			eventProcess.UnRegEventReceiverFun(eventTyp, handler)
		}
	}
}

func (processor *EventProcessor) EventHandler(ev IEvent) {
	defer func() {
		if r := recover(); r != nil {
			buf := make([]byte, 4096)
			l := runtime.Stack(buf, false)
			log.SError("core dump info:",r,":",string(buf[:l]))
		}
	}()

	mapCallBack,ok := processor.mapBindHandlerEvent[ev.GetEventType()]
	if ok == false {
		return
	}
	for _,callback := range mapCallBack {
		callback(ev)
	}
}


func (processor *EventProcessor) castEvent(event IEvent){
	if processor.mapListenerEvent == nil {
		log.SError("mapListenerEvent not init!")
		return
	}

	eventProcessor,ok := processor.mapListenerEvent[event.GetEventType()]
	if ok == false || processor == nil{
		log.SDebug("event type ",event.GetEventType()," not listen.")
		return
	}

	for proc := range eventProcessor {
		proc.PushEvent(event)
	}
}