-
Notifications
You must be signed in to change notification settings - Fork 3
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #7 from amazeeio/retry
Add retry cron task to controller to attempt retry paused
- Loading branch information
Showing
6 changed files
with
160 additions
and
5 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,105 @@ | ||
| package handlers | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/json" | ||
| "fmt" | ||
| "strconv" | ||
| "strings" | ||
|
|
||
| corev1 "k8s.io/api/core/v1" | ||
| networkv1beta1 "k8s.io/api/networking/v1beta1" | ||
| "k8s.io/apimachinery/pkg/types" | ||
| ctrl "sigs.k8s.io/controller-runtime" | ||
| "sigs.k8s.io/controller-runtime/pkg/client" | ||
| ) | ||
|
|
||
| type cleanup interface { | ||
| CheckPausedCertStatus() | ||
| } | ||
|
|
||
| // Cleanup is used for cleaning up old pods or resources. | ||
| type Cleanup struct { | ||
| Client client.Client | ||
| EnableDebug bool | ||
| } | ||
|
|
||
| // NewCleanup returns a cleanup with controller-runtime client. | ||
| func NewCleanup(client client.Client, enableDebug bool) *Cleanup { | ||
| return &Cleanup{ | ||
| Client: client, | ||
| EnableDebug: enableDebug, | ||
| } | ||
| } | ||
|
|
||
| // CheckPausedCertStatus is a cronjob that will periodically check for paused status on ingresses that have failed to | ||
| // upload their certificate to fastly and will unpause them to allow them to retry | ||
| // after 5 attempts it will give up | ||
| func (h *Cleanup) CheckPausedCertStatus() { | ||
| fmt.Println("paused") | ||
| opLog := ctrl.Log.WithName("handlers").WithName("PausedCertStatusCheck") | ||
| namespaces := &corev1.NamespaceList{} | ||
| if err := h.Client.List(context.Background(), namespaces); err != nil { | ||
| opLog.Error(err, fmt.Sprintf("Unable to list namespaces created by Lagoon, there may be none or something went wrong")) | ||
| return | ||
| } | ||
| for _, ns := range namespaces.Items { | ||
| opLog.Info(fmt.Sprintf("Checking LagoonBuilds in namespace %s", ns.ObjectMeta.Name)) | ||
| ingresses := &networkv1beta1.IngressList{} | ||
| listOption := (&client.ListOptions{}).ApplyOptions([]client.ListOption{ | ||
| client.InNamespace(ns.ObjectMeta.Name), | ||
| client.MatchingLabels(map[string]string{ | ||
| "fastly.amazee.io/paused": "true", | ||
| }), | ||
| }) | ||
| if err := h.Client.List(context.Background(), ingresses, listOption); err != nil { | ||
| opLog.Error(err, fmt.Sprintf("Unable to list Ingress resource in namespace, there may be none or something went wrong")) | ||
| return | ||
| } | ||
| for _, ingress := range ingresses.Items { | ||
| // check if a reason exists | ||
| if reason, ok := ingress.ObjectMeta.Annotations["fastly.amazee.io/paused-reason"]; ok { | ||
| // always set retryCount to 0 | ||
| retryCount := 0 | ||
| // then read the value in from the annotation if it is present and set the value to it | ||
| if retryValue, ok := ingress.ObjectMeta.Annotations["fastly.amazee.io/paused-retry-count"]; ok { | ||
| if i, err := strconv.Atoi(retryValue); err == nil { | ||
| retryCount = i | ||
| } | ||
| } | ||
| // and if the reason is unable to find a secret | ||
| if strings.Contains(reason, "Unable to find secret of") { | ||
| // then attempt the process to fix it, but give up after 5 attempts | ||
| if retryCount <= 5 { | ||
| //increment the retry count by 1 | ||
| retryCount = retryCount + 1 | ||
|
|
||
| // set the paused status to false, and upsert the retry-count into the annotations | ||
| mergePatch, err := json.Marshal(map[string]interface{}{ | ||
| "metadata": map[string]interface{}{ | ||
| "annotations": map[string]interface{}{ | ||
| "fastly.amazee.io/paused": nil, | ||
| "fastly.amazee.io/paused-retry-count": fmt.Sprintf("%d", retryCount), | ||
| }, | ||
| "labels": map[string]interface{}{ | ||
| "fastly.amazee.io/paused": "false", | ||
| }, | ||
| }, | ||
| }) | ||
| if err != nil { | ||
| opLog.Info(fmt.Sprintf("Unable to create mergepatch for %s, error was: %v", ingress.ObjectMeta.Name, err)) | ||
| continue | ||
| } | ||
| // patch the ingress so that the controller will attemp to run through its process | ||
| if err := h.Client.Patch(context.Background(), &ingress, client.ConstantPatch(types.MergePatchType, mergePatch)); err != nil { | ||
| opLog.Info(fmt.Sprintf("Unable to patch ingress %s, error was: %v", ingress.ObjectMeta.Name, err)) | ||
| continue | ||
| } | ||
| opLog.Info(fmt.Sprintf("Patched ingress %s", ingress.ObjectMeta.Name)) | ||
| } | ||
| } | ||
| } | ||
| } | ||
| } | ||
| return | ||
| } |
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