forked from pneymrl2f/nightingale
UlricQin
4 years ago
parent
920dd9a947
commit
d32f9ef763
@ -1,124 +0,0 @@
|
||||
package http
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"strings"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/toolkits/pkg/logger"
|
||||
"github.com/toolkits/pkg/slice"
|
||||
|
||||
"github.com/didi/nightingale/src/common/address"
|
||||
"github.com/didi/nightingale/src/models"
|
||||
"github.com/didi/nightingale/src/modules/rdb/config"
|
||||
)
|
||||
|
||||
func shouldBeLogin() gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
c.Set("username", mustUsername(c))
|
||||
c.Next()
|
||||
}
|
||||
}
|
||||
|
||||
func shouldBeRoot() gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
username := mustUsername(c)
|
||||
|
||||
user, err := models.UserGet("username=?", username)
|
||||
dangerous(err)
|
||||
|
||||
if user.IsRoot != 1 {
|
||||
bomb("forbidden")
|
||||
}
|
||||
|
||||
c.Set("username", username)
|
||||
c.Set("user", user)
|
||||
c.Next()
|
||||
}
|
||||
}
|
||||
|
||||
func shouldBeService() gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
remoteAddr := c.Request.RemoteAddr
|
||||
idx := strings.LastIndex(remoteAddr, ":")
|
||||
ip := ""
|
||||
if idx > 0 {
|
||||
ip = remoteAddr[0:idx]
|
||||
}
|
||||
|
||||
if ip == "127.0.0.1" {
|
||||
c.Next()
|
||||
return
|
||||
}
|
||||
|
||||
if ip != "" && slice.ContainsString(address.GetAddresses("rdb"), ip) {
|
||||
c.Next()
|
||||
return
|
||||
}
|
||||
|
||||
token := c.GetHeader("X-Srv-Token")
|
||||
if token == "" {
|
||||
c.AbortWithError(http.StatusForbidden, fmt.Errorf("X-Srv-Token is blank"))
|
||||
return
|
||||
}
|
||||
|
||||
if !slice.ContainsString(config.Config.Tokens, token) {
|
||||
c.AbortWithError(http.StatusForbidden, fmt.Errorf("X-Srv-Token[%s] invalid", token))
|
||||
return
|
||||
}
|
||||
|
||||
c.Next()
|
||||
}
|
||||
}
|
||||
|
||||
func mustUsername(c *gin.Context) string {
|
||||
username := cookieUsername(c)
|
||||
if username == "" {
|
||||
username = headerUsername(c)
|
||||
}
|
||||
|
||||
if username == "" {
|
||||
bomb("unauthorized")
|
||||
}
|
||||
|
||||
return username
|
||||
}
|
||||
|
||||
func cookieUsername(c *gin.Context) string {
|
||||
return models.UsernameByUUID(readCookieUser(c))
|
||||
}
|
||||
|
||||
func headerUsername(c *gin.Context) string {
|
||||
token := c.GetHeader("X-User-Token")
|
||||
if token == "" {
|
||||
return ""
|
||||
}
|
||||
|
||||
ut, err := models.UserTokenGet("token=?", token)
|
||||
if err != nil {
|
||||
logger.Warningf("UserTokenGet[%s] fail: %v", token, err)
|
||||
return ""
|
||||
}
|
||||
|
||||
if ut == nil {
|
||||
return ""
|
||||
}
|
||||
|
||||
return ut.Username
|
||||
}
|
||||
|
||||
// ------------
|
||||
|
||||
func readCookieUser(c *gin.Context) string {
|
||||
uuid, err := c.Cookie(config.Config.HTTP.CookieName)
|
||||
if err != nil {
|
||||
return ""
|
||||
}
|
||||
|
||||
return uuid
|
||||
}
|
||||
|
||||
func writeCookieUser(c *gin.Context, uuid string) {
|
||||
c.SetCookie(config.Config.HTTP.CookieName, uuid, 3600*24, "/", config.Config.HTTP.CookieDomain, false, true)
|
||||
}
|
||||
@ -1,157 +0,0 @@
|
||||
package routes
|
||||
|
||||
import (
|
||||
"github.com/didi/nightingale/src/common/dataobj"
|
||||
"github.com/didi/nightingale/src/modules/transfer/backend"
|
||||
"github.com/didi/nightingale/src/toolkits/http/render"
|
||||
"github.com/didi/nightingale/src/toolkits/stats"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/toolkits/pkg/errors"
|
||||
"github.com/toolkits/pkg/logger"
|
||||
)
|
||||
|
||||
func QueryData(c *gin.Context) {
|
||||
stats.Counter.Set("data.api.qp10s", 1)
|
||||
|
||||
dataSource, err := backend.GetDataSourceFor("")
|
||||
if err != nil {
|
||||
logger.Warningf("could not find datasource")
|
||||
render.Message(c, err)
|
||||
return
|
||||
}
|
||||
|
||||
var input []dataobj.QueryData
|
||||
errors.Dangerous(c.ShouldBindJSON(&input))
|
||||
resp := dataSource.QueryData(input)
|
||||
render.Data(c, resp, nil)
|
||||
}
|
||||
|
||||
func QueryDataForUI(c *gin.Context) {
|
||||
stats.Counter.Set("data.ui.qp10s", 1)
|
||||
var input dataobj.QueryDataForUI
|
||||
var respData []*dataobj.QueryDataForUIResp
|
||||
errors.Dangerous(c.ShouldBindJSON(&input))
|
||||
start := input.Start
|
||||
end := input.End
|
||||
|
||||
dataSource, err := backend.GetDataSourceFor("")
|
||||
if err != nil {
|
||||
logger.Warningf("could not find datasource")
|
||||
render.Message(c, err)
|
||||
return
|
||||
}
|
||||
resp := dataSource.QueryDataForUI(input)
|
||||
for _, d := range resp {
|
||||
data := &dataobj.QueryDataForUIResp{
|
||||
Start: d.Start,
|
||||
End: d.End,
|
||||
Endpoint: d.Endpoint,
|
||||
Nid: d.Nid,
|
||||
Counter: d.Counter,
|
||||
DsType: d.DsType,
|
||||
Step: d.Step,
|
||||
Values: d.Values,
|
||||
}
|
||||
respData = append(respData, data)
|
||||
}
|
||||
|
||||
if len(input.Comparisons) > 1 {
|
||||
for i := 1; i < len(input.Comparisons); i++ {
|
||||
comparison := input.Comparisons[i]
|
||||
input.Start = start - comparison
|
||||
input.End = end - comparison
|
||||
res := dataSource.QueryDataForUI(input)
|
||||
for _, d := range res {
|
||||
for j := range d.Values {
|
||||
d.Values[j].Timestamp += comparison
|
||||
}
|
||||
|
||||
data := &dataobj.QueryDataForUIResp{
|
||||
Start: d.Start,
|
||||
End: d.End,
|
||||
Endpoint: d.Endpoint,
|
||||
Nid: d.Nid,
|
||||
Counter: d.Counter,
|
||||
DsType: d.DsType,
|
||||
Step: d.Step,
|
||||
Values: d.Values,
|
||||
Comparison: comparison,
|
||||
}
|
||||
respData = append(respData, data)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
render.Data(c, respData, nil)
|
||||
}
|
||||
|
||||
func GetMetrics(c *gin.Context) {
|
||||
stats.Counter.Set("metric.qp10s", 1)
|
||||
recv := dataobj.EndpointsRecv{}
|
||||
errors.Dangerous(c.ShouldBindJSON(&recv))
|
||||
|
||||
dataSource, err := backend.GetDataSourceFor("")
|
||||
if err != nil {
|
||||
logger.Warningf("could not find datasource")
|
||||
render.Message(c, err)
|
||||
return
|
||||
}
|
||||
|
||||
resp := dataSource.QueryMetrics(recv)
|
||||
|
||||
render.Data(c, resp, nil)
|
||||
}
|
||||
|
||||
func GetTagPairs(c *gin.Context) {
|
||||
stats.Counter.Set("tag.qp10s", 1)
|
||||
recv := dataobj.EndpointMetricRecv{}
|
||||
errors.Dangerous(c.ShouldBindJSON(&recv))
|
||||
|
||||
dataSource, err := backend.GetDataSourceFor("")
|
||||
if err != nil {
|
||||
logger.Warningf("could not find datasource")
|
||||
render.Message(c, err)
|
||||
return
|
||||
}
|
||||
|
||||
resp := dataSource.QueryTagPairs(recv)
|
||||
render.Data(c, resp, nil)
|
||||
}
|
||||
|
||||
func GetIndexByClude(c *gin.Context) {
|
||||
stats.Counter.Set("xclude.qp10s", 1)
|
||||
recvs := make([]dataobj.CludeRecv, 0)
|
||||
errors.Dangerous(c.ShouldBindJSON(&recvs))
|
||||
|
||||
dataSource, err := backend.GetDataSourceFor("")
|
||||
if err != nil {
|
||||
logger.Warningf("could not find datasource")
|
||||
render.Message(c, err)
|
||||
return
|
||||
}
|
||||
|
||||
resp := dataSource.QueryIndexByClude(recvs)
|
||||
render.Data(c, resp, nil)
|
||||
}
|
||||
|
||||
func GetIndexByFullTags(c *gin.Context) {
|
||||
stats.Counter.Set("counter.qp10s", 1)
|
||||
recvs := make([]dataobj.IndexByFullTagsRecv, 0)
|
||||
errors.Dangerous(c.ShouldBindJSON(&recvs))
|
||||
|
||||
dataSource, err := backend.GetDataSourceFor("")
|
||||
if err != nil {
|
||||
logger.Warningf("could not find datasource")
|
||||
render.Message(c, err)
|
||||
return
|
||||
}
|
||||
|
||||
resp := dataSource.QueryIndexByFullTags(recvs)
|
||||
render.Data(c, &listResp{List: resp, Count: len(resp)}, nil)
|
||||
}
|
||||
|
||||
type listResp struct {
|
||||
List interface{} `json:"list"`
|
||||
Count int `json:"count"`
|
||||
}
|
||||
Loading…
Reference in new issue