linux/tools/cgroup/memcg_shrinker.py
Alexander Pantyukhin d526643f15 tools:cgroup:memcg_shrinker remove redundant import
Remove redundant import of the sys module.

Also use the sort function instead of sorted.  It sorts the direct array
without create the new one in memory.

Link: https://lkml.kernel.org/r/20230108105023.4289-1-apantykhin@gmail.com
Signed-off-by: Alexander Pantyukhin <apantykhin@gmail.com>
Cc: Roman Gushchin <roman.gushchin@linux.dev>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
2023-01-18 17:12:59 -08:00

70 lines
1.8 KiB
Python

#!/usr/bin/env python3
#
# Copyright (C) 2022 Roman Gushchin <roman.gushchin@linux.dev>
# Copyright (C) 2022 Meta
import os
import argparse
def scan_cgroups(cgroup_root):
cgroups = {}
for root, subdirs, _ in os.walk(cgroup_root):
for cgroup in subdirs:
path = os.path.join(root, cgroup)
ino = os.stat(path).st_ino
cgroups[ino] = path
# (memcg ino, path)
return cgroups
def scan_shrinkers(shrinker_debugfs):
shrinkers = []
for root, subdirs, _ in os.walk(shrinker_debugfs):
for shrinker in subdirs:
count_path = os.path.join(root, shrinker, "count")
with open(count_path) as f:
for line in f.readlines():
items = line.split(' ')
ino = int(items[0])
# (count, shrinker, memcg ino)
shrinkers.append((int(items[1]), shrinker, ino))
return shrinkers
def main():
parser = argparse.ArgumentParser(description='Display biggest shrinkers')
parser.add_argument('-n', '--lines', type=int, help='Number of lines to print')
args = parser.parse_args()
cgroups = scan_cgroups("/sys/fs/cgroup/")
shrinkers = scan_shrinkers("/sys/kernel/debug/shrinker/")
shrinkers.sort(reverse = True, key = lambda x: x[0])
n = 0
for s in shrinkers:
count, name, ino = (s[0], s[1], s[2])
if count == 0:
break
if ino == 0 or ino == 1:
cg = "/"
else:
try:
cg = cgroups[ino]
except KeyError:
cg = "unknown (%d)" % ino
print("%-8s %-20s %s" % (count, name, cg))
n += 1
if args.lines and n >= args.lines:
break
if __name__ == '__main__':
main()