添加resource event

Author: yumaojun03

http/context/context.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"net/http"
 
+	"github.com/infraboard/mcube/http/router"
 	"github.com/julienschmidt/httprouter"
 )
 
@@ -36,6 +37,7 @@ func NewReqContext() *ReqContext {
 
 // ReqContext context
 type ReqContext struct {
+	Entry    *router.Entry
 	PS       httprouter.Params
 	AuthInfo interface{}
 }

http/response/event.go

@@ -0,0 +1,11 @@
+package response
+
+// ResourceEvent 资源事件
+type ResourceEvent interface {
+	ResourceType() string
+	ResourceUUID() string
+	ResourceDomain() string
+	ResourceNamespace() string
+	ResourceName() string
+	ResourceAction() string
+}

http/response/reporter.go

@@ -0,0 +1,41 @@
+package response
+
+import (
+	"github.com/infraboard/mcube/bus"
+	"github.com/infraboard/mcube/bus/event"
+	"github.com/infraboard/mcube/logger"
+	"github.com/infraboard/mcube/logger/zap"
+)
+
+var (
+	eReporter bus.Publisher
+	log       logger.Logger
+)
+
+func getLog() logger.Logger {
+	if log == nil {
+		log = zap.L().Named("Response")
+	}
+
+	return log
+}
+
+// SetEventReporter tood
+func SetEventReporter(pub bus.Publisher) {
+	eReporter = pub
+}
+
+// HasEventReporter 是否已经初始化
+func HasEventReporter() bool {
+	return eReporter != nil
+}
+
+func sendEvent(re ResourceEvent) {
+	if !HasEventReporter() {
+		getLog().Errorf("event reporter is nil")
+		return
+	}
+
+	e := event.NewEvent()
+	eReporter.Pub("xxx", e)
+}

http/router/httprouter/httprouter.go

@@ -136,14 +136,21 @@ func (r *httpRouter) add(e *entry) {
 // 在添加路由时 装饰了认证逻辑
 func (r *httpRouter) addHandler(method, path string, h http.Handler) {
 	wrapper := func(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
+		rc := context.GetContext(req)
+
+		// 路由前钩子
 		// 使用auther进行认证
-		var authInfo interface{}
+		var (
+			authInfo interface{}
+			entry    *entry
+		)
 		if r.auther != nil {
-			entry := r.findEntry(method, path)
+			entry = r.findEntry(method, path)
 			if entry == nil {
 				r.notFound.ServeHTTP(w, req)
 				return
 			}
+			rc.Entry = entry.Entry
 
 			ai, err := r.auther.Auth(req, *entry.Entry)
 			if err != nil {
@@ -153,11 +160,11 @@ func (r *httpRouter) addHandler(method, path string, h http.Handler) {
 			authInfo = ai
 		}
 
-		rc := context.GetContext(req)
 		rc.AuthInfo = authInfo
 		rc.PS = ps
 		req = context.WithContext(req, rc)
 		h.ServeHTTP(w, req)
+		// 路由后钩子
 	}
 	r.r.Handle(method, path, wrapper)
 }

http/router/httprouter/httprouter_test.go

@@ -32,6 +32,10 @@ func WithContextHandler(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	if rctx.Entry == nil {
+		response.Failed(w, exception.NewBadRequest("no entry"))
+	}
+
 	response.Failed(w, exception.NewBadRequest("failed"))
 	return
 }