Function findGrootAndGuardians

dgraphtest/load.go:283–327

findGrootAndGuardians returns the UIDs of groot user and guardians group

(c *LocalCluster)

// findGrootAndGuardians returns the UIDs of groot user and guardians group
func findGrootAndGuardians(c *LocalCluster) (string, string, error) {
	gc, cleanup, err := c.Client()
	if err != nil {
		return "", "", errors.Wrapf(err, "error creating grpc client")
	}
	defer cleanup()

	if c.conf.acl {
		ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
		defer cancel()
		err = gc.LoginIntoNamespace(ctx, dgraphapi.DefaultUser, dgraphapi.DefaultPassword, x.RootNamespace)
		if err != nil {
			return "", "", errors.Wrapf(err, "error logging in as groot")
		}
	}

	query := `{
		q(func: eq(dgraph.xid, "groot")){
			uid
			dgraph.user.group @filter(eq(dgraph.xid, "guardians")) {
				uid
			}
		}
	}`
	resp, err := gc.Query(query)
	if err != nil {
		return "", "", errors.Wrapf(err, "error querying groot & guardians UIDs")
	}
	var result struct {
		Q []struct {
			Uid    string
			Groups []struct {
				Uid string
			} `json:"dgraph.user.group"`
		}
	}
	if err := json.Unmarshal(resp.Json, &result); err != nil {
		return "", "", errors.Wrapf(err, "error unmarshalling resp: [%v]", string(resp.Json))
	}
	if len(result.Q) != 1 || result.Q[0].Uid == "" ||
		len(result.Q[0].Groups) != 1 || result.Q[0].Groups[0].Uid == "" {
		return "", "", errors.Errorf("unable to find groot & guardians, resp: [%+v]", resp)
	}
	return result.Q[0].Uid, result.Q[0].Groups[0].Uid, nil
}

// modifyACLEntries replaces groot's and guardians' UIDs
// in the exported files to the UIDs in this dgraph cluster.