Support both 1 and 2 args for Loop

lucapette · Aug 21, 2017 · 736c246 · 736c246
1 parent 436dd2c
commit 736c246
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pkg/fakedata/template.go b/pkg/fakedata/template.go
@@ -18,12 +18,14 @@ func newTemplateFactory() *templateFactory {
 
 func (tf templateFactory) getFunctions() template.FuncMap {
 	funcMap := template.FuncMap{
-		"Loop": func(min int, max int) []int {
+		"Loop": func(minmax ...int) []int {
 			var size int;
 
-			if max == 0 {
-				size = min;
+			if len(minmax) == 1 {
+				size = minmax[0];
 			} else {
+				min := minmax[0];
+				max := minmax[1];
 				size = rand.Intn(max - min) + min;
 			}