路由独立文件

cmd/search_http_server.go

@@ -5,9 +5,9 @@ import (
 	"github.com/gin-gonic/gin"
 	"github.com/micro/go-micro/v2/web"
 	"search_server/boot"
-	"search_server/controller"
 	_ "search_server/controller"
 	"search_server/pkg/config"
+	"search_server/routes"
 )
 
 func main() {
@@ -18,12 +18,8 @@ func main() {
 		panic(err)
 	}
 	gin.SetMode(config.Get("web.mode").String())
-	r := gin.New()
-
-	//路由
-	r.POST("/search/bom/autospu", controller.AutoSpu)
-	r.POST("/search/bom/recommend", controller.Recommend)
 
+	r := routes.InitRouter()
 
 	port := config.Get("web.port").String()
 	//web改成micro 就是grpc,并直接注册到etcd里面

routes/router.go

+package routes
+
+import (
+	"github.com/gin-gonic/gin"
+	"search_server/controller"
+)
+
+//初始化路由
+func InitRouter() *gin.Engine {
+	r := gin.New()
+	r.Use(gin.Logger())
+	r.Use(gin.Recovery())
+
+	//路由
+	r.POST("/search/bom/autospu", controller.AutoSpu)
+	r.POST("/search/bom/recommend", controller.Recommend)
+
+	return r
+}