fix(Dgraph): Forward task status requests to correct Alpha (#7774)

graphql/admin/backup.go

@@ -62,9 +62,9 @@ func resolveBackup(ctx context.Context, m schema.Mutation) (*resolve.Resolved, b
 		return resolve.EmptyResult(m, err), false
 	}
 
-	msg := fmt.Sprintf("Backup queued with ID %s", taskId)
+	msg := fmt.Sprintf("Backup queued with ID %#x", taskId)
 	data := response("Success", msg)
-	data["taskId"] = taskId
+	data["taskId"] = fmt.Sprintf("%#x", taskId)
 	return resolve.DataResult(
 		m,
 		map[string]interface{}{m.Name(): data},

graphql/admin/export.go

@@ -97,9 +97,9 @@ func resolveExport(ctx context.Context, m schema.Mutation) (*resolve.Resolved, b
 		return resolve.EmptyResult(m, err), false
 	}
 
-	msg := fmt.Sprintf("Export queued with ID %s", taskId)
+	msg := fmt.Sprintf("Export queued with ID %#x", taskId)
 	data := response("Success", msg)
-	data["taskId"] = taskId
+	data["taskId"] = fmt.Sprintf("%#x", taskId)
 	return resolve.DataResult(
 		m,
 		map[string]interface{}{m.Name(): data},

graphql/admin/task.go

@@ -20,30 +20,42 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
+	"strconv"
 	"time"
 
 	"github.com/dgraph-io/dgraph/graphql/resolve"
 	"github.com/dgraph-io/dgraph/graphql/schema"
+	"github.com/dgraph-io/dgraph/protos/pb"
 	"github.com/dgraph-io/dgraph/worker"
+	"github.com/pkg/errors"
 )
 
 type taskInput struct {
 	Id string
 }
 
 func resolveTask(ctx context.Context, q schema.Query) *resolve.Resolved {
+	// Get Task ID.
 	input, err := getTaskInput(q)
 	if err != nil {
 		return resolve.EmptyResult(q, err)
 	}
 	if input.Id == "" {
 		return resolve.EmptyResult(q, fmt.Errorf("task ID is missing"))
 	}
+	taskId, err := strconv.ParseUint(input.Id, 0, 64)
+	if err != nil {
+		err = errors.Wrapf(err, "invalid task ID: %s", input.Id)
+		return resolve.EmptyResult(q, err)
+	}
 
-	meta, err := worker.Tasks.Get(input.Id)
+	// Get TaskMeta from network.
+	req := &pb.TaskStatusRequest{TaskId: taskId}
+	resp, err := worker.TaskStatusOverNetwork(context.Background(), req)
 	if err != nil {
 		return resolve.EmptyResult(q, err)
 	}
+	meta := worker.TaskMeta(resp.GetTaskMeta())
 	return resolve.DataResult(
 		q,
 		map[string]interface{}{q.Name(): map[string]interface{}{