-
Notifications
You must be signed in to change notification settings - Fork 149
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
7bc11f0
commit 6a3293d
Showing
10 changed files
with
302 additions
and
17 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,94 @@ | ||
| package handler | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/base64" | ||
| "encoding/json" | ||
| "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" | ||
| ) | ||
|
|
||
| type Idled struct { | ||
| Idled bool `json:"idled"` | ||
| } | ||
|
|
||
| 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", m.LagoonAPI.Version, &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 | ||
| } | ||
| decodeData, _ := base64.StdEncoding.DecodeString(message.Meta.AdvancedData) | ||
| idled := &Idled{} | ||
| json.Unmarshal(decodeData, idled) | ||
| updateEnvironmentPatch := schema.UpdateEnvironmentPatchInput{ | ||
| Idled: &idled.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
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/20241127000000_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
Oops, something went wrong.