Trending: Memory consumption

[csit.git] / resources / tools / presentation / input_data_parser.py

diff --git a/resources/tools/presentation/input_data_parser.py b/resources/tools/presentation/input_data_parser.py
index 8356091..37532c8 100644 (file)
--- a/resources/tools/presentation/input_data_parser.py
+++ b/resources/tools/presentation/input_data_parser.py
@@ -22,6 +22,8 @@
 import multiprocessing
 import os
 import re
+import resource
+import objgraph
 import pandas as pd
 import logging
 
@@ -1307,6 +1309,7 @@ class InputData(object):
             for build in builds:
 
                 result = self._download_and_parse_build(job, build, repeat)
+                build_nr = result["build"]["build"]
 
                 if result["data"]:
                     data = result["data"]
@@ -1321,12 +1324,17 @@ class InputData(object):
 
                     if self._input_data.get(job, None) is None:
                         self._input_data[job] = pd.Series()
-                    self._input_data[job][str(build)] = build_data
+                    self._input_data[job][str(build_nr)] = build_data
 
                     self._cfg.set_input_file_name(
-                        job, build, result["build"]["file-name"])
+                        job, build_nr, result["build"]["file-name"])
 
-                self._cfg.set_input_state(job, build, result["state"])
+                self._cfg.set_input_state(job, build_nr, result["state"])
+
+                logging.info("ru_maxrss = {0}".format(
+                    resource.getrusage(resource.RUSAGE_SELF).ru_maxrss))
+
+                logging.info(objgraph.most_common_types())
 
         logging.info("Done.")