Method executeGraphQLRoleQuery

core/role_query.go:73–131

(
	c context.Context,
	conn *sql.Conn,
	vmap map[string]json.RawMessage,
	rc *RequestConfig,
)

}

func (gj *graphjinEngine) executeGraphQLRoleQuery(
	c context.Context,
	conn *sql.Conn,
	vmap map[string]json.RawMessage,
	rc *RequestConfig,
) (role string, err error) {
	role = "user"

	pdb := gj.primaryDB()
	if pdb == nil || pdb.psqlCompiler == nil {
		return "", fmt.Errorf("roles_query: primary database not initialized")
	}
	if gj.roleGraphQLStmt.qc == nil {
		return "", fmt.Errorf("roles_query: graphql roles_query not initialized")
	}

	vars := roleQueryVars(vmap, gj.roleGraphQLStmt.qc.Vars)
	ar, err := gj.argList(c,
		gj.roleGraphQLStmt.md,
		vars,
		rc,
		false,
		pdb.psqlCompiler)
	if err != nil {
		return "", err
	}

	query, queryArgs, err := prepareQueryArgsForDB(pdb.dbtype, gj.roleGraphQLStmt.sql, ar.values)
	if err != nil {
		return "", err
	}

	raw, err := scanJSONRow(c, pdb.dbtype, conn, requestTx(rc), query, queryArgs)
	if err == sql.ErrNoRows {
		return role, nil
	}
	if err != nil {
		return "", err
	}

	attrs, empty, err := extractRoleQueryAttrs(raw)
	if err != nil {
		return "", err
	}
	if empty {
		return role, nil
	}

	for _, rm := range gj.roleGraphQLMatches {
		ok, err := rm.expr.eval(attrs)
		if err != nil {
			return "", fmt.Errorf("roles_query: role %q match failed: %w", rm.role, err)
		}
		if ok {
			return rm.role, nil
		}
	}
	return role, nil