Search in sources :

Example 1 with ThrowStatement

use of com.google.devtools.j2objc.ast.ThrowStatement in project j2objc by google.

the class Rewriter method visit.

@Override
public boolean visit(TryStatement node) {
    // This visit rewrites try-with-resources constructs into regular try statements according to
    // JLS 14.20.3. The rewriting is done in a visit instead of endVisit because the mutations may
    // result in more try-with-resources constructs that need to be rewritten recursively.
    List<TreeNode> resources = node.getResources();
    if (resources.isEmpty()) {
        return true;
    }
    if (!node.getCatchClauses().isEmpty() || node.getFinally() != null) {
        // Extended try-with-resources. (JLS 14.20.3.2)
        // The new innerTry statement will be a "Basic try-with-resources" and will be rewritten by
        // the code below when it is visited.
        TryStatement innerTry = new TryStatement().setBody(TreeUtil.remove(node.getBody()));
        ;
        TreeUtil.moveList(resources, innerTry.getResources());
        node.setBody(new Block().addStatement(innerTry));
        return true;
    }
    // Basic try-with-resources. (JLS 14.20.3.1)
    DeclaredType throwableType = (DeclaredType) typeUtil.getJavaThrowable().asType();
    VariableElement primaryException = GeneratedVariableElement.newLocalVar("__primaryException" + resources.size(), throwableType, null);
    TreeNode resource = resources.remove(0);
    VariableElement resourceVar = null;
    VariableDeclarationFragment resourceFrag = null;
    if (resource.getKind() == Kind.VARIABLE_DECLARATION_EXPRESSION) {
        List<VariableDeclarationFragment> resourceFrags = ((VariableDeclarationExpression) resource).getFragments();
        assert resourceFrags.size() == 1;
        resourceFrag = resourceFrags.get(0);
        resourceVar = resourceFrag.getVariableElement();
    } else {
        resourceVar = TreeUtil.getVariableElement((Expression) resource);
    }
    assert resourceVar != null;
    DeclaredType closeableType = typeUtil.findSupertype(resourceVar.asType(), "java.lang.AutoCloseable");
    ExecutablePair closeMethod = typeUtil.findMethod(closeableType, "close");
    ExecutablePair addSuppressedMethod = typeUtil.findMethod(throwableType, "addSuppressed", "java.lang.Throwable");
    Block block = new Block();
    if (resourceFrag != null) {
        block.addStatement(new VariableDeclarationStatement(resourceVar, TreeUtil.remove(resourceFrag.getInitializer())));
    }
    block.addStatement(new VariableDeclarationStatement(primaryException, new NullLiteral(typeUtil.getNull())));
    // If the current try node is the only statement in its parent block then replace the parent
    // block instead of the try node to avoid extra nesting of braces.
    TreeNode parent = node.getParent();
    if (parent instanceof Block && ((Block) parent).getStatements().size() == 1) {
        parent.replaceWith(block);
    } else {
        node.replaceWith(block);
    }
    block.addStatement(TreeUtil.remove(node));
    VariableElement caughtException = GeneratedVariableElement.newLocalVar("e", throwableType, null);
    Block catchBlock = new Block().addStatement(new ExpressionStatement(new Assignment(new SimpleName(primaryException), new SimpleName(caughtException)))).addStatement(new ThrowStatement(new SimpleName(caughtException)));
    node.addCatchClause(new CatchClause().setException(new SingleVariableDeclaration(caughtException)).setBody(catchBlock));
    Statement closeResource = new ExpressionStatement(new MethodInvocation(closeMethod, typeUtil.getVoid(), new SimpleName(resourceVar)));
    VariableElement suppressedException = GeneratedVariableElement.newLocalVar("e", throwableType, null);
    node.setFinally(new Block().addStatement(new IfStatement().setExpression(new InfixExpression(typeUtil.getBoolean(), InfixExpression.Operator.NOT_EQUALS, new SimpleName(resourceVar), new NullLiteral(typeUtil.getNull()))).setThenStatement(new Block().addStatement(new IfStatement().setExpression(new InfixExpression(typeUtil.getBoolean(), InfixExpression.Operator.NOT_EQUALS, new SimpleName(primaryException), new NullLiteral(typeUtil.getNull()))).setThenStatement(new Block().addStatement(new TryStatement().setBody(new Block().addStatement(closeResource)).addCatchClause(new CatchClause().setException(new SingleVariableDeclaration(suppressedException)).setBody(new Block().addStatement(new ExpressionStatement(new MethodInvocation(addSuppressedMethod, typeUtil.getVoid(), new SimpleName(primaryException)).addArgument(new SimpleName(suppressedException)))))))).setElseStatement(new Block().addStatement(closeResource.copy()))))));
    // Visit the new block instead of the current node because some of content of the node (eg. the
    // resource initializer) has been moved outside of the try node.
    block.accept(this);
    return false;
}
Also used : ExecutablePair(com.google.devtools.j2objc.types.ExecutablePair) SingleVariableDeclaration(com.google.devtools.j2objc.ast.SingleVariableDeclaration) ExpressionStatement(com.google.devtools.j2objc.ast.ExpressionStatement) ForStatement(com.google.devtools.j2objc.ast.ForStatement) ThrowStatement(com.google.devtools.j2objc.ast.ThrowStatement) TryStatement(com.google.devtools.j2objc.ast.TryStatement) VariableDeclarationStatement(com.google.devtools.j2objc.ast.VariableDeclarationStatement) IfStatement(com.google.devtools.j2objc.ast.IfStatement) Statement(com.google.devtools.j2objc.ast.Statement) VariableDeclarationExpression(com.google.devtools.j2objc.ast.VariableDeclarationExpression) SimpleName(com.google.devtools.j2objc.ast.SimpleName) MethodInvocation(com.google.devtools.j2objc.ast.MethodInvocation) GeneratedVariableElement(com.google.devtools.j2objc.types.GeneratedVariableElement) VariableElement(javax.lang.model.element.VariableElement) CatchClause(com.google.devtools.j2objc.ast.CatchClause) Assignment(com.google.devtools.j2objc.ast.Assignment) IfStatement(com.google.devtools.j2objc.ast.IfStatement) TryStatement(com.google.devtools.j2objc.ast.TryStatement) Expression(com.google.devtools.j2objc.ast.Expression) VariableDeclarationExpression(com.google.devtools.j2objc.ast.VariableDeclarationExpression) InfixExpression(com.google.devtools.j2objc.ast.InfixExpression) ParenthesizedExpression(com.google.devtools.j2objc.ast.ParenthesizedExpression) TreeNode(com.google.devtools.j2objc.ast.TreeNode) VariableDeclarationFragment(com.google.devtools.j2objc.ast.VariableDeclarationFragment) ExpressionStatement(com.google.devtools.j2objc.ast.ExpressionStatement) InfixExpression(com.google.devtools.j2objc.ast.InfixExpression) Block(com.google.devtools.j2objc.ast.Block) VariableDeclarationStatement(com.google.devtools.j2objc.ast.VariableDeclarationStatement) ThrowStatement(com.google.devtools.j2objc.ast.ThrowStatement) NullLiteral(com.google.devtools.j2objc.ast.NullLiteral) DeclaredType(javax.lang.model.type.DeclaredType)

Aggregations

Assignment (com.google.devtools.j2objc.ast.Assignment)1 Block (com.google.devtools.j2objc.ast.Block)1 CatchClause (com.google.devtools.j2objc.ast.CatchClause)1 Expression (com.google.devtools.j2objc.ast.Expression)1 ExpressionStatement (com.google.devtools.j2objc.ast.ExpressionStatement)1 ForStatement (com.google.devtools.j2objc.ast.ForStatement)1 IfStatement (com.google.devtools.j2objc.ast.IfStatement)1 InfixExpression (com.google.devtools.j2objc.ast.InfixExpression)1 MethodInvocation (com.google.devtools.j2objc.ast.MethodInvocation)1 NullLiteral (com.google.devtools.j2objc.ast.NullLiteral)1 ParenthesizedExpression (com.google.devtools.j2objc.ast.ParenthesizedExpression)1 SimpleName (com.google.devtools.j2objc.ast.SimpleName)1 SingleVariableDeclaration (com.google.devtools.j2objc.ast.SingleVariableDeclaration)1 Statement (com.google.devtools.j2objc.ast.Statement)1 ThrowStatement (com.google.devtools.j2objc.ast.ThrowStatement)1 TreeNode (com.google.devtools.j2objc.ast.TreeNode)1 TryStatement (com.google.devtools.j2objc.ast.TryStatement)1 VariableDeclarationExpression (com.google.devtools.j2objc.ast.VariableDeclarationExpression)1 VariableDeclarationFragment (com.google.devtools.j2objc.ast.VariableDeclarationFragment)1 VariableDeclarationStatement (com.google.devtools.j2objc.ast.VariableDeclarationStatement)1