Give parallelStream() the same treatment as stream()

cushon · ronshapiro · commit aee9cd85cfbe · 2019-04-29T11:40:49.000-04:00
Fixes google#365 ------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=244264487
diff --git a/core/src/main/java/com/google/googlejavaformat/java/JavaInputAstVisitor.java b/core/src/main/java/com/google/googlejavaformat/java/JavaInputAstVisitor.java
@@ -1564,7 +1564,8 @@ private static List<Long> handleStream(List<ExpressionTree> parts) {
                 return false;
               }
               Name name = getMethodName((MethodInvocationTree) p);
-              return Stream.of("stream", "toBuilder").anyMatch(name::contentEquals);
+              return Stream.of("stream", "parallelStream", "toBuilder")
+                  .anyMatch(name::contentEquals);
             })
         .collect(toList());
   }
diff --git a/core/src/test/resources/com/google/googlejavaformat/java/testdata/I365.input b/core/src/test/resources/com/google/googlejavaformat/java/testdata/I365.input
@@ -0,0 +1,10 @@
+class I365 {
+  {
+    return foo____________
+        .bar__________()
+        .baz____________()
+        .parallelStream()
+        .map(Baz::getId)
+        .collect(toList());
+  }
+}
diff --git a/core/src/test/resources/com/google/googlejavaformat/java/testdata/I365.output b/core/src/test/resources/com/google/googlejavaformat/java/testdata/I365.output
@@ -0,0 +1,7 @@
+class I365 {
+  {
+    return foo____________.bar__________().baz____________().parallelStream()
+        .map(Baz::getId)
+        .collect(toList());
+  }
+}

Original file line number	Diff line number	Diff line change
`@@ -1564,7 +1564,8 @@ private static List<Long> handleStream(List<ExpressionTree> parts) {`
`1564`	`1564`	`return false;`
`1565`	`1565`	`}`
`1566`	`1566`	`Name name = getMethodName((MethodInvocationTree) p);`
`1567`		`- return Stream.of("stream", "toBuilder").anyMatch(name::contentEquals);`
	`1567`	`+ return Stream.of("stream", "parallelStream", "toBuilder")`
	`1568`	`+ .anyMatch(name::contentEquals);`
`1568`	`1569`	`})`
`1569`	`1570`	`.collect(toList());`
`1570`	`1571`	`}`