feat/ansible_pb: Add IncludeRole implments 'ITask' in pkg/util/ansiblev2

yunionio · May 9, 2020 · faf3241 · faf3241
1 parent 331ba9b
commit faf3241
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/pkg/util/ansiblev2/playbook.go b/pkg/util/ansiblev2/playbook.go
@@ -104,6 +104,29 @@ func (t *ShellTask) MarshalYAML() (interface{}, error) {
 	return r, nil
 }
 
+type IncludeRole struct {
+	Name string
+	Tags []string
+	Vars map[string]interface{}
+	When string
+}
+
+func (r *IncludeRole) MarshalYAML() (interface{}, error) {
+	out := map[string]interface{}{
+		"include_role": map[string]interface{}{"name": r.Name},
+	}
+	if len(r.Tags) > 0 {
+		out["tags"] = r.Tags
+	}
+	if len(r.Vars) > 0 {
+		out["vars"] = r.Vars
+	}
+	if len(r.When) > 0 {
+		out["when"] = r.When
+	}
+	return out, nil
+}
+
 type Block struct {
 	Name          string
 	WithPlugin    string

diff --git a/pkg/util/ansiblev2/playbook_test.go b/pkg/util/ansiblev2/playbook_test.go
@@ -59,6 +59,14 @@ func TestPlaybookString(t *testing.T) {
 				"group": "root",
 			},
 		},
+		&IncludeRole{
+			Name: "foo_app_instance",
+			Vars: map[string]interface{}{
+				"dir": "/opt/a",
+			},
+			Tags: []string{"bar", "baz"},
+			When: "ansible_facts['os_family'] == 'RedHat'",
+		},
 	)
 	play.Hosts = "all"
 	configureBlock := NewBlock(