change controller router style

mylxsw · Aug 6, 2019 · 08a11ff · 08a11ff
1 parent 434cdad
commit 08a11ff
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 3 deletions.
diff --git a/api/controller/controller.go b/api/controller/controller.go
@@ -0,0 +1,9 @@
+package controller
+
+import (
+	"github.com/mylxsw/go-toolkit/web"
+)
+
+type Controller interface {
+	Register(router *web.Router)
+}
diff --git a/api/controller/welcome.go b/api/controller/welcome.go
@@ -10,6 +10,10 @@ func NewWelcomeController() *WelcomeController {
 	return &WelcomeController{}
 }
 
+func (controller *WelcomeController) Register(router *web.Router) {
+	router.Get("/", controller.Home)
+}
+
 func (*WelcomeController) Home(ctx *web.WebContext, req *web.Request) web.HTTPResponse {
 	return ctx.API("0000", "hello, world", nil)
 }
diff --git a/api/router.go b/api/router.go
@@ -6,11 +6,9 @@ import (
 )
 
 func routers(router *web.Router, mw web.RequestMiddleware) {
-	welcome := controller.NewWelcomeController()
 
 	router.Group("/", func(router *web.Router) {
-		router.Get("/", welcome.Home)
-
+		controller.NewWelcomeController().Register(router)
 
 	}, mw.AccessLog(), mw.JSONExceptionHandler(), mw.CORS("*"))
 }