feat(event): Add SetResultErr func

This commit is contained in:
HuangJiaLuo 2021-10-07 16:30:23 +08:00
parent 60abd336b1
commit e3171a8b3e
2 changed files with 33 additions and 5 deletions

View File

@ -57,10 +57,6 @@ func (e *Event) GetValue(key string) (interface{}, bool) {
return val, ok
}
func (e *Event) SetErr(err error) {
e.err = err
}
// InitWaitEvent 初始化 wait event 必须调用才拥有等待特性
func (e *Event) InitWaitEvent() {
e.muClose.Lock()
@ -98,14 +94,28 @@ func (e *Event) ExecWorkAndSendResult(work EventWorkFunc) (interface{}, error) {
e.eventStatus = workEventState
res, err := work()
e.waitResult <- res
e.err = err
e.waitResult <- res
close(e.waitResult)
e.eventStatus = closeEventState
return res, err
}
func (e *Event) SetResultErr(err error) {
e.muClose.Lock()
defer e.muClose.Unlock()
if e.eventStatus != waitEventState {
return
}
e.eventStatus = workEventState
e.err = err
e.waitResult <- nil
close(e.waitResult)
e.eventStatus = closeEventState
}
func NewEvent(eventName string) *Event {
return &Event{
eventName: eventName,

View File

@ -3,6 +3,7 @@ package event
import (
"context"
"fmt"
"gitee.com/timedb/wheatCache/pkg/errorx"
"testing"
"time"
@ -79,3 +80,20 @@ func waitConsumer(t *testing.T, ctx context.Context, v ConsumerInterface) {
})
}
}
func TestEvent_SetResultErr(t *testing.T) {
ctx := context.Background()
event := NewEvent("dddd")
driver := NewDriver(100)
produce := NewProduce(driver)
consumer := NewConsumer(driver)
go func() {
event := consumer.Receive(ctx)
event.SetResultErr(errorx.New("err"))
}()
event.InitWaitEvent()
produce.Call(ctx, event)
_, err := event.StartWaitEvent(2 * time.Second)
fmt.Println(err)
require.Error(t, err)
}