RFR: 8303078: Reduce allocations when pretty printing JCTree during compilation [v2]

[Vicente Romero]

On Wed, 22 Feb 2023 16:36:50 GMT, Christoph Dreis <duke at openjdk.org> wrote:

>> Hi,
>> 
>> I've been recently optimizing a few project pipelines and always noticed that their compilation step is somewhat "expensive".  Zooming into the issue always revealed Lombok to be a larger contributor. Unfortunately, I can't get rid of Lombok in this customer project (before you ask).
>> 
>> Apparently, Lombok is printing the respective `JCTree` here and there to check for type matches. I can't imagine this to be super efficient, but that's how it is at the moment.
>> <img width="948" alt="image" src="https://user-images.githubusercontent.com/6304496/220134737-bd9b508c-a908-448b-b0d1-e960db28b24a.png">
>> 
>> Anyhow, regardless of the Lombok inefficiencies I think there are some optimization opportunities in the JDK itself. 
>> 
>> 1. Overall, `Pretty.visitSelect` accounts for 8-10% of the total allocations in this project. And among those there are StringBuilder allocations coming from the following:
>> 
>> 
>>     public void visitSelect(JCFieldAccess tree) {
>>         try {
>>             printExpr(tree.selected, TreeInfo.postfixPrec);
>>             // StringBuilder allocations hiding in here.
>>             print("." + tree.name);
>>         } catch (IOException e) {
>>             throw new UncheckedIOException(e);
>>         }
>>     }
>> 
>> 
>> This PR splits the `print` calls into two separate ones to avoid this String concatenation.
>> 
>> ...
>>             printExpr(tree.selected, TreeInfo.postfixPrec);
>>             print('.');
>>             print(tree.name);
>> ...
>> 
>> 
>> Secondly, the `print` method takes an `Object` which seems like a good fit for another (private?) variant of it that only takes a `char`. By this we would probably avoid any eventual boxing and avoid any conversion with `Convert.escapeUnicode(s.toString())` that seems superfluous for chars like `.`, ` `, or any braces like `(`, `{` etc. 
>> 
>> This is currently a draft PR as long as the scope is not clarified. It currently only includes the necessary changes that would optimize the particular use-case. But there are more cases where e.g. the new `char` variant could be used and/or any String concatenation could be split into separate `print` calls.
>> 
>> Let me know what you think and if I should include the other cases as well. If you think this is worthwhile, I'd appreciate if this is is sponsored. (Including creating an issue as I can't do this myself apparently. I will of course squash everything together with the proper issue ID once available.)
>> 
>> I've contributed before, so the CLA should be signed.
>> 
>> Cheers,
>> Christoph
>
> Christoph Dreis has updated the pull request incrementally with one additional commit since the last revision:
> 
>   8303078: Reduce allocations for single characters in DocPretty

src/jdk.compiler/share/classes/com/sun/tools/javac/tree/Pretty.java line 1246:

> 1244:             if (tree.paramKind == JCLambda.ParameterKind.EXPLICIT) {
> 1245:                 printExprs(tree.params);
> 1246:             } else {

nit, just below: `String sep = "";` -> `char sep = '\0';` also `sep = ","` -> `sep = ','`?

src/jdk.compiler/share/classes/com/sun/tools/javac/tree/Pretty.java line 1456:

> 1454:                     break;
> 1455:                 case CHAR:
> 1456:                     print("\'");

`print(''');` ?

src/jdk.compiler/share/classes/com/sun/tools/javac/tree/Pretty.java line 1458:

> 1456:                     print("\'");
> 1457:                     print(Convert.quote(String.valueOf((char)((Number)tree.value).intValue())));
> 1458:                     print("\'");

same: `print(''');` ?

src/jdk.compiler/share/classes/com/sun/tools/javac/tree/Pretty.java line 1467:

> 1465:                     break;
> 1466:                 default:
> 1467:                     print("\"");

`print('"');`?

src/jdk.compiler/share/classes/com/sun/tools/javac/tree/Pretty.java line 1469:

> 1467:                     print("\"");
> 1468:                     print(Convert.quote(tree.value.toString()));
> 1469:                     print("\"");

same as above

-------------

PR: https://git.openjdk.org/jdk/pull/12667