-
Notifications
You must be signed in to change notification settings - Fork 151
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: support idling and unidling from api with idled status on envir…
…onment
- Loading branch information
1 parent
38ca596
commit 94e80d3
Showing
10 changed files
with
614 additions
and
429 deletions.
There are no files selected for viewing
Large diffs are not rendered by default.
Oops, something went wrong.
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,85 @@ | ||
| package handler | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
| "log" | ||
| "time" | ||
|
|
||
| mq "github.com/cheshir/go-mq/v2" | ||
| "github.com/uselagoon/machinery/api/lagoon" | ||
| lclient "github.com/uselagoon/machinery/api/lagoon/client" | ||
| "github.com/uselagoon/machinery/api/schema" | ||
| "github.com/uselagoon/machinery/utils/jwt" | ||
| ) | ||
|
|
||
| func (m *Messenger) handleIdling(ctx context.Context, messageQueue *mq.MessageQueue, message *schema.LagoonMessage, messageID string) error { | ||
| prefix := fmt.Sprintf("(messageid:%s) %s: ", messageID, message.Namespace) | ||
| log.Println(fmt.Sprintf("%sreceived idling environment status update", prefix)) | ||
| // generate a lagoon token with a expiry of 60 seconds from now | ||
| token, err := jwt.GenerateAdminToken(m.LagoonAPI.TokenSigningKey, m.LagoonAPI.JWTAudience, m.LagoonAPI.JWTSubject, m.LagoonAPI.JWTIssuer, time.Now().Unix(), 60) | ||
| if err != nil { | ||
| // the token wasn't generated | ||
| if m.EnableDebug { | ||
| log.Println(fmt.Sprintf("ERROR: unable to generate token: %v", err)) | ||
| } | ||
| return nil | ||
| } | ||
| // set up a lagoon client for use in the following process | ||
| l := lclient.New(m.LagoonAPI.Endpoint, "actions-handler", &token, false) | ||
| var environmentID uint | ||
| // determine the environment id from the message | ||
| if message.Meta.ProjectID == nil && message.Meta.EnvironmentID == nil { | ||
| project, err := lagoon.GetMinimalProjectByName(ctx, message.Meta.Project, l) | ||
| if err != nil { | ||
| // send the log to the lagoon-logs exchange to be processed | ||
| m.toLagoonLogs(messageQueue, map[string]interface{}{ | ||
| "severity": "error", | ||
| "event": fmt.Sprintf("actions-handler:%s:failed", "updateEnvironment"), | ||
| "meta": project, | ||
| "message": err.Error(), | ||
| }) | ||
| if m.EnableDebug { | ||
| log.Println(fmt.Sprintf("%sERROR: unable to get project - %v", prefix, err)) | ||
| } | ||
| return err | ||
| } | ||
| environment, err := lagoon.GetEnvironmentByName(ctx, message.Meta.Environment, project.ID, l) | ||
| if err != nil { | ||
| // send the log to the lagoon-logs exchange to be processed | ||
| m.toLagoonLogs(messageQueue, map[string]interface{}{ | ||
| "severity": "error", | ||
| "event": fmt.Sprintf("actions-handler:%s:failed", "updateEnvironment"), | ||
| "meta": project, | ||
| "message": err.Error(), | ||
| }) | ||
| if m.EnableDebug { | ||
| log.Println(fmt.Sprintf("%sERROR: unable to get environment - %v", prefix, err)) | ||
| } | ||
| return err | ||
| } | ||
| environmentID = environment.ID | ||
| } else { | ||
| // pull the id from the message | ||
| environmentID = *message.Meta.EnvironmentID | ||
| } | ||
| updateEnvironmentPatch := schema.UpdateEnvironmentPatchInput{ | ||
| Idled: &message.Idled, | ||
| } | ||
| updateEnvironment, err := lagoon.UpdateEnvironment(ctx, environmentID, updateEnvironmentPatch, l) | ||
| if err != nil { | ||
| // send the log to the lagoon-logs exchange to be processed | ||
| m.toLagoonLogs(messageQueue, map[string]interface{}{ | ||
| "severity": "error", | ||
| "event": fmt.Sprintf("actions-handler:%s:failed", "updateDeployment"), | ||
| "meta": updateEnvironment, | ||
| "message": err.Error(), | ||
| }) | ||
| if m.EnableDebug { | ||
| log.Println(fmt.Sprintf("%sERROR: unable to update environment - %v", prefix, err)) | ||
| } | ||
| return err | ||
| } | ||
| log.Println(fmt.Sprintf("%supdated environment", prefix)) | ||
| return nil | ||
| } |
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
21 changes: 21 additions & 0 deletions
21
services/api/database/migrations/20231215000000_environment_idling_status.js
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,21 @@ | ||
| /** | ||
| * @param { import("knex").Knex } knex | ||
| * @returns { Promise<void> } | ||
| */ | ||
| exports.up = async function(knex) { | ||
| return knex.schema | ||
| .alterTable('environment', (table) => { | ||
| table.boolean('idled').notNullable().defaultTo(0); | ||
| }) | ||
| }; | ||
|
|
||
| /** | ||
| * @param { import("knex").Knex } knex | ||
| * @returns { Promise<void> } | ||
| */ | ||
| exports.down = async function(knex) { | ||
| return knex.schema | ||
| .alterTable('environment', (table) => { | ||
| table.dropColumn('idled'); | ||
| }) | ||
| }; |
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