Include the total number of touched nodes in the query metrics. (#5073)

Total is stored under a key called "_total".
dgraph-io · Apr 1, 2020 · e3118ec · e3118ec
1 parent a3f9647
commit e3118ec
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/edgraph/server.go b/edgraph/server.go
@@ -987,6 +987,11 @@ func processQuery(ctx context.Context, qc *queryContext) (*api.Response, error)
 	resp.Metrics = &api.Metrics{
 		NumUids: er.Metrics,
 	}
+	var total uint64
+	for _, num := range resp.Metrics.NumUids {
+		total += num
+	}
+	resp.Metrics.NumUids["_total"] = total
 
 	return resp, err
 }

diff --git a/query/query4_test.go b/query/query4_test.go
@@ -1545,4 +1545,5 @@ func TestNumUids(t *testing.T) {
 	metrics := processQueryForMetrics(t, query)
 	require.Equal(t, metrics.NumUids["friend"], uint64(10))
 	require.Equal(t, metrics.NumUids["name"], uint64(16))
+	require.Equal(t, metrics.NumUids["_total"], uint64(26))
 }