From 92f7603051aca10f672c454f2ed56c034104f789 Mon Sep 17 00:00:00 2001
From: Eric Fenderbosch <eric.fenderbosch@revenueconduit.com>
Date: Fri, 24 Mar 2017 14:05:39 -0400
Subject: [PATCH] added DecorateConsumer syntatic sugar

---
 .../resilience4j/decorators/Decorators.java   | 36 +++++++++++++++++--
 1 file changed, 34 insertions(+), 2 deletions(-)

diff --git a/resilience4j-all/src/main/java/io/github/resilience4j/decorators/Decorators.java b/resilience4j-all/src/main/java/io/github/resilience4j/decorators/Decorators.java
index d8d5bf6406..f0de30a25b 100644
--- a/resilience4j-all/src/main/java/io/github/resilience4j/decorators/Decorators.java
+++ b/resilience4j-all/src/main/java/io/github/resilience4j/decorators/Decorators.java
@@ -9,11 +9,13 @@
 
 import java.util.concurrent.CompletionStage;
 import java.util.concurrent.ScheduledExecutorService;
+import java.util.function.Consumer;
 import java.util.function.Function;
 import java.util.function.Supplier;
 
 /**
- * A Decorator builder which can be used to apply multiple decorators to a (Checked-)Supplier, (Checked-)Function or (Checked-)Runnable.
+ * A Decorator builder which can be used to apply multiple decorators to a (Checked-)Supplier, (Checked-)Function,
+ * (Checked-)Runnable, (Checked-)CompletionStage or (Checked-)Consumer
  */
 public interface Decorators{
 
@@ -45,6 +47,9 @@ static <T> DecorateCompletionStage<T> ofCompletionStage(Supplier<CompletionStage
         return new DecorateCompletionStage<>(stageSupplier);
     }
 
+    public static <T> DecorateConsumer<T> ofConsumer(Consumer<T> consumer) {
+        return new DecorateConsumer<>(consumer);
+    }
 
     class DecorateSupplier<T>{
         private Supplier<T> supplier;
@@ -268,4 +273,31 @@ public CompletionStage<T> get() {
             return stageSupplier.get();
         }
     }
-}
\ No newline at end of file
+
+    class DecorateConsumer<T> {
+
+        private Consumer<T> consumer;
+
+        private DecorateConsumer(Consumer<T> consumer) {
+            this.consumer = consumer;
+        }
+
+        public DecorateConsumer<T> withCircuitBreaker(CircuitBreaker circuitBreaker) {
+            consumer = CircuitBreaker.decorateConsumer(circuitBreaker, consumer);
+            return this;
+        }
+
+        public DecorateConsumer<T> withRateLimiter(RateLimiter rateLimiter) {
+            consumer = RateLimiter.decorateConsumer(rateLimiter, consumer);
+            return this;
+        }
+
+        public Consumer<T> decorate() {
+            return consumer;
+        }
+
+        public void accept(T obj) {
+            consumer.accept(obj);
+        }
+    }
+}