init commit

air/controller/controller.go

@@ -0,0 +1,48 @@
+package controller
+
+import (
+	"air/service"
+	"flswld.com/common/config"
+	"github.com/gin-gonic/gin"
+	"strconv"
+)
+
+type Controller struct {
+	service *service.Service
+}
+
+func NewController(service *service.Service) (r *Controller) {
+	r = new(Controller)
+	r.service = service
+	go r.registerRouter()
+	return r
+}
+
+func (c *Controller) registerRouter() {
+	if config.CONF.Logger.Level == "DEBUG" {
+		gin.SetMode(gin.DebugMode)
+	} else {
+		gin.SetMode(gin.ReleaseMode)
+	}
+	engine := gin.Default()
+	// HTTP
+	engine.GET("/http/fetch", c.fetchHttpService)
+	engine.GET("/http/fetch/all", c.fetchAllHttpService)
+	engine.POST("/http/reg", c.registerHttpService)
+	engine.POST("/http/cancel", c.cancelHttpService)
+	engine.POST("/http/ka", c.httpKeepalive)
+	// RPC
+	engine.GET("/rpc/fetch", c.fetchRpcService)
+	engine.GET("/rpc/fetch/all", c.fetchAllRpcService)
+	engine.POST("/rpc/reg", c.registerRpcService)
+	engine.POST("/rpc/cancel", c.cancelRpcService)
+	engine.POST("/rpc/ka", c.rpcKeepalive)
+	// 长轮询
+	engine.GET("/poll/http", c.pollHttpService)
+	engine.GET("/poll/http/all", c.pollAllHttpService)
+	engine.GET("/poll/rpc", c.pollRpcService)
+	engine.GET("/poll/rpc/all", c.pollAllRpcService)
+	port := strconv.FormatInt(int64(config.CONF.HttpPort), 10)
+	portStr := ":" + port
+	_ = engine.Run(portStr)
+}