Archived
1
0
Fork 0

feat: add filter on price range

WIP

Signed-off-by: Julien Riou <julien@riou.xyz>
This commit is contained in:
Julien Riou 2021-05-20 18:03:45 +02:00
parent 27d9f4976e
commit ba791435b7
No known key found for this signature in database
GPG key ID: FF42D23B580C89F7
2 changed files with 93 additions and 0 deletions

48
filter_range.go Normal file
View file

@ -0,0 +1,48 @@
package main
import (
"regexp"
log "github.com/sirupsen/logrus"
)
// RangeFilter to store the pattern to match product model and price limits
type RangeFilter struct {
model *regexp.Regexp
min float64
max float64
}
// NewRangeFilter to create a RangeFilter
func NewRangeFilter(regex string, min float64, max float64) (*RangeFilter, error) {
var err error
var compiledRegex *regexp.Regexp
log.Debugf("compiling model filter regex")
if regex != "" {
compiledRegex, err = regexp.Compile(regex)
if err != nil {
return nil, err
}
}
return &RangeFilter{
model: compiledRegex,
min: min,
max: max,
}, nil
}
// Include returns false when a product name matches the model regex and price is outside of the range
// implements the Filter interface
func (f *RangeFilter) Include(product *Product) bool {
if f.model == nil {
return true
}
if f.model.MatchString(product.Name) && product.Price < f.min || product.Price > f.max {
log.Debugf("product %s excluded because price for the model is outside of the range [%.2f-%.2f]", product.Name, f.min, f.max)
return false
}
log.Debugf("product %s included because price range filter is not applicable", product.Name)
return true
}