Introduce StreamMapToValuesFromMap Refaster rule

PicnicSupermarket · Dec 18, 2024 · cb2e17d · cb2e17d
1 parent 67bf5b9
commit cb2e17d
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 0 deletions.
diff --git a/error-prone-contrib/src/main/java/tech/picnic/errorprone/refasterrules/StreamRules.java b/error-prone-contrib/src/main/java/tech/picnic/errorprone/refasterrules/StreamRules.java
@@ -297,6 +297,18 @@ boolean after(Stream<T> stream) {
     }
   }
 
+  static final class StreamMapToValuesFromMap<K, V> {
+    @BeforeTemplate
+    Stream<V> before(Stream<K> stream, Map<K, V> map) {
+      return stream.filter(map::containsKey).map(map::get);
+    }
+
+    @AfterTemplate
+    Stream<V> after(Stream<K> stream, Map<K, V> map) {
+      return stream.map(map::get).filter(Objects::nonNull);
+    }
+  }
+
   static final class StreamMin<T> {
     @BeforeTemplate
     @SuppressWarnings("java:S4266" /* This violation will be rewritten. */)

diff --git a/...contrib/src/test/resources/tech/picnic/errorprone/refasterrules/StreamRulesTestInput.java b/...contrib/src/test/resources/tech/picnic/errorprone/refasterrules/StreamRulesTestInput.java
@@ -141,6 +141,11 @@ boolean testStreamFindAnyIsPresent() {
     return Stream.of(1).findFirst().isPresent();
   }
 
+  Stream<String> testStreamMapToValuesFromMap() {
+    ImmutableMap<String, String> map = ImmutableMap.of("foo", "bar");
+    return Stream.of("foo").filter(map::containsKey).map(map::get);
+  }
+
   ImmutableSet<Optional<String>> testStreamMin() {
     return ImmutableSet.of(
         Stream.of("foo").max(comparingInt(String::length).reversed()),

diff --git a/...ontrib/src/test/resources/tech/picnic/errorprone/refasterrules/StreamRulesTestOutput.java b/...ontrib/src/test/resources/tech/picnic/errorprone/refasterrules/StreamRulesTestOutput.java
@@ -141,6 +141,11 @@ boolean testStreamFindAnyIsPresent() {
     return Stream.of(1).findAny().isPresent();
   }
 
+  Stream<String> testStreamMapToValuesFromMap() {
+    ImmutableMap<String, String> map = ImmutableMap.of("foo", "bar");
+    return Stream.of("foo").map(map::get).filter(Objects::nonNull);
+  }
+
   ImmutableSet<Optional<String>> testStreamMin() {
     return ImmutableSet.of(
         Stream.of("foo").min(comparingInt(String::length)),