Update contains method

Author: nmayorsplit

storage/inmemory/mutexmap/rulebasedsegment.go

@@ -101,14 +101,16 @@ func (r *RuleBasedSegmentsStorageImpl) GetSegments() *set.ThreadUnsafeSet {
 	return segments
 }
 
-// Contains returns true or false if all the segment names are in the rule-based segments
+// Contains returns true or false if all the rule-based segment names are present
 func (r *RuleBasedSegmentsStorageImpl) Contains(ruleBasedSegmentNames []string) bool {
 	if len(ruleBasedSegmentNames) == 0 {
 		return false
 	}
-	segments := r.GetSegments()
+	r.mutex.RLock()
+	defer r.mutex.RUnlock()
 	for _, name := range ruleBasedSegmentNames {
-		if !segments.Has(name) {
+		_, exists := r.data[name]
+		if !exists {
 			return false
 		}
 	}

storage/inmemory/mutexmap/rulebasedsegment_test.go

@@ -79,8 +79,8 @@ func TestRuleBasedSegmentsStorage(t *testing.T) {
 	assert.True(t, segments.Has("excluded1"), "excluded1 should be in segments")
 
 	// Test Contains
-	assert.True(t, storage.Contains([]string{"segment1", "excluded1"}), "should contain segment1 and excluded1")
-	assert.True(t, storage.Contains([]string{"segment2"}), "should contain segment2")
+	assert.True(t, storage.Contains([]string{"rule1", "rule2"}), "should contain rule1 and rule2")
+	assert.True(t, storage.Contains([]string{"rule1"}), "should contain rule1")
 	assert.False(t, storage.Contains([]string{"nonexistent"}))
 
 	// Test Remove