Class TryCatch

src/main/java/io/airlift/bytecode/control/TryCatch.java:32–154

import static java.util.Objects.requireNonNull;

public class TryCatch
        implements FlowControl
{
    private final String comment;
    private final BytecodeNode tryNode;
    private final List<CatchBlock> catchBlocks;

    public TryCatch(BytecodeNode tryNode, List<CatchBlock> catchBlocks)
    {
        this(null, tryNode, catchBlocks);
    }

    public TryCatch(String comment, BytecodeNode tryNode, List<CatchBlock> catchBlocks)
    {
        this.comment = comment;
        this.tryNode = requireNonNull(tryNode, "tryNode is null");
        this.catchBlocks = ImmutableList.copyOf(requireNonNull(catchBlocks, "catchBlocks is null"));
    }

    @Override
    public String getComment()
    {
        return comment;
    }

    public BytecodeNode getTryNode()
    {
        return tryNode;
    }

    public List<CatchBlock> getCatchBlocks()
    {
        return catchBlocks;
    }

    @Override
    public void accept(MethodVisitor visitor, MethodGenerationContext generationContext)
    {
        LabelNode tryStart = new LabelNode("tryStart");
        LabelNode tryEnd = new LabelNode("tryEnd");
        List<LabelNode> handlers = new ArrayList<>();
        LabelNode done = new LabelNode("done");

        BytecodeBlock block = new BytecodeBlock();

        // try block
        block.visitLabel(tryStart)
                .append(tryNode)
                .visitLabel(tryEnd)
                .gotoLabel(done);

        // catch blocks
        for (int i = 0; i < catchBlocks.size(); i++) {
            BytecodeNode handlerBlock = catchBlocks.get(i).getHandler();
            LabelNode handler = new LabelNode("handler" + i);
            handlers.add(handler);
            block.visitLabel(handler)
                    .append(handlerBlock);