mirror of
https://github.com/openfaas/faas.git
synced 2025-06-09 08:46:48 +00:00
7df548668f
Trivial change to add logging around scale from zero events in scaling.go. Previously scale from zero events were not logged in the same way that normal scaling events are. This change adds log writes to show when a scale from zero was requested and when a function successfully moved to > 0 replicas. Signed-off-by: Richard Gee <richard@technologee.co.uk>
97 lines
2.6 KiB
Go
97 lines
2.6 KiB
Go
// Copyright (c) OpenFaaS Author(s). All rights reserved.
|
|
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
|
|
|
|
package handlers
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
// ScalingConfig for scaling behaviours
|
|
type ScalingConfig struct {
|
|
MaxPollCount uint
|
|
FunctionPollInterval time.Duration
|
|
CacheExpiry time.Duration
|
|
ServiceQuery ServiceQuery
|
|
}
|
|
|
|
// MakeScalingHandler creates handler which can scale a function from
|
|
// zero to 1 replica(s).
|
|
func MakeScalingHandler(next http.HandlerFunc, upstream http.HandlerFunc, config ScalingConfig) http.HandlerFunc {
|
|
cache := FunctionCache{
|
|
Cache: make(map[string]*FunctionMeta),
|
|
Expiry: config.CacheExpiry,
|
|
}
|
|
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
functionName := getServiceName(r.URL.String())
|
|
|
|
if serviceQueryResponse, hit := cache.Get(functionName); hit && serviceQueryResponse.AvailableReplicas > 0 {
|
|
next.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
|
|
queryResponse, err := config.ServiceQuery.GetReplicas(functionName)
|
|
|
|
if err != nil {
|
|
var errStr string
|
|
errStr = fmt.Sprintf("error finding function %s: %s", functionName, err.Error())
|
|
|
|
log.Printf(errStr)
|
|
w.WriteHeader(http.StatusNotFound)
|
|
w.Write([]byte(errStr))
|
|
return
|
|
}
|
|
|
|
cache.Set(functionName, queryResponse)
|
|
|
|
if queryResponse.AvailableReplicas == 0 {
|
|
minReplicas := uint64(1)
|
|
if queryResponse.MinReplicas > 0 {
|
|
minReplicas = queryResponse.MinReplicas
|
|
}
|
|
|
|
log.Printf("[Scale] function=%s 0 => %d requested", functionName, minReplicas)
|
|
scalingStartTime := time.Now()
|
|
|
|
err := config.ServiceQuery.SetReplicas(functionName, minReplicas)
|
|
if err != nil {
|
|
errStr := fmt.Errorf("unable to scale function [%s], err: %s", functionName, err)
|
|
log.Printf(errStr.Error())
|
|
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
w.Write([]byte(errStr.Error()))
|
|
return
|
|
}
|
|
|
|
for i := 0; i < int(config.MaxPollCount); i++ {
|
|
queryResponse, err := config.ServiceQuery.GetReplicas(functionName)
|
|
cache.Set(functionName, queryResponse)
|
|
|
|
if err != nil {
|
|
errStr := fmt.Sprintf("error: %s", err.Error())
|
|
log.Printf(errStr)
|
|
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
w.Write([]byte(errStr))
|
|
return
|
|
}
|
|
|
|
if queryResponse.AvailableReplicas > 0 {
|
|
scalingDuration := time.Since(scalingStartTime)
|
|
log.Printf("[Scale] function=%s 0 => %d successful - %f seconds", functionName, queryResponse.AvailableReplicas, scalingDuration.Seconds())
|
|
break
|
|
}
|
|
|
|
time.Sleep(config.FunctionPollInterval)
|
|
}
|
|
}
|
|
|
|
next.ServeHTTP(w, r)
|
|
}
|
|
}
|