Allows credit flow Credits to be set via configuration

Fixes rabbitmq/rabbitmq-server#143

1 files changed, 14 insertions, 1 deletions

diff --git a/src/credit_flow.erl b/src/credit_flow.erl
index fc233b87c3..f8e991ede8 100644
--- a/src/credit_flow.erl
+++ b/src/credit_flow.erl
@@ -28,7 +28,20 @@
 %% is itself blocked - thus the only processes that need to check
 %% blocked/0 are ones that read from network sockets.
 
--define(DEFAULT_CREDIT, {200, 50}).
+-define(DEFAULT_INITIAL_CREDIT, 200).
+-define(DEFAULT_MORE_CREDIT_AFTER, 50).
+
+-define(DEFAULT_CREDIT,
+        case get(credit_flow_default_credit) of
+            undefined ->
+                Val = {rabbit_misc:get_env(rabbit, credit_flow_initial_credit,
+                                           ?DEFAULT_INITIAL_CREDIT),
+                       rabbit_misc:get_env(rabbit, credit_flow_more_credit_after,
+                                           ?DEFAULT_MORE_CREDIT_AFTER)},
+                put(credit_flow_default_credit, Val),
+                Val;
+            Val       -> Val
+        end).
 
 -export([send/1, send/2, ack/1, ack/2, handle_bump_msg/1, blocked/0, state/0]).
 -export([peer_down/1]).