diff --git a/pkg/middle-msg/logx.go b/pkg/middle-msg/logx.go index 67d887d..dcd9a83 100644 --- a/pkg/middle-msg/logx.go +++ b/pkg/middle-msg/logx.go @@ -3,7 +3,7 @@ package middle_msg import "time" var ( - EventNameLog = "LogContext" + EventNameLog = "logcontext" EventKeyLog = "LogContext" ) diff --git a/pkg/middle/worker.go b/pkg/middle/worker.go index 67eadb1..d617901 100644 --- a/pkg/middle/worker.go +++ b/pkg/middle/worker.go @@ -2,15 +2,23 @@ package middle import ( "context" + + msg "gitee.com/timedb/wheatCache/pkg/middle-msg" ) func (m *MiddleWare) middleWorker() { ctx := context.Background() workEvent := m.eventConsumer.Receive(ctx) m.loadPlugins() - for _, singles := range m.plugins { - for _, single := range singles { - single.Exec(workEvent) + switch workEvent.GetEventName() { + case msg.EventNameLog: + for middleMsg, singles := range m.plugins { + + if middleMsg == msg.EventNameLog { + for _, single := range singles { + single.Exec(workEvent) + } + } } } diff --git a/pkg/middle/worker_test.go b/pkg/middle/worker_test.go index 332d7ef..0f36a26 100644 --- a/pkg/middle/worker_test.go +++ b/pkg/middle/worker_test.go @@ -7,7 +7,7 @@ import ( ) func TestWorker(t *testing.T) { - event := event.NewEvent("LogContext") + event := event.NewEvent("logcontext") m := NewMiddleWare() m.eventDriver.Put(event)