mattermost-community-enterprise/vendor/github.com/splitio/go-split-commons/v7/engine/grammar/equaltoset.go

package grammar

import (
	"fmt"

	"github.com/splitio/go-toolkit/v5/datastructures/set"
)

// EqualToSetMatcher matches if the set supplied to the getTreatment is equal to the one in the feature flag
type EqualToSetMatcher struct {
	Matcher
	comparisonSet *set.ThreadUnsafeSet
}

// Match returns true if the match provided and the one in the feature flag are equal
func (m *EqualToSetMatcher) Match(key string, attributes map[string]interface{}, bucketingKey *string) bool {
	matchingKey, err := m.matchingKey(key, attributes)
	if err != nil {
		m.logger.Warning(fmt.Sprintf("EqualToSetMatcher: %s", err.Error()))
		return false
	}

	conv, ok := matchingKey.([]string)
	if !ok {
		m.logger.Error("EqualToSetMatcher: Cannot type assert to []string")
		return false
	}

	matchingSet := set.NewSet()
	for _, x := range conv {
		matchingSet.Add(x)
	}

	return matchingSet.IsEqual(m.comparisonSet)

}

// NewEqualToSetMatcher returns a pointer to a new instance of EqualToSetMatcher
func NewEqualToSetMatcher(negate bool, setItems []string, attributeName *string) *EqualToSetMatcher {
	setObj := set.NewSet()
	for _, item := range setItems {
		setObj.Add(item)
	}

	return &EqualToSetMatcher{
		Matcher: Matcher{
			negate:        negate,
			attributeName: attributeName,
		},
		comparisonSet: setObj,
	}
}