From cd3b62e1238d1974598ee5ce5a0f31f296c78c33 Mon Sep 17 00:00:00 2001
From: Dhruv Parthasarathy <dhroovie@gmail.com>
Date: Thu, 1 Aug 2013 17:01:50 -0700
Subject: [PATCH] fix to runtime params

---
 .../druid/master/DruidMasterRuntimeParams.java      | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/server/src/main/java/com/metamx/druid/master/DruidMasterRuntimeParams.java b/server/src/main/java/com/metamx/druid/master/DruidMasterRuntimeParams.java
index 4b46a8174ed..7e6229374e9 100644
--- a/server/src/main/java/com/metamx/druid/master/DruidMasterRuntimeParams.java
+++ b/server/src/main/java/com/metamx/druid/master/DruidMasterRuntimeParams.java
@@ -207,7 +207,8 @@ public class DruidMasterRuntimeParams
         mergeSegmentsLimit,
         maxSegmentsToMove,
         balancerReferenceTimestamp,
-        emitStats
+        emitStats,
+        strategyFactory
     );
   }
 
@@ -268,7 +269,8 @@ public class DruidMasterRuntimeParams
         int mergeSegmentsLimit,
         int maxSegmentsToMove,
         DateTime balancerReferenceTimestamp,
-        boolean emitBalancingCostParams
+        boolean emitBalancingCostParams,
+        BalancerStrategyFactory strategyFactory
     )
     {
       this.startTime = startTime;
@@ -287,6 +289,7 @@ public class DruidMasterRuntimeParams
       this.maxSegmentsToMove = maxSegmentsToMove;
       this.balancerReferenceTimestamp = balancerReferenceTimestamp;
       this.emitBalancingCostParams = emitBalancingCostParams;
+      this.strategyFactory=strategyFactory;
     }
 
     public DruidMasterRuntimeParams build()
@@ -413,5 +416,11 @@ public class DruidMasterRuntimeParams
       this.balancerReferenceTimestamp = balancerReferenceTimestamp;
       return this;
     }
+
+    public Builder withBalancerStrategyFactory(BalancerStrategyFactory strategyFactory)
+    {
+      this.strategyFactory=strategyFactory;
+      return this;
+    }
   }
 }