tmp: Group filter for graphs uses the whole path instead of just the filename

This commit is contained in:
Nimrod Gilboa Markevich 2022-05-25 00:28:47 +03:00
parent 9f867f1cf1
commit 26272c582b

View File

@ -87,7 +87,7 @@ def plot(ax, df: pd.DataFrame, title: str, xlabel: str, ylabel: str, group_patte
def get_group_color(names, pattern): def get_group_color(names, pattern):
props = [int(re.findall(pattern, pathlib.Path(name).name)[0]) for name in names] props = [re.findall(pattern, name)[0] for name in names]
key = dict(zip(sorted(list(set(props))), range(len(set(props))))) key = dict(zip(sorted(list(set(props))), range(len(set(props)))))
n_colors = len(key) n_colors = len(key)
color_options = plt.get_cmap('jet')(np.linspace(0, 1, n_colors)) color_options = plt.get_cmap('jet')(np.linspace(0, 1, n_colors))
@ -113,14 +113,15 @@ if __name__ == '__main__':
with open(filename, 'r') as f: with open(filename, 'r') as f:
cpu_samples, rss_samples, count_samples, matched_samples, live_samples, processed_samples, heap_samples, goroutines_samples = extract_samples(f) cpu_samples, rss_samples, count_samples, matched_samples, live_samples, processed_samples, heap_samples, goroutines_samples = extract_samples(f)
cpu_samples.name = pathlib.Path(filename).name name = os.path.join(*pathlib.Path(filename).parts[-2:])
rss_samples.name = pathlib.Path(filename).name cpu_samples.name = name
count_samples.name = pathlib.Path(filename).name rss_samples.name = name
matched_samples.name = pathlib.Path(filename).name count_samples.name = name
live_samples.name = pathlib.Path(filename).name matched_samples.name = name
processed_samples.name = pathlib.Path(filename).name live_samples.name = name
heap_samples.name = pathlib.Path(filename).name processed_samples.name = name
goroutines_samples.name = pathlib.Path(filename).name heap_samples.name = name
goroutines_samples.name = name
cpu_samples_all_files.append(cpu_samples) cpu_samples_all_files.append(cpu_samples)
rss_samples_all_files.append(rss_samples) rss_samples_all_files.append(rss_samples)
@ -140,7 +141,7 @@ if __name__ == '__main__':
heap_samples_df = pd.concat(heap_samples_all_files, axis=1) heap_samples_df = pd.concat(heap_samples_all_files, axis=1)
goroutines_samples_df = pd.concat(goroutines_samples_all_files, axis=1) goroutines_samples_df = pd.concat(goroutines_samples_all_files, axis=1)
group_pattern = r'^\d+' group_pattern = r'(\d+)[\w-]*\.\w+$'
cpu_plot = plt.subplot(8, 2, 1) cpu_plot = plt.subplot(8, 2, 1)
plot(cpu_plot, cpu_samples_df, 'cpu', '', 'cpu (%)', group_pattern) plot(cpu_plot, cpu_samples_df, 'cpu', '', 'cpu (%)', group_pattern)
@ -179,4 +180,4 @@ if __name__ == '__main__':
print('Saving graph to graph.png') print('Saving graph to graph.png')
plt.savefig('graph.png', bbox_inches='tight') plt.savefig('graph.png', bbox_inches='tight')