def __init__(self, args, cluster):
super(ActionBalance, self).__init__(args, cluster)
self.balance_types = dict((cls.name, cls) for cls in get_modules(kafka.tools.assigner.actions.balancemodules, ActionBalanceModule))
# Initialize all the modules to use
self.modules = []
for bmodule in args.types:
self.modules.append(self.balance_types[bmodule](args, cluster))
def __init__(self, args, cluster):
super(ActionBalance, self).__init__(args, cluster)
self.balance_types = dict((cls.name, cls) for cls in get_modules(kafka.tools.assigner.actions.balancemodules, ActionBalanceModule))
# Initialize all the modules to use
self.modules = []
for bmodule in args.types:
self.modules.append(self.balance_types[bmodule](args, cluster))
def _add_args(cls, parser):
# We'll need to build this list here as well as in the constructor
balance_actions = get_modules(
kafka.tools.assigner.actions.balancemodules, ActionBalanceModule)
parser.add_argument(
'-t',
'--types',
help=
"Balance types to perform. Multiple may be specified and they will be run in order",
required=True,
choices=[klass.name for klass in balance_actions],
nargs='*')
def create_action_map(self):
self.action_map = dict(
(cls.name, cls)
for cls in get_modules(kafka.tools.assigner.actions,
kafka.tools.assigner.actions.ActionModule))
def get_plugins_list():
return get_modules(kafka.tools.assigner.plugins,
kafka.tools.assigner.plugins.PluginModule)
def get_module_map(base_module_name, base_module_class):
return dict((cls.name, cls)
for cls in get_modules(base_module_name, base_module_class))
def create_action_map(self):
self.action_map = dict((cls.name, cls) for cls in get_modules(kafka.tools.assigner.actions, kafka.tools.assigner.actions.ActionModule))
def test_get_modules(self):
module_list = get_modules(tests.tools.assigner.fixturemodules, tests.tools.assigner.fixturemodules.ModuleTesterBaseClass)
assert module_list == [ModuleInheritedFromBase]
def get_plugins_list():
return get_modules(kafka.tools.assigner.plugins, kafka.tools.assigner.plugins.PluginModule)
def get_module_map(base_module_name, base_module_class):
return dict((cls.name, cls) for cls in get_modules(base_module_name, base_module_class))
def get_sizer_map():
return dict((cls.name, cls) for cls in get_modules(kafka.tools.assigner.sizers, kafka.tools.assigner.sizers.SizerModule))
def get_action_map():
return dict((cls.name, cls) for cls in get_modules(kafka.tools.assigner.actions, kafka.tools.assigner.actions.ActionModule))
def _add_args(cls, parser):
# We'll need to build this list here as well as in the constructor
balance_actions = get_modules(kafka.tools.assigner.actions.balancemodules, ActionBalanceModule)
parser.add_argument('-t', '--types', help="Balance types to perform. Multiple may be specified and they will be run in order", required=True,
choices=[klass.name for klass in balance_actions], nargs='*')
