ccfos
/
nightingale
Not watched
1
0
Fork 0
Code
Releases 20 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: eebf2cff49
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'eebf2cff49'
${ noResults }
nightingale/src/webapi/router/router.go

345 lines
16 kB
Raw Blame History 

  1. package router

  2. 

  3. import (

  4. 	"fmt"

  5. 	"os"

  6. 	"path"

  7. 	"strings"

  8. 	"time"

  9. 

  10. 	"github.com/gin-contrib/pprof"

  11. 	"github.com/gin-gonic/gin"

  12. 	"github.com/prometheus/client_golang/prometheus/promhttp"

  13. 	"github.com/toolkits/pkg/ginx"

  14. 

  15. 	"github.com/didi/nightingale/v5/src/pkg/aop"

  16. 	"github.com/didi/nightingale/v5/src/webapi/config"

  17. 	promstat "github.com/didi/nightingale/v5/src/webapi/stat"

  18. )

  19. 

  20. func stat() gin.HandlerFunc {

  21. 	return func(c *gin.Context) {

  22. 		start := time.Now()

  23. 		c.Next()

  24. 

  25. 		code := fmt.Sprintf("%d", c.Writer.Status())

  26. 		method := c.Request.Method

  27. 		labels := []string{promstat.Service, code, c.FullPath(), method}

  28. 

  29. 		promstat.RequestCounter.WithLabelValues(labels...).Inc()

  30. 		promstat.RequestDuration.WithLabelValues(labels...).Observe(float64(time.Since(start).Seconds()))

  31. 	}

  32. }

  33. 

  34. func languageDetector() gin.HandlerFunc {

  35. 	headerKey := config.C.I18NHeaderKey

  36. 	return func(c *gin.Context) {

  37. 		if headerKey != "" {

  38. 			lang := c.GetHeader(headerKey)

  39. 			if lang != "" {

  40. 				if strings.HasPrefix(lang, "*") || strings.HasPrefix(lang, "zh") {

  41. 					c.Request.Header.Set("X-Language", "zh")

  42. 				} else if strings.HasPrefix(lang, "en") {

  43. 					c.Request.Header.Set("X-Language", "en")

  44. 				} else {

  45. 					c.Request.Header.Set("X-Language", lang)

  46. 				}

  47. 			}

  48. 		}

  49. 		c.Next()

  50. 	}

  51. }

  52. 

  53. func New(version string) *gin.Engine {

  54. 	gin.SetMode(config.C.RunMode)

  55. 

  56. 	if strings.ToLower(config.C.RunMode) == "release" {

  57. 		aop.DisableConsoleColor()

  58. 	}

  59. 

  60. 	r := gin.New()

  61. 

  62. 	r.Use(stat())

  63. 	r.Use(languageDetector())

  64. 	r.Use(aop.Recovery())

  65. 

  66. 	// whether print access log

  67. 	if config.C.HTTP.PrintAccessLog {

  68. 		r.Use(aop.Logger())

  69. 	}

  70. 

  71. 	configRoute(r, version)

  72. 	configNoRoute(r)

  73. 

  74. 	return r

  75. }

  76. 

  77. func configNoRoute(r *gin.Engine) {

  78. 	r.NoRoute(func(c *gin.Context) {

  79. 		arr := strings.Split(c.Request.URL.Path, ".")

  80. 		suffix := arr[len(arr)-1]

  81. 		switch suffix {

  82. 		case "png", "jpeg", "jpg", "svg", "ico", "gif", "css", "js", "html", "htm", "gz", "zip", "map":

  83. 			c.File(path.Join(strings.Split("pub/"+c.Request.URL.Path, "/")...))

  84. 		default:

  85. 			c.File(path.Join("pub", "index.html"))

  86. 		}

  87. 	})

  88. }

  89. 

  90. func configRoute(r *gin.Engine, version string) {

  91. 	if config.C.HTTP.PProf {

  92. 		pprof.Register(r, "/api/debug/pprof")

  93. 	}

  94. 

  95. 	r.GET("/ping", func(c *gin.Context) {

  96. 		c.String(200, "pong")

  97. 	})

  98. 

  99. 	r.GET("/pid", func(c *gin.Context) {

  100. 		c.String(200, fmt.Sprintf("%d", os.Getpid()))

  101. 	})

  102. 

  103. 	r.GET("/addr", func(c *gin.Context) {

  104. 		c.String(200, c.Request.RemoteAddr)

  105. 	})

  106. 

  107. 	r.GET("/version", func(c *gin.Context) {

  108. 		c.String(200, version)

  109. 	})

  110. 

  111. 	r.GET("/i18n", func(c *gin.Context) {

  112. 		ginx.NewRender(c).Message("just a test: %s", "by ulric")

  113. 	})

  114. 

  115. 	r.GET("/metrics", gin.WrapH(promhttp.Handler()))

  116. 

  117. 	pagesPrefix := "/api/n9e"

  118. 

  119. 	pages := r.Group(pagesPrefix)

  120. 	{

  121. 		if config.C.AnonymousAccess.PromQuerier {

  122. 			pages.Any("/prometheus/*url", prometheusProxy)

  123. 			pages.POST("/query-range-batch", promBatchQueryRange)

  124. 		} else {

  125. 			pages.Any("/prometheus/*url", auth(), prometheusProxy)

  126. 			pages.POST("/query-range-batch", auth(), promBatchQueryRange)

  127. 		}

  128. 

  129. 		pages.GET("/version", func(c *gin.Context) {

  130. 			c.String(200, version)

  131. 		})

  132. 

  133. 		pages.POST("/auth/login", jwtMock(), loginPost)

  134. 		pages.POST("/auth/logout", jwtMock(), logoutPost)

  135. 		pages.POST("/auth/refresh", jwtMock(), refreshPost)

  136. 

  137. 		pages.GET("/auth/sso-config", ssoConfigGet)

  138. 		pages.GET("/auth/redirect", loginRedirect)

  139. 		pages.GET("/auth/redirect/cas", loginRedirectCas)

  140. 		pages.GET("/auth/redirect/oauth", loginRedirectOAuth)

  141. 		pages.GET("/auth/callback", loginCallback)

  142. 		pages.GET("/auth/callback/cas", loginCallbackCas)

  143. 		pages.GET("/auth/callback/oauth", loginCallbackOAuth)

  144. 

  145. 		pages.GET("/metrics/desc", metricsDescGetFile)

  146. 		pages.POST("/metrics/desc", metricsDescGetMap)

  147. 

  148. 		pages.GET("/roles", rolesGets)

  149. 		pages.GET("/notify-channels", notifyChannelsGets)

  150. 		pages.GET("/contact-keys", contactKeysGets)

  151. 		pages.GET("/clusters", clustersGets)

  152. 

  153. 		pages.GET("/self/perms", auth(), user(), permsGets)

  154. 		pages.GET("/self/profile", auth(), user(), selfProfileGet)

  155. 		pages.PUT("/self/profile", auth(), user(), selfProfilePut)

  156. 		pages.PUT("/self/password", auth(), user(), selfPasswordPut)

  157. 

  158. 		pages.GET("/users", auth(), user(), perm("/users"), userGets)

  159. 		pages.POST("/users", auth(), admin(), userAddPost)

  160. 		pages.GET("/user/:id/profile", auth(), userProfileGet)

  161. 		pages.PUT("/user/:id/profile", auth(), admin(), userProfilePut)

  162. 		pages.PUT("/user/:id/password", auth(), admin(), userPasswordPut)

  163. 		pages.DELETE("/user/:id", auth(), admin(), userDel)

  164. 

  165. 		pages.GET("/metric-views", auth(), metricViewGets)

  166. 		pages.DELETE("/metric-views", auth(), user(), metricViewDel)

  167. 		pages.POST("/metric-views", auth(), user(), metricViewAdd)

  168. 		pages.PUT("/metric-views", auth(), user(), metricViewPut)

  169. 

  170. 		pages.GET("/user-groups", auth(), user(), userGroupGets)

  171. 		pages.POST("/user-groups", auth(), user(), perm("/user-groups/add"), userGroupAdd)

  172. 		pages.GET("/user-group/:id", auth(), user(), userGroupGet)

  173. 		pages.PUT("/user-group/:id", auth(), user(), perm("/user-groups/put"), userGroupWrite(), userGroupPut)

  174. 		pages.DELETE("/user-group/:id", auth(), user(), perm("/user-groups/del"), userGroupWrite(), userGroupDel)

  175. 		pages.POST("/user-group/:id/members", auth(), user(), perm("/user-groups/put"), userGroupWrite(), userGroupMemberAdd)

  176. 		pages.DELETE("/user-group/:id/members", auth(), user(), perm("/user-groups/put"), userGroupWrite(), userGroupMemberDel)

  177. 

  178. 		pages.GET("/busi-groups", auth(), user(), busiGroupGets)

  179. 		pages.POST("/busi-groups", auth(), user(), perm("/busi-groups/add"), busiGroupAdd)

  180. 		pages.GET("/busi-groups/alertings", auth(), busiGroupAlertingsGets)

  181. 		pages.GET("/busi-group/:id", auth(), user(), bgro(), busiGroupGet)

  182. 		pages.PUT("/busi-group/:id", auth(), user(), perm("/busi-groups/put"), bgrw(), busiGroupPut)

  183. 		pages.POST("/busi-group/:id/members", auth(), user(), perm("/busi-groups/put"), bgrw(), busiGroupMemberAdd)

  184. 		pages.DELETE("/busi-group/:id/members", auth(), user(), perm("/busi-groups/put"), bgrw(), busiGroupMemberDel)

  185. 		pages.DELETE("/busi-group/:id", auth(), user(), perm("/busi-groups/del"), bgrw(), busiGroupDel)

  186. 		pages.GET("/busi-group/:id/perm/:perm", auth(), user(), checkBusiGroupPerm)

  187. 

  188. 		pages.GET("/targets", auth(), user(), targetGets)

  189. 		pages.DELETE("/targets", auth(), user(), perm("/targets/del"), targetDel)

  190. 		pages.GET("/targets/tags", auth(), user(), targetGetTags)

  191. 		pages.POST("/targets/tags", auth(), user(), perm("/targets/put"), targetBindTagsByFE)

  192. 		pages.DELETE("/targets/tags", auth(), user(), perm("/targets/put"), targetUnbindTagsByFE)

  193. 		pages.PUT("/targets/note", auth(), user(), perm("/targets/put"), targetUpdateNote)

  194. 		pages.PUT("/targets/bgid", auth(), user(), perm("/targets/put"), targetUpdateBgid)

  195. 

  196. 		pages.GET("/builtin-boards", builtinBoardGets)

  197. 		pages.GET("/builtin-board/:name", builtinBoardGet)

  198. 

  199. 		pages.GET("/busi-group/:id/boards", auth(), user(), perm("/dashboards"), bgro(), boardGets)

  200. 		pages.POST("/busi-group/:id/boards", auth(), user(), perm("/dashboards/add"), bgrw(), boardAdd)

  201. 		pages.POST("/busi-group/:id/board/:bid/clone", auth(), user(), perm("/dashboards/add"), bgrw(), boardClone)

  202. 

  203. 		pages.GET("/board/:bid", boardGet)

  204. 		pages.GET("/board/:bid/pure", boardPureGet)

  205. 		pages.PUT("/board/:bid", auth(), user(), perm("/dashboards/put"), boardPut)

  206. 		pages.PUT("/board/:bid/configs", auth(), user(), perm("/dashboards/put"), boardPutConfigs)

  207. 		pages.PUT("/board/:bid/public", auth(), user(), perm("/dashboards/put"), boardPutPublic)

  208. 		pages.DELETE("/boards", auth(), user(), perm("/dashboards/del"), boardDel)

  209. 

  210. 		// migrate v5.8.0

  211. 		pages.GET("/dashboards", auth(), admin(), migrateDashboards)

  212. 		pages.GET("/dashboard/:id", auth(), admin(), migrateDashboardGet)

  213. 		pages.PUT("/dashboard/:id/migrate", auth(), admin(), migrateDashboard)

  214. 

  215. 		// deprecated ↓

  216. 		pages.GET("/dashboards/builtin/list", builtinBoardGets)

  217. 		pages.POST("/busi-group/:id/dashboards/builtin", auth(), user(), perm("/dashboards/add"), bgrw(), dashboardBuiltinImport)

  218. 		pages.GET("/busi-group/:id/dashboards", auth(), user(), perm("/dashboards"), bgro(), dashboardGets)

  219. 		pages.POST("/busi-group/:id/dashboards", auth(), user(), perm("/dashboards/add"), bgrw(), dashboardAdd)

  220. 		pages.POST("/busi-group/:id/dashboards/export", auth(), user(), perm("/dashboards"), bgro(), dashboardExport)

  221. 		pages.POST("/busi-group/:id/dashboards/import", auth(), user(), perm("/dashboards/add"), bgrw(), dashboardImport)

  222. 		pages.POST("/busi-group/:id/dashboard/:did/clone", auth(), user(), perm("/dashboards/add"), bgrw(), dashboardClone)

  223. 		pages.GET("/busi-group/:id/dashboard/:did", auth(), user(), perm("/dashboards"), bgro(), dashboardGet)

  224. 		pages.PUT("/busi-group/:id/dashboard/:did", auth(), user(), perm("/dashboards/put"), bgrw(), dashboardPut)

  225. 		pages.DELETE("/busi-group/:id/dashboard/:did", auth(), user(), perm("/dashboards/del"), bgrw(), dashboardDel)

  226. 

  227. 		pages.GET("/busi-group/:id/chart-groups", auth(), user(), bgro(), chartGroupGets)

  228. 		pages.POST("/busi-group/:id/chart-groups", auth(), user(), bgrw(), chartGroupAdd)

  229. 		pages.PUT("/busi-group/:id/chart-groups", auth(), user(), bgrw(), chartGroupPut)

  230. 		pages.DELETE("/busi-group/:id/chart-groups", auth(), user(), bgrw(), chartGroupDel)

  231. 

  232. 		pages.GET("/busi-group/:id/charts", auth(), user(), bgro(), chartGets)

  233. 		pages.POST("/busi-group/:id/charts", auth(), user(), bgrw(), chartAdd)

  234. 		pages.PUT("/busi-group/:id/charts", auth(), user(), bgrw(), chartPut)

  235. 		pages.DELETE("/busi-group/:id/charts", auth(), user(), bgrw(), chartDel)

  236. 		// deprecated ↑

  237. 

  238. 		pages.GET("/share-charts", chartShareGets)

  239. 		pages.POST("/share-charts", auth(), chartShareAdd)

  240. 

  241. 		pages.GET("/alert-rules/builtin/list", alertRuleBuiltinList)

  242. 		pages.POST("/busi-group/:id/alert-rules/builtin", auth(), user(), perm("/alert-rules/add"), bgrw(), alertRuleBuiltinImport)

  243. 		pages.GET("/busi-group/:id/alert-rules", auth(), user(), perm("/alert-rules"), alertRuleGets)

  244. 		pages.POST("/busi-group/:id/alert-rules", auth(), user(), perm("/alert-rules/add"), bgrw(), alertRuleAddByFE)

  245. 		pages.DELETE("/busi-group/:id/alert-rules", auth(), user(), perm("/alert-rules/del"), bgrw(), alertRuleDel)

  246. 		pages.PUT("/busi-group/:id/alert-rules/fields", auth(), user(), perm("/alert-rules/put"), bgrw(), alertRulePutFields)

  247. 		pages.PUT("/busi-group/:id/alert-rule/:arid", auth(), user(), perm("/alert-rules/put"), alertRulePutByFE)

  248. 		pages.GET("/alert-rule/:arid", auth(), user(), perm("/alert-rules"), alertRuleGet)

  249. 

  250. 		pages.GET("/busi-group/:id/recording-rules", auth(), user(), perm("/recording-rules"), recordingRuleGets)

  251. 		pages.POST("/busi-group/:id/recording-rules", auth(), user(), perm("/recording-rules/add"), bgrw(), recordingRuleAddByFE)

  252. 		pages.DELETE("/busi-group/:id/recording-rules", auth(), user(), perm("/recording-rules/del"), bgrw(), recordingRuleDel)

  253. 		pages.PUT("/busi-group/:id/recording-rule/:rrid", auth(), user(), perm("/recording-rules/put"), bgrw(), recordingRulePutByFE)

  254. 		pages.GET("/recording-rule/:rrid", auth(), user(), perm("/recording-rules"), recordingRuleGet)

  255. 		pages.PUT("/busi-group/:id/recording-rules/fields", auth(), user(), perm("/recording-rules/put"), recordingRulePutFields)

  256. 

  257. 		pages.GET("/busi-group/:id/alert-mutes", auth(), user(), perm("/alert-mutes"), bgro(), alertMuteGetsByBG)

  258. 		pages.POST("/busi-group/:id/alert-mutes", auth(), user(), perm("/alert-mutes/add"), bgrw(), alertMuteAdd)

  259. 		pages.DELETE("/busi-group/:id/alert-mutes", auth(), user(), perm("/alert-mutes/del"), bgrw(), alertMuteDel)

  260. 		pages.PUT("/busi-group/:id/alert-mute/:amid", auth(), user(), perm("/alert-mutes/put"), alertMutePutByFE)

  261. 		pages.PUT("/busi-group/:id/alert-mutes/fields", auth(), user(), perm("/alert-mutes/put"), bgrw(), alertMutePutFields)

  262. 

  263. 		pages.GET("/busi-group/:id/alert-subscribes", auth(), user(), perm("/alert-subscribes"), bgro(), alertSubscribeGets)

  264. 		pages.GET("/alert-subscribe/:sid", auth(), user(), perm("/alert-subscribes"), alertSubscribeGet)

  265. 		pages.POST("/busi-group/:id/alert-subscribes", auth(), user(), perm("/alert-subscribes/add"), bgrw(), alertSubscribeAdd)

  266. 		pages.PUT("/busi-group/:id/alert-subscribes", auth(), user(), perm("/alert-subscribes/put"), bgrw(), alertSubscribePut)

  267. 		pages.DELETE("/busi-group/:id/alert-subscribes", auth(), user(), perm("/alert-subscribes/del"), bgrw(), alertSubscribeDel)

  268. 

  269. 		if config.C.AnonymousAccess.AlertDetail {

  270. 			pages.GET("/alert-cur-event/:eid", alertCurEventGet)

  271. 			pages.GET("/alert-his-event/:eid", alertHisEventGet)

  272. 		} else {

  273. 			pages.GET("/alert-cur-event/:eid", auth(), alertCurEventGet)

  274. 			pages.GET("/alert-his-event/:eid", auth(), alertHisEventGet)

  275. 		}

  276. 

  277. 		// card logic

  278. 		pages.GET("/alert-cur-events/list", auth(), alertCurEventsList)

  279. 		pages.GET("/alert-cur-events/card", auth(), alertCurEventsCard)

  280. 		pages.POST("/alert-cur-events/card/details", auth(), alertCurEventsCardDetails)

  281. 		pages.GET("/alert-his-events/list", auth(), alertHisEventsList)

  282. 		pages.DELETE("/alert-cur-events", auth(), user(), perm("/alert-cur-events/del"), alertCurEventDel)

  283. 

  284. 		pages.GET("/alert-aggr-views", auth(), alertAggrViewGets)

  285. 		pages.DELETE("/alert-aggr-views", auth(), user(), alertAggrViewDel)

  286. 		pages.POST("/alert-aggr-views", auth(), user(), alertAggrViewAdd)

  287. 		pages.PUT("/alert-aggr-views", auth(), user(), alertAggrViewPut)

  288. 

  289. 		pages.GET("/busi-group/:id/task-tpls", auth(), user(), perm("/job-tpls"), bgro(), taskTplGets)

  290. 		pages.POST("/busi-group/:id/task-tpls", auth(), user(), perm("/job-tpls/add"), bgrw(), taskTplAdd)

  291. 		pages.DELETE("/busi-group/:id/task-tpl/:tid", auth(), user(), perm("/job-tpls/del"), bgrw(), taskTplDel)

  292. 		pages.POST("/busi-group/:id/task-tpls/tags", auth(), user(), perm("/job-tpls/put"), bgrw(), taskTplBindTags)

  293. 		pages.DELETE("/busi-group/:id/task-tpls/tags", auth(), user(), perm("/job-tpls/put"), bgrw(), taskTplUnbindTags)

  294. 		pages.GET("/busi-group/:id/task-tpl/:tid", auth(), user(), perm("/job-tpls"), bgro(), taskTplGet)

  295. 		pages.PUT("/busi-group/:id/task-tpl/:tid", auth(), user(), perm("/job-tpls/put"), bgrw(), taskTplPut)

  296. 

  297. 		pages.GET("/busi-group/:id/tasks", auth(), user(), perm("/job-tasks"), bgro(), taskGets)

  298. 		pages.POST("/busi-group/:id/tasks", auth(), user(), perm("/job-tasks/add"), bgrw(), taskAdd)

  299. 		pages.GET("/busi-group/:id/task/*url", auth(), user(), perm("/job-tasks"), taskProxy)

  300. 		pages.PUT("/busi-group/:id/task/*url", auth(), user(), perm("/job-tasks/put"), bgrw(), taskProxy)

  301. 

  302. 		pages.GET("/servers", auth(), admin(), serversGet)

  303. 		pages.PUT("/server/:id", auth(), admin(), serverBindCluster)

  304. 	}

  305. 

  306. 	service := r.Group("/v1/n9e")

  307. 	if len(config.C.BasicAuth) > 0 {

  308. 		service.Use(gin.BasicAuth(config.C.BasicAuth))

  309. 	}

  310. 	{

  311. 		service.Any("/prometheus/*url", prometheusProxy)

  312. 		service.POST("/users", userAddPost)

  313. 		service.GET("/users", userFindAll)

  314. 

  315. 		service.GET("/targets", targetGets)

  316. 		service.GET("/targets/tags", targetGetTags)

  317. 		service.POST("/targets/tags", targetBindTagsByService)

  318. 		service.DELETE("/targets/tags", targetUnbindTagsByService)

  319. 		service.PUT("/targets/note", targetUpdateNoteByService)

  320. 

  321. 		service.POST("/alert-rules", alertRuleAddByService)

  322. 		service.DELETE("/alert-rules", alertRuleDelByService)

  323. 		service.PUT("/alert-rule/:arid", alertRulePutByService)

  324. 		service.GET("/alert-rule/:arid", alertRuleGet)

  325. 		service.GET("/alert-rules", alertRulesGetByService)

  326. 

  327. 		service.GET("/alert-mutes", alertMuteGets)

  328. 		service.POST("/alert-mutes", alertMuteAddByService)

  329. 		service.DELETE("/alert-mutes", alertMuteDel)

  330. 

  331. 		service.GET("/alert-cur-events", alertCurEventsList)

  332. 		service.GET("/alert-his-events", alertHisEventsList)

  333. 		service.GET("/alert-his-event/:eid", alertHisEventGet)

  334. 

  335. 		service.GET("/config/:id", configGet)

  336. 		service.GET("/configs", configsGet)

  337. 		service.PUT("/configs", configsPut)

  338. 		service.POST("/configs", configsPost)

  339. 		service.DELETE("/configs", configsDel)

  340. 

  341. 		service.POST("/conf-prop/encrypt", confPropEncrypt)

  342. 		service.POST("/conf-prop/decrypt", confPropDecrypt)

  343. 	}

  344. }