feature: add SetCategories func in action model

Signed-off-by: xcaspar <[email protected]>
chaosblade-io · Mar 5, 2021 · 2c541c4 · 2c541c4
1 parent a444216
commit 2c541c4
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/spec/model.go b/spec/model.go
@@ -85,6 +85,9 @@ type ExpActionCommandSpec interface {
 
 	// Scenario categories
 	Categories() []string
+
+	// SetCategories
+	SetCategories(categories []string)
 }
 
 type ExpFlagSpec interface {
@@ -208,6 +211,10 @@ func (b *BaseExpActionCommandSpec) Categories() []string {
 	return b.ActionCategories
 }
 
+func (b *BaseExpActionCommandSpec) SetCategories(categories []string) {
+	b.ActionCategories = categories
+}
+
 // ActionModel for yaml file
 type ActionModel struct {
 	ActionName       string    `yaml:"action"`
@@ -282,6 +289,10 @@ func (am *ActionModel) Categories() []string {
 	return am.ActionCategories
 }
 
+func (am *ActionModel) SetCategories(categories []string) {
+	am.ActionCategories = categories
+}
+
 type ExpPrepareModel struct {
 	PrepareType     string    `yaml:"type"`
 	PrepareFlags    []ExpFlag `yaml:"flags"`