Skip to content

Commit

Permalink
fix memory leak error
Browse files Browse the repository at this point in the history
  • Loading branch information
alirezade authored and alirezade committed Jan 31, 2024
1 parent 80233a3 commit 5e659a2
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 12 deletions.
18 changes: 9 additions & 9 deletions src/AgentProcessor.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,15 +9,15 @@ class AgentProcessor:

def __init__(self, predictor_address, attacks_address, file_address, server_connection):
self.__processes = []
annotator = ProcessAnnotator(predictor_address, attacks_address)
flow_file_writer = ProcessFlowFileWriter(file_address)
status_sender = ProcessStatusSenderMQTT(server_connection, voting_interval=Config.StatusSender.voting_interval)
flow_sender = ProcessFlowSenderMQTT(server_connection)

self.__processes.append(annotator)
self.__processes.append(flow_file_writer)
if annotator.is_prediction_enabled():
self.__processes.append(status_sender)
# annotator = ProcessAnnotator(predictor_address, attacks_address)
# flow_file_writer = ProcessFlowFileWriter(file_address)
# status_sender = ProcessStatusSenderMQTT(server_connection, voting_interval=Config.StatusSender.voting_interval)
# flow_sender = ProcessFlowSenderMQTT(server_connection)
#
# self.__processes.append(annotator)
# self.__processes.append(flow_file_writer)
# if annotator.is_prediction_enabled():
# self.__processes.append(status_sender)
#self.__processes.append(flow_sender)

def process(self, flow):
Expand Down
7 changes: 4 additions & 3 deletions src/Helper.py
Original file line number Diff line number Diff line change
Expand Up @@ -117,11 +117,12 @@ def configure_log_files (directory, separate_event_log):

@staticmethod
def log(text, level):

msg = f"{text} [{inspect.currentframe().f_back}]"
frame = f"[{inspect.currentframe().f_back}]"
msg = f"{text} {frame}"
print_msg = f"{Log.log_colors[level]}[{logging.getLevelName(level)}] {text}\033[0m {frame}"

logger = Log.event_logger if Log.event_logger else logging
logger.log(level, msg)

print_msg = Log.log_colors[level] + f'[{logging.getLevelName(level)}] {msg}' + '\033[0m'

print(print_msg)
1 change: 1 addition & 0 deletions start.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ printStep(){

printStep "Getting last changes pull request"
sudo git pull
sleep 1

printStep "Check Requirement.txt"
pip install -r requirements.txt
Expand Down

0 comments on commit 5e659a2

Please sign in to comment.