| /* |
| Licensed to the Apache Software Foundation (ASF) under one or more |
| contributor license agreements. See the NOTICE file distributed with |
| this work for additional information regarding copyright ownership. |
| The ASF licenses this file to You under the Apache License, Version 2.0 |
| (the "License"); you may not use this file except in compliance with |
| the License. You may obtain a copy of the License at |
| |
| http://www.apache.org/licenses/LICENSE-2.0 |
| |
| Unless required by applicable law or agreed to in writing, software |
| distributed under the License is distributed on an "AS IS" BASIS, |
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| See the License for the specific language governing permissions and |
| limitations under the License. |
| */ |
| |
| package trait |
| |
| import ( |
| "fmt" |
| "strconv" |
| |
| corev1 "k8s.io/api/core/v1" |
| metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" |
| "k8s.io/utils/pointer" |
| |
| serving "knative.dev/serving/pkg/apis/serving/v1" |
| |
| v1 "github.com/apache/camel-k/pkg/apis/camel/v1" |
| traitv1 "github.com/apache/camel-k/pkg/apis/camel/v1/trait" |
| "github.com/apache/camel-k/pkg/metadata" |
| "github.com/apache/camel-k/pkg/util/kubernetes" |
| "github.com/apache/camel-k/pkg/util/label" |
| ) |
| |
| const ( |
| knativeServiceTraitID = "knative-service" |
| |
| // Auto-scaling annotations. |
| knativeServingClassAnnotation = "autoscaling.knative.dev/class" |
| knativeServingMetricAnnotation = "autoscaling.knative.dev/metric" |
| knativeServingTargetAnnotation = "autoscaling.knative.dev/target" |
| knativeServingMinScaleAnnotation = "autoscaling.knative.dev/minScale" |
| knativeServingMaxScaleAnnotation = "autoscaling.knative.dev/maxScale" |
| // Rollout annotation. |
| knativeServingRolloutDurationAnnotation = "serving.knative.dev/rolloutDuration" |
| // visibility label. |
| knativeServingVisibilityLabel = "networking.knative.dev/visibility" |
| ) |
| |
| type knativeServiceTrait struct { |
| BaseTrait |
| traitv1.KnativeServiceTrait `property:",squash"` |
| } |
| |
| var _ ControllerStrategySelector = &knativeServiceTrait{} |
| |
| func newKnativeServiceTrait() Trait { |
| return &knativeServiceTrait{ |
| BaseTrait: NewBaseTrait(knativeServiceTraitID, 1400), |
| } |
| } |
| |
| // IsAllowedInProfile overrides default. |
| func (t *knativeServiceTrait) IsAllowedInProfile(profile v1.TraitProfile) bool { |
| return profile.Equal(v1.TraitProfileKnative) |
| } |
| |
| func (t *knativeServiceTrait) Configure(e *Environment) (bool, error) { |
| if e.Integration == nil || !pointer.BoolDeref(t.Enabled, true) { |
| if e.Integration != nil { |
| e.Integration.Status.SetCondition( |
| v1.IntegrationConditionKnativeServiceAvailable, |
| corev1.ConditionFalse, |
| v1.IntegrationConditionKnativeServiceNotAvailableReason, |
| "explicitly disabled", |
| ) |
| } |
| |
| return false, nil |
| } |
| |
| if !e.IntegrationInRunningPhases() { |
| return false, nil |
| } |
| |
| if e.Resources.GetDeploymentForIntegration(e.Integration) != nil { |
| e.Integration.Status.SetCondition( |
| v1.IntegrationConditionKnativeServiceAvailable, |
| corev1.ConditionFalse, |
| v1.IntegrationConditionKnativeServiceNotAvailableReason, |
| fmt.Sprintf("different controller strategy used (%s)", string(ControllerStrategyDeployment)), |
| ) |
| |
| // A controller is already present for the integration |
| return false, nil |
| } |
| |
| strategy, err := e.DetermineControllerStrategy() |
| if err != nil { |
| e.Integration.Status.SetErrorCondition( |
| v1.IntegrationConditionKnativeServiceAvailable, |
| v1.IntegrationConditionKnativeServiceNotAvailableReason, |
| err, |
| ) |
| |
| return false, err |
| } |
| if strategy != ControllerStrategyKnativeService { |
| e.Integration.Status.SetCondition( |
| v1.IntegrationConditionKnativeServiceAvailable, |
| corev1.ConditionFalse, |
| v1.IntegrationConditionKnativeServiceNotAvailableReason, |
| fmt.Sprintf("different controller strategy used (%s)", string(strategy)), |
| ) |
| |
| return false, nil |
| } |
| |
| if pointer.BoolDeref(t.Auto, true) { |
| // Check the right value for minScale, as not all services are allowed to scale down to 0 |
| if t.MinScale == nil { |
| sources, err := kubernetes.ResolveIntegrationSources(e.Ctx, t.Client, e.Integration, e.Resources) |
| if err != nil { |
| e.Integration.Status.SetErrorCondition( |
| v1.IntegrationConditionKnativeServiceAvailable, |
| v1.IntegrationConditionKnativeServiceNotAvailableReason, |
| err, |
| ) |
| |
| return false, err |
| } |
| |
| meta := metadata.ExtractAll(e.CamelCatalog, sources) |
| if !meta.ExposesHTTPServices || !meta.PassiveEndpoints { |
| single := 1 |
| t.MinScale = &single |
| } |
| } |
| } |
| |
| if e.IntegrationInPhase(v1.IntegrationPhaseRunning, v1.IntegrationPhaseError) { |
| condition := e.Integration.Status.GetCondition(v1.IntegrationConditionKnativeServiceAvailable) |
| return condition != nil && condition.Status == corev1.ConditionTrue, nil |
| } |
| |
| return true, nil |
| } |
| |
| func (t *knativeServiceTrait) Apply(e *Environment) error { |
| ksvc, err := t.getServiceFor(e) |
| if err != nil { |
| return err |
| } |
| e.Resources.Add(ksvc) |
| |
| e.Integration.Status.SetCondition( |
| v1.IntegrationConditionKnativeServiceAvailable, |
| corev1.ConditionTrue, |
| v1.IntegrationConditionKnativeServiceAvailableReason, |
| fmt.Sprintf("Knative service name is %s", ksvc.Name), |
| ) |
| |
| return nil |
| } |
| |
| func (t *knativeServiceTrait) SelectControllerStrategy(e *Environment) (*ControllerStrategy, error) { |
| knativeServiceStrategy := ControllerStrategyKnativeService |
| if t.Enabled != nil { |
| if *t.Enabled { |
| return &knativeServiceStrategy, nil |
| } |
| return nil, nil |
| } |
| |
| var sources []v1.SourceSpec |
| var err error |
| if sources, err = kubernetes.ResolveIntegrationSources(e.Ctx, t.Client, e.Integration, e.Resources); err != nil { |
| return nil, err |
| } |
| |
| meta := metadata.ExtractAll(e.CamelCatalog, sources) |
| if meta.ExposesHTTPServices { |
| return &knativeServiceStrategy, nil |
| } |
| return nil, nil |
| } |
| |
| func (t *knativeServiceTrait) ControllerStrategySelectorOrder() int { |
| return 100 |
| } |
| |
| func (t *knativeServiceTrait) getServiceFor(e *Environment) (*serving.Service, error) { |
| serviceAnnotations := make(map[string]string) |
| if e.Integration.Annotations != nil { |
| for k, v := range e.Integration.Annotations { |
| serviceAnnotations[k] = v |
| } |
| } |
| // Set Knative rollout |
| if t.RolloutDuration != "" { |
| serviceAnnotations[knativeServingRolloutDurationAnnotation] = t.RolloutDuration |
| } |
| |
| revisionAnnotations := make(map[string]string) |
| if e.Integration.Annotations != nil { |
| for k, v := range filterTransferableAnnotations(e.Integration.Annotations) { |
| revisionAnnotations[k] = v |
| } |
| } |
| // Set Knative auto-scaling |
| if t.Class != "" { |
| revisionAnnotations[knativeServingClassAnnotation] = t.Class |
| } |
| if t.Metric != "" { |
| revisionAnnotations[knativeServingMetricAnnotation] = t.Metric |
| } |
| if t.Target != nil { |
| revisionAnnotations[knativeServingTargetAnnotation] = strconv.Itoa(*t.Target) |
| } |
| if t.MinScale != nil && *t.MinScale > 0 { |
| revisionAnnotations[knativeServingMinScaleAnnotation] = strconv.Itoa(*t.MinScale) |
| } |
| if t.MaxScale != nil && *t.MaxScale > 0 { |
| revisionAnnotations[knativeServingMaxScaleAnnotation] = strconv.Itoa(*t.MaxScale) |
| } |
| |
| serviceLabels := map[string]string{ |
| v1.IntegrationLabel: e.Integration.Name, |
| } |
| if t.Visibility != "" { |
| serviceLabels[knativeServingVisibilityLabel] = t.Visibility |
| } |
| |
| svc := serving.Service{ |
| TypeMeta: metav1.TypeMeta{ |
| Kind: "Service", |
| APIVersion: serving.SchemeGroupVersion.String(), |
| }, |
| ObjectMeta: metav1.ObjectMeta{ |
| Name: e.Integration.Name, |
| Namespace: e.Integration.Namespace, |
| Labels: serviceLabels, |
| Annotations: serviceAnnotations, |
| }, |
| Spec: serving.ServiceSpec{ |
| ConfigurationSpec: serving.ConfigurationSpec{ |
| Template: serving.RevisionTemplateSpec{ |
| ObjectMeta: metav1.ObjectMeta{ |
| Labels: label.AddLabels(e.Integration.Name), |
| Annotations: revisionAnnotations, |
| }, |
| Spec: serving.RevisionSpec{ |
| PodSpec: corev1.PodSpec{ |
| ServiceAccountName: e.Integration.Spec.ServiceAccountName, |
| }, |
| }, |
| }, |
| }, |
| }, |
| } |
| |
| replicas := e.Integration.Spec.Replicas |
| |
| isUpdateRequired := false |
| minScale, ok := svc.Spec.Template.Annotations[knativeServingMinScaleAnnotation] |
| if ok { |
| min, err := strconv.Atoi(minScale) |
| if err != nil { |
| return nil, err |
| } |
| if replicas == nil || min != int(*replicas) { |
| isUpdateRequired = true |
| } |
| } else if replicas != nil { |
| isUpdateRequired = true |
| } |
| |
| maxScale, ok := svc.Spec.Template.Annotations[knativeServingMaxScaleAnnotation] |
| if ok { |
| max, err := strconv.Atoi(maxScale) |
| if err != nil { |
| return nil, err |
| } |
| if replicas == nil || max != int(*replicas) { |
| isUpdateRequired = true |
| } |
| } else if replicas != nil { |
| isUpdateRequired = true |
| } |
| |
| if isUpdateRequired { |
| if replicas == nil { |
| if t.MinScale != nil && *t.MinScale > 0 { |
| svc.Spec.Template.Annotations[knativeServingMinScaleAnnotation] = strconv.Itoa(*t.MinScale) |
| } else { |
| delete(svc.Spec.Template.Annotations, knativeServingMinScaleAnnotation) |
| } |
| if t.MaxScale != nil && *t.MaxScale > 0 { |
| svc.Spec.Template.Annotations[knativeServingMaxScaleAnnotation] = strconv.Itoa(*t.MaxScale) |
| } else { |
| delete(svc.Spec.Template.Annotations, knativeServingMaxScaleAnnotation) |
| } |
| } else { |
| scale := strconv.Itoa(int(*replicas)) |
| svc.Spec.Template.Annotations[knativeServingMinScaleAnnotation] = scale |
| svc.Spec.Template.Annotations[knativeServingMaxScaleAnnotation] = scale |
| } |
| } |
| |
| return &svc, nil |
| } |