Introduce MonoSingleOptional Refaster rule

PicnicSupermarket · Feb 20, 2023 · 5ee0703 · 5ee0703
1 parent 208659d
commit 5ee0703
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/error-prone-contrib/src/main/java/tech/picnic/errorprone/refasterrules/ReactorRules.java b/error-prone-contrib/src/main/java/tech/picnic/errorprone/refasterrules/ReactorRules.java
@@ -717,6 +717,19 @@ Mono<Optional<T>> after(Mono<T> mono) {
     }
   }
 
+  /** Prefer {@link Mono#singleOptional()} over more contrived alternatives. */
+  static final class MonoSingleOptional<T> {
+    @BeforeTemplate
+    Mono<Optional<T>> before(Mono<T> mono) {
+      return mono.flux().collect(toOptional());
+    }
+
+    @AfterTemplate
+    Mono<Optional<T>> after(Mono<T> mono) {
+      return mono.singleOptional();
+    }
+  }
+
   /** Prefer {@link Mono#cast(Class)} over {@link Mono#map(Function)} with a cast. */
   static final class MonoCast<T, S> {
     @BeforeTemplate