added filter for vector lengths

Author: darkcoderrises

posting/index.go

@@ -316,7 +316,7 @@ func (mp *MutationPipeline) ProcessList(ctx context.Context, pipeline *Predicate
 	}
 
 	if info.count {
-		if err := mp.ProcessCount(ctx, pipeline, &postings, info); err != nil {
+		if err := mp.ProcessCount(ctx, pipeline, &postings, info, false); err != nil {
 			return err
 		}
 	}
@@ -378,9 +378,15 @@ func (mp *MutationPipeline) ProcessReverse(ctx context.Context, pipeline *Predic
 	}
 
 	if info.count {
-		newInfo := info
-		newInfo.isList = true
-		return mp.ProcessCount(ctx, pipeline, &reverseredMap, newInfo)
+		newInfo := predicateInfo{
+			isList:     true,
+			index:      info.index,
+			reverse:    info.reverse,
+			count:      info.count,
+			noConflict: info.noConflict,
+			hasUpsert:  info.hasUpsert,
+		}
+		return mp.ProcessCount(ctx, pipeline, &reverseredMap, newInfo, true)
 	}
 
 	for uid, pl := range reverseredMap {
@@ -469,9 +475,9 @@ func (txn *Txn) addConflictKeyWithUid(key []byte, pl *pb.PostingList, hasUpsert
 	}
 }
 
-func (mp *MutationPipeline) ProcessCount(ctx context.Context, pipeline *PredicatePipeline, postings *map[uint64]*pb.PostingList, info predicateInfo) error {
+func (mp *MutationPipeline) ProcessCount(ctx context.Context, pipeline *PredicatePipeline, postings *map[uint64]*pb.PostingList, info predicateInfo, isReverseEdge bool) error {
 	dataKey := x.DataKey(pipeline.attr, 0)
-	if info.reverse {
+	if isReverseEdge {
 		dataKey = x.ReverseKey(pipeline.attr, 0)
 	}
 	edge := pb.DirectedEdge{
@@ -548,7 +554,7 @@ func (mp *MutationPipeline) ProcessCount(ctx context.Context, pipeline *Predicat
 
 	for c, pl := range countMap {
 		//fmt.Println("COUNT", c, pl)
-		ck := x.CountKey(pipeline.attr, uint32(c), info.reverse)
+		ck := x.CountKey(pipeline.attr, uint32(c), isReverseEdge)
 		if newPl, err := mp.txn.AddDelta(string(ck), *pl); err != nil {
 			return err
 		} else {
@@ -660,7 +666,7 @@ func (mp *MutationPipeline) ProcessSingle(ctx context.Context, pipeline *Predica
 
 	if info.count {
 		// Count should take care of updating the posting list
-		return mp.ProcessCount(ctx, pipeline, &postings, info)
+		return mp.ProcessCount(ctx, pipeline, &postings, info, false)
 	}
 
 	baseKey := string(dataKey[:len(dataKey)-8]) // Avoid repeated conversion