// Package event ... // // Description : event ... // // Author : go_developer@163.com<白茶清欢> // // Date : 2024-07-17 11:21 package event import ( "context" "errors" "git.zhangdeman.cn/zhangdeman/consts" "git.zhangdeman.cn/zhangdeman/event/abstract" "git.zhangdeman.cn/zhangdeman/event/define" ) var ( MemoryEventClient abstract.IEvent ) // InitMemoryEvent 初始化事件实例 // // Author : go_developer@163.com<白茶清欢> // // Date : 11:24 2024/7/17 func InitMemoryEvent(cfg *define.MemoryEventConfig) { instance := &MemoryEvent{ base: &base{ panicCallback: define.DefaultPanicCallback, parseFailCallback: define.DefaultParseFailCallbackFunc, }, } instance.Init(cfg) MemoryEventClient = instance } // MemoryEvent 基于内存实现的消息队列(仅适用于单机) // // Author : go_developer@163.com<白茶清欢> // // Date : 11:22 2024/7/17 type MemoryEvent struct { *base cfg *define.MemoryEventConfig messageChannel chan *define.EventData } // SendEvent 发送事件 // // Author : go_developer@163.com<白茶清欢> // // Date : 12:28 2024/7/17 func (m *MemoryEvent) SendEvent(ctx context.Context, eventData *define.EventData) (*define.SendResult, error) { if nil == eventData { return nil, errors.New("event data is nil") } m.messageChannel <- eventData return &define.SendResult{ Data: eventData, Partition: 0, Topic: "", IsSuccess: true, FailReason: "success", Extension: nil, }, nil } // SendEventAsync 异步发送事件 // // Author : go_developer@163.com<白茶清欢> // // Date : 12:29 2024/7/17 func (m *MemoryEvent) SendEventAsync(ctx context.Context, eventData *define.EventData, sendSuccessCallback abstract.SendSuccessCallback, sendFailCallback abstract.SendFailCallback) { if nil == sendFailCallback { sendFailCallback = define.DefaultSendFailCallback } if nil == sendSuccessCallback { sendSuccessCallback = define.DefaultSendSuccessCallback } go func() { var ( res *define.SendResult err error ) defer func() { if r := recover(); nil != r { m.panicCallback(err, eventData, map[string]any{ "send_result": res, }) } }() res, err = m.SendEvent(ctx, eventData) if nil != err { sendFailCallback(ctx, eventData, res, err) return } else { sendSuccessCallback(ctx, res) } }() } // GetConsumeEventChan 消费的消息chan // // Author : go_developer@163.com<白茶清欢> // // Date : 12:41 2024/7/17 func (m *MemoryEvent) GetConsumeEventChan() (<-chan *define.EventData, error) { return m.messageChannel, nil } func (m *MemoryEvent) ConsumeEvent(handler abstract.EventHandler, successCallback abstract.ConsumeSuccessCallback, failureCallback abstract.ConsumeFailCallbackHandler) error { //TODO implement me panic("implement me") } func (m *MemoryEvent) Destroy() { //TODO implement me panic("implement me") } func (m *MemoryEvent) DriverType() string { return consts.EventDriverMemory } // Init 初始化 // // Author : go_developer@163.com<白茶清欢> // // Date : 12:08 2024/7/17 func (m *MemoryEvent) Init(cfg *define.MemoryEventConfig) { if nil == cfg { cfg = &define.MemoryEventConfig{} } if cfg.MessageBufferSize <= 0 { cfg.MessageBufferSize = define.DefaultMemoryChannelSize } if cfg.CloseMaxWaitTime <= 0 { cfg.CloseMaxWaitTime = define.DefaultMemoryCloseMaxWaitTime } m.cfg = cfg // 初始化内存 channel m.messageChannel = make(chan *define.EventData, cfg.MessageBufferSize) }