def __init__(self, script, plugin_config): """ Init function Args: script - Health check script Return: None Raise: None """ HealthCheck.__init__(self, script, plugin_config)
def __init__(self, shell_script, plugin_config): """ Init function for the KafkaHealthCheck class Args: shell_script - script that is used to check the health of the cluster Return: None Raise: None """ HealthCheck.__init__(self, shell_script, plugin_config) self.config_file = plugin_config['topology_config'] with open(self.config_file, 'r') as file_desc: doc = yaml.load(file_desc) # Declare the config variables self.zookeeper = None self.zookeeper_paths = None self.kafka_host_resolution = None for key, val in doc.iteritems(): setattr(self, key, val) if self.zookeeper: zook = KazooClient(hosts=self.zookeeper.get("host"), read_only=True) zook.start() # read ZK Paths into a dictionary zk_paths = {} if self.zookeeper_paths: for path_type, path in self.zookeeper_paths.items(): zk_paths[path_type] = path # create zk_helper object helper = KafkaZkHelper(zook, zk_paths) self.helper = helper