测试resource event

Author: yumaojun03

bus/broker/mock/mock.go

@@ -0,0 +1,21 @@
+package broker
+
+import (
+	"fmt"
+
+	"github.com/infraboard/mcube/bus"
+	"github.com/infraboard/mcube/bus/event"
+)
+
+// NewBroker todo
+func NewBroker() bus.Publisher {
+	return &mockBroker{}
+}
+
+type mockBroker struct {
+}
+
+func (m *mockBroker) Pub(topic string, e *event.Event) error {
+	fmt.Println(topic, e)
+	return nil
+}

bus/event/type.go

@@ -1,7 +1,9 @@
+//go:generate mcube enum -m
+
 package event
 
 const (
-	// OperateEventType (operate_event) 资源操作事件
+	// OperateEventType (operate) 资源操作事件
 	OperateEventType Type = iota
 	// AlertEventType (alert) 告警事件
 	AlertEventType

bus/event/type_generate.go

@@ -7,17 +7,29 @@ import (
 // ResourceEvent 资源事件
 type ResourceEvent interface {
 	ResourceType() string
+	ResourceAction() string
+
 	ResourceUUID() string
 	ResourceDomain() string
 	ResourceNamespace() string
 	ResourceName() string
-	ResourceAction() string
 	ResourceData() interface{}
 }
 
-func newEvent(re ResourceEvent) *event.Event {
+func newResourceEvent(re ResourceEvent) *event.Event {
 	e := event.NewEvent()
 	e.Label["domain"] = re.ResourceDomain()
 	e.Label["namespace"] = re.ResourceNamespace()
+	e.Type = event.OperateEventType
+
+	body := &event.OperateEvent{
+		ResourceType: re.ResourceType(),
+		ResourceUUID: re.ResourceUUID(),
+		ResourceName: re.ResourceName(),
+		Action:       re.ResourceAction(),
+		Data:         re.ResourceData(),
+	}
+	e.Body = body
+
 	return e
 }

http/response/event.go

@@ -35,7 +35,8 @@ func sendEvent(re ResourceEvent) {
 		return
 	}
 
-	if err := eReporter.Pub("xxx", newEvent(re)); err != nil {
+	e := newResourceEvent(re)
+	if err := eReporter.Pub(e.Type.String(), e); err != nil {
 		getLog().Errorf("send event error, %s", err)
 		return
 	}

http/response/reporter.go

@@ -80,5 +80,10 @@ func Success(w http.ResponseWriter, data interface{}) {
 
 	w.WriteHeader(http.StatusOK)
 	w.Write(respBytes)
+
+	// 发送事件
+	if re, ok := data.(ResourceEvent); HasEventReporter() && ok {
+		sendEvent(re)
+	}
 	return
 }

http/response/send.go

@@ -39,6 +39,15 @@ func (e *Entry) AddLabel(labels ...*Label) EntryDecorator {
 	return e
 }
 
+// GetLableValue 获取Lable的值
+func (e *Entry) GetLableValue(key string) string {
+	v, ok := e.Labels[key]
+	if ok {
+		return v
+	}
+	return ""
+}
+
 // EnableAuth 启动身份验证
 func (e *Entry) EnableAuth() EntryDecorator {
 	e.AuthEnable = true

http/router/entry.go

@@ -5,8 +5,10 @@ import (
 	"net/http/httptest"
 	"testing"
 
+	broker "github.com/infraboard/mcube/bus/broker/mock"
 	"github.com/infraboard/mcube/exception"
 	"github.com/infraboard/mcube/http/context"
+	"github.com/infraboard/mcube/http/label"
 	"github.com/infraboard/mcube/http/mock"
 	"github.com/infraboard/mcube/http/response"
 	"github.com/infraboard/mcube/http/router"
@@ -40,6 +42,56 @@ func WithContextHandler(w http.ResponseWriter, r *http.Request) {
 	return
 }
 
+func ResourceEventHandler(w http.ResponseWriter, r *http.Request) {
+	rctx := context.GetContext(r)
+
+	if rctx.Entry == nil {
+		response.Failed(w, exception.NewBadRequest("no entry"))
+	}
+
+	data := &mockResourceEvent{entry: rctx.Entry}
+	response.Success(w, data)
+	return
+}
+
+type mockResourceEvent struct {
+	entry *router.Entry
+}
+
+func (mre *mockResourceEvent) ResourceType() string {
+	if mre.entry == nil {
+		return ""
+	}
+	return mre.entry.Resource
+}
+
+func (mre *mockResourceEvent) ResourceAction() string {
+	if mre.entry == nil {
+		return ""
+	}
+	return mre.entry.GetLableValue(label.ActionLableKey)
+}
+
+func (mre *mockResourceEvent) ResourceUUID() string {
+	return "mock-resource-event"
+}
+
+func (mre *mockResourceEvent) ResourceDomain() string {
+	return "mock-domain"
+}
+
+func (mre *mockResourceEvent) ResourceNamespace() string {
+	return "mock-namespace"
+}
+
+func (mre *mockResourceEvent) ResourceName() string {
+	return "mock-name"
+}
+
+func (mre *mockResourceEvent) ResourceData() interface{} {
+	return nil
+}
+
 func TestBase(t *testing.T) {
 	should := assert.New(t)
 
@@ -166,3 +218,18 @@ func TestAPIRootOrderOK(t *testing.T) {
 		should.Equal("/test3", es.Items[2].Path)
 	}
 }
+
+func TestResourceEventOk(t *testing.T) {
+	should := assert.New(t)
+
+	response.SetEventReporter(broker.NewBroker())
+
+	r := httprouter.New()
+	req, _ := http.NewRequest("GET", "/", nil)
+	w := httptest.NewRecorder()
+
+	r.Handle("GET", "/", ResourceEventHandler)
+	r.ServeHTTP(w, req)
+
+	should.Equal(200, w.Code)
+}