-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
7c5b6bf
commit 8f55b92
Showing
9 changed files
with
858 additions
and
10 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,39 @@ | ||
| /* | ||
| Copyright (c) Microsoft Corporation. | ||
| Licensed under the MIT license. | ||
| */ | ||
|
|
||
| // Package azureerrors defines shared azure error util functions. | ||
| package azureerrors | ||
|
|
||
| import ( | ||
| "errors" | ||
| "net/http" | ||
|
|
||
| "github.com/Azure/azure-sdk-for-go/sdk/azcore" | ||
| ) | ||
|
|
||
| // IsNotFound returns true if the error is a http 404 error returned by the azure server. | ||
| func IsNotFound(err error) bool { | ||
| var responseError *azcore.ResponseError | ||
| return errors.As(err, &responseError) && responseError.StatusCode == http.StatusNotFound | ||
| } | ||
|
|
||
| // IsClientError returns true if the error is a client error (400-499) returned by the azure server. | ||
| func IsClientError(err error) bool { | ||
| var responseError *azcore.ResponseError | ||
| return errors.As(err, &responseError) && | ||
| responseError.StatusCode >= http.StatusBadRequest && responseError.StatusCode < http.StatusInternalServerError | ||
| } | ||
|
|
||
| // IsConflict determines if the error is a http 409 error returned by the azure server. | ||
| func IsConflict(err error) bool { | ||
| var responseError *azcore.ResponseError | ||
| return errors.As(err, &responseError) && responseError.StatusCode == http.StatusConflict | ||
| } | ||
|
|
||
| // IsThrottled determines if the error is a http 429 error returned by the azure server. | ||
| func IsThrottled(err error) bool { | ||
| var responseError *azcore.ResponseError | ||
| return errors.As(err, &responseError) && responseError.StatusCode == http.StatusTooManyRequests | ||
| } |
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
196 changes: 196 additions & 0 deletions
196
pkg/controllers/hub/trafficmanagerprofile/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,196 @@ | ||
| /* | ||
| Copyright (c) Microsoft Corporation. | ||
| Licensed under the MIT license. | ||
| */ | ||
|
|
||
| // Package trafficmanagerprofile features the TrafficManagerProfile controller to reconcile TrafficManagerProfile CRs. | ||
| package trafficmanagerprofile | ||
|
|
||
| import ( | ||
| "context" | ||
| "errors" | ||
| "fmt" | ||
| "time" | ||
|
|
||
| "github.com/Azure/azure-sdk-for-go/sdk/azcore" | ||
| "github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/trafficmanager/armtrafficmanager" | ||
| apierrors "k8s.io/apimachinery/pkg/api/errors" | ||
| "k8s.io/apimachinery/pkg/api/meta" | ||
| metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" | ||
| "k8s.io/klog/v2" | ||
| "k8s.io/utils/ptr" | ||
| ctrl "sigs.k8s.io/controller-runtime" | ||
| "sigs.k8s.io/controller-runtime/pkg/client" | ||
| "sigs.k8s.io/controller-runtime/pkg/controller/controllerutil" | ||
|
|
||
| "go.goms.io/fleet/pkg/utils/controller" | ||
|
|
||
| fleetnetv1alpha1 "go.goms.io/fleet-networking/api/v1alpha1" | ||
| "go.goms.io/fleet-networking/pkg/common/azureerrors" | ||
| "go.goms.io/fleet-networking/pkg/common/objectmeta" | ||
| ) | ||
|
|
||
| // Reconciler reconciles a TrafficManagerProfile object. | ||
| type Reconciler struct { | ||
| client.Client | ||
|
|
||
| ProfilesClient *armtrafficmanager.ProfilesClient | ||
| ResourceGroupName string // default resource group name to create azure traffic manager profiles | ||
| } | ||
|
|
||
| //+kubebuilder:rbac:groups=networking.fleet.azure.com,resources=trafficmanagerprofiles,verbs=get;list;watch;create;update;patch;delete | ||
| //+kubebuilder:rbac:groups=networking.fleet.azure.com,resources=trafficmanagerprofiles/status,verbs=get;update;patch | ||
| //+kubebuilder:rbac:groups=networking.fleet.azure.com,resources=trafficmanagerprofiles/finalizers,verbs=get;update | ||
| //+kubebuilder:rbac:groups="",resources=events,verbs=create;patch | ||
|
|
||
| // Reconcile triggers a single reconcile round. | ||
| func (r *Reconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) { | ||
| name := req.NamespacedName | ||
| profileKRef := klog.KRef(name.Namespace, name.Name) | ||
|
|
||
| startTime := time.Now() | ||
| klog.V(2).InfoS("Reconciliation starts", "trafficManagerProfile", profileKRef) | ||
| defer func() { | ||
| latency := time.Since(startTime).Milliseconds() | ||
| klog.V(2).InfoS("Reconciliation ends", "trafficManagerProfile", profileKRef, "latency", latency) | ||
| }() | ||
|
|
||
| profile := &fleetnetv1alpha1.TrafficManagerProfile{} | ||
| if err := r.Client.Get(ctx, name, profile); err != nil { | ||
| if apierrors.IsNotFound(err) { | ||
| klog.V(4).InfoS("Ignoring NotFound trafficManagerProfile", "trafficManagerProfile", profileKRef) | ||
| return ctrl.Result{}, nil | ||
| } | ||
| klog.ErrorS(err, "Failed to get trafficManagerProfile", "trafficManagerProfile", profileKRef) | ||
| return ctrl.Result{}, controller.NewAPIServerError(true, err) | ||
| } | ||
|
|
||
| if !profile.ObjectMeta.DeletionTimestamp.IsZero() { | ||
| return r.handleDelete(ctx, profile) | ||
| } | ||
|
|
||
| // register finalizer | ||
| if !controllerutil.ContainsFinalizer(profile, objectmeta.TrafficManagerProfileFinalizer) { | ||
| controllerutil.AddFinalizer(profile, objectmeta.TrafficManagerProfileFinalizer) | ||
| if err := r.Update(ctx, profile); err != nil { | ||
| klog.ErrorS(err, "Failed to add finalizer to trafficManagerProfile", "trafficManagerProfile", profileKRef) | ||
| return ctrl.Result{}, controller.NewUpdateIgnoreConflictError(err) | ||
| } | ||
| } | ||
|
|
||
| return r.handleUpdate(ctx, profile) | ||
| } | ||
|
|
||
| func (r *Reconciler) handleDelete(ctx context.Context, profile *fleetnetv1alpha1.TrafficManagerProfile) (ctrl.Result, error) { | ||
| profileKObj := klog.KObj(profile) | ||
| // The profile is being deleted | ||
| if !controllerutil.ContainsFinalizer(profile, objectmeta.TrafficManagerProfileFinalizer) { | ||
| klog.V(4).InfoS("TrafficManagerProfile is being deleted", "trafficManagerProfile", profileKObj) | ||
| return ctrl.Result{}, nil | ||
| } | ||
|
|
||
| klog.V(2).InfoS("Deleting Azure Traffic Manager profile", "trafficManagerProfile", profileKObj) | ||
| if _, err := r.ProfilesClient.Delete(ctx, r.ResourceGroupName, profile.Name, nil); err != nil { | ||
| if !azureerrors.IsNotFound(err) { | ||
| klog.ErrorS(err, "Failed to delete Azure Traffic Manager profile", "trafficManagerProfile", profileKObj) | ||
| return ctrl.Result{}, err | ||
| } | ||
| } | ||
|
|
||
| controllerutil.RemoveFinalizer(profile, objectmeta.TrafficManagerProfileFinalizer) | ||
| if err := r.Client.Update(ctx, profile); err != nil { | ||
| klog.ErrorS(err, "Failed to remove trafficManagerProfile finalizer", "trafficManagerProfile", profileKObj) | ||
| return ctrl.Result{}, err | ||
| } | ||
| return ctrl.Result{}, nil | ||
| } | ||
|
|
||
| func (r *Reconciler) handleUpdate(ctx context.Context, profile *fleetnetv1alpha1.TrafficManagerProfile) (ctrl.Result, error) { | ||
| profileKObj := klog.KObj(profile) | ||
| res, updateErr := r.ProfilesClient.CreateOrUpdate(ctx, r.ResourceGroupName, profile.Name, generateAzureTrafficManagerProfile(profile), nil) | ||
| var responseError *azcore.ResponseError | ||
| if updateErr != nil { | ||
| if !errors.As(updateErr, &responseError) { | ||
| klog.ErrorS(updateErr, "Failed to send the createOrUpdate request", "trafficManagerProfile", profileKObj) | ||
| return ctrl.Result{}, updateErr | ||
| } | ||
| klog.ErrorS(updateErr, "Failed to create or update a profile", "trafficManagerProfile", profileKObj, "errorCode", responseError.ErrorCode, "statusCode", responseError.StatusCode) | ||
| } else { | ||
| // res.Profile.Properties.DNSConfig.Fqdn should not be nil | ||
| if res.Profile.Properties != nil && res.Profile.Properties.DNSConfig != nil { | ||
| profile.Status.DNSName = res.Profile.Properties.DNSConfig.Fqdn | ||
| } else { | ||
| err := fmt.Errorf("got nil DNSConfig for Azure Traffic Manager profile") | ||
| klog.ErrorS(controller.NewUnexpectedBehaviorError(err), "Unexpected value returned by the Azure Traffic Manager", "trafficManagerProfile", profileKObj) | ||
| } | ||
| } | ||
| cond := metav1.Condition{ | ||
| Type: string(fleetnetv1alpha1.TrafficManagerProfileConditionProgrammed), | ||
| Status: metav1.ConditionTrue, | ||
| ObservedGeneration: profile.Generation, | ||
| Reason: string(fleetnetv1alpha1.TrafficManagerProfileReasonProgrammed), | ||
| Message: "Successfully configured the Azure Traffic Manager profile", | ||
| } | ||
| if azureerrors.IsConflict(updateErr) { | ||
| cond = metav1.Condition{ | ||
| Type: string(fleetnetv1alpha1.TrafficManagerProfileConditionProgrammed), | ||
| Status: metav1.ConditionFalse, | ||
| ObservedGeneration: profile.Generation, | ||
| Reason: string(fleetnetv1alpha1.TrafficManagerProfileReasonDNSNameNotAvailable), | ||
| Message: "Domain name is not available. Please choose a different profile name", | ||
| } | ||
| } else if azureerrors.IsClientError(updateErr) && !azureerrors.IsThrottled(updateErr) { | ||
| cond = metav1.Condition{ | ||
| Type: string(fleetnetv1alpha1.TrafficManagerProfileConditionProgrammed), | ||
| Status: metav1.ConditionFalse, | ||
| ObservedGeneration: profile.Generation, | ||
| Reason: string(fleetnetv1alpha1.TrafficManagerProfileReasonInvalid), | ||
| Message: fmt.Sprintf("Invalid profile: %v", updateErr), | ||
| } | ||
| } else if updateErr != nil { | ||
| cond = metav1.Condition{ | ||
| Type: string(fleetnetv1alpha1.TrafficManagerProfileConditionProgrammed), | ||
| Status: metav1.ConditionFalse, | ||
| ObservedGeneration: profile.Generation, | ||
| Reason: string(fleetnetv1alpha1.TrafficManagerProfileReasonPending), | ||
| Message: fmt.Sprintf("Failed to configure profile and retyring: %v", updateErr), | ||
| } | ||
| } | ||
| meta.SetStatusCondition(&profile.Status.Conditions, cond) | ||
| if err := r.Client.Status().Update(ctx, profile); err != nil { | ||
| klog.ErrorS(err, "Failed to update trafficManagerProfile status", "trafficManagerProfile", profileKObj) | ||
| return ctrl.Result{}, controller.NewUpdateIgnoreConflictError(err) | ||
| } | ||
| klog.V(2).InfoS("Updated the trafficProfile status", "trafficManagerProfile", profileKObj, "status", profile.Status) | ||
| return ctrl.Result{}, updateErr | ||
| } | ||
|
|
||
| func generateAzureTrafficManagerProfile(profile *fleetnetv1alpha1.TrafficManagerProfile) armtrafficmanager.Profile { | ||
| mc := profile.Spec.MonitorConfig | ||
| return armtrafficmanager.Profile{ | ||
| Location: ptr.To("global"), | ||
| Properties: &armtrafficmanager.ProfileProperties{ | ||
| DNSConfig: &armtrafficmanager.DNSConfig{ | ||
| RelativeName: ptr.To(profile.Name), | ||
| }, | ||
| MonitorConfig: &armtrafficmanager.MonitorConfig{ | ||
| IntervalInSeconds: mc.IntervalInSeconds, | ||
| Path: mc.Path, | ||
| Port: mc.Port, | ||
| Protocol: ptr.To(armtrafficmanager.MonitorProtocol(*mc.Protocol)), | ||
| TimeoutInSeconds: mc.TimeoutInSeconds, | ||
| ToleratedNumberOfFailures: mc.ToleratedNumberOfFailures, | ||
| }, | ||
| ProfileStatus: ptr.To(armtrafficmanager.ProfileStatusEnabled), | ||
| // By default, the routing method is set to Weighted. | ||
| TrafficRoutingMethod: ptr.To(armtrafficmanager.TrafficRoutingMethodWeighted), | ||
| }, | ||
| } | ||
| } | ||
|
|
||
| // SetupWithManager sets up the controller with the Manager. | ||
| func (r *Reconciler) SetupWithManager(mgr ctrl.Manager) error { | ||
| return ctrl.NewControllerManagedBy(mgr). | ||
| For(&fleetnetv1alpha1.TrafficManagerProfile{}). | ||
| Complete(r) | ||
| } |
Oops, something went wrong.