added test fixed review

Author: darkcoderrises

dql/parser.go

@@ -2691,7 +2691,7 @@ func validKeyAtRoot(k string) bool {
 	switch k {
 	case "func", "orderasc", "orderdesc", "first", "offset", "after":
 		return true
-	case "from", "to", "numpaths", "minweight", "maxweight", "maxheapsize":
+	case "from", "to", "numpaths", "minweight", "maxweight", "maxfrontiersize":
 		// Specific to shortest path
 		return true
 	case "depth":

dql/parser_test.go

@@ -1302,7 +1302,7 @@ func TestParseQueryWithMultipleVar(t *testing.T) {
 func TestParseShortestPath(t *testing.T) {
 	query := `
 	{
-		shortest(from:0x0a, to:0x0b, numpaths: 3, minweight: 3, maxweight: 6, maxheapsize: 1) {
+		shortest(from:0x0a, to:0x0b, numpaths: 3, minweight: 3, maxweight: 6, maxfrontiersize: 1) {
 			friends
 			name
 		}
@@ -1317,7 +1317,7 @@ func TestParseShortestPath(t *testing.T) {
 	require.Equal(t, "3", res.Query[0].Args["numpaths"])
 	require.Equal(t, "3", res.Query[0].Args["minweight"])
 	require.Equal(t, "6", res.Query[0].Args["maxweight"])
-	require.Equal(t, "1", res.Query[0].Args["maxheapsize"])
+	require.Equal(t, "1", res.Query[0].Args["maxfrontiersize"])
 }
 
 func TestParseShortestPathWithUidVars(t *testing.T) {

query/query.go

@@ -165,8 +165,8 @@ type params struct {
 	MaxWeight float64
 	// MinWeight is the min weight allowed in a path returned by the shortest path algorithm.
 	MinWeight float64
-	// MaxHeapSize is the maximum size of the heap that can be made while querying shortest path
-	MaxHeapSize int
+	// MaxFrontierSize is the maximum size of the frontier that can be made while querying shortest path
+	MaxFrontierSize int64
 
 	// ExploreDepth is used by recurse and shortest path queries to specify the maximum graph
 	// depth to explore.
@@ -716,14 +716,14 @@ func (args *params) fill(gq *dql.GraphQuery) error {
 			args.MinWeight = -math.MaxFloat64
 		}
 
-		if v, ok := gq.Args["maxheapsize"]; ok {
-			maxHeapSize, err := strconv.ParseInt(v, 0, 64)
+		if v, ok := gq.Args["maxfrontiersize"]; ok {
+			maxfrontiersize, err := strconv.ParseInt(v, 0, 64)
 			if err != nil {
 				return err
 			}
-			args.MaxHeapSize = int(maxHeapSize)
+			args.MaxFrontierSize = maxfrontiersize
 		} else if !ok {
-			args.MaxHeapSize = math.MaxInt64
+			args.MaxFrontierSize = math.MaxInt64
 		}
 
 		if gq.ShortestPathArgs.From == nil || gq.ShortestPathArgs.To == nil {
@@ -2652,7 +2652,7 @@ func (sg *SubGraph) sortAndPaginateUsingVar(ctx context.Context) error {
 func isValidArg(a string) bool {
 	switch a {
 	case "numpaths", "from", "to", "orderasc", "orderdesc", "first", "offset", "after", "depth",
-		"minweight", "maxweight", "maxheapsize":
+		"minweight", "maxweight", "maxfrontiersize":
 		return true
 	}
 	return false

query/shortest.go

@@ -405,7 +405,7 @@ func runKShortestPaths(ctx context.Context, sg *SubGraph) ([]*SubGraph, error) {
 				hop:  item.hop + 1,
 				path: route{route: curPath},
 			}
-			if pq.Len() > sg.Params.MaxHeapSize {
+			if int64(pq.Len()) > sg.Params.MaxFrontierSize {
 				pq.Pop()
 			}
 			heap.Push(&pq, node)
@@ -561,7 +561,7 @@ func shortestPath(ctx context.Context, sg *SubGraph) ([]*SubGraph, error) {
 					cost: nodeCost,
 					hop:  item.hop + 1,
 				}
-				if pq.Len() > sg.Params.MaxHeapSize {
+				if int64(pq.Len()) > sg.Params.MaxFrontierSize {
 					pq.Pop()
 				}
 				heap.Push(&pq, node)

systest/shortest-path/graph.rdf.gz

@@ -0,0 +1,60 @@
+//go:build integration
+
+/*
+ * SPDX-FileCopyrightText: © Hypermode Inc. <hello@hypermode.com>
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package main
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/hypermodeinc/dgraph/v25/dgraphapi"
+	"github.com/hypermodeinc/dgraph/v25/dgraphtest"
+	"github.com/hypermodeinc/dgraph/v25/x"
+	"github.com/stretchr/testify/require"
+)
+
+func TestShortestPath(t *testing.T) {
+	conf := dgraphtest.NewClusterConfig().WithNumAlphas(1).WithNumZeros(1).WithReplicas(1).WithACL(time.Hour)
+	c, err := dgraphtest.NewLocalCluster(conf)
+	require.NoError(t, err)
+	defer func() { c.Cleanup(t.Failed()) }()
+	require.NoError(t, c.Start())
+
+	err = c.LiveLoad(dgraphtest.LiveOpts{
+		DataFiles:      []string{"graph.rdf.gz"},
+		SchemaFiles:    []string{"graph.schema.gz"},
+		GqlSchemaFiles: []string{},
+	})
+	require.NoError(t, err)
+
+	gc, cleanup, err := c.Client()
+	require.NoError(t, err)
+	defer cleanup()
+	require.NoError(t, gc.LoginIntoNamespace(context.Background(),
+		dgraphapi.DefaultUser, dgraphapi.DefaultPassword, x.GalaxyNamespace))
+
+	_, err = gc.Query(`
+	{
+		q(func: eq(guid, "85270d10-560e-4cc8-8703-4b4c563a2f4e")) {
+		 	a as uid
+  		}
+  		q1(func: eq(guid, "4a520068-80b6-42f2-9019-4e6ef8a02bb3")) {
+			b as uid
+  		}
+  
+ 		path as shortest(from: uid(a), to: uid(b), numpaths: 5, maxfrontiersize: 10000) {
+  			connected_to @facets(weight)
+ 		}
+
+ 		path(func: uid(path)) {
+   			uid
+ 		}
+	}
+	`)
+	require.NoError(t, err)
+}