feat(mutation): mutation now shows changed UIDs

unigraph-dev · Mar 18, 2022 · 999d68c · 999d68c
1 parent 8f0c539
commit 999d68c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/dgraph/cmd/alpha/http.go b/dgraph/cmd/alpha/http.go
@@ -437,6 +437,7 @@ func mutationHandler(w http.ResponseWriter, r *http.Request) {
 	resp.Latency.ParsingNs = uint64(parseEnd.Sub(parseStart).Nanoseconds())
 	e := query.Extensions{
 		Txn:     resp.Txn,
+		Metrics: resp.Metrics,
 		Latency: resp.Latency,
 	}
 	sort.Strings(e.Txn.Keys)

diff --git a/edgraph/server.go b/edgraph/server.go
@@ -816,6 +816,11 @@ func (s *Server) doMutate(ctx context.Context, qc *queryContext, resp *api.Respo
 		cost := uint64(len(newUids) + len(edges))
 		resp.Metrics.NumUids["mutation_cost"] = cost
 		resp.Metrics.NumUids["_total"] = resp.Metrics.NumUids["_total"] + cost
+
+		for _, edge := range m.Edges {
+			uidKey := fmt.Sprintf("_u-0x%x", edge.Entity)
+			resp.Metrics.NumUids[uidKey] = 1
+		}
 	}
 	if !qc.req.CommitNow {
 		calculateMutationMetrics()