-
Notifications
You must be signed in to change notification settings - Fork 107
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
gateway server and client controllers
- Loading branch information
Showing
14 changed files
with
924 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
205 changes: 205 additions & 0 deletions
205
pkg/liqo-controller-manager/external-network/client-operator/client_controller.go
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,205 @@ | ||
| // Copyright 2019-2023 The Liqo Authors | ||
| // | ||
| // Licensed 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 clientoperator | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
| "strings" | ||
|
|
||
| corev1 "k8s.io/api/core/v1" | ||
| apierrors "k8s.io/apimachinery/pkg/api/errors" | ||
| metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" | ||
| "k8s.io/apimachinery/pkg/apis/meta/v1/unstructured" | ||
| "k8s.io/apimachinery/pkg/runtime" | ||
| "k8s.io/apimachinery/pkg/runtime/schema" | ||
| "k8s.io/client-go/dynamic" | ||
| "k8s.io/klog/v2" | ||
| "k8s.io/utils/pointer" | ||
| ctrl "sigs.k8s.io/controller-runtime" | ||
| "sigs.k8s.io/controller-runtime/pkg/client" | ||
|
|
||
| networkingv1alpha1 "github.com/liqotech/liqo/apis/networking/v1alpha1" | ||
| enutils "github.com/liqotech/liqo/pkg/liqo-controller-manager/external-network/utils" | ||
| ) | ||
|
|
||
| // ClientReconciler manage GatewayClient lifecycle. | ||
| type ClientReconciler struct { | ||
| client.Client | ||
| Scheme *runtime.Scheme | ||
| DynClient dynamic.Interface | ||
| Factory *enutils.RunnableFactory | ||
| ClientResources []string | ||
| } | ||
|
|
||
| // NewClientReconciler returns a new ClientReconciler. | ||
| func NewClientReconciler(ctx context.Context, | ||
| cl client.Client, dynClient dynamic.Interface, | ||
| factory *enutils.RunnableFactory, s *runtime.Scheme, | ||
| clientResources []string) *ClientReconciler { | ||
| return &ClientReconciler{ | ||
| Client: cl, | ||
| Scheme: s, | ||
| DynClient: dynClient, | ||
| Factory: factory, | ||
| ClientResources: clientResources, | ||
| } | ||
| } | ||
|
|
||
| // cluster-role | ||
| // +kubebuilder:rbac:groups=networking.liqo.io,resources=gatewayclients,verbs=get;list;watch;delete;create;update;patch | ||
|
|
||
| // Reconcile manage GatewayClient lifecycle. | ||
| func (r *ClientReconciler) Reconcile(ctx context.Context, req ctrl.Request) (res ctrl.Result, err error) { | ||
| gwClient := &networkingv1alpha1.GatewayClient{} | ||
| if err = r.Get(ctx, req.NamespacedName, gwClient); err != nil { | ||
| if apierrors.IsNotFound(err) { | ||
| klog.Infof("Gateway client %q not found", req.NamespacedName) | ||
| return ctrl.Result{}, nil | ||
| } | ||
| klog.Errorf("Unable to get the gateway client %q: %s", req.NamespacedName, err) | ||
| return ctrl.Result{}, err | ||
| } | ||
|
|
||
| defer func() { | ||
| newErr := r.Status().Update(ctx, gwClient) | ||
| if newErr != nil { | ||
| if err != nil { | ||
| klog.Errorf("Error reconciling the gateway client %q: %s", req.NamespacedName, err) | ||
| } | ||
| klog.Errorf("Unable to update the gateway client %q: %s", req.NamespacedName, newErr) | ||
| err = newErr | ||
| } | ||
| }() | ||
|
|
||
| if err = r.EnsureGatewayClient(ctx, gwClient); err != nil { | ||
| klog.Errorf("Unable to ensure the gateway client %q: %s", req.NamespacedName, err) | ||
| return ctrl.Result{}, err | ||
| } | ||
|
|
||
| return ctrl.Result{}, nil | ||
| } | ||
|
|
||
| // EnsureGatewayClient ensures the GatewayClient is correctly configured. | ||
| func (r *ClientReconciler) EnsureGatewayClient(ctx context.Context, gwClient *networkingv1alpha1.GatewayClient) error { | ||
| templateGV, err := schema.ParseGroupVersion(gwClient.Spec.ClientTemplateRef.APIVersion) | ||
| if err != nil { | ||
| return fmt.Errorf("unable to parse the client template group version: %w", err) | ||
| } | ||
|
|
||
| templateGVR := schema.GroupVersionResource{ | ||
| Group: templateGV.Group, | ||
| Version: templateGV.Version, | ||
| Resource: enutils.KindToResource(gwClient.Spec.ClientTemplateRef.Kind), | ||
| } | ||
| template, err := r.DynClient.Resource(templateGVR). | ||
| Namespace(gwClient.Spec.ClientTemplateRef.Namespace). | ||
| Get(ctx, gwClient.Spec.ClientTemplateRef.Name, metav1.GetOptions{}) | ||
| if err != nil { | ||
| return fmt.Errorf("unable to get the client template: %w", err) | ||
| } | ||
|
|
||
| templateSpec, ok := template.Object["spec"].(map[string]interface{}) | ||
| if !ok { | ||
| return fmt.Errorf("unable to get the spec of the client template") | ||
| } | ||
| objectKindInt, ok := templateSpec["objectKind"].(map[string]interface{}) | ||
| if !ok { | ||
| return fmt.Errorf("unable to get the object kind of the client template") | ||
| } | ||
| objectKind := metav1.TypeMeta{ | ||
| Kind: objectKindInt["kind"].(string), | ||
| APIVersion: objectKindInt["apiVersion"].(string), | ||
| } | ||
| objectTemplate, ok := templateSpec["template"].(map[string]interface{}) | ||
| if !ok { | ||
| return fmt.Errorf("unable to get the template of the client template") | ||
| } | ||
| objectTemplateMetadataInt, ok := objectTemplate["metadata"].(map[string]interface{}) | ||
| if !ok { | ||
| return fmt.Errorf("unable to get the metadata of the client template") | ||
| } | ||
| objectTemplateMetadata := metav1.ObjectMeta{ | ||
| Name: enutils.GetValueOrDefault(objectTemplateMetadataInt, "name", gwClient.Name), | ||
| Namespace: enutils.GetValueOrDefault(objectTemplateMetadataInt, "namespace", gwClient.Namespace), | ||
| Labels: enutils.TranslateMap(objectTemplateMetadataInt["labels"]), | ||
| Annotations: enutils.TranslateMap(objectTemplateMetadataInt["annotations"]), | ||
| } | ||
| objectTemplateSpec, ok := objectTemplate["spec"].(map[string]interface{}) | ||
| if !ok { | ||
| return fmt.Errorf("unable to get the spec of the client template") | ||
| } | ||
|
|
||
| unstructuredObject, err := enutils.CreateOrPatch(ctx, r.DynClient.Resource(objectKind.GroupVersionKind().GroupVersion().WithResource(enutils.KindToResource(objectKind.Kind))). | ||
| Namespace(gwClient.Namespace), gwClient.Name, func(obj *unstructured.Unstructured) error { | ||
| obj.SetGroupVersionKind(objectKind.GroupVersionKind()) | ||
| obj.SetName(gwClient.Name) | ||
| obj.SetNamespace(gwClient.Namespace) | ||
| obj.SetLabels(objectTemplateMetadata.Labels) | ||
| obj.SetAnnotations(objectTemplateMetadata.Annotations) | ||
| obj.SetOwnerReferences([]metav1.OwnerReference{ | ||
| { | ||
| APIVersion: gwClient.APIVersion, | ||
| Kind: gwClient.Kind, | ||
| Name: gwClient.Name, | ||
| UID: gwClient.UID, | ||
| Controller: pointer.Bool(true), | ||
| }, | ||
| }) | ||
| spec, err := enutils.RenderTemplate(objectTemplateSpec, gwClient.Spec) | ||
| if err != nil { | ||
| return fmt.Errorf("unable to render the template: %w", err) | ||
| } | ||
| obj.Object["spec"] = spec | ||
| return nil | ||
| }) | ||
| if err != nil { | ||
| return fmt.Errorf("unable to update the client: %w", err) | ||
| } | ||
|
|
||
| gwClient.Status.ClientRef = corev1.ObjectReference{ | ||
| APIVersion: unstructuredObject.GetAPIVersion(), | ||
| Kind: unstructuredObject.GetKind(), | ||
| Name: unstructuredObject.GetName(), | ||
| Namespace: unstructuredObject.GetNamespace(), | ||
| UID: unstructuredObject.GetUID(), | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| // SetupWithManager register the ClientReconciler to the manager. | ||
| func (r *ClientReconciler) SetupWithManager(mgr ctrl.Manager) error { | ||
| ownerEnqueuer := enutils.NewOwnerEnqueuer(networkingv1alpha1.GatewayClientKind) | ||
| factorySource := enutils.NewFactorySource(r.Factory) | ||
|
|
||
| for _, resource := range r.ClientResources { | ||
| tmp := strings.Split(resource, "/") | ||
| if len(tmp) != 3 { | ||
| return fmt.Errorf("invalid resource %q", resource) | ||
| } | ||
| gvr := schema.GroupVersionResource{ | ||
| Group: tmp[0], | ||
| Version: tmp[1], | ||
| Resource: tmp[2], | ||
| } | ||
| factorySource.ForResource(gvr) | ||
| } | ||
|
|
||
| return ctrl.NewControllerManagedBy(mgr). | ||
| WatchesRawSource(factorySource.Source(), ownerEnqueuer). | ||
| For(&networkingv1alpha1.GatewayClient{}). | ||
| Complete(r) | ||
| } |
Oops, something went wrong.