Fixed metrics on/off setting
This commit is contained in:
parent
43676f045c
commit
1d98e4cabe
2 changed files with 65 additions and 62 deletions
|
@ -211,7 +211,7 @@ func GetUserFromClaims(claims jwt.MapClaims) (user *User, err error) {
|
||||||
func UpdateActiveUsersFromContext(c echo.Context) (err error) {
|
func UpdateActiveUsersFromContext(c echo.Context) (err error) {
|
||||||
user, err := GetCurrentUser(c)
|
user, err := GetCurrentUser(c)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
allActiveUsers, err := metrics.GetActiveUsers()
|
allActiveUsers, err := metrics.GetActiveUsers()
|
||||||
|
|
|
@ -27,72 +27,75 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
func setupMetrics(a *echo.Group) {
|
func setupMetrics(a *echo.Group) {
|
||||||
if config.ServiceEnableMetrics.GetBool() {
|
if !config.ServiceEnableMetrics.GetBool() {
|
||||||
|
return
|
||||||
if !config.RedisEnabled.GetBool() {
|
|
||||||
log.Fatal("You have to enable redis in order to use metrics")
|
|
||||||
}
|
|
||||||
|
|
||||||
metrics.InitMetrics()
|
|
||||||
|
|
||||||
type countable struct {
|
|
||||||
Rediskey string
|
|
||||||
Type interface{}
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, c := range []countable{
|
|
||||||
{
|
|
||||||
metrics.ListCountKey,
|
|
||||||
models.List{},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
metrics.UserCountKey,
|
|
||||||
models.User{},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
metrics.NamespaceCountKey,
|
|
||||||
models.Namespace{},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
metrics.TaskCountKey,
|
|
||||||
models.Task{},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
metrics.TeamCountKey,
|
|
||||||
models.Team{},
|
|
||||||
},
|
|
||||||
} {
|
|
||||||
// Set initial totals
|
|
||||||
total, err := models.GetTotalCount(c.Type)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Could not set initial count for %v, error was %s", c.Type, err)
|
|
||||||
}
|
|
||||||
if err := metrics.SetCount(total, c.Rediskey); err != nil {
|
|
||||||
log.Fatalf("Could not set initial count for %v, error was %s", c.Type, err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// init active users, sometimes we'll have garbage from previous runs in redis instead
|
|
||||||
if err := metrics.SetActiveUsers([]*metrics.ActiveUser{}); err != nil {
|
|
||||||
log.Fatalf("Could not set initial count for active users, error was %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
a.GET("/metrics", echo.WrapHandler(promhttp.Handler()))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if !config.RedisEnabled.GetBool() {
|
||||||
|
log.Fatal("You have to enable redis in order to use metrics")
|
||||||
|
}
|
||||||
|
|
||||||
|
metrics.InitMetrics()
|
||||||
|
|
||||||
|
type countable struct {
|
||||||
|
Rediskey string
|
||||||
|
Type interface{}
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, c := range []countable{
|
||||||
|
{
|
||||||
|
metrics.ListCountKey,
|
||||||
|
models.List{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
metrics.UserCountKey,
|
||||||
|
models.User{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
metrics.NamespaceCountKey,
|
||||||
|
models.Namespace{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
metrics.TaskCountKey,
|
||||||
|
models.Task{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
metrics.TeamCountKey,
|
||||||
|
models.Team{},
|
||||||
|
},
|
||||||
|
} {
|
||||||
|
// Set initial totals
|
||||||
|
total, err := models.GetTotalCount(c.Type)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Could not get initial count for %v, error was %s", c.Type, err)
|
||||||
|
}
|
||||||
|
if err := metrics.SetCount(total, c.Rediskey); err != nil {
|
||||||
|
log.Fatalf("Could not set initial count for %v, error was %s", c.Type, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// init active users, sometimes we'll have garbage from previous runs in redis instead
|
||||||
|
if err := metrics.SetActiveUsers([]*metrics.ActiveUser{}); err != nil {
|
||||||
|
log.Fatalf("Could not set initial count for active users, error was %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
a.GET("/metrics", echo.WrapHandler(promhttp.Handler()))
|
||||||
}
|
}
|
||||||
|
|
||||||
func setupMetricsMiddleware(a *echo.Group) {
|
func setupMetricsMiddleware(a *echo.Group) {
|
||||||
if config.ServiceJWTSecret.GetBool() {
|
if !config.ServiceEnableMetrics.GetBool() {
|
||||||
a.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
|
return
|
||||||
return func(c echo.Context) error {
|
}
|
||||||
|
|
||||||
// Update currently active users
|
a.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
|
||||||
if err := models.UpdateActiveUsersFromContext(c); err != nil {
|
return func(c echo.Context) error {
|
||||||
log.Error(err)
|
|
||||||
return next(c)
|
// Update currently active users
|
||||||
}
|
if err := models.UpdateActiveUsersFromContext(c); err != nil {
|
||||||
|
log.Error(err)
|
||||||
return next(c)
|
return next(c)
|
||||||
}
|
}
|
||||||
})
|
return next(c)
|
||||||
}
|
}
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue