!53 更新driver, 重构了worker

Merge pull request !53 from K-on/fix-driver

pkg/event/driver.go

@@ -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,

pkg/event/event_test.go

@@ -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)
+}

pkg/lru/worker.go

@@ -2,8 +2,8 @@ package lru
 
 import (
 	"context"
+	"gitee.com/timedb/wheatCache/pkg/errorx"
 	"gitee.com/timedb/wheatCache/pkg/event"
-	"log"
 )
 
 func (lru *SingleCache) lruSingleWork() interface{} {
@@ -11,16 +11,20 @@ func (lru *SingleCache) lruSingleWork() interface{} {
 	for {
 		workEvent := lru.lruConsumer.Receive(ctx)
 
+		switch workEvent.GetEventName() {
+		case OptionEventName:
 			workFunc, ok := workEvent.GetValue(WorkFuncEventKey)
 			if !ok {
+				workEvent.ExecWorkAndSendResult(func() (interface{}, error) {
+					return nil, errorx.New("the event haven't work of function")
+				})
 				continue
 			}
 
-		switch workFunc.(type) {
-		case event.EventWorkFunc:
-			workEvent.ExecWorkAndSendResult(workFunc.(event.EventWorkFunc))
-		default:
-			log.Print("this is debug ")
+			if work, ok := workFunc.(event.EventWorkFunc); ok {
+				workEvent.ExecWorkAndSendResult(work)
+			}
+		case CleanEventName:
 		}
 	}
 }