Add missing sync=true during cloud role update (#334)

astronomer · May 15, 2020 · c4fdeda · c4fdeda
1 parent 2ec2450
commit c4fdeda
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/deployment/deployment.go b/deployment/deployment.go
@@ -149,9 +149,16 @@ func List(ws string, all bool, client *houston.Client, out io.Writer) error {
 
 // Update an airflow deployment
 func Update(id, cloudRole string, args map[string]string) error {
+	vars := map[string]interface{}{"deploymentId": id, "payload": args, "cloudRole": cloudRole}
+
+	// sync with commander only when we have cloudRole
+	if cloudRole != "" {
+		vars["sync"] = true
+	}
+
 	req := houston.Request{
 		Query:     houston.DeploymentUpdateRequest,
-		Variables: map[string]interface{}{"deploymentId": id, "payload": args, "cloudRole": cloudRole},
+		Variables: vars,
 	}
 
 	r, err := req.Do()

diff --git a/houston/queries.go b/houston/queries.go
@@ -90,8 +90,8 @@ var (
 	}`
 
 	DeploymentUpdateRequest = `
-	mutation UpdateDeployment($deploymentId: Uuid!, $payload: JSON!, $cloudRole: String) {
-		updateDeployment(deploymentUuid: $deploymentId, payload: $payload, cloudRole: $cloudRole) {
+	mutation UpdateDeployment($deploymentId: Uuid!, $payload: JSON!, $cloudRole: String, $sync: Boolean) {
+		updateDeployment(deploymentUuid: $deploymentId, payload: $payload, cloudRole: $cloudRole, sync: $sync) {
 			id
 			type
 			label