faas-rs/faas
1
0
Fork 0
mirror of https://github.com/openfaas/faas.git synced 2025-06-16 12:16:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Extract magic variables

Browse Source
This commit is contained in:
Alex Ellis 2017-04-08 09:53:10 +01:00 committed by Alex Ellis
parent a71329de4a
commit 4a0c712c0f
1 changed files with 26 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
gateway/handlers/alerthandler.go
View File

@ -14,6 +14,9 @@ import (
"github.com/docker/docker/client" "github.com/docker/docker/client"
) )
// DefaultMaxReplicas is the amount of replicas a service will auto-scale up to.
const DefaultMaxReplicas = 20
// MakeAlertHandler handles alerts from Prometheus Alertmanager // MakeAlertHandler handles alerts from Prometheus Alertmanager
func MakeAlertHandler(c *client.Client) http.HandlerFunc { func MakeAlertHandler(c *client.Client) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) { return func(w http.ResponseWriter, r *http.Request) {
@ -45,26 +48,6 @@ func MakeAlertHandler(c *client.Client) http.HandlerFunc {
} }
} }
// CalculateReplicas decides what replica count to set depending on a Prometheus alert
func CalculateReplicas(status string, currentReplicas uint64, maxReplicas uint64) uint64 {
newReplicas := currentReplicas
if status == "firing" {
if currentReplicas == 1 {
newReplicas = 5
} else {
if currentReplicas+5 > maxReplicas {
newReplicas = maxReplicas
} else {
newReplicas = currentReplicas + 5
}
}
} else { // Resolved event.
newReplicas = 1
}
return newReplicas
}
func scaleService(req requests.PrometheusAlert, c *client.Client) error { func scaleService(req requests.PrometheusAlert, c *client.Client) error {
var err error var err error
serviceName := req.Alerts[0].Labels.FunctionName serviceName := req.Alerts[0].Labels.FunctionName
@ -78,7 +61,7 @@ func scaleService(req requests.PrometheusAlert, c *client.Client) error {
status := req.Status status := req.Status
replicaLabel := service.Spec.TaskTemplate.ContainerSpec.Labels["com.faas.max_replicas"] replicaLabel := service.Spec.TaskTemplate.ContainerSpec.Labels["com.faas.max_replicas"]
maxReplicas := 20 maxReplicas := DefaultMaxReplicas
if len(replicaLabel) > 0 { if len(replicaLabel) > 0 {
maxReplicas, err = strconv.Atoi(replicaLabel) maxReplicas, err = strconv.Atoi(replicaLabel)
if err != nil { if err != nil {
@ -108,3 +91,25 @@ func scaleService(req requests.PrometheusAlert, c *client.Client) error {
} }
return err return err
} }
// CalculateReplicas decides what replica count to set depending on current/desired amount
func CalculateReplicas(status string, currentReplicas uint64, maxReplicas uint64) uint64 {
newReplicas := currentReplicas
const step = 5
if status == "firing" {
if currentReplicas == 1 {
// First jump is from 1 to "step" i.e. 1->5
newReplicas = step
} else {
if currentReplicas+step > maxReplicas {
newReplicas = maxReplicas
} else {
newReplicas = currentReplicas + step
}
}
} else { // Resolved event.
newReplicas = 1
}
return newReplicas
}