Examples with BLangReturn org.wso2.ballerinalang.compiler.tree.statements.BLangReturn used on opensource projects

Example 6 with BLangReturn

use of org.wso2.ballerinalang.compiler.tree.statements.BLangReturn in project ballerina by ballerina-lang.

the class IterableCodeDesugar method generateSimpleIteratorBlock.

private void generateSimpleIteratorBlock(IterableContext ctx, BLangFunction funcNode) {
    final Operation firstOperation = ctx.getFirstOperation();
    final DiagnosticPos pos = firstOperation.pos;
    // return result;
    if (isReturningIteratorFunction(ctx)) {
        createCounterVarDefStmt(funcNode, ctx);
        createResultVarDefStmt(funcNode, ctx);
    }
    // Create variables required.
    final List<BLangVariable> foreachVariables = createForeachVariables(ctx, ctx.getFirstOperation().argVar, funcNode);
    ctx.iteratorResultVariables = foreachVariables;
    final BLangForeach foreachStmt = ASTBuilderUtil.createForeach(pos, funcNode.body, ASTBuilderUtil.createVariableRef(pos, ctx.collectionVar.symbol), ASTBuilderUtil.createVariableRefList(pos, foreachVariables), ctx.foreachTypes);
    if (isReturningIteratorFunction(ctx)) {
        generateAggregator(foreachStmt.body, ctx);
        generateFinalResult(funcNode.body, ctx);
    }
    final BLangReturn returnStmt = ASTBuilderUtil.createReturnStmt(firstOperation.pos, funcNode.body);
    if (isReturningIteratorFunction(ctx)) {
        returnStmt.addExpression(ASTBuilderUtil.createVariableRef(pos, ctx.resultVar.symbol));
    }
}

Example 7 with BLangReturn

use of org.wso2.ballerinalang.compiler.tree.statements.BLangReturn in project ballerina by ballerina-lang.

the class CodeGenerator method visitReturnStatementsExprs.

private void visitReturnStatementsExprs(BLangReturn returnNode) {
    int i = 0;
    while (i < returnNode.exprs.size()) {
        BLangExpression expr = returnNode.exprs.get(i);
        this.genNode(expr, this.env);
        emit(this.typeTagToInstr(expr.type.tag), getOperand(i), expr.regIndex);
        i++;
    }
    generateFinallyInstructions(returnNode);
}

Example 8 with BLangReturn

use of org.wso2.ballerinalang.compiler.tree.statements.BLangReturn in project ballerina by ballerina-lang.

the class ASTBuilderUtil method createReturnStmt.

static BLangReturn createReturnStmt(DiagnosticPos pos, BLangBlockStmt target) {
    final BLangReturn returnStmt = (BLangReturn) TreeBuilder.createReturnNode();
    returnStmt.pos = pos;
    target.addStatement(returnStmt);
    return returnStmt;
}

Example 9 with BLangReturn

use of org.wso2.ballerinalang.compiler.tree.statements.BLangReturn in project ballerina by ballerina-lang.

the class Desugar method getSafeAssignErrorPattern.

private BLangMatchStmtPatternClause getSafeAssignErrorPattern(DiagnosticPos pos, BSymbol invokableSymbol) {
    // From here onwards we assume that this function has only one return type
    // Owner of the variable symbol must be an invokable symbol
    boolean noRetParams = ((BInvokableType) invokableSymbol.type).retTypes.isEmpty();
    boolean returnErrorType = false;
    if (!noRetParams) {
        BType retType = ((BInvokableType) invokableSymbol.type).retTypes.get(0);
        Set<BType> returnTypeSet = retType.tag == TypeTags.UNION ? ((BUnionType) retType).memberTypes : new HashSet<BType>() {

            {
                add(retType);
            }
        };
        returnErrorType = returnTypeSet.stream().anyMatch(type -> types.isAssignable(type, symTable.errStructType));
    }
    // Create the pattern to match the error type
    // 1) Create the pattern variable
    String patternFailureCaseVarName = GEN_VAR_PREFIX.value + "t_failure";
    BLangVariable patternFailureCaseVar = ASTBuilderUtil.createVariable(pos, patternFailureCaseVarName, symTable.errStructType, null, new BVarSymbol(0, names.fromString(patternFailureCaseVarName), this.env.scope.owner.pkgID, symTable.errStructType, this.env.scope.owner));
    // 2) Create the pattern block
    BLangVariableReference patternFailureCaseVarRef = ASTBuilderUtil.createVariableRef(pos, patternFailureCaseVar.symbol);
    BLangBlockStmt patternBlockFailureCase = (BLangBlockStmt) TreeBuilder.createBlockNode();
    patternBlockFailureCase.pos = pos;
    if (noRetParams || !returnErrorType) {
        // throw e
        BLangThrow throwStmt = (BLangThrow) TreeBuilder.createThrowNode();
        throwStmt.pos = pos;
        throwStmt.expr = patternFailureCaseVarRef;
        patternBlockFailureCase.stmts.add(throwStmt);
    } else {
        // return e;
        BLangReturn returnStmt = (BLangReturn) TreeBuilder.createReturnNode();
        returnStmt.pos = pos;
        returnStmt.exprs = new ArrayList<BLangExpression>() {

            {
                add(patternFailureCaseVarRef);
            }
        };
        patternBlockFailureCase.stmts.add(returnStmt);
    }
    return ASTBuilderUtil.createMatchStatementPattern(pos, patternFailureCaseVar, patternBlockFailureCase);
}

Example 10 with BLangReturn

use of org.wso2.ballerinalang.compiler.tree.statements.BLangReturn in project ballerina by ballerina-lang.

the class Desugar method addTransformerReturn.

private void addTransformerReturn(BLangTransformer transformerNode) {
    // This will only check whether last statement is a return and just add a return statement.
    // This won't analyze if else blocks etc to see whether return statements are present
    BLangBlockStmt blockStmt = transformerNode.body;
    if (transformerNode.workers.size() == 0 && (blockStmt.stmts.size() < 1 || blockStmt.stmts.get(blockStmt.stmts.size() - 1).getKind() != NodeKind.RETURN)) {
        BLangReturn returnStmt = (BLangReturn) TreeBuilder.createReturnNode();
        DiagnosticPos invPos = transformerNode.pos;
        DiagnosticPos pos = new DiagnosticPos(invPos.src, invPos.eLine, invPos.eLine, invPos.sCol, invPos.sCol);
        returnStmt.pos = pos;
        if (!transformerNode.retParams.isEmpty()) {
            returnStmt.namedReturnVariables = transformerNode.retParams;
        }
        blockStmt.addStatement(returnStmt);
    }
}