exprlib.go 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  1. package exprhelpers
  2. import (
  3. "bufio"
  4. "fmt"
  5. "os"
  6. "path"
  7. "regexp"
  8. "strconv"
  9. "strings"
  10. log "github.com/sirupsen/logrus"
  11. )
  12. var dataFile map[string][]string
  13. var dataFileRegex map[string][]*regexp.Regexp
  14. func Atof(x string) float64 {
  15. log.Debugf("debug atof %s", x)
  16. ret, err := strconv.ParseFloat(x, 64)
  17. if err != nil {
  18. log.Warningf("Atof : can't convert float '%s' : %v", x, err)
  19. }
  20. return ret
  21. }
  22. func StartsWith(s string, pref string) bool {
  23. return strings.HasPrefix(s, pref)
  24. }
  25. func EndsWith(s string, suff string) bool {
  26. return strings.HasSuffix(s, suff)
  27. }
  28. func Upper(s string) string {
  29. return strings.ToUpper(s)
  30. }
  31. func GetExprEnv(ctx map[string]interface{}) map[string]interface{} {
  32. var ExprLib = map[string]interface{}{
  33. "Atof": Atof,
  34. "StartsWith": StartsWith,
  35. "EndsWith": EndsWith,
  36. "JsonExtract": JsonExtract,
  37. "JsonExtractLib": JsonExtractLib,
  38. "File": File,
  39. "RegexpInFile": RegexpInFile,
  40. "Upper": Upper,
  41. }
  42. for k, v := range ctx {
  43. ExprLib[k] = v
  44. }
  45. return ExprLib
  46. }
  47. func Init() error {
  48. log.Infof("Expr helper initiated")
  49. dataFile = make(map[string][]string)
  50. dataFileRegex = make(map[string][]*regexp.Regexp)
  51. return nil
  52. }
  53. func FileInit(fileFolder string, filename string, fileType string) error {
  54. filepath := path.Join(fileFolder, filename)
  55. file, err := os.Open(filepath)
  56. if err != nil {
  57. return err
  58. }
  59. defer file.Close()
  60. if _, ok := dataFile[filename]; !ok {
  61. dataFile[filename] = []string{}
  62. }
  63. if fileType == "" {
  64. fileType = "string"
  65. }
  66. scanner := bufio.NewScanner(file)
  67. for scanner.Scan() {
  68. switch fileType {
  69. case "regex", "regexp":
  70. dataFileRegex[filename] = append(dataFileRegex[filename], regexp.MustCompile(scanner.Text()))
  71. case "string":
  72. dataFile[filename] = append(dataFile[filename], scanner.Text())
  73. default:
  74. return fmt.Errorf("unknown data type '%s' for : '%s'", fileType, filename)
  75. }
  76. }
  77. if err := scanner.Err(); err != nil {
  78. return err
  79. }
  80. return nil
  81. }
  82. func File(filename string) []string {
  83. if _, ok := dataFile[filename]; ok {
  84. return dataFile[filename]
  85. }
  86. log.Errorf("file '%s' not found for expr library", filename)
  87. return []string{}
  88. }
  89. func RegexpInFile(data string, filename string) bool {
  90. if _, ok := dataFileRegex[filename]; ok {
  91. for _, re := range dataFileRegex[filename] {
  92. if re.Match([]byte(data)) {
  93. return true
  94. }
  95. }
  96. } else {
  97. log.Errorf("file '%s' not found for expr library", filename)
  98. }
  99. return false
  100. }