EVOLUTION-NINJA

Edit File: apache_zookeeper.py

import agent_util import sys import socket import traceback def netcat(hostname, port, content): s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.connect((hostname, port)) s.sendall(content) s.shutdown(socket.SHUT_WR) while 1: data = s.recv(1024) if data == "": break response = data break s.close() return response def nc_to_dict(data): parsed = {} a = data.split("\n") for l in a: try: b = l.split("\t") parsed[b[0]] = b[1] except: continue return parsed class ApacheZookeeperPlugin(agent_util.Plugin): textkey = "apache_zookeeper" label = "Apache Zookeeper" @classmethod def get_metadata(self, config): status = agent_util.SUPPORTED msg = None if status == agent_util.SUPPORTED and ( not "host" in config or not "port" in config ): msg = "The host and port entries were not found in the [apache_zookeeper] block in the agent config file." self.log.info(msg) status = agent_util.MISCONFIGURED return {} if status == agent_util.SUPPORTED: response = None try: response = netcat(config["host"], int(config["port"]), "envi") except: self.log.exception("Error running Zookeeper hello") self.log.debug(traceback.format_exc()) status = agent_util.MISCONFIGURED msg = "" return {} if response is None or response == "": self.log.exception("Bad response running Zookeeper hello") return {} data = { # basic "avg_latency": { "label": "Average request latency", "options": None, "status": status, "error_message": msg, "unit": "ms", }, "max_latency": { "label": "Maximum request latency", "options": None, "status": status, "error_message": msg, "unit": "ms", }, "packets_received": { "label": "Packets received", "options": None, "status": status, "error_message": msg, "unit": "packets", }, "packets_sent": { "label": "Packets sent", "options": None, "status": status, "error_message": msg, "unit": "packets", }, "packets_received_per_sec": { "label": "Packets received/sec", "options": None, "status": status, "error_message": msg, "unit": "packets/sec", }, "packets_sent_per_sec": { "label": "Packets sent/sec", "options": None, "status": status, "error_message": msg, "unit": "packets/sec", }, "outstanding_requests": { "label": "Outstanding Requests", "options": None, "status": status, "error_message": msg, "unit": "", }, "server_state": { "label": "Server Mode", "options": None, "status": status, "error_message": msg, "unit": "", }, "znode_count": { "label": "Node count", "options": None, "status": status, "error_message": msg, "unit": "", }, "watch_count": { "label": "Watcher Count", "options": None, "status": status, "error_message": msg, "unit": "", }, "approximate_data_size": { "label": "Approximate data size", "options": None, "status": status, "error_message": msg, "unit": "bytes", }, "open_file_descriptor_count": { "label": "Open file descriptors", "options": None, "status": status, "error_message": msg, "unit": "files", }, "fsync_threshold_exceed_count": { "label": "Slow fsync count", "options": None, "status": status, "error_message": msg, "unit": "", }, "ruok": { "label": "Node error state", "options": None, "status": status, "error_message": msg, "unit": "", }, } return data def check(self, textkey, data, config): if textkey == "ruok": output = netcat(config["host"], int(config["port"]), "ruok") if output == "imok": return 0 else: return 1 else: output = netcat(config["host"], int(config["port"]), "mntr") data = nc_to_dict(output) key = "zk_" + str(textkey.replace("_per_sec", "")) value = data.get(key, False) if textkey == "server_state": if value == "follower": return 0 if value == "leader": return 1 if ( textkey == "packets_received_per_sec" or textkey == "packets_sent_per_sec" ): self.log.debug(data) if value > 0: cached = self.get_cache_results(textkey, None) self.cache_result(textkey, None, value, replace=True) self.log.debug("####\nGot cached result!\n%s" % cached) print("####\nGot cached result!\n%s" % cached) delta, c = cached[0] rate = (float(value) / float(c)) / float(delta) return rate else: return 0 if value: return float(value) else: return None