class DriverStatsHandler(object):
def __init__(self, group_type, driver_name):
if group_type == "destination":
statement_short_name = "dst"
self.component = "{}.{}".format(statement_short_name, driver_name)
elif group_type == "source":
statement_short_name = "src"
self.component = "{}.{}".format(statement_short_name, driver_name)
elif group_type == "parser":
self.component = "{}".format(group_type)
else:
raise Exception("Unknown group_type: {}".format(group_type))
self.syslog_ng_ctl = SyslogNgCtl(tc_parameters.INSTANCE_PATH)
def build_query_pattern(self):
return self.build_pattern(delimiter=".")
def build_stats_pattern(self):
return self.build_pattern(delimiter=";")
def build_pattern(self, delimiter):
return "{}{}".format(
self.component,
delimiter,
)
def parse_result(self, result, result_type):
statistical_elements = [
"memory_usage", "written", "processed", "dropped", "queued",
"stamp", "discarded"
]
parsed_output = {}
for stat_element in statistical_elements:
for line in result:
if stat_element in line:
if result_type == "query":
parsed_output.update({
stat_element:
int(line.split(".")[-1].split("=")[-1])
})
elif result_type == "stats":
parsed_output.update({
stat_element:
int(line.split(".")[-1].split(";")[-1])
})
else:
raise Exception(
"Unknown result_type: {}".format(result_type))
return parsed_output
def filter_stats_result(self, stats_result, stats_pattern):
filtered_list = []
for stats_line in stats_result:
if stats_pattern in stats_line:
filtered_list.append(stats_line)
return filtered_list
def get_query(self):
query_pattern = self.build_query_pattern()
query_result = self.syslog_ng_ctl.query(
pattern="*{}*".format(query_pattern))['stdout'].splitlines()
return self.parse_result(result=query_result, result_type="query")
def get_stats(self):
stats_pattern = self.build_stats_pattern()
stats_result = self.syslog_ng_ctl.stats()['stdout'].splitlines()
filtered_stats_result = self.filter_stats_result(
stats_result, stats_pattern)
return self.parse_result(result=filtered_stats_result,
result_type="stats")
class DriverStatsHandler(object):
def __init__(self, group_type, driver_name):
if group_type == "destination":
statement_short_name = "dst"
self.component = "{}.{}".format(statement_short_name, driver_name)
elif group_type == "source":
statement_short_name = "src"
self.component = "{}.{}".format(statement_short_name, driver_name)
elif group_type == "parser":
self.component = "{}".format(group_type)
elif group_type == "filter":
self.component = "{}".format(group_type)
else:
raise Exception("Unknown group_type: {}".format(group_type))
self.syslog_ng_ctl = SyslogNgCtl(tc_parameters.INSTANCE_PATH)
def build_query_pattern(self):
return self.build_pattern(delimiter=".")
def build_stats_pattern(self):
return self.build_pattern(delimiter=";")
def build_pattern(self, delimiter):
return "{}{}".format(
self.component,
delimiter,
)
def __parse_query_result(self, line: str):
split = re.split(r"\.|=", line)
if len(split) < 2:
return dict()
key, value = split[-2:]
return {key: int(value)}
def __parse_stats_result(self, line):
split = line.split(";")
if len(split) < 2:
return dict()
key, value = split[-2:]
return {key: int(value)}
def parse_result(self, result, result_type):
parsed_output = {}
for line in result:
if result_type == "query":
parsed_output.update(self.__parse_query_result(line))
elif result_type == "stats":
parsed_output.update(self.__parse_stats_result(line))
else:
raise Exception("Unknown result_type: {}".format(result_type))
return parsed_output
def filter_stats_result(self, stats_result, stats_pattern):
filtered_list = []
for stats_line in stats_result:
if stats_pattern in stats_line:
filtered_list.append(stats_line)
return filtered_list
def get_query(self):
query_pattern = self.build_query_pattern()
query_result = self.syslog_ng_ctl.query(
pattern="*{}*".format(query_pattern))['stdout'].splitlines()
return self.parse_result(result=query_result, result_type="query")
def get_stats(self):
stats_pattern = self.build_stats_pattern()
stats_result = self.syslog_ng_ctl.stats()['stdout'].splitlines()
filtered_stats_result = self.filter_stats_result(
stats_result, stats_pattern)
return self.parse_result(result=filtered_stats_result,
result_type="stats")