Add filter for workload type (#321)

* add filter for workload type * resolve pr comment
threefoldtecharchive · May 31, 2021 · 1ce2486 · 1ce2486
1 parent 9c6f130
commit 1ce2486
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/pkg/workloads/types/workload.go b/pkg/workloads/types/workload.go
@@ -43,7 +43,15 @@ func ApplyQueryFilterWorkload(r *http.Request, filter WorkloadFilter) (WorkloadF
 		if err != nil {
 			return nil, errors.Wrap(err, "next_action should be an integer")
 		}
-		filter = filter.WithNextAction(generated.NextActionEnum(nextAction))
+		filter = filter.WithNextAction(workloads.NextActionEnum(nextAction))
+	}
+	sWorkloadType := r.FormValue("workload_type")
+	if len(sWorkloadType) != 0 {
+		workloadType, err := strconv.ParseUint(sWorkloadType, 10, 8)
+		if err != nil {
+			return nil, errors.Wrap(err, "workload_type should be an integer")
+		}
+		filter = filter.WithWorkloadType(workloads.WorkloadTypeEnum(workloadType))
 	}
 	return filter, nil
 }