Rainbond/node/api/router/router.go
2020-09-06 11:09:48 +08:00

101 lines
3.9 KiB
Go

// Copyright (C) 2014-2018 Goodrain Co., Ltd.
// RAINBOND, Application Management Platform
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version. For any non-GPL usage of Rainbond,
// one or multiple Commercial Licenses authorized by Goodrain Co., Ltd.
// must be obtained first.
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
package router
import (
"time"
"github.com/sirupsen/logrus"
"github.com/goodrain/rainbond/node/api/controller"
"github.com/goodrain/rainbond/util/log"
"github.com/go-chi/chi"
"github.com/go-chi/chi/middleware"
)
//Routers 路由
func Routers(mode string) *chi.Mux {
r := chi.NewRouter()
r.Use(middleware.RequestID) //每个请求的上下文中注册一个id
//Sets a http.Request's RemoteAddr to either X-Forwarded-For or X-Real-IP
r.Use(middleware.RealIP)
//Logs the start and end of each request with the elapsed processing time
logger := logrus.New()
logger.SetLevel(logrus.GetLevel())
r.Use(log.NewStructuredLogger(logger))
//Gracefully absorb panics and prints the stack trace
r.Use(middleware.Recoverer)
//request time out
r.Use(middleware.Timeout(time.Second * 5))
r.Mount("/v1", DisconverRoutes())
r.Route("/v2", func(r chi.Router) {
r.Get("/ping", controller.Ping)
r.Route("/apps", func(r chi.Router) {
r.Get("/{app_name}/register", controller.APPRegister)
r.Get("/{app_name}/discover", controller.APPDiscover)
r.Get("/", controller.APPList)
})
r.Route("/localvolumes", func(r chi.Router) {
r.Post("/create", controller.CreateLocalVolume)
r.Delete("/", controller.DeleteLocalVolume)
})
//以下只有管理节点具有的API
if mode == "master" {
r.Route("/configs", func(r chi.Router) {
r.Get("/datacenter", controller.GetDatacenterConfig)
r.Put("/datacenter", controller.PutDatacenterConfig)
})
r.Route("/cluster", func(r chi.Router) {
r.Get("/", controller.ClusterInfo)
r.Get("/service-health", controller.GetServicesHealthy)
})
r.Route("/nodes", func(r chi.Router) {
// abandoned
r.Get("/fullres", controller.ClusterInfo)
r.Get("/{node_id}/node_resource", controller.GetNodeResource)
r.Get("/resources", controller.Resources)
r.Get("/capres", controller.CapRes)
r.Get("/", controller.GetNodes)
r.Get("/all_node_health", controller.GetAllNodeHealth)
r.Get("/rule/{rule}", controller.GetRuleNodes)
r.Get("/{node_id}", controller.GetNode)
r.Put("/{node_id}/status", controller.UpdateNodeStatus)
r.Put("/{node_id}/unschedulable", controller.Cordon)
r.Put("/{node_id}/reschedulable", controller.UnCordon)
r.Post("/{node_id}/labels", controller.PutLabel)
r.Get("/{node_id}/labels", controller.GetLabel)
r.Delete("/{node_id}/labels", controller.DeleteLabel)
r.Post("/{node_id}/down", controller.DownNode)
r.Post("/{node_id}/up", controller.UpNode)
r.Get("/{node_id}/instance", controller.Instances)
r.Get("/{node_id}/check", controller.CheckNode)
r.Get("/{node_id}/resource", controller.Resource)
r.Get("/{node_id}/conditions", controller.ListNodeCondition)
r.Delete("/{node_id}/conditions/{condition}", controller.DeleteNodeCondition)
// about node install
r.Post("/{node_id}/install", controller.InstallNode) //install node
r.Post("/", controller.AddNode) //add node
r.Delete("/{node_id}", controller.DeleteRainbondNode) //delete node
})
}
})
return r
}