Ability to retain specialised pods for poolmanager functions (#2830)

- added retainPods flag to take in the number of specialized pods to retain
- add retainPods in both the create function and update function command
- modify crd keys to be typed instead of string
- keep track of function generation in case of update function operation
- add delete handler function to make sure specialized pods are deleted in case function is deleted

---------

Signed-off-by: Sanket Sudake <sanketsudake@gmail.com>
Signed-off-by: Pranoy Kundu <pranoy1998k@gmail.com>
Co-authored-by: Pranoy Kundu <pranoy1998k@gmail.com>
This commit is contained in:
Sanket Sudake
2023-09-27 13:33:19 +05:30
committed by GitHub
co-authored by Pranoy Kundu
parent 657aee7cc2
commit 56b49dcee8
26 changed files with 405 additions and 211 deletions
+2 -2
View File
@@ -51,7 +51,7 @@ test-run: code-checks
### Binaries ### Binaries
build-fission-cli: build-fission-cli:
@GOOS=$(GOOS) GOARCH=$(GOARCH) GOAMD64=$(GOAMD64) GORELEASER_CURRENT_TAG=$(VERSION) goreleaser build --snapshot --rm-dist --single-target --id fission-cli @GOOS=$(GOOS) GOARCH=$(GOARCH) GOAMD64=$(GOAMD64) GORELEASER_CURRENT_TAG=$(VERSION) goreleaser build --snapshot --clean --single-target --id fission-cli
install-fission-cli: install-fission-cli:
# TODO: Fix this hack, replace v1 with GOAMD64 # TODO: Fix this hack, replace v1 with GOAMD64
@@ -111,7 +111,7 @@ generate-crd-ref-docs: install-crd-ref-docs
all-generators: codegen generate-crds generate-swagger-doc generate-cli-docs generate-crd-ref-docs all-generators: codegen generate-crds generate-swagger-doc generate-cli-docs generate-crd-ref-docs
skaffold-prebuild: skaffold-prebuild:
@GOOS=linux GOARCH=amd64 GORELEASER_CURRENT_TAG=$(VERSION) goreleaser build --snapshot --rm-dist --single-target @GOOS=linux GOARCH=amd64 GORELEASER_CURRENT_TAG=$(VERSION) goreleaser build --snapshot --clean --single-target
@cp -v cmd/builder/Dockerfile dist/builder_linux_amd64_v1/Dockerfile @cp -v cmd/builder/Dockerfile dist/builder_linux_amd64_v1/Dockerfile
@cp -v cmd/fetcher/Dockerfile dist/fetcher_linux_amd64_v1/Dockerfile @cp -v cmd/fetcher/Dockerfile dist/fetcher_linux_amd64_v1/Dockerfile
@cp -v cmd/fission-bundle/Dockerfile dist/fission-bundle_linux_amd64_v1/Dockerfile @cp -v cmd/fission-bundle/Dockerfile dist/fission-bundle_linux_amd64_v1/Dockerfile
+5
View File
@@ -8047,6 +8047,11 @@ spec:
More info: https://kubernetes.io/docs/concepts/configuration/manage-resources-containers/' More info: https://kubernetes.io/docs/concepts/configuration/manage-resources-containers/'
type: object type: object
type: object type: object
retainPods:
description: RetainPods specifies the number of specialized pods that
should be retained after serving requests This is optional. If not
specified default value will be taken as 0
type: integer
secrets: secrets:
description: Reference to a list of secrets. description: Reference to a list of secrets.
items: items:
+9
View File
@@ -399,6 +399,11 @@ type (
// +optional // +optional
OnceOnly bool `json:"onceOnly,omitempty"` OnceOnly bool `json:"onceOnly,omitempty"`
// RetainPods specifies the number of specialized pods that should be retained after serving requests
// This is optional. If not specified default value will be taken as 0
// +optional
RetainPods int `json:"retainPods,omitempty"`
// Podspec specifies podspec to use for executor type container based functions // Podspec specifies podspec to use for executor type container based functions
// Different arguments mentioned for container based function are populated inside a pod. // Different arguments mentioned for container based function are populated inside a pod.
// +optional // +optional
@@ -876,6 +881,10 @@ func (fn Function) GetConcurrency() int {
return fn.Spec.Concurrency return fn.Spec.Concurrency
} }
func (fn Function) GetRetainPods() int {
return fn.Spec.RetainPods
}
func (fn Function) GetRequestPerPod() int { func (fn Function) GetRequestPerPod() int {
if fn.Spec.RequestsPerPod == 0 { if fn.Spec.RequestsPerPod == 0 {
return DefaultRequestsPerPod return DefaultRequestsPerPod
@@ -203,6 +203,7 @@ var map_FunctionSpec = map[string]string{
"concurrency": "Maximum number of pods to be specialized which will serve requests This is optional. If not specified default value will be taken as 500", "concurrency": "Maximum number of pods to be specialized which will serve requests This is optional. If not specified default value will be taken as 500",
"requestsPerPod": "RequestsPerPod indicates the maximum number of concurrent requests that can be served by a specialized pod This is optional. If not specified default value will be taken as 1", "requestsPerPod": "RequestsPerPod indicates the maximum number of concurrent requests that can be served by a specialized pod This is optional. If not specified default value will be taken as 1",
"onceOnly": "OnceOnly specifies if specialized pod will serve exactly one request in its lifetime and would be garbage collected after serving that one request This is optional. If not specified default value will be taken as false", "onceOnly": "OnceOnly specifies if specialized pod will serve exactly one request in its lifetime and would be garbage collected after serving that one request This is optional. If not specified default value will be taken as false",
"retainPods": "RetainPods specifies the number of specialized pods that should be retained after serving requests This is optional. If not specified default value will be taken as 0",
"podspec": "Podspec specifies podspec to use for executor type container based functions Different arguments mentioned for container based function are populated inside a pod.", "podspec": "Podspec specifies podspec to use for executor type container based functions Different arguments mentioned for container based function are populated inside a pod.",
} }
+10 -9
View File
@@ -28,6 +28,7 @@ import (
apiv1 "k8s.io/api/core/v1" apiv1 "k8s.io/api/core/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/labels" "k8s.io/apimachinery/pkg/labels"
"k8s.io/apimachinery/pkg/types"
"k8s.io/apimachinery/pkg/util/intstr" "k8s.io/apimachinery/pkg/util/intstr"
"k8s.io/client-go/kubernetes" "k8s.io/client-go/kubernetes"
k8sCache "k8s.io/client-go/tools/cache" k8sCache "k8s.io/client-go/tools/cache"
@@ -62,7 +63,7 @@ type (
environmentWatcher struct { environmentWatcher struct {
logger *zap.Logger logger *zap.Logger
cache map[string]*builderInfo cache map[types.UID]*builderInfo
fissionClient versioned.Interface fissionClient versioned.Interface
kubernetesClient kubernetes.Interface kubernetesClient kubernetes.Interface
nsResolver *utils.NamespaceResolver nsResolver *utils.NamespaceResolver
@@ -96,7 +97,7 @@ func makeEnvironmentWatcher(
envWatcher := &environmentWatcher{ envWatcher := &environmentWatcher{
logger: logger.Named("environment_watcher"), logger: logger.Named("environment_watcher"),
cache: make(map[string]*builderInfo), cache: make(map[types.UID]*builderInfo),
fissionClient: fissionClient, fissionClient: fissionClient,
kubernetesClient: kubernetesClient, kubernetesClient: kubernetesClient,
nsResolver: utils.DefaultNSResolver(), nsResolver: utils.DefaultNSResolver(),
@@ -165,13 +166,13 @@ func (envw *environmentWatcher) EnvWatchEventHandlers(ctx context.Context) error
func (envw *environmentWatcher) AddUpdateBuilder(ctx context.Context, env *fv1.Environment) { func (envw *environmentWatcher) AddUpdateBuilder(ctx context.Context, env *fv1.Environment) {
//builder is not supported with v1 interface and ignore env without builder image //builder is not supported with v1 interface and ignore env without builder image
if env.Spec.Version != 1 && len(env.Spec.Builder.Image) != 0 { if env.Spec.Version != 1 && len(env.Spec.Builder.Image) != 0 {
if _, ok := envw.cache[crd.CacheKeyUID(&env.ObjectMeta)]; !ok { if _, ok := envw.cache[crd.CacheKeyUIDFromMeta(&env.ObjectMeta)]; !ok {
builderInfo, err := envw.createBuilder(ctx, env, envw.nsResolver.GetBuilderNS(env.ObjectMeta.Namespace)) builderInfo, err := envw.createBuilder(ctx, env, envw.nsResolver.GetBuilderNS(env.ObjectMeta.Namespace))
if err != nil { if err != nil {
envw.logger.Error("error creating builder service", zap.Error(err)) envw.logger.Error("error creating builder service", zap.Error(err))
return return
} }
envw.cache[crd.CacheKeyUID(&env.ObjectMeta)] = builderInfo envw.cache[crd.CacheKeyUIDFromMeta(&env.ObjectMeta)] = builderInfo
} else { } else {
envw.DeleteBuilder(ctx, env) envw.DeleteBuilder(ctx, env)
// once older builder deleted then add new builder service // once older builder deleted then add new builder service
@@ -180,16 +181,16 @@ func (envw *environmentWatcher) AddUpdateBuilder(ctx context.Context, env *fv1.E
envw.logger.Error("error updating builder service", zap.Error(err)) envw.logger.Error("error updating builder service", zap.Error(err))
return return
} }
envw.cache[crd.CacheKeyUID(&env.ObjectMeta)] = builderInfo envw.cache[crd.CacheKeyUIDFromMeta(&env.ObjectMeta)] = builderInfo
} }
} }
} }
func (envw *environmentWatcher) DeleteBuilder(ctx context.Context, env *fv1.Environment) { func (envw *environmentWatcher) DeleteBuilder(ctx context.Context, env *fv1.Environment) {
if _, ok := envw.cache[crd.CacheKeyUID(&env.ObjectMeta)]; ok { if _, ok := envw.cache[crd.CacheKeyUIDFromMeta(&env.ObjectMeta)]; ok {
envw.DeleteBuilderService(ctx, env) envw.DeleteBuilderService(ctx, env)
envw.DeleteBuilderDeployment(ctx, env) envw.DeleteBuilderDeployment(ctx, env)
delete(envw.cache, crd.CacheKeyUID(&env.ObjectMeta)) delete(envw.cache, crd.CacheKeyUIDFromMeta(&env.ObjectMeta))
envw.logger.Info("builder service deleted", zap.String("env_name", env.ObjectMeta.Name), zap.String("namespace", envw.nsResolver.GetBuilderNS(env.ObjectMeta.Namespace))) envw.logger.Info("builder service deleted", zap.String("env_name", env.ObjectMeta.Name), zap.String("namespace", envw.nsResolver.GetBuilderNS(env.ObjectMeta.Namespace)))
} else { } else {
envw.logger.Debug("builder service not found", zap.String("env_name", env.ObjectMeta.Name), zap.String("namespace", envw.nsResolver.GetBuilderNS(env.ObjectMeta.Namespace))) envw.logger.Debug("builder service not found", zap.String("env_name", env.ObjectMeta.Name), zap.String("namespace", envw.nsResolver.GetBuilderNS(env.ObjectMeta.Namespace)))
@@ -204,7 +205,7 @@ func (envw *environmentWatcher) DeleteBuilderService(ctx context.Context, env *f
} }
for _, svc := range svcList { for _, svc := range svcList {
envName := svc.ObjectMeta.Labels[LABEL_ENV_NAME] envName := svc.ObjectMeta.Labels[LABEL_ENV_NAME]
if _, ok := envw.cache[crd.CacheKeyUID(&env.ObjectMeta)]; ok { if _, ok := envw.cache[crd.CacheKeyUIDFromMeta(&env.ObjectMeta)]; ok {
err := envw.deleteBuilderServiceByName(ctx, svc.ObjectMeta.Name, svc.ObjectMeta.Namespace) err := envw.deleteBuilderServiceByName(ctx, svc.ObjectMeta.Name, svc.ObjectMeta.Namespace)
if err != nil { if err != nil {
envw.logger.Error("error removing builder service", zap.Error(err), envw.logger.Error("error removing builder service", zap.Error(err),
@@ -228,7 +229,7 @@ func (envw *environmentWatcher) DeleteBuilderDeployment(ctx context.Context, env
envw.logger.Error("error getting the builder deployment list", zap.Error(err)) envw.logger.Error("error getting the builder deployment list", zap.Error(err))
} }
for _, deploy := range deployList { for _, deploy := range deployList {
if _, ok := envw.cache[crd.CacheKeyUID(&env.ObjectMeta)]; ok { if _, ok := envw.cache[crd.CacheKeyUIDFromMeta(&env.ObjectMeta)]; ok {
err := envw.deleteBuilderDeploymentByName(ctx, deploy.ObjectMeta.Name, deploy.ObjectMeta.Namespace) err := envw.deleteBuilderDeploymentByName(ctx, deploy.ObjectMeta.Name, deploy.ObjectMeta.Namespace)
if err != nil { if err != nil {
envw.logger.Error("error removing builder deployment", zap.Error(err), envw.logger.Error("error removing builder deployment", zap.Error(err),
+46 -10
View File
@@ -20,21 +20,57 @@ import (
"fmt" "fmt"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/types"
) )
// CacheKey : Given metadata, create a key that uniquely identifies the contents type CacheKeyUR struct {
// of the object. Since resourceVersion changes on every update and UID types.UID
// UIDs are unique, uid+resourceVersion identifies the ResourceVersion string
// content. (ResourceVersion may also update on status updates, so }
// this will result in some unnecessary cache misses. That should be
// ok.) func (ck CacheKeyUR) String() string {
func CacheKey(metadata *metav1.ObjectMeta) string { return fmt.Sprintf("%v_%v", ck.UID, ck.ResourceVersion)
return fmt.Sprintf("%v_%v", metadata.UID, metadata.ResourceVersion) }
type CacheKeyURG struct {
UID types.UID
ResourceVersion string
Generation int64
}
func (ck CacheKeyURG) String() string {
return fmt.Sprintf("%v_%v_%v", ck.UID, ck.ResourceVersion, ck.Generation)
} }
// CacheKeyForUID create a key that uniquely identifies the // CacheKeyForUID create a key that uniquely identifies the
// of the object. Since resourceVersion changes on every update and // of the object. Since resourceVersion changes on every update and
// UIDs are unique, we don't use resource version here // UIDs are unique, we don't use resource version here
func CacheKeyUID(metadata *metav1.ObjectMeta) string { func CacheKeyUIDFromMeta(metadata *metav1.ObjectMeta) types.UID {
return fmt.Sprintf("%v", metadata.UID) return metadata.UID
}
// CacheKeyURFromMeta : Given metadata, create a key that uniquely identifies the contents
// of the object. Since resourceVersion changes on every update and
// UIDs are unique, uid+resourceVersion identifies the
// content. (ResourceVersion may also update on status updates, so
// this will result in some unnecessary cache misses. That should be
// ok.)
func CacheKeyURFromMeta(metadata *metav1.ObjectMeta) CacheKeyUR {
return CacheKeyUR{
UID: metadata.UID,
ResourceVersion: metadata.ResourceVersion,
}
}
// CacheKeyURGFromMeta : Given metadata, create a key that uniquely identifies the contents
// of the object. Since resourceVersion changes on every update and
// UIDs are unique, uid+resourceVersion identifies the
// content.
// Generation is also included to identify latest generation of the object.
func CacheKeyURGFromMeta(metadata *metav1.ObjectMeta) CacheKeyURG {
return CacheKeyURG{
UID: metadata.UID,
ResourceVersion: metadata.ResourceVersion,
Generation: metadata.Generation,
}
} }
+3 -5
View File
@@ -31,7 +31,6 @@ import (
"go.uber.org/zap" "go.uber.org/zap"
fv1 "github.com/fission/fission/pkg/apis/core/v1" fv1 "github.com/fission/fission/pkg/apis/core/v1"
"github.com/fission/fission/pkg/crd"
ferror "github.com/fission/fission/pkg/error" ferror "github.com/fission/fission/pkg/error"
"github.com/fission/fission/pkg/executor/client" "github.com/fission/fission/pkg/executor/client"
"github.com/fission/fission/pkg/executor/fscache" "github.com/fission/fission/pkg/executor/fscache"
@@ -157,9 +156,9 @@ func (executor *Executor) getServiceForFunction(ctx context.Context, fn *fv1.Fun
return return
} }
if funcSvc != nil { if funcSvc != nil {
et.UnTapService(ctx, crd.CacheKey(funcSvc.Function), resp.funcSvc.Address) et.UnTapService(ctx, funcSvc.Function, resp.funcSvc.Address)
} else { } else {
et.MarkSpecializationFailure(ctx, crd.CacheKey(&fn.ObjectMeta)) et.MarkSpecializationFailure(ctx, &fn.ObjectMeta)
} }
} }
if errors.Is(ctx.Err(), context.Canceled) { if errors.Is(ctx.Err(), context.Canceled) {
@@ -248,7 +247,6 @@ func (executor *Executor) unTapService(w http.ResponseWriter, r *http.Request) {
http.Error(w, "Failed to parse request", http.StatusBadRequest) http.Error(w, "Failed to parse request", http.StatusBadRequest)
return return
} }
key := fmt.Sprintf("%v_%v", tapSvcReq.FnMetadata.UID, tapSvcReq.FnMetadata.ResourceVersion)
t := tapSvcReq.FnExecutorType t := tapSvcReq.FnExecutorType
if t != fv1.ExecutorTypePoolmgr { if t != fv1.ExecutorTypePoolmgr {
msg := fmt.Sprintf("Unknown executor type '%v'", t) msg := fmt.Sprintf("Unknown executor type '%v'", t)
@@ -258,7 +256,7 @@ func (executor *Executor) unTapService(w http.ResponseWriter, r *http.Request) {
et := executor.executorTypes[t] et := executor.executorTypes[t]
et.UnTapService(ctx, key, tapSvcReq.ServiceURL) et.UnTapService(ctx, &tapSvcReq.FnMetadata, tapSvcReq.ServiceURL)
w.WriteHeader(http.StatusOK) w.WriteHeader(http.StatusOK)
} }
+3 -3
View File
@@ -138,13 +138,13 @@ func (executor *Executor) serveCreateFuncServices() {
} }
// Cache miss -- is this first one to request the func? // Cache miss -- is this first one to request the func?
wg, found := executor.fsCreateWg.Load(crd.CacheKey(fnMetadata)) wg, found := executor.fsCreateWg.Load(crd.CacheKeyURFromMeta(fnMetadata))
if !found { if !found {
// create a waitgroup for other requests for // create a waitgroup for other requests for
// the same function to wait on // the same function to wait on
wg := &sync.WaitGroup{} wg := &sync.WaitGroup{}
wg.Add(1) wg.Add(1)
executor.fsCreateWg.Store(crd.CacheKey(fnMetadata), wg) executor.fsCreateWg.Store(crd.CacheKeyURFromMeta(fnMetadata), wg)
// launch a goroutine for each request, to parallelize // launch a goroutine for each request, to parallelize
// the specialization of different functions // the specialization of different functions
@@ -176,7 +176,7 @@ func (executor *Executor) serveCreateFuncServices() {
funcSvc: fsvc, funcSvc: fsvc,
err: err, err: err,
} }
executor.fsCreateWg.Delete(crd.CacheKey(fnMetadata)) executor.fsCreateWg.Delete(crd.CacheKeyURFromMeta(fnMetadata))
wg.Done() wg.Done()
}() }()
} else { } else {
@@ -176,12 +176,12 @@ func (caaf *Container) GetTotalAvailable(fn *fv1.Function) int {
} }
// UnTapService has not been implemented for CaaF. // UnTapService has not been implemented for CaaF.
func (caaf *Container) UnTapService(ctx context.Context, key string, svcHost string) { func (caaf *Container) UnTapService(ctx context.Context, fnMeta *metav1.ObjectMeta, svcHost string) {
// Not Implemented for CaaF. // Not Implemented for CaaF.
} }
// MarkSpecializationFailure has not been implemented for CaaF. // MarkSpecializationFailure has not been implemented for CaaF.
func (caaf *Container) MarkSpecializationFailure(ctx context.Context, key string) { func (caaf *Container) MarkSpecializationFailure(ctx context.Context, fnMeta *metav1.ObjectMeta) {
// Not Implemented for CaaF. // Not Implemented for CaaF.
} }
+4 -2
View File
@@ -21,6 +21,8 @@ import (
"go.uber.org/zap" "go.uber.org/zap"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
fv1 "github.com/fission/fission/pkg/apis/core/v1" fv1 "github.com/fission/fission/pkg/apis/core/v1"
"github.com/fission/fission/pkg/executor/fscache" "github.com/fission/fission/pkg/executor/fscache"
) )
@@ -49,10 +51,10 @@ type ExecutorType interface {
TapService(ctx context.Context, serviceUrl string) error TapService(ctx context.Context, serviceUrl string) error
// UnTapService updates the isActive to false // UnTapService updates the isActive to false
UnTapService(ctx context.Context, key string, svcHost string) UnTapService(ctx context.Context, fnMeta *metav1.ObjectMeta, svcHost string)
// ReduceSpecializationInProgress updates the svcWaiting count in funcSvcGroup // ReduceSpecializationInProgress updates the svcWaiting count in funcSvcGroup
MarkSpecializationFailure(ctx context.Context, key string) MarkSpecializationFailure(ctx context.Context, fnMeta *metav1.ObjectMeta)
// IsValid returns true if a function service is valid. Different executor types // IsValid returns true if a function service is valid. Different executor types
// use distinct ways to examine the function service. // use distinct ways to examine the function service.
@@ -201,12 +201,12 @@ func (deploy *NewDeploy) DeleteFuncSvcFromCache(ctx context.Context, fsvc *fscac
} }
// UnTapService has not been implemented for NewDeployment. // UnTapService has not been implemented for NewDeployment.
func (deploy *NewDeploy) UnTapService(ctx context.Context, key string, svcHost string) { func (deploy *NewDeploy) UnTapService(ctx context.Context, fnMeta *metav1.ObjectMeta, svcHost string) {
// Not Implemented for NewDeployment. Will be used when support of concurrent specialization of same function is added. // Not Implemented for NewDeployment. Will be used when support of concurrent specialization of same function is added.
} }
// MarkSpecializationFailure has not been implemented for NewDeployment. // MarkSpecializationFailure has not been implemented for NewDeployment.
func (deploy *NewDeploy) MarkSpecializationFailure(ctx context.Context, key string) { func (deploy *NewDeploy) MarkSpecializationFailure(ctx context.Context, fnMeta *metav1.ObjectMeta) {
// Not Implemented for NewDeployment. Will be used when support of concurrent specialization of same function is added. // Not Implemented for NewDeployment. Will be used when support of concurrent specialization of same function is added.
} }
+3 -3
View File
@@ -202,7 +202,7 @@ func (gp *GenericPool) updateCPUUtilizationSvc(ctx context.Context) {
if value, ok := gp.podFSVCMap.Load(val.ObjectMeta.Name); ok { if value, ok := gp.podFSVCMap.Load(val.ObjectMeta.Name); ok {
if valArray, ok1 := value.([]interface{}); ok1 { if valArray, ok1 := value.([]interface{}); ok1 {
function, address := valArray[0], valArray[1] function, address := valArray[0], valArray[1]
gp.fsCache.SetCPUUtilizaton(function.(string), address.(string), p) gp.fsCache.SetCPUUtilizaton(function.(crd.CacheKeyURG), address.(string), p)
gp.logger.Info(fmt.Sprintf("updated function %s, address %s, cpuUsage %+v", function.(string), address.(string), p)) gp.logger.Info(fmt.Sprintf("updated function %s, address %s, cpuUsage %+v", function.(string), address.(string), p))
} }
} }
@@ -622,8 +622,8 @@ func (gp *GenericPool) getFuncSvc(ctx context.Context, fn *fv1.Function) (*fscac
} }
gp.fsCache.PodToFsvc.Store(pod.GetObjectMeta().GetName(), fsvc) gp.fsCache.PodToFsvc.Store(pod.GetObjectMeta().GetName(), fsvc)
gp.podFSVCMap.Store(pod.ObjectMeta.Name, []interface{}{crd.CacheKey(fsvc.Function), fsvc.Address}) gp.podFSVCMap.Store(pod.ObjectMeta.Name, []interface{}{crd.CacheKeyURGFromMeta(fsvc.Function), fsvc.Address})
gp.fsCache.AddFunc(ctx, *fsvc, fn.GetRequestPerPod()) gp.fsCache.AddFunc(ctx, *fsvc, fn.GetRequestPerPod(), fn.GetRetainPods())
logger.Info("added function service", logger.Info("added function service",
zap.String("pod", pod.ObjectMeta.Name), zap.String("pod", pod.ObjectMeta.Name),
+13 -11
View File
@@ -71,7 +71,7 @@ type (
GenericPoolManager struct { GenericPoolManager struct {
logger *zap.Logger logger *zap.Logger
pools map[string]*GenericPool pools map[k8sTypes.UID]*GenericPool
kubernetesClient kubernetes.Interface kubernetesClient kubernetes.Interface
metricsClient metricsclient.Interface metricsClient metricsclient.Interface
nsResolver *utils.NamespaceResolver nsResolver *utils.NamespaceResolver
@@ -141,7 +141,7 @@ func MakeGenericPoolManager(ctx context.Context,
} }
gpm := &GenericPoolManager{ gpm := &GenericPoolManager{
logger: gpmLogger, logger: gpmLogger,
pools: make(map[string]*GenericPool), pools: make(map[k8sTypes.UID]*GenericPool),
kubernetesClient: kubernetesClient, kubernetesClient: kubernetesClient,
nsResolver: utils.DefaultNSResolver(), nsResolver: utils.DefaultNSResolver(),
metricsClient: metricsClient, metricsClient: metricsClient,
@@ -237,9 +237,10 @@ func (gpm *GenericPoolManager) DeleteFuncSvcFromCache(ctx context.Context, fsvc
gpm.fsCache.DeleteFunctionSvc(ctx, fsvc) gpm.fsCache.DeleteFunctionSvc(ctx, fsvc)
} }
func (gpm *GenericPoolManager) UnTapService(ctx context.Context, key string, svcHost string) { func (gpm *GenericPoolManager) UnTapService(ctx context.Context, fnMeta *metav1.ObjectMeta, svcHost string) {
key := crd.CacheKeyURGFromMeta(fnMeta)
otelUtils.SpanTrackEvent(ctx, "UnTapService", otelUtils.SpanTrackEvent(ctx, "UnTapService",
attribute.KeyValue{Key: "key", Value: attribute.StringValue(key)}, attribute.KeyValue{Key: "key", Value: attribute.StringValue(key.String())},
attribute.KeyValue{Key: "svcHost", Value: attribute.StringValue(svcHost)}) attribute.KeyValue{Key: "svcHost", Value: attribute.StringValue(svcHost)})
gpm.fsCache.MarkAvailable(key, svcHost) gpm.fsCache.MarkAvailable(key, svcHost)
} }
@@ -254,9 +255,10 @@ func (gpm *GenericPoolManager) TapService(ctx context.Context, svcHost string) e
return nil return nil
} }
func (gpm *GenericPoolManager) MarkSpecializationFailure(ctx context.Context, key string) { func (gpm *GenericPoolManager) MarkSpecializationFailure(ctx context.Context, fnMeta *metav1.ObjectMeta) {
key := crd.CacheKeyURGFromMeta(fnMeta)
otelUtils.SpanTrackEvent(ctx, "MarkSpecializationFailure", otelUtils.SpanTrackEvent(ctx, "MarkSpecializationFailure",
attribute.KeyValue{Key: "key", Value: attribute.StringValue(key)}) attribute.KeyValue{Key: "key", Value: attribute.StringValue(key.String())})
logger := otelUtils.LoggerWithTraceID(ctx, gpm.logger) logger := otelUtils.LoggerWithTraceID(ctx, gpm.logger)
logger.Info("marking specialization failure", zap.Any("key", key)) logger.Info("marking specialization failure", zap.Any("key", key))
gpm.fsCache.MarkSpecializationFailure(key) gpm.fsCache.MarkSpecializationFailure(key)
@@ -503,7 +505,7 @@ func (gpm *GenericPoolManager) service() {
// just because they are missing in the cache, we end up creating another duplicate pool. // just because they are missing in the cache, we end up creating another duplicate pool.
var err error var err error
created := false created := false
pool, ok := gpm.pools[crd.CacheKeyUID(&req.env.ObjectMeta)] pool, ok := gpm.pools[crd.CacheKeyUIDFromMeta(&req.env.ObjectMeta)]
if !ok { if !ok {
// To support backward compatibility, if envs are created in default ns, we go ahead // To support backward compatibility, if envs are created in default ns, we go ahead
// and create pools in fission-function ns as earlier. // and create pools in fission-function ns as earlier.
@@ -516,7 +518,7 @@ func (gpm *GenericPoolManager) service() {
req.responseChannel <- &response{error: err} req.responseChannel <- &response{error: err}
continue continue
} }
gpm.pools[crd.CacheKeyUID(&req.env.ObjectMeta)] = pool gpm.pools[crd.CacheKeyUIDFromMeta(&req.env.ObjectMeta)] = pool
created = true created = true
} }
req.responseChannel <- &response{pool: pool, created: created} req.responseChannel <- &response{pool: pool, created: created}
@@ -526,7 +528,7 @@ func (gpm *GenericPoolManager) service() {
zap.String("environment", env.ObjectMeta.Name), zap.String("environment", env.ObjectMeta.Name),
zap.String("namespace", env.ObjectMeta.Namespace)) zap.String("namespace", env.ObjectMeta.Namespace))
key := crd.CacheKeyUID(&req.env.ObjectMeta) key := crd.CacheKeyUIDFromMeta(&req.env.ObjectMeta)
pool, ok := gpm.pools[key] pool, ok := gpm.pools[key]
if !ok { if !ok {
gpm.logger.Error("Could not find pool", zap.String("environment", env.ObjectMeta.Name), zap.String("namespace", env.ObjectMeta.Namespace)) gpm.logger.Error("Could not find pool", zap.String("environment", env.ObjectMeta.Name), zap.String("namespace", env.ObjectMeta.Namespace))
@@ -573,7 +575,7 @@ func (gpm *GenericPoolManager) getFunctionEnv(ctx context.Context, fn *fv1.Funct
// Cached ? // Cached ?
// TODO: the cache should be able to search by <env name, fn namespace> instead of function metadata. // TODO: the cache should be able to search by <env name, fn namespace> instead of function metadata.
result, err := gpm.functionEnv.Get(crd.CacheKey(&fn.ObjectMeta)) result, err := gpm.functionEnv.Get(crd.CacheKeyURFromMeta(&fn.ObjectMeta))
if err == nil { if err == nil {
env = result.(*fv1.Environment) env = result.(*fv1.Environment)
return env, nil return env, nil
@@ -591,7 +593,7 @@ func (gpm *GenericPoolManager) getFunctionEnv(ctx context.Context, fn *fv1.Funct
// cache for future lookups // cache for future lookups
m := fn.ObjectMeta m := fn.ObjectMeta
_, err = gpm.functionEnv.Set(crd.CacheKey(&m), env) _, err = gpm.functionEnv.Set(crd.CacheKeyURFromMeta(&m), env)
if err != nil { if err != nil {
gpm.logger.Error( gpm.logger.Error(
"failed to set the key", "failed to set the key",
@@ -36,6 +36,7 @@ import (
"k8s.io/client-go/util/workqueue" "k8s.io/client-go/util/workqueue"
fv1 "github.com/fission/fission/pkg/apis/core/v1" fv1 "github.com/fission/fission/pkg/apis/core/v1"
"github.com/fission/fission/pkg/crd"
"github.com/fission/fission/pkg/executor/fscache" "github.com/fission/fission/pkg/executor/fscache"
genInformer "github.com/fission/fission/pkg/generated/informers/externalversions" genInformer "github.com/fission/fission/pkg/generated/informers/externalversions"
flisterv1 "github.com/fission/fission/pkg/generated/listers/core/v1" flisterv1 "github.com/fission/fission/pkg/generated/listers/core/v1"
@@ -94,6 +95,14 @@ func NewPoolPodController(ctx context.Context, logger *zap.Logger,
} }
} }
} }
for _, factory := range finformerFactory {
_, err := factory.Core().V1().Functions().Informer().AddEventHandler(k8sCache.ResourceEventHandlerFuncs{
DeleteFunc: p.handleFuncDelete,
})
if err != nil {
return nil, err
}
}
for ns, informer := range finformerFactory { for ns, informer := range finformerFactory {
_, err := informer.Core().V1().Environments().Informer().AddEventHandler(k8sCache.ResourceEventHandlerFuncs{ _, err := informer.Core().V1().Environments().Informer().AddEventHandler(k8sCache.ResourceEventHandlerFuncs{
AddFunc: p.enqueueEnvAdd, AddFunc: p.enqueueEnvAdd,
@@ -133,6 +142,11 @@ func IsPodActive(p *v1.Pod) bool {
p.DeletionTimestamp == nil p.DeletionTimestamp == nil
} }
func (p *PoolPodController) handleFuncDelete(obj interface{}) {
fn := obj.(*fv1.Function)
p.gpm.fsCache.MarkFuncDeleted(crd.CacheKeyURGFromMeta(&fn.ObjectMeta))
}
func (p *PoolPodController) processRS(rs *apps.ReplicaSet) { func (p *PoolPodController) processRS(rs *apps.ReplicaSet) {
if *(rs.Spec.Replicas) != 0 { if *(rs.Spec.Replicas) != 0 {
return return
+21 -17
View File
@@ -134,9 +134,9 @@ func (fsc *FunctionServiceCache) service() {
funcObjects := make([]*FuncSvc, 0) funcObjects := make([]*FuncSvc, 0)
for _, funcSvc := range fscs { for _, funcSvc := range fscs {
mI := funcSvc.(metav1.ObjectMeta) mI := funcSvc.(metav1.ObjectMeta)
fsvcI, err := fsc.byFunction.Get(crd.CacheKey(&mI)) fsvcI, err := fsc.byFunction.Get(crd.CacheKeyURFromMeta(&mI))
if err != nil { if err != nil {
fsc.logger.Error("error while getting service", zap.Any("error", err)) fsc.logger.Error("error while getting service", zap.String("error", err.Error()))
return return
} }
fsvc := fsvcI.(*FuncSvc) fsvc := fsvcI.(*FuncSvc)
@@ -194,7 +194,7 @@ func (fsc *FunctionServiceCache) DumpDebugInfo(ctx context.Context) error {
// GetByFunction gets a function service from cache using function key. // GetByFunction gets a function service from cache using function key.
func (fsc *FunctionServiceCache) GetByFunction(m *metav1.ObjectMeta) (*FuncSvc, error) { func (fsc *FunctionServiceCache) GetByFunction(m *metav1.ObjectMeta) (*FuncSvc, error) {
key := crd.CacheKey(m) key := crd.CacheKeyURFromMeta(m)
fsvcI, err := fsc.byFunction.Get(key) fsvcI, err := fsc.byFunction.Get(key)
if err != nil { if err != nil {
@@ -211,7 +211,7 @@ func (fsc *FunctionServiceCache) GetByFunction(m *metav1.ObjectMeta) (*FuncSvc,
// GetFuncSvc gets a function service from pool cache using function key and returns number of active instances of function pod // GetFuncSvc gets a function service from pool cache using function key and returns number of active instances of function pod
func (fsc *FunctionServiceCache) GetFuncSvc(ctx context.Context, m *metav1.ObjectMeta, requestsPerPod int, concurrency int) (*FuncSvc, error) { func (fsc *FunctionServiceCache) GetFuncSvc(ctx context.Context, m *metav1.ObjectMeta, requestsPerPod int, concurrency int) (*FuncSvc, error) {
key := crd.CacheKey(m) key := crd.CacheKeyURGFromMeta(m)
fsvc, err := fsc.connFunctionCache.GetSvcValue(ctx, key, requestsPerPod, concurrency) fsvc, err := fsc.connFunctionCache.GetSvcValue(ctx, key, requestsPerPod, concurrency)
if err != nil { if err != nil {
@@ -235,7 +235,7 @@ func (fsc *FunctionServiceCache) GetByFunctionUID(uid types.UID) (*FuncSvc, erro
m := mI.(metav1.ObjectMeta) m := mI.(metav1.ObjectMeta)
fsvcI, err := fsc.byFunction.Get(crd.CacheKey(&m)) fsvcI, err := fsc.byFunction.Get(crd.CacheKeyURFromMeta(&m))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@@ -249,30 +249,34 @@ func (fsc *FunctionServiceCache) GetByFunctionUID(uid types.UID) (*FuncSvc, erro
} }
// AddFunc adds a function service to pool cache. // AddFunc adds a function service to pool cache.
func (fsc *FunctionServiceCache) AddFunc(ctx context.Context, fsvc FuncSvc, requestsPerPod int) { func (fsc *FunctionServiceCache) AddFunc(ctx context.Context, fsvc FuncSvc, requestsPerPod, svcsRetain int) {
fsc.connFunctionCache.SetSvcValue(ctx, crd.CacheKey(fsvc.Function), fsvc.Address, &fsvc, fsvc.CPULimit, requestsPerPod) fsc.connFunctionCache.SetSvcValue(ctx, crd.CacheKeyURGFromMeta(fsvc.Function), fsvc.Address, &fsvc, fsvc.CPULimit, requestsPerPod, svcsRetain)
now := time.Now() now := time.Now()
fsvc.Ctime = now fsvc.Ctime = now
fsvc.Atime = now fsvc.Atime = now
} }
func (fsc *FunctionServiceCache) MarkFuncDeleted(key crd.CacheKeyURG) {
fsc.connFunctionCache.MarkFuncDeleted(key)
}
// SetCPUUtilizaton updates/sets CPUutilization in the pool cache // SetCPUUtilizaton updates/sets CPUutilization in the pool cache
func (fsc *FunctionServiceCache) SetCPUUtilizaton(key string, svcHost string, cpuUsage resource.Quantity) { func (fsc *FunctionServiceCache) SetCPUUtilizaton(key crd.CacheKeyURG, svcHost string, cpuUsage resource.Quantity) {
fsc.connFunctionCache.SetCPUUtilization(key, svcHost, cpuUsage) fsc.connFunctionCache.SetCPUUtilization(key, svcHost, cpuUsage)
} }
// MarkAvailable marks the value at key [function][address] as available. // MarkAvailable marks the value at key [function][address] as available.
func (fsc *FunctionServiceCache) MarkAvailable(key string, svcHost string) { func (fsc *FunctionServiceCache) MarkAvailable(key crd.CacheKeyURG, svcHost string) {
fsc.connFunctionCache.MarkAvailable(key, svcHost) fsc.connFunctionCache.MarkAvailable(key, svcHost)
} }
func (fsc *FunctionServiceCache) MarkSpecializationFailure(key string) { func (fsc *FunctionServiceCache) MarkSpecializationFailure(key crd.CacheKeyURG) {
fsc.connFunctionCache.MarkSpecializationFailure(key) fsc.connFunctionCache.MarkSpecializationFailure(key)
} }
// Add adds a function service to cache if it does not exist already. // Add adds a function service to cache if it does not exist already.
func (fsc *FunctionServiceCache) Add(fsvc FuncSvc) (*FuncSvc, error) { func (fsc *FunctionServiceCache) Add(fsvc FuncSvc) (*FuncSvc, error) {
existing, err := fsc.byFunction.Set(crd.CacheKey(fsvc.Function), &fsvc) existing, err := fsc.byFunction.Set(crd.CacheKeyURFromMeta(fsvc.Function), &fsvc)
if err != nil { if err != nil {
if IsNameExistError(err) { if IsNameExistError(err) {
f := existing.(*FuncSvc) f := existing.(*FuncSvc)
@@ -334,7 +338,7 @@ func (fsc *FunctionServiceCache) _touchByAddress(address string) error {
return err return err
} }
m := mI.(metav1.ObjectMeta) m := mI.(metav1.ObjectMeta)
fsvcI, err := fsc.byFunction.Get(crd.CacheKey(&m)) fsvcI, err := fsc.byFunction.Get(crd.CacheKeyURFromMeta(&m))
if err != nil { if err != nil {
return err return err
} }
@@ -346,7 +350,7 @@ func (fsc *FunctionServiceCache) _touchByAddress(address string) error {
// DeleteEntry deletes a function service from cache. // DeleteEntry deletes a function service from cache.
func (fsc *FunctionServiceCache) DeleteEntry(fsvc *FuncSvc) { func (fsc *FunctionServiceCache) DeleteEntry(fsvc *FuncSvc) {
msg := "error deleting function service" msg := "error deleting function service"
err := fsc.byFunction.Delete(crd.CacheKey(fsvc.Function)) err := fsc.byFunction.Delete(crd.CacheKeyURFromMeta(fsvc.Function))
if err != nil { if err != nil {
fsc.logger.Error( fsc.logger.Error(
msg, msg,
@@ -378,18 +382,18 @@ func (fsc *FunctionServiceCache) DeleteEntry(fsvc *FuncSvc) {
// DeleteFunctionSvc deletes a function service at key composed of [function][address]. // DeleteFunctionSvc deletes a function service at key composed of [function][address].
func (fsc *FunctionServiceCache) DeleteFunctionSvc(ctx context.Context, fsvc *FuncSvc) { func (fsc *FunctionServiceCache) DeleteFunctionSvc(ctx context.Context, fsvc *FuncSvc) {
err := fsc.connFunctionCache.DeleteValue(ctx, crd.CacheKey(fsvc.Function), fsvc.Address) err := fsc.connFunctionCache.DeleteValue(ctx, crd.CacheKeyURGFromMeta(fsvc.Function), fsvc.Address)
if err != nil { if err != nil {
fsc.logger.Error( fsc.logger.Error(
"error deleting function service", "error deleting function service",
zap.Any("function", fsvc.Function.Name), zap.String("function", fsvc.Function.Name),
zap.Any("address", fsvc.Address), zap.String("address", fsvc.Address),
zap.Error(err), zap.Error(err),
) )
} }
} }
func (fsc *FunctionServiceCache) SetCPUUtilization(key string, svcHost string, cpuUsage resource.Quantity) { func (fsc *FunctionServiceCache) SetCPUUtilization(key crd.CacheKeyURG, svcHost string, cpuUsage resource.Quantity) {
fsc.connFunctionCache.SetCPUUtilization(key, svcHost, cpuUsage) fsc.connFunctionCache.SetCPUUtilization(key, svcHost, cpuUsage)
} }
@@ -2,11 +2,11 @@ package fscache
import ( import (
"context" "context"
"fmt"
"log" "log"
"testing" "testing"
"time" "time"
"github.com/stretchr/testify/require"
"go.uber.org/zap" "go.uber.org/zap"
"go.uber.org/zap/zapcore" "go.uber.org/zap/zapcore"
apiv1 "k8s.io/api/core/v1" apiv1 "k8s.io/api/core/v1"
@@ -14,6 +14,7 @@ import (
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
fv1 "github.com/fission/fission/pkg/apis/core/v1" fv1 "github.com/fission/fission/pkg/apis/core/v1"
"github.com/fission/fission/pkg/crd"
) )
func panicIf(err error) { func panicIf(err error) {
@@ -29,9 +30,7 @@ func TestFunctionServiceCache(t *testing.T) {
panicIf(err) panicIf(err)
fsc := MakeFunctionServiceCache(logger) fsc := MakeFunctionServiceCache(logger)
if fsc == nil { require.NotNil(t, fsc)
log.Panicf("error creating cache")
}
var fsvc *FuncSvc var fsvc *FuncSvc
now := time.Now() now := time.Now()
@@ -75,55 +74,30 @@ func TestFunctionServiceCache(t *testing.T) {
Atime: now, Atime: now,
} }
_, err = fsc.Add(*fsvc) _, err = fsc.Add(*fsvc)
if err != nil { require.NoError(t, err)
fsc.Log()
log.Panicf("Failed to add fsvc: %v", err)
}
_, err = fsc.GetByFunction(fsvc.Function) _, err = fsc.GetByFunction(fsvc.Function)
if err != nil { require.NoError(t, err)
fsc.Log()
log.Panicf("Failed to get fsvc: %v", err)
}
f, err := fsc.GetByFunctionUID(fsvc.Function.UID) f, err := fsc.GetByFunctionUID(fsvc.Function.UID)
if err != nil { require.NoError(t, err)
fsc.Log()
log.Panicf("Failed to get fsvc by function uid: %v", err)
}
fsvc.Atime = f.Atime fsvc.Atime = f.Atime
fsvc.Ctime = f.Ctime fsvc.Ctime = f.Ctime
if f.Address != fsvc.Address { require.Equal(t, fsvc.Address, f.Address)
fsc.Log()
log.Panicf("Incorrect fsvc \n(expected: %#v)\n (found: %#v)", fsvc, f)
}
err = fsc.TouchByAddress(fsvc.Address) err = fsc.TouchByAddress(fsvc.Address)
if err != nil { require.NoError(t, err)
fsc.Log()
log.Panicf("Failed to touch fsvc: %v", err)
}
deleted, err := fsc.DeleteOld(fsvc, 0) deleted, err := fsc.DeleteOld(fsvc, 0)
if err != nil { require.NoError(t, err)
fsc.Log() require.False(t, deleted)
log.Panicf("Failed to delete fsvc: %v", err)
}
if !deleted {
fsc.Log()
log.Panicf("Did not delete fsvc")
}
_, err = fsc.GetByFunction(fsvc.Function) _, err = fsc.GetByFunction(fsvc.Function)
if err == nil { require.NoError(t, err)
fsc.Log()
log.Panicf("found fsvc while expecting empty cache: %v", err)
}
_, err = fsc.GetByFunctionUID(fsvc.Function.UID) _, err = fsc.GetByFunctionUID(fsvc.Function.UID)
if err == nil { require.NoError(t, err)
fsc.Log()
log.Panicf("found fsvc by function uid while expecting empty cache: %v", err)
}
} }
func TestFunctionServiceNewCache(t *testing.T) { func TestFunctionServiceNewCache(t *testing.T) {
@@ -131,9 +105,7 @@ func TestFunctionServiceNewCache(t *testing.T) {
panicIf(err) panicIf(err)
fsc := MakeFunctionServiceCache(logger) fsc := MakeFunctionServiceCache(logger)
if fsc == nil { require.NotNil(t, fsc)
log.Panicf("error creating cache")
}
var fsvc *FuncSvc var fsvc *FuncSvc
now := time.Now() now := time.Now()
@@ -174,8 +146,8 @@ func TestFunctionServiceNewCache(t *testing.T) {
Address: "xxx", Address: "xxx",
KubernetesObjects: objects, KubernetesObjects: objects,
CPULimit: resource.MustParse("5m"), CPULimit: resource.MustParse("5m"),
Ctime: now, Ctime: now.Add(-2 * time.Minute),
Atime: now, Atime: now.Add(-1 * time.Minute),
} }
fn := &fv1.Function{ fn := &fv1.Function{
ObjectMeta: metav1.ObjectMeta{ ObjectMeta: metav1.ObjectMeta{
@@ -187,27 +159,34 @@ func TestFunctionServiceNewCache(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
fsc.AddFunc(ctx, *fsvc, 10) fsc.AddFunc(ctx, *fsvc, 10, fn.GetRetainPods())
concurrency := 10 concurrency := 10
_, err = fsc.GetFuncSvc(ctx, fsvc.Function, 5, concurrency) _, err = fsc.GetFuncSvc(ctx, fsvc.Function, 5, concurrency)
if err != nil { require.NoError(t, err)
logger.Panic("received error while retrieving value from cache")
}
key := fmt.Sprintf("%v_%v", fn.ObjectMeta.UID, fn.ObjectMeta.ResourceVersion) //key := fmt.Sprintf("%v_%v", cancel.UID, fn.ObjectMeta.ResourceVersion)
key := crd.CacheKeyURGFromMeta(&fn.ObjectMeta)
fsc.MarkAvailable(key, fsvc.Address) fsc.MarkAvailable(key, fsvc.Address)
_, err = fsc.GetFuncSvc(ctx, fsvc.Function, 5, concurrency) _, err = fsc.GetFuncSvc(ctx, fsvc.Function, 5, concurrency)
if err != nil { require.NoError(t, err)
logger.Panic("received error while retrieving value from cache")
}
for i := 0; i < 2; i++ {
fsc.MarkAvailable(key, fsvc.Address)
}
vals, err := fsc.ListOldForPool(30 * time.Second) vals, err := fsc.ListOldForPool(30 * time.Second)
if err != nil { require.NoError(t, err)
logger.Panic("received error while get list of old values") require.Equal(t, 0, len(vals))
}
if len(vals) != 0 { vals, err = fsc.ListOldForPool(0)
logger.Panic(fmt.Sprintln("list of old values didn't matched the expected: 1", "received", len(vals))) require.NoError(t, err)
} require.Equal(t, 1, len(vals))
fsc.DeleteFunctionSvc(ctx, fsvc)
fsvc.Address = "xxx2"
fn.Spec.RetainPods = 2
fsc.AddFunc(ctx, *fsvc, 10, fn.GetRetainPods())
vals, err = fsc.ListOldForPool(0)
require.NoError(t, err)
require.Equal(t, 0, len(vals))
} }
+67 -17
View File
@@ -24,7 +24,9 @@ import (
"go.uber.org/zap" "go.uber.org/zap"
"k8s.io/apimachinery/pkg/api/resource" "k8s.io/apimachinery/pkg/api/resource"
"k8s.io/apimachinery/pkg/types"
"github.com/fission/fission/pkg/crd"
ferror "github.com/fission/fission/pkg/error" ferror "github.com/fission/fission/pkg/error"
otelUtils "github.com/fission/fission/pkg/utils/otel" otelUtils "github.com/fission/fission/pkg/utils/otel"
) )
@@ -40,6 +42,7 @@ const (
setCPUUtilization setCPUUtilization
markSpecializationFailure markSpecializationFailure
logFuncSvc logFuncSvc
markDeleted
) )
type ( type (
@@ -52,14 +55,16 @@ type (
funcSvcGroup struct { funcSvcGroup struct {
svcWaiting int svcWaiting int
svcRetain int
svcs map[string]*funcSvcInfo svcs map[string]*funcSvcInfo
queue *Queue queue *Queue
deleted bool
} }
// PoolCache implements a simple cache implementation having values mapped by two keys [function][address]. // PoolCache implements a simple cache implementation having values mapped by two keys [function][address].
// As of now PoolCache is only used by poolmanager executor // As of now PoolCache is only used by poolmanager executor
PoolCache struct { PoolCache struct {
cache map[string]*funcSvcGroup cache map[crd.CacheKeyURG]*funcSvcGroup
requestChannel chan *request requestChannel chan *request
logger *zap.Logger logger *zap.Logger
} }
@@ -67,7 +72,7 @@ type (
request struct { request struct {
requestType requestType
ctx context.Context ctx context.Context
function string function crd.CacheKeyURG
address string address string
dumpWriter io.Writer dumpWriter io.Writer
value *FuncSvc value *FuncSvc
@@ -75,6 +80,7 @@ type (
cpuUsage resource.Quantity cpuUsage resource.Quantity
responseChannel chan *response responseChannel chan *response
concurrency int concurrency int
svcsRetain int
} }
response struct { response struct {
error error
@@ -92,7 +98,7 @@ type (
func NewPoolCache(logger *zap.Logger) *PoolCache { func NewPoolCache(logger *zap.Logger) *PoolCache {
c := &PoolCache{ c := &PoolCache{
cache: make(map[string]*funcSvcGroup), cache: make(map[crd.CacheKeyURG]*funcSvcGroup),
requestChannel: make(chan *request), requestChannel: make(chan *request),
logger: logger, logger: logger,
} }
@@ -131,7 +137,7 @@ func (c *PoolCache) service() {
// mark active // mark active
funcSvcGroup.svcs[addr].activeRequests++ funcSvcGroup.svcs[addr].activeRequests++
if c.logger.Core().Enabled(zap.DebugLevel) { if c.logger.Core().Enabled(zap.DebugLevel) {
otelUtils.LoggerWithTraceID(req.ctx, c.logger).Debug("Increase active requests with getValue", zap.String("function", req.function), zap.String("address", addr), zap.Int("activeRequests", funcSvcGroup.svcs[addr].activeRequests)) otelUtils.LoggerWithTraceID(req.ctx, c.logger).Debug("Increase active requests with getValue", zap.String("function", req.function.String()), zap.String("address", addr), zap.Int("activeRequests", funcSvcGroup.svcs[addr].activeRequests))
} }
resp.value = funcSvcGroup.svcs[addr].val resp.value = funcSvcGroup.svcs[addr].val
found = true found = true
@@ -172,6 +178,7 @@ func (c *PoolCache) service() {
if _, ok := c.cache[req.function].svcs[req.address]; !ok { if _, ok := c.cache[req.function].svcs[req.address]; !ok {
c.cache[req.function].svcs[req.address] = &funcSvcInfo{} c.cache[req.function].svcs[req.address] = &funcSvcInfo{}
} }
c.cache[req.function].svcRetain = req.svcsRetain
c.cache[req.function].svcs[req.address].val = req.value c.cache[req.function].svcs[req.address].val = req.value
c.cache[req.function].svcs[req.address].activeRequests++ c.cache[req.function].svcs[req.address].activeRequests++
if c.cache[req.function].svcWaiting > 0 { if c.cache[req.function].svcWaiting > 0 {
@@ -196,22 +203,52 @@ func (c *PoolCache) service() {
} }
} }
if c.logger.Core().Enabled(zap.DebugLevel) { if c.logger.Core().Enabled(zap.DebugLevel) {
otelUtils.LoggerWithTraceID(req.ctx, c.logger).Debug("Increase active requests with setValue", zap.String("function", req.function), zap.String("address", req.address), zap.Int("activeRequests", c.cache[req.function].svcs[req.address].activeRequests)) otelUtils.LoggerWithTraceID(req.ctx, c.logger).Debug("Increase active requests with setValue", zap.String("function", req.function.String()), zap.String("address", req.address), zap.Int("activeRequests", c.cache[req.function].svcs[req.address].activeRequests))
} }
c.cache[req.function].svcs[req.address].cpuLimit = req.cpuUsage c.cache[req.function].svcs[req.address].cpuLimit = req.cpuUsage
case markDeleted:
for key := range c.cache {
if key.UID == req.function.UID {
c.cache[key].deleted = true
break
}
}
case listAvailableValue: case listAvailableValue:
vals := make([]*FuncSvc, 0) vals := make([]*FuncSvc, 0)
latestFuncGen := make(map[types.UID]int64)
// find the latest generation of each function
for key := range c.cache {
if currentFuncGen, ok := latestFuncGen[key.UID]; ok {
if key.Generation > currentFuncGen {
latestFuncGen[key.UID] = key.Generation
}
} else {
latestFuncGen[key.UID] = key.Generation
}
}
for key1, values := range c.cache { for key1, values := range c.cache {
svcRetain := values.svcRetain
// if the function is not latest generation, then we don't need to retain any pods
if latestFuncGen[key1.UID] != key1.Generation || values.deleted {
svcRetain = 0
}
svcCleanQuota := len(values.svcs) - svcRetain
if svcCleanQuota <= 0 {
continue
}
for key2, value := range values.svcs { for key2, value := range values.svcs {
debugLevel := c.logger.Core().Enabled(zap.DebugLevel) debugLevel := c.logger.Core().Enabled(zap.DebugLevel)
if debugLevel { if debugLevel {
otelUtils.LoggerWithTraceID(req.ctx, c.logger).Debug("Reading active requests", zap.String("function", key1), zap.String("address", key2), zap.Int("activeRequests", value.activeRequests)) otelUtils.LoggerWithTraceID(req.ctx, c.logger).Debug("Reading active requests", zap.String("function", key1.String()), zap.String("address", key2), zap.Int("activeRequests", value.activeRequests))
} }
if value.activeRequests == 0 { if value.activeRequests == 0 && svcCleanQuota > 0 {
if debugLevel { if debugLevel {
otelUtils.LoggerWithTraceID(req.ctx, c.logger).Debug("Function service with no active requests", zap.String("function", key1), zap.String("address", key2), zap.Int("activeRequests", value.activeRequests)) otelUtils.LoggerWithTraceID(req.ctx, c.logger).Debug("Function service with no active requests", zap.String("function", key1.String()), zap.String("address", key2), zap.Int("activeRequests", value.activeRequests))
} }
vals = append(vals, value.val) vals = append(vals, value.val)
svcCleanQuota--
} }
} }
} }
@@ -230,10 +267,10 @@ func (c *PoolCache) service() {
if c.cache[req.function].svcs[req.address].activeRequests > 0 { if c.cache[req.function].svcs[req.address].activeRequests > 0 {
c.cache[req.function].svcs[req.address].activeRequests-- c.cache[req.function].svcs[req.address].activeRequests--
if c.logger.Core().Enabled(zap.DebugLevel) { if c.logger.Core().Enabled(zap.DebugLevel) {
otelUtils.LoggerWithTraceID(req.ctx, c.logger).Debug("Decrease active requests", zap.String("function", req.function), zap.String("address", req.address), zap.Int("activeRequests", c.cache[req.function].svcs[req.address].activeRequests)) otelUtils.LoggerWithTraceID(req.ctx, c.logger).Debug("Decrease active requests", zap.String("function", req.function.String()), zap.String("address", req.address), zap.Int("activeRequests", c.cache[req.function].svcs[req.address].activeRequests))
} }
} else { } else {
otelUtils.LoggerWithTraceID(req.ctx, c.logger).Error("Invalid request to decrease active requests", zap.String("function", req.function), zap.String("address", req.address), zap.Int("activeRequests", c.cache[req.function].svcs[req.address].activeRequests)) otelUtils.LoggerWithTraceID(req.ctx, c.logger).Error("Invalid request to decrease active requests", zap.String("function", req.function.String()), zap.String("address", req.address), zap.Int("activeRequests", c.cache[req.function].svcs[req.address].activeRequests))
} }
} }
} }
@@ -246,7 +283,12 @@ func (c *PoolCache) service() {
} }
} }
case deleteValue: case deleteValue:
delete(c.cache[req.function].svcs, req.address) if funcSvcGroup, ok := c.cache[req.function]; ok {
delete(c.cache[req.function].svcs, req.address)
if funcSvcGroup.deleted && len(c.cache[req.function].svcs) == 0 {
delete(c.cache, req.function)
}
}
req.responseChannel <- resp req.responseChannel <- resp
case logFuncSvc: case logFuncSvc:
datawriter := bufio.NewWriter(req.dumpWriter) datawriter := bufio.NewWriter(req.dumpWriter)
@@ -298,8 +340,15 @@ func (c *PoolCache) service() {
} }
} }
func (c *PoolCache) MarkFuncDeleted(function crd.CacheKeyURG) {
c.requestChannel <- &request{
requestType: markDeleted,
function: function,
}
}
// GetValue returns a function service with status in Active else return error // GetValue returns a function service with status in Active else return error
func (c *PoolCache) GetSvcValue(ctx context.Context, function string, requestsPerPod int, concurrency int) (*FuncSvc, error) { func (c *PoolCache) GetSvcValue(ctx context.Context, function crd.CacheKeyURG, requestsPerPod int, concurrency int) (*FuncSvc, error) {
respChannel := make(chan *response) respChannel := make(chan *response)
c.requestChannel <- &request{ c.requestChannel <- &request{
ctx: ctx, ctx: ctx,
@@ -334,7 +383,7 @@ func (c *PoolCache) ListAvailableValue() []*FuncSvc {
} }
// SetValue marks the value at key [function][address] as active(begin used) // SetValue marks the value at key [function][address] as active(begin used)
func (c *PoolCache) SetSvcValue(ctx context.Context, function, address string, value *FuncSvc, cpuLimit resource.Quantity, requestsPerPod int) { func (c *PoolCache) SetSvcValue(ctx context.Context, function crd.CacheKeyURG, address string, value *FuncSvc, cpuLimit resource.Quantity, requestsPerPod, svcsRetain int) {
respChannel := make(chan *response) respChannel := make(chan *response)
c.requestChannel <- &request{ c.requestChannel <- &request{
ctx: ctx, ctx: ctx,
@@ -344,12 +393,13 @@ func (c *PoolCache) SetSvcValue(ctx context.Context, function, address string, v
value: value, value: value,
cpuUsage: cpuLimit, cpuUsage: cpuLimit,
requestsPerPod: requestsPerPod, requestsPerPod: requestsPerPod,
svcsRetain: svcsRetain,
responseChannel: respChannel, responseChannel: respChannel,
} }
} }
// SetCPUUtilization updates/sets the CPU utilization limit for the pod // SetCPUUtilization updates/sets the CPU utilization limit for the pod
func (c *PoolCache) SetCPUUtilization(function, address string, cpuUsage resource.Quantity) { func (c *PoolCache) SetCPUUtilization(function crd.CacheKeyURG, address string, cpuUsage resource.Quantity) {
c.requestChannel <- &request{ c.requestChannel <- &request{
requestType: setCPUUtilization, requestType: setCPUUtilization,
function: function, function: function,
@@ -360,7 +410,7 @@ func (c *PoolCache) SetCPUUtilization(function, address string, cpuUsage resourc
} }
// MarkAvailable marks the value at key [function][address] as available // MarkAvailable marks the value at key [function][address] as available
func (c *PoolCache) MarkAvailable(function, address string) { func (c *PoolCache) MarkAvailable(function crd.CacheKeyURG, address string) {
respChannel := make(chan *response) respChannel := make(chan *response)
c.requestChannel <- &request{ c.requestChannel <- &request{
requestType: markAvailable, requestType: markAvailable,
@@ -371,7 +421,7 @@ func (c *PoolCache) MarkAvailable(function, address string) {
} }
// DeleteValue deletes the value at key composed of [function][address] // DeleteValue deletes the value at key composed of [function][address]
func (c *PoolCache) DeleteValue(ctx context.Context, function, address string) error { func (c *PoolCache) DeleteValue(ctx context.Context, function crd.CacheKeyURG, address string) error {
respChannel := make(chan *response) respChannel := make(chan *response)
c.requestChannel <- &request{ c.requestChannel <- &request{
ctx: ctx, ctx: ctx,
@@ -385,7 +435,7 @@ func (c *PoolCache) DeleteValue(ctx context.Context, function, address string) e
} }
// ReduceSpecializationInProgress reduces the svcWaiting count // ReduceSpecializationInProgress reduces the svcWaiting count
func (c *PoolCache) MarkSpecializationFailure(function string) { func (c *PoolCache) MarkSpecializationFailure(function crd.CacheKeyURG) {
c.requestChannel <- &request{ c.requestChannel <- &request{
requestType: markSpecializationFailure, requestType: markSpecializationFailure,
function: function, function: function,
+143 -59
View File
@@ -12,6 +12,7 @@ import (
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
"k8s.io/apimachinery/pkg/api/resource" "k8s.io/apimachinery/pkg/api/resource"
"github.com/fission/fission/pkg/crd"
ferror "github.com/fission/fission/pkg/error" ferror "github.com/fission/fission/pkg/error"
"github.com/fission/fission/pkg/utils/loggerfactory" "github.com/fission/fission/pkg/utils/loggerfactory"
) )
@@ -26,77 +27,128 @@ func TestPoolCache(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
logger := loggerfactory.GetLogger() logger := loggerfactory.GetLogger()
c := NewPoolCache(logger)
concurrency := 5 concurrency := 5
requestsPerPod := 2 requestsPerPod := 2
// should return err since no svc is present keyFunc := crd.CacheKeyURG{
_, err := c.GetSvcValue(ctx, "func", requestsPerPod, concurrency) UID: "func",
if err == nil { }
log.Panicf("found value when expected it to be nil") keyFunc2 := crd.CacheKeyURG{
UID: "func2",
} }
c.SetSvcValue(ctx, "func", "ip", &FuncSvc{ t.Run("Test create new svc ", func(t *testing.T) {
Name: "value", c1 := NewPoolCache(logger)
}, resource.MustParse("45m"), 10)
// should not return any error since we added a svc // should return err since no svc is present
_, err = c.GetSvcValue(ctx, "func", requestsPerPod, concurrency) _, err := c1.GetSvcValue(ctx, keyFunc, requestsPerPod, concurrency)
checkErr(err) if err == nil {
log.Panicf("found value when expected it to be nil")
}
c.SetSvcValue(ctx, "func", "ip", &FuncSvc{ c1.SetSvcValue(ctx, keyFunc, "ip", &FuncSvc{
Name: "value", Name: "value",
}, resource.MustParse("45m"), 10) }, resource.MustParse("45m"), 10, 0)
// should return err since all functions are busy // should not return any error since we added a svc
_, err = c.GetSvcValue(ctx, "func", requestsPerPod, concurrency) _, err = c1.GetSvcValue(ctx, keyFunc, requestsPerPod, concurrency)
if err == nil { checkErr(err)
log.Panicf("found value when expected it to be nil") })
}
c.SetSvcValue(ctx, "func", "ip", &FuncSvc{ t.Run("Test return error when functions are busy", func(t *testing.T) {
Name: "value", c2 := NewPoolCache(logger)
}, resource.MustParse("45m"), 10) c2.SetSvcValue(ctx, keyFunc, "ip", &FuncSvc{
Name: "value",
}, resource.MustParse("45m"), 10, 0)
c2.SetSvcValue(ctx, keyFunc, "ip", &FuncSvc{
Name: "value",
}, resource.MustParse("45m"), 10, 0)
// should return err since all functions are busy
_, err := c2.GetSvcValue(ctx, keyFunc, requestsPerPod, concurrency)
if err == nil {
log.Panicf("found value when expected it to be nil")
}
})
c.SetSvcValue(ctx, "func2", "ip2", &FuncSvc{ t.Run("Test does not list available values when a function svc is deleted", func(t *testing.T) {
Name: "value2", c3 := NewPoolCache(logger)
}, resource.MustParse("50m"), 10) c3.SetSvcValue(ctx, keyFunc, "ip", &FuncSvc{
Name: "value",
}, resource.MustParse("45m"), 10, 0)
c.SetSvcValue(ctx, "func2", "ip22", &FuncSvc{ c3.SetSvcValue(ctx, keyFunc2, "ip2", &FuncSvc{
Name: "value22", Name: "value2",
}, resource.MustParse("33m"), 10) }, resource.MustParse("50m"), 10, 0)
checkErr(c.DeleteValue(ctx, "func2", "ip2")) checkErr(c3.DeleteValue(ctx, keyFunc2, "ip2"))
cc := c.ListAvailableValue() cc := c3.ListAvailableValue()
if len(cc) != 0 { if len(cc) != 0 {
log.Panicf("expected 0 available items") log.Panicf("expected 0 available items")
} }
_, err := c3.GetSvcValue(ctx, keyFunc2, requestsPerPod, concurrency)
if err == nil {
log.Panicf("found deleted element")
}
})
c.MarkAvailable("func", "ip") t.Run("Test return error when current CPU usage is more then permissible", func(t *testing.T) {
c4 := NewPoolCache(logger)
_, err := c4.GetSvcValue(ctx, keyFunc, requestsPerPod, concurrency)
if err == nil {
log.Panicf("found value when expected it to be nil")
}
checkErr(c.DeleteValue(ctx, "func", "ip")) c4.SetSvcValue(ctx, keyFunc, "ip", &FuncSvc{
Name: "value",
}, resource.MustParse("45m"), 10, 0)
_, err = c.GetSvcValue(ctx, "func", requestsPerPod, concurrency) // should not return any error since we added a svc
if err == nil { _, err = c4.GetSvcValue(ctx, keyFunc, requestsPerPod, concurrency)
log.Panicf("found deleted element") checkErr(err)
}
c.SetSvcValue(ctx, "cpulimit", "100", &FuncSvc{ c4.SetCPUUtilization(keyFunc, "ip", resource.MustParse("4m"))
Name: "value",
}, resource.MustParse("3m"), 10) _, err = c4.GetSvcValue(ctx, keyFunc, requestsPerPod, concurrency)
c.SetCPUUtilization("cpulimit", "100", resource.MustParse("4m")) if err == nil {
log.Panicf("found value when expected it to be nil")
}
})
t.Run("Test function should not exist when mark deleted is called", func(t *testing.T) {
c5 := NewPoolCache(logger)
c5.SetSvcValue(ctx, keyFunc, "ip", &FuncSvc{
Name: "value",
}, resource.MustParse("45m"), 10, 0)
// should not return any error since we added a svc
_, err := c5.GetSvcValue(ctx, keyFunc, requestsPerPod, concurrency)
checkErr(err)
c5.MarkFuncDeleted(keyFunc)
checkErr(c5.DeleteValue(ctx, keyFunc, "ip"))
_, err = c5.GetSvcValue(ctx, keyFunc, requestsPerPod, concurrency)
if err == nil {
log.Panicf("found value when expected it to be nil")
}
})
} }
func TestPoolCacheRequests(t *testing.T) { func TestPoolCacheRequests(t *testing.T) {
key := crd.CacheKeyURG{
UID: "func",
Generation: 1,
}
type structForTest struct { type structForTest struct {
name string name string
requests int requests int
concurrency int concurrency int
rpp int rpp int
simultaneous int simultaneous int
failedRequests int failedRequests int
retainPods int
generationUpdate bool
} }
for _, tt := range []structForTest{ for _, tt := range []structForTest{
@@ -118,7 +170,6 @@ func TestPoolCacheRequests(t *testing.T) {
concurrency: 5, concurrency: 5,
rpp: 60, rpp: 60,
}, },
{ {
name: "test4", name: "test4",
requests: 6, requests: 6,
@@ -149,11 +200,19 @@ func TestPoolCacheRequests(t *testing.T) {
failedRequests: 10, failedRequests: 10,
}, },
{ {
name: "test8", name: "test8",
requests: 10, requests: 2,
concurrency: 10, concurrency: 2,
rpp: 1, rpp: 1,
simultaneous: 10, retainPods: 1,
},
{
name: "test9",
requests: 10,
concurrency: 5,
rpp: 2,
retainPods: 2,
generationUpdate: true,
}, },
} { } {
t.Run(fmt.Sprintf("scenario-%s", tt.name), func(t *testing.T) { t.Run(fmt.Sprintf("scenario-%s", tt.name), func(t *testing.T) {
@@ -168,13 +227,13 @@ func TestPoolCacheRequests(t *testing.T) {
wg.Add(1) wg.Add(1)
go func(reqno int) { go func(reqno int) {
defer wg.Done() defer wg.Done()
svc, err := p.GetSvcValue(context.Background(), "func", tt.rpp, tt.concurrency) svc, err := p.GetSvcValue(context.Background(), key, tt.rpp, tt.concurrency)
if err != nil { if err != nil {
code, _ := ferror.GetHTTPError(err) code, _ := ferror.GetHTTPError(err)
if code == http.StatusNotFound { if code == http.StatusNotFound {
p.SetSvcValue(context.Background(), "func", fmt.Sprintf("svc-%d", svcCounter), &FuncSvc{ p.SetSvcValue(context.Background(), key, fmt.Sprintf("svc-%d", svcCounter), &FuncSvc{
Name: "value", Name: "value",
}, resource.MustParse("45m"), tt.rpp) }, resource.MustParse("45m"), tt.rpp, tt.retainPods)
atomic.AddUint64(&svcCounter, 1) atomic.AddUint64(&svcCounter, 1)
} else { } else {
t.Log(reqno, "=>", err) t.Log(reqno, "=>", err)
@@ -195,6 +254,31 @@ func TestPoolCacheRequests(t *testing.T) {
require.Equal(t, tt.failedRequests, int(atomic.LoadUint64(&failedRequests))) require.Equal(t, tt.failedRequests, int(atomic.LoadUint64(&failedRequests)))
require.Equal(t, tt.concurrency, int(atomic.LoadUint64(&svcCounter))) require.Equal(t, tt.concurrency, int(atomic.LoadUint64(&svcCounter)))
for i := 0; i < tt.concurrency; i++ {
for j := 0; j < tt.rpp; j++ {
wg.Add(1)
go func(i int) {
defer wg.Done()
p.MarkAvailable(key, fmt.Sprintf("svc-%d", i))
}(i)
}
}
wg.Wait()
if tt.generationUpdate {
newKey := crd.CacheKeyURG{
UID: "func",
Generation: 2,
}
p.SetSvcValue(context.Background(), newKey, fmt.Sprintf("svc-%d", svcCounter), &FuncSvc{
Name: "value",
}, resource.MustParse("45m"), tt.rpp, tt.retainPods)
funcSvc := p.ListAvailableValue()
require.Equal(t, tt.concurrency, len(funcSvc))
} else {
funcSvc := p.ListAvailableValue()
require.Equal(t, tt.concurrency-tt.retainPods, len(funcSvc))
}
}) })
} }
} }
+2 -2
View File
@@ -36,7 +36,7 @@ func Commands() *cobra.Command {
flag.FnExecutorType, flag.FnCfgMap, flag.FnSecret, flag.FnExecutorType, flag.FnCfgMap, flag.FnSecret,
flag.FnSpecializationTimeout, flag.FnExecutionTimeout, flag.FnSpecializationTimeout, flag.FnExecutionTimeout,
flag.FnIdleTimeout, flag.FnConcurrency, flag.FnRequestsPerPod, flag.FnIdleTimeout, flag.FnConcurrency, flag.FnRequestsPerPod,
flag.FnOnceOnly, flag.Labels, flag.Annotation, flag.FnOnceOnly, flag.Labels, flag.Annotation, flag.FnRetainPods,
// TODO retired pkg & trigger related flags from function cmd // TODO retired pkg & trigger related flags from function cmd
flag.PkgCode, flag.PkgSrcArchive, flag.PkgDeployArchive, flag.PkgCode, flag.PkgSrcArchive, flag.PkgDeployArchive,
@@ -87,7 +87,7 @@ func Commands() *cobra.Command {
flag.FnExecutorType, flag.FnSecret, flag.FnCfgMap, flag.FnExecutorType, flag.FnSecret, flag.FnCfgMap,
flag.FnSpecializationTimeout, flag.FnExecutionTimeout, flag.FnSpecializationTimeout, flag.FnExecutionTimeout,
flag.FnIdleTimeout, flag.FnConcurrency, flag.FnRequestsPerPod, flag.FnIdleTimeout, flag.FnConcurrency, flag.FnRequestsPerPod,
flag.FnOnceOnly, flag.Labels, flag.Annotation, flag.FnOnceOnly, flag.Labels, flag.Annotation, flag.FnRetainPods,
flag.PkgCode, flag.PkgSrcArchive, flag.PkgDeployArchive, flag.PkgCode, flag.PkgSrcArchive, flag.PkgDeployArchive,
flag.PkgSrcChecksum, flag.PkgDeployChecksum, flag.PkgInsecure, flag.PkgSrcChecksum, flag.PkgDeployChecksum, flag.PkgInsecure,
+2
View File
@@ -104,6 +104,7 @@ func (opts *CreateSubCommand) complete(input cli.Input) error {
} }
requestsPerPod := input.Int(flagkey.FnRequestsPerPod) requestsPerPod := input.Int(flagkey.FnRequestsPerPod)
retainPods := input.Int(flagkey.FnRetainPods)
fnOnceOnly := input.Bool(flagkey.FnOnceOnly) fnOnceOnly := input.Bool(flagkey.FnOnceOnly)
@@ -306,6 +307,7 @@ func (opts *CreateSubCommand) complete(input cli.Input) error {
IdleTimeout: &fnIdleTimeout, IdleTimeout: &fnIdleTimeout,
Concurrency: fnConcurrency, Concurrency: fnConcurrency,
RequestsPerPod: requestsPerPod, RequestsPerPod: requestsPerPod,
RetainPods: retainPods,
OnceOnly: fnOnceOnly, OnceOnly: fnOnceOnly,
}, },
} }
+4
View File
@@ -162,6 +162,10 @@ func (opts *UpdateSubCommand) complete(input cli.Input) error {
function.Spec.RequestsPerPod = input.Int(flagkey.FnRequestsPerPod) function.Spec.RequestsPerPod = input.Int(flagkey.FnRequestsPerPod)
} }
if input.IsSet(flagkey.FnRetainPods) {
function.Spec.RetainPods = input.Int(flagkey.FnRetainPods)
}
if input.IsSet(flagkey.FnOnceOnly) { if input.IsSet(flagkey.FnOnceOnly) {
function.Spec.OnceOnly = input.Bool(flagkey.FnOnceOnly) function.Spec.OnceOnly = input.Bool(flagkey.FnOnceOnly)
} }
+1
View File
@@ -132,6 +132,7 @@ var (
FnOnceOnly = Flag{Type: Bool, Name: flagkey.FnOnceOnly, Aliases: []string{"yolo"}, Usage: "Specifies if specialized pod will serve exactly one request in its lifetime"} FnOnceOnly = Flag{Type: Bool, Name: flagkey.FnOnceOnly, Aliases: []string{"yolo"}, Usage: "Specifies if specialized pod will serve exactly one request in its lifetime"}
FnSubPath = Flag{Type: String, Name: flagkey.FnSubPath, Usage: "Sub Path to check if function internally supports routing"} FnSubPath = Flag{Type: String, Name: flagkey.FnSubPath, Usage: "Sub Path to check if function internally supports routing"}
FnLogAllPods = Flag{Type: Bool, Name: flagkey.FnLogAllPods, Usage: "Get all pod's logs in the function."} FnLogAllPods = Flag{Type: Bool, Name: flagkey.FnLogAllPods, Usage: "Get all pod's logs in the function."}
FnRetainPods = Flag{Type: Int, Name: flagkey.FnRetainPods, Usage: "Number of pods to retain after pods specialization.", DefaultValue: 0}
// Termination Grace Period configurable at function creation/update only for container functions // Termination Grace Period configurable at function creation/update only for container functions
FnTerminationGracePeriod = Flag{Type: Int64, Name: flagkey.FnGracePeriod, Usage: "Grace time (in seconds) for pod to perform connection draining before termination (only non-negative values considered)", DefaultValue: 360} FnTerminationGracePeriod = Flag{Type: Int64, Name: flagkey.FnGracePeriod, Usage: "Grace time (in seconds) for pod to perform connection draining before termination (only non-negative values considered)", DefaultValue: 360}
+1
View File
@@ -86,6 +86,7 @@ const (
FnSubPath = "subpath" FnSubPath = "subpath"
FnGracePeriod = "graceperiod" FnGracePeriod = "graceperiod"
FnLogAllPods = "all-pods" FnLogAllPods = "all-pods"
FnRetainPods = "retainpods"
HtName = resourceName HtName = resourceName
HtMethod = "method" HtMethod = "method"
+1 -1
View File
@@ -684,7 +684,7 @@ func (fh functionHandler) getServiceEntry(ctx context.Context) (svcURL *url.URL,
fnMeta := &fh.function.ObjectMeta fnMeta := &fh.function.ObjectMeta
recordObj, err := fh.svcAddrUpdateThrottler.RunOnce( recordObj, err := fh.svcAddrUpdateThrottler.RunOnce(
crd.CacheKey(fnMeta), crd.CacheKeyURFromMeta(fnMeta).String(),
func(firstToTheLock bool) (interface{}, error) { func(firstToTheLock bool) (interface{}, error) {
if !firstToTheLock { if !firstToTheLock {
svcURL, err := fh.getServiceEntryFromCache() svcURL, err := fh.getServiceEntryFromCache()
+3 -2
View File
@@ -21,6 +21,7 @@ import (
"github.com/robfig/cron/v3" "github.com/robfig/cron/v3"
"go.uber.org/zap" "go.uber.org/zap"
"k8s.io/apimachinery/pkg/types"
fv1 "github.com/fission/fission/pkg/apis/core/v1" fv1 "github.com/fission/fission/pkg/apis/core/v1"
"github.com/fission/fission/pkg/publisher" "github.com/fission/fission/pkg/publisher"
@@ -36,7 +37,7 @@ const (
type ( type (
Timer struct { Timer struct {
logger *zap.Logger logger *zap.Logger
triggers map[string]*timerTriggerWithCron triggers map[types.UID]*timerTriggerWithCron
publisher *publisher.Publisher publisher *publisher.Publisher
} }
@@ -49,7 +50,7 @@ type (
func MakeTimer(logger *zap.Logger, publisher publisher.Publisher) *Timer { func MakeTimer(logger *zap.Logger, publisher publisher.Publisher) *Timer {
timer := &Timer{ timer := &Timer{
logger: logger.Named("timer"), logger: logger.Named("timer"),
triggers: make(map[string]*timerTriggerWithCron), triggers: make(map[types.UID]*timerTriggerWithCron),
publisher: &publisher, publisher: &publisher,
} }
return timer return timer
+4 -4
View File
@@ -63,7 +63,7 @@ func (ws *TimerSync) AddUpdateTimeTrigger(timeTrigger *fv1.TimeTrigger) {
ws.logger.Debug("cron event") ws.logger.Debug("cron event")
if item, ok := ws.timer.triggers[crd.CacheKeyUID(&timeTrigger.ObjectMeta)]; ok { if item, ok := ws.timer.triggers[crd.CacheKeyUIDFromMeta(&timeTrigger.ObjectMeta)]; ok {
if item.cron != nil { if item.cron != nil {
item.cron.Stop() item.cron.Stop()
} }
@@ -71,7 +71,7 @@ func (ws *TimerSync) AddUpdateTimeTrigger(timeTrigger *fv1.TimeTrigger) {
item.cron = ws.timer.newCron(*timeTrigger) item.cron = ws.timer.newCron(*timeTrigger)
logger.Debug("cron updated") logger.Debug("cron updated")
} else { } else {
ws.timer.triggers[crd.CacheKeyUID(&timeTrigger.ObjectMeta)] = &timerTriggerWithCron{ ws.timer.triggers[crd.CacheKeyUIDFromMeta(&timeTrigger.ObjectMeta)] = &timerTriggerWithCron{
trigger: *timeTrigger, trigger: *timeTrigger,
cron: ws.timer.newCron(*timeTrigger), cron: ws.timer.newCron(*timeTrigger),
} }
@@ -82,12 +82,12 @@ func (ws *TimerSync) AddUpdateTimeTrigger(timeTrigger *fv1.TimeTrigger) {
func (ws *TimerSync) DeleteTimeTrigger(timeTrigger *fv1.TimeTrigger) { func (ws *TimerSync) DeleteTimeTrigger(timeTrigger *fv1.TimeTrigger) {
logger := ws.logger.With(zap.String("trigger_name", timeTrigger.Name), zap.String("trigger_namespace", timeTrigger.Namespace)) logger := ws.logger.With(zap.String("trigger_name", timeTrigger.Name), zap.String("trigger_namespace", timeTrigger.Namespace))
if item, ok := ws.timer.triggers[crd.CacheKeyUID(&timeTrigger.ObjectMeta)]; ok { if item, ok := ws.timer.triggers[crd.CacheKeyUIDFromMeta(&timeTrigger.ObjectMeta)]; ok {
if item.cron != nil { if item.cron != nil {
item.cron.Stop() item.cron.Stop()
logger.Info("cron for time trigger stopped") logger.Info("cron for time trigger stopped")
} }
delete(ws.timer.triggers, crd.CacheKeyUID(&timeTrigger.ObjectMeta)) delete(ws.timer.triggers, crd.CacheKeyUIDFromMeta(&timeTrigger.ObjectMeta))
logger.Debug("cron deleted") logger.Debug("cron deleted")
} }
} }