fix: Handle location in PostNewVersion method

stenic · Dec 14, 2022 · dc74d58 · dc74d58
1 parent 60eb516
commit dc74d58
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/internal/agent/agent.go b/internal/agent/agent.go
@@ -61,14 +61,15 @@ func Run(endpoint, namespace, location string) {
 func getNotifyFunc(lc client.LedgerClient, location string) kubernetes.NotifyFunc {
 	return func(events []kubernetes.Event) error {
 		for _, e := range events {
+			if e.Location == "" {
+				e.Location = location
+			}
 			logrus.WithFields(logrus.Fields{
 				"application": e.Application,
+				"location":    e.Location,
 				"environment": e.Environment,
 				"version":     e.Version,
 			}).Info("Notifying ledger")
-			if e.Location == "" {
-				e.Location = location
-			}
 			if err := lc.PostNewVersion(e.Application, e.Location, e.Environment, e.Version); err != nil {
 				logrus.Error(err)
 			}

diff --git a/internal/client/client.go b/internal/client/client.go
@@ -24,12 +24,13 @@ func (c *LedgerClient) PostNewVersion(app, location, env, version string) error
 				createVersion(input: {
 					application: "%s",
 					environment: "%s",
-					version: "%s"
+					version: "%s",
+					location: "%s",
 				}) {
 					id
 				}
 			}
-		`, app, env, version),
+		`, app, env, version, location),
 	}
 	jsonValue, _ := json.Marshal(jsonData)
 	r := regexp.MustCompile(`(\\[tn]|\s+)`)