Merge pull request #33 from HubSpot/if-ok-or-else-err

Add ifOkOrElseError

Author: mikemedina

algebra/src/main/java/com/hubspot/algebra/Result.java

@@ -44,6 +44,11 @@ public void ifErr(Consumer<? super ERROR_TYPE> consumer) {
     Results.getErr(this).ifPresent(consumer);
   }
 
+  public void consume(Consumer<? super ERROR_TYPE> errConsumer, Consumer<? super SUCCESS_TYPE> okConsumer) {
+    ifOk(okConsumer);
+    ifErr(errConsumer);
+  }
+
   public <NEW_ERROR_TYPE> Result<SUCCESS_TYPE, NEW_ERROR_TYPE> mapErr(Function<ERROR_TYPE, NEW_ERROR_TYPE> mapper) {
     return Results.<SUCCESS_TYPE, ERROR_TYPE, NEW_ERROR_TYPE>modErr(mapper).apply(this);
   }

algebra/src/test/java/com/hubspot/algebra/ResultTest.java

@@ -4,6 +4,7 @@
 import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
 
 import java.util.ArrayList;
+import java.util.Collections;
 import java.util.List;
 import java.util.function.Consumer;
 import java.util.function.Function;
@@ -172,4 +173,22 @@ public void itAllowsConsumerOfErrSuperType() {
     result.ifErr(consumer);
     assertThat(!check.isEmpty());
   }
+
+  @Test
+  public void itConsumesOks() throws Exception {
+    List<String> okResults = new ArrayList<>();
+    List<SampleError> errorResults = Collections.emptyList();
+    OK_RESULT.consume(errorResults::add, okResults::add);
+    assertThat(okResults).contains(OK_RESULT.unwrapOrElseThrow());
+    assertThat(errorResults).isEmpty();
+  }
+
+  @Test
+  public void itConsumesErrors() throws Exception {
+    List<String> okResults = Collections.emptyList();
+    List<SampleError> errorResults = new ArrayList<>();
+    ERR_RESULT.consume(errorResults::add, okResults::add);
+    assertThat(okResults).isEmpty();
+    assertThat(errorResults).contains(ERR_RESULT.unwrapErrOrElseThrow());
+  }
 }