mirror of
https://github.com/grafana/grafana.git
synced 2024-11-26 19:00:54 -06:00
31 lines
852 B
Go
31 lines
852 B
Go
package middleware
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/services/quota"
|
|
"github.com/grafana/grafana/pkg/web"
|
|
)
|
|
|
|
// Quota returns a function that returns a function used to call quotaservice based on target name
|
|
func Quota(quotaService quota.Service) func(string) web.Handler {
|
|
if quotaService == nil {
|
|
panic("quotaService is nil")
|
|
}
|
|
//https://open.spotify.com/track/7bZSoBEAEEUsGEuLOf94Jm?si=T1Tdju5qRSmmR0zph_6RBw fuuuuunky
|
|
return func(targetSrv string) web.Handler {
|
|
return func(c *models.ReqContext) {
|
|
limitReached, err := quotaService.QuotaReached(c, quota.TargetSrv(targetSrv))
|
|
if err != nil {
|
|
c.JsonApiErr(500, "Failed to get quota", err)
|
|
return
|
|
}
|
|
if limitReached {
|
|
c.JsonApiErr(403, fmt.Sprintf("%s Quota reached", targetSrv), nil)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|