Skip to content

Commit

Permalink
Reconcile Sub Component Limitador CR (#350)
Browse files Browse the repository at this point in the history
* Add limitador types

* Reconcile limitador spec from kuadrant CR

* Fix Integration Test
  • Loading branch information
Boomatang committed Jan 23, 2024
1 parent 5d4f2a8 commit 624f58b
Show file tree
Hide file tree
Showing 10 changed files with 2,539 additions and 24 deletions.
22 changes: 22 additions & 0 deletions api/v1beta1/kuadrant_types.go
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,8 @@ package v1beta1
import (
"github.com/go-logr/logr"
"github.com/google/go-cmp/cmp"
limitadorv1alpha1 "github.com/kuadrant/limitador-operator/api/v1alpha1"
corev1 "k8s.io/api/core/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"

"github.com/kuadrant/kuadrant-operator/pkg/common"
Expand All @@ -29,6 +31,26 @@ import (

// KuadrantSpec defines the desired state of Kuadrant
type KuadrantSpec struct {
// +optional
Limitador *LimitadorSpec `json:"limitador,omitempty"`
}

type LimitadorSpec struct {

// +optional
Affinity *corev1.Affinity `json:"affinity,omitempty"`

// +optional
Replicas *int `json:"replicas,omitempty"`

// +optional
ResourceRequirements *corev1.ResourceRequirements `json:"resourceRequirements,omitempty"`

// +optional
PodDisruptionBudget *limitadorv1alpha1.PodDisruptionBudgetType `json:"pdb,omitempty"`

// +optional
Storage *limitadorv1alpha1.Storage `json:"storage,omitempty"`
}

// KuadrantStatus defines the observed state of Kuadrant
Expand Down
53 changes: 50 additions & 3 deletions api/v1beta1/zz_generated.deepcopy.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

1,036 changes: 1,036 additions & 0 deletions bundle/manifests/kuadrant.io_kuadrants.yaml

Large diffs are not rendered by default.

1,036 changes: 1,036 additions & 0 deletions config/crd/bases/kuadrant.io_kuadrants.yaml

Large diffs are not rendered by default.

8 changes: 6 additions & 2 deletions controllers/gateway_kuadrant_controller_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -69,19 +69,21 @@ var _ = Describe("Kuadrant Gateway controller", func() {
})

Context("Two kuadrant instances", func() {
var secondNamespace string

BeforeEach(func() {
CreateNamespace(&secondNamespace)
newKuadrantName := "second"
newKuadrant := &kuadrantv1beta1.Kuadrant{
TypeMeta: metav1.TypeMeta{APIVersion: "v1beta1", Kind: "Kuadrant"},
ObjectMeta: metav1.ObjectMeta{Name: newKuadrantName, Namespace: testNamespace},
ObjectMeta: metav1.ObjectMeta{Name: newKuadrantName, Namespace: secondNamespace},
}
err := testClient().Create(context.Background(), newKuadrant)
Expect(err).ToNot(HaveOccurred())

Eventually(func() bool {
kuadrant := &kuadrantv1beta1.Kuadrant{}
err := k8sClient.Get(context.Background(), client.ObjectKey{Name: newKuadrantName, Namespace: testNamespace}, kuadrant)
err := k8sClient.Get(context.Background(), client.ObjectKey{Name: newKuadrantName, Namespace: secondNamespace}, kuadrant)
if err != nil {
return false
}
Expand All @@ -92,6 +94,8 @@ var _ = Describe("Kuadrant Gateway controller", func() {
}, time.Minute, 5*time.Second).Should(BeTrue())
})

AfterEach(DeleteNamespaceCallback(&secondNamespace))

It("new gateway should not be annotated", func() {
gateway := testBuildBasicGateway(gwName, testNamespace)
err := k8sClient.Create(context.Background(), gateway)
Expand Down
5 changes: 1 addition & 4 deletions controllers/helper_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -62,10 +62,7 @@ func CreateNamespace(namespace *string) {
existingNamespace := &v1.Namespace{}
Eventually(func() bool {
err := testClient().Get(context.Background(), types.NamespacedName{Name: generatedTestNamespace}, existingNamespace)
if err != nil {
return false
}
return true
return err == nil
}, time.Minute, 5*time.Second).Should(BeTrue())

*namespace = existingNamespace.Name
Expand Down
58 changes: 43 additions & 15 deletions controllers/kuadrant_controller.go
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ import (
"context"
"encoding/json"

"github.com/kuadrant/kuadrant-operator/pkg/kuadranttools"
corev1 "k8s.io/api/core/v1"
"k8s.io/utils/env"

Expand Down Expand Up @@ -469,27 +470,54 @@ func (r *KuadrantReconciler) removeAnnotationFromGateways(ctx context.Context, k
}

func (r *KuadrantReconciler) reconcileLimitador(ctx context.Context, kObj *kuadrantv1beta1.Kuadrant) error {
limitador := &limitadorv1alpha1.Limitador{
TypeMeta: metav1.TypeMeta{
Kind: "Limitador",
APIVersion: "limitador.kuadrant.io/v1alpha1",
},
ObjectMeta: metav1.ObjectMeta{
Name: common.LimitadorName,
Namespace: kObj.Namespace,
},
Spec: limitadorv1alpha1.LimitadorSpec{
RateLimitHeaders: &[]limitadorv1alpha1.RateLimitHeadersType{limitadorv1alpha1.RateLimitHeadersTypeDraft03}[0],
Telemetry: &[]limitadorv1alpha1.Telemetry{limitadorv1alpha1.TelemetryExhaustive}[0],
},
limitadorKey := client.ObjectKey{Name: common.LimitadorName, Namespace: kObj.Namespace}
limitador := &limitadorv1alpha1.Limitador{}
err := r.Client().Get(ctx, limitadorKey, limitador)
if err != nil {
if apierrors.IsNotFound(err) {
limitador = &limitadorv1alpha1.Limitador{
TypeMeta: metav1.TypeMeta{
Kind: "Limitador",
APIVersion: "limitador.kuadrant.io/v1alpha1",
},
ObjectMeta: metav1.ObjectMeta{
Name: common.LimitadorName,
Namespace: kObj.Namespace,
},
Spec: limitadorv1alpha1.LimitadorSpec{
RateLimitHeaders: &[]limitadorv1alpha1.RateLimitHeadersType{limitadorv1alpha1.RateLimitHeadersTypeDraft03}[0],
Telemetry: &[]limitadorv1alpha1.Telemetry{limitadorv1alpha1.TelemetryExhaustive}[0],
},
}
} else {
return err
}
}

if kObj.Spec.Limitador != nil {
if kObj.Spec.Limitador.Affinity != nil {
limitador.Spec.Affinity = kObj.Spec.Limitador.Affinity
}
if kObj.Spec.Limitador.PodDisruptionBudget != nil {
limitador.Spec.PodDisruptionBudget = kObj.Spec.Limitador.PodDisruptionBudget
}
if kObj.Spec.Limitador.Replicas != nil {
limitador.Spec.Replicas = kObj.Spec.Limitador.Replicas
}
if kObj.Spec.Limitador.ResourceRequirements != nil {
limitador.Spec.ResourceRequirements = kObj.Spec.Limitador.ResourceRequirements
}
if kObj.Spec.Limitador.Storage != nil {
limitador.Spec.Storage = kObj.Spec.Limitador.Storage
}
}

err := r.SetOwnerReference(kObj, limitador)
err = r.SetOwnerReference(kObj, limitador)
if err != nil {
return err
}

return r.ReconcileResource(ctx, &limitadorv1alpha1.Limitador{}, limitador, reconcilers.CreateOnlyMutator)
return r.ReconcileResource(ctx, &limitadorv1alpha1.Limitador{}, limitador, kuadranttools.LimitadorMutator)
}

func (r *KuadrantReconciler) reconcileAuthorino(ctx context.Context, kObj *kuadrantv1beta1.Kuadrant) error {
Expand Down
94 changes: 94 additions & 0 deletions doc/reference/kuadrant.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,94 @@
# The Kuadrant Custom Resource Definition (CRD)

## kuadrant

<details>
<summary>Note on Limitador</summary>
The Kuadrant operator creates a Limitador CR named `limitador` in the same namespace as the Kuadrant CR. If there is a pre-existing Limitador CR of the same name the kuadrant operator will take ownership of that Limitador CR.
</details>

| **Field** | **Type** | **Required** | **Description** |
|-----------|-----------------------------------|:------------:|-------------------------------------------------|
| `spec` | [KuadrantSpec](#kuadrantspec) | No | The specification for Kuadrant custom resource. |
| `status` | [KuadrantStatus](#kuadrantstatus) | No | The status for the custom resources. |

## KuadrantSpec

| **Field** | **Type** | **Required** | **Description** |
|-------------|-------------------------|:------------:|----------------------------------|
| `limitador` | [Limitador](#limitador) | No | Configure limitador deployments. |

### Limitador

| **Field** | **Type** | **Required** | **Description** |
|------------------------|------------------------------------------------------------------------------------|:------------:|----------------------------------------------------|
| `affinity` | [Affinity](https://pkg.go.dev/k8s.io/api/core/v1#Affinity) | No | Describes the scheduling rules for limitador pods. |
| `replicas` | Number | No | Sets the number of limitador replicas to deploy. |
| `resourceRequirements` | [ResourceRequirements](https://pkg.go.dev/k8s.io/api/core/v1#ResourceRequirements) | No | Set the resource requirements for limitador pods. |
| `pdb` | [PodDisruptionBudgetType](#poddisruptionbudgettype) | No | Configure allowed pod disruption budget fields. |
| `storage` | [Storage](#storage) | No | Define backend storage option for limitador. |

### PodDisruptionBudgetType

| **Field** | **Type** | **Required** | **Description** |
|------------------|----------|:------------:|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
| `maxUnavailable` | Number | No | An eviction is allowed if at most "maxUnavailable" limitador pods are unavailable after the eviction, i.e. even in absence of the evicted pod. For example, one can prevent all voluntary evictions by specifying 0. This is a mutually exclusive setting with "minAvailable". |
| `minAvailable` | Number | No | An eviction is allowed if at least "minAvailable" limitador pods will still be available after the eviction, i.e. even in the absence of the evicted pod. So for example you can prevent all voluntary evictions by specifying "100%". |

### Storage

| **Field** | **Type** | **Required** | **Description** |
|----------------|-----------------------------|:------------:|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
| `redis` | [Redis](#redis) | No | Uses Redis to store limitador counters. |
| `redis-cached` | [RedisCached](#redisCached) | No | Uses Redis to store limitador counters, with an in-memory cache |
| `disk` | [Disk](#disk) | No | Counters are held on disk (persistent). Kubernetes [Persistent Volumes](https://kubernetes.io/docs/concepts/storage/persistent-volumes/) will be used to store counters. |

#### Redis

| **Field** | **Type** | **Required** | **Description** |
|-------------------|------------------------------------------------------------------------------------|:------------:|-----------------------------------------------------------------|
| `configSecretRef` | [LocalObjectReference](https://pkg.go.dev/k8s.io/api/core/v1#LocalObjectReference) | No | ConfigSecretRef refers to the secret holding the URL for Redis. |

#### RedisCached

| **Field** | **Type** | **Required** | **Description** |
|-------------------|------------------------------------------------------------------------------------|:------------:|-----------------------------------------------------------------|
| `configSecretRef` | [LocalObjectReference](https://pkg.go.dev/k8s.io/api/core/v1#LocalObjectReference) | No | ConfigSecretRef refers to the secret holding the URL for Redis. |
| `options` | [Options](#options) | No | Configures a number of caching options for limitador. |

##### Options

| **Field** | **Type** | **Required** | **Description** |
|----------------|----------|:------------:|----------------------------------------------------------------------------|
| `ttl` | Number | No | TTL for cached counters in milliseconds [default: 5000] |
| `ratio` | Number | No | Ratio to apply to the TTL from Redis on cached counters [default: 10] |
| `flush-period` | Number | No | FlushPeriod for counters in milliseconds [default: 1000] |
| `max-cached` | Number | No | MaxCached refers to the maximum amount of counters cached [default: 10000] |

#### Disk

| **Field** | **Type** | **Required** | **Description** |
|-------------------------|-----------------------------------|:------------:|-----------------------------------------------------------------------------------------------|
| `persistentVolumeClaim` | [PVCGenericSpec](#pvcgenericspec) | No | Configure resources for PVC. |
| `Optimize` | String | No | Defines optimization option of the disk persistence type. Valid options: "throughput", "disk" |

##### PVCGenericSpec

| **Field** | **Type** | **Required** | **Description** |
|--------------------|-------------------------------------------------------------------|:------------:|-------------------------------------------------------------------------------|
| `storageClassName` | String | No | Storage class name |
| `resources` | [PersistentVolumeClaimResources](#persistentvolumeclaimresources) | No | Resources represent the minimum resources the volume should have |
| `volumeName` | String | No | VolumeName is the binding reference to the PersistentVolume backing the claim |

###### PersistentVolumeClaimResources

| **Field** | **Type** | **Required** | **Description** |
|------------|--------------------------------------------------------------------------------------|:------------:|---------------------------------------------------------------------|
| `requests` | [Quantity](https://pkg.go.dev/k8s.io/apimachinery@v0.28.4/pkg/api/resource#Quantity) | Yes | Storage resources requests to be used on the persisitentVolumeClaim |

## KuadrantStatus

| **Field** | **Type** | **Description** |
|----------------------|----------------------------------------------------------------------------------------------|-------------------------------------------------------------------------------------------------------------------------------------|
| `observedGeneration` | String | Number of the last observed generation of the resource. Use it to check if the status info is up to date with latest resource spec. |
| `conditions` | [][ConditionSpec](https://pkg.go.dev/k8s.io/apimachinery@v0.28.4/pkg/apis/meta/v1#Condition) | List of conditions that define that status of the resource. |
53 changes: 53 additions & 0 deletions pkg/kuadranttools/limitador_tools.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
package kuadranttools

import (
"fmt"
"reflect"

"github.com/kuadrant/kuadrant-operator/api/v1beta1"
limitadorv1alpha1 "github.com/kuadrant/limitador-operator/api/v1alpha1"
"sigs.k8s.io/controller-runtime/pkg/client"
)

func LimitadorMutator(existingObj, desiredObj client.Object) (bool, error) {
update := false
existing, ok := existingObj.(*limitadorv1alpha1.Limitador)
if !ok {
return false, fmt.Errorf("existingObj %T is not a *limitadorv1alpha1.Limitador", existingObj)
}
desired, ok := desiredObj.(*limitadorv1alpha1.Limitador)
if !ok {
return false, fmt.Errorf("desireObj %T is not a *limitadorv1alpha1.Limitador", desiredObj)
}

if !reflect.DeepEqual(existing.OwnerReferences, desired.OwnerReferences) {
update = true
existing.OwnerReferences = desired.OwnerReferences
}

existingSpec := limitadorSpecSubSet(existing.Spec)
desiredSpec := limitadorSpecSubSet(desired.Spec)

if !reflect.DeepEqual(existingSpec, desiredSpec) {
update = true
existing.Spec.Affinity = desired.Spec.Affinity
existing.Spec.PodDisruptionBudget = desired.Spec.PodDisruptionBudget
existing.Spec.Replicas = desired.Spec.Replicas
existing.Spec.ResourceRequirements = desired.Spec.ResourceRequirements
existing.Spec.Storage = desired.Spec.Storage
}

return update, nil
}

func limitadorSpecSubSet(spec limitadorv1alpha1.LimitadorSpec) v1beta1.LimitadorSpec {
out := v1beta1.LimitadorSpec{}

out.Affinity = spec.Affinity
out.PodDisruptionBudget = spec.PodDisruptionBudget
out.Replicas = spec.Replicas
out.ResourceRequirements = spec.ResourceRequirements
out.Storage = spec.Storage

return out
}
Loading

0 comments on commit 624f58b

Please sign in to comment.