Merge pull request #31 from zalando-incubator/fix-daemon

Fix daemon
zalando-nakadi · Aug 5, 2016 · ef2dc75 · ef2dc75
2 parents 69906cf + 1ae08a7
commit ef2dc75
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/bubuku/daemon.py b/bubuku/daemon.py
@@ -15,7 +15,6 @@
 from bubuku.features.terminate import register_terminate_on_interrupt
 from bubuku.id_generator import get_broker_id_policy
 from bubuku.utils import CmdHelper
-from bubuku.zookeeper import load_exhibitor_proxy, BukuExhibitor
 from bubuku.zookeeper import BukuExhibitor, load_exhibitor_proxy
 
 _LOG = logging.getLogger('bubuku.main')
@@ -31,9 +30,9 @@ def apply_features(features: str, controller: Controller, buku_proxy: BukuExhibi
         elif feature == 'rebalance_on_brokers_change':
             controller.add_check(RebalanceOnBrokerListChange(buku_proxy, broker))
         elif feature == 'rebalance_by_size':
-            controller.add_check(GenerateDataSizeStatistics(exhibitor, broker, CmdHelper(),
+            controller.add_check(GenerateDataSizeStatistics(buku_proxy, broker, CmdHelper(),
                                                             kafka_properties.get_property("log.dirs").split(",")))
-            controller.add_check(RebalanceBySize(exhibitor, broker))
+            controller.add_check(RebalanceBySize(buku_proxy, broker))
         elif feature == 'graceful_terminate':
             register_terminate_on_interrupt(controller, broker)
         elif feature == 'use_ip_address':