extend enum support (#144)

Author: gpop63

pkg/genlib/generator_interface.go

@@ -9,16 +9,17 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
-	"github.com/Pallinder/go-randomdata"
-	"github.com/elastic/elastic-integration-corpus-generator-tool/pkg/genlib/config"
-	"github.com/elastic/elastic-integration-corpus-generator-tool/pkg/genlib/fields"
 	"math"
 	"regexp"
 	"strconv"
 	"strings"
 	"sync"
 	"testing"
 	"time"
+
+	"github.com/Pallinder/go-randomdata"
+	"github.com/elastic/elastic-integration-corpus-generator-tool/pkg/genlib/config"
+	"github.com/elastic/elastic-integration-corpus-generator-tool/pkg/genlib/fields"
 )
 
 var timeNowToBind time.Time
@@ -970,6 +971,23 @@ func bindLongWithReturn(fieldCfg ConfigField, field Field, fieldMap map[string]a
 		return err
 	}
 
+	if len(fieldCfg.Enum) > 0 {
+		var emitF emitF
+		idx := customRand.Intn(len(fieldCfg.Enum))
+		f, err := strconv.ParseInt(fieldCfg.Enum[idx], 10, 64)
+		if err != nil {
+			return fmt.Errorf("field %s enum value is not a long: %w", fieldCfg.Name, err)
+		}
+
+		emitF = func(state *genState) any {
+			return f
+		}
+
+		fieldMap[field.Name] = emitF
+
+		return nil
+	}
+
 	if fieldCfg.Counter {
 		var emitF emitF
 
@@ -1072,6 +1090,23 @@ func bindDoubleWithReturn(fieldCfg ConfigField, field Field, fieldMap map[string
 		return err
 	}
 
+	if len(fieldCfg.Enum) > 0 {
+		var emitF emitF
+		idx := customRand.Intn(len(fieldCfg.Enum))
+		f, err := strconv.ParseFloat(fieldCfg.Enum[idx], 64)
+		if err != nil {
+			return fmt.Errorf("field %s enum value is not a double: %w", fieldCfg.Name, err)
+		}
+
+		emitF = func(state *genState) any {
+			return f
+		}
+
+		fieldMap[field.Name] = emitF
+
+		return nil
+	}
+
 	if fieldCfg.Counter {
 		var emitF emitF