public static Collection<JetReturnExpression> collectReturns(@NotNull JetExpression expression) {
Collection<JetReturnExpression> result = Lists.newArrayList();
expression.accept(
new JetTreeVisitor<Collection<JetReturnExpression>>() {
@Override
public Void visitReturnExpression(
@NotNull JetReturnExpression expression, Collection<JetReturnExpression> data) {
data.add(expression);
return null;
}
},
result);
return result;
}
