[FFM-9476]: Optimize memory usage

evaluation/evaluator.go

@@ -2,6 +2,7 @@ package evaluation
 
 import (
 	"encoding/json"
+	"errors"
 	"fmt"
 	"regexp"
 	"sort"
@@ -27,11 +28,16 @@ const (
 	equalSensitiveOperator = "equal_sensitive"
 )
 
+var (
+	ErrNilFlag = errors.New("flag is nil")
+)
+
 // Query provides methods for segment and flag retrieval
 type Query interface {
 	GetSegment(identifier string) (rest.Segment, error)
 	GetFlag(identifier string) (rest.FeatureConfig, error)
 	GetFlags() ([]rest.FeatureConfig, error)
+	GetFlagMap() (map[string]*rest.FeatureConfig, error)
 }
 
 // FlagVariations list of FlagVariations
@@ -329,7 +335,7 @@ func (e Evaluator) EvaluateAll(target *Target) (FlagVariations, error) {
 // takes uses feature store.List function to get all the flags.
 func (e Evaluator) evaluateAll(target *Target) ([]FlagVariation, error) {
 	var variations []FlagVariation
-	flags, err := e.query.GetFlags()
+	flags, err := e.query.GetFlagMap()
 	if err != nil {
 		return variations, err
 	}
@@ -358,29 +364,32 @@ func (e Evaluator) evaluate(identifier string, target *Target) (FlagVariation, e
 		return FlagVariation{}, err
 	}
 
-	variation, err := e.getVariationForTheFlag(flag, target)
+	variation, err := e.getVariationForTheFlag(&flag, target)
 	if err != nil {
 		return FlagVariation{}, err
 	}
 	return FlagVariation{flag.Feature, flag.Kind, variation}, nil
 }
 
 // evaluates the flag and returns a proper variation.
-func (e Evaluator) getVariationForTheFlag(flag rest.FeatureConfig, target *Target) (rest.Variation, error) {
+func (e Evaluator) getVariationForTheFlag(flag *rest.FeatureConfig, target *Target) (rest.Variation, error) {
+	if flag == nil {
+		return rest.Variation{}, ErrNilFlag
+	}
 
 	if flag.Prerequisites != nil {
-		prereq, err := e.checkPreRequisite(&flag, target)
+		prereq, err := e.checkPreRequisite(flag, target)
 		if err != nil || !prereq {
 			return findVariation(flag.Variations, flag.OffVariation)
 		}
 	}
-	variation, err := e.evaluateFlag(flag, target)
+	variation, err := e.evaluateFlag(*flag, target)
 	if err != nil {
 		return rest.Variation{}, err
 	}
 	if e.postEvalCallback != nil {
 		data := PostEvalData{
-			FeatureConfig: &flag,
+			FeatureConfig: flag,
 			Target:        target,
 			Variation:     &variation,
 		}

evaluation/evaluator_test.go

@@ -312,6 +312,14 @@ func (m TestRepository) GetFlags() ([]rest.FeatureConfig, error) {
 	return flags, nil
 }
 
+func (m TestRepository) GetFlagMap() (map[string]*rest.FeatureConfig, error) {
+	var flags map[string]*rest.FeatureConfig
+	for _, f := range m.flags {
+		flags[f.Feature] = &f
+	}
+	return flags, nil
+}
+
 func TestNewEvaluator(t *testing.T) {
 	noOpLogger := logger.NewNoOpLogger()
 	eval, _ := NewEvaluator(testRepo, nil, noOpLogger)

pkg/repository/repository.go

@@ -15,6 +15,7 @@ type Repository interface {
 	GetFlag(identifier string) (rest.FeatureConfig, error)
 	GetSegment(identifier string) (rest.Segment, error)
 	GetFlags() ([]rest.FeatureConfig, error)
+	GetFlagMap() (map[string]*rest.FeatureConfig, error)
 
 	SetFlag(featureConfig rest.FeatureConfig, initialLoad bool)
 	SetFlags(initialLoad bool, envID string, featureConfig ...rest.FeatureConfig)
@@ -120,6 +121,11 @@ func (r FFRepository) GetFlags() ([]rest.FeatureConfig, error) {
 	return []rest.FeatureConfig{}, nil
 }
 
+// GetFlagMap returns all flags as a mao /* Not Implemented *.
+func (r FFRepository) GetFlagMap() (map[string]*rest.FeatureConfig, error) {
+	return map[string]*rest.FeatureConfig{}, nil
+}
+
 func (r FFRepository) getSegmentAndCache(identifier string, cacheable bool) (rest.Segment, error) {
 	segmentKey := formatSegmentKey(identifier)
 	flag, ok := r.cache.Get(segmentKey)