Add docker registry dirsize

Author: Daniel Widerin

README.md

@@ -30,6 +30,9 @@ user git
 [gitlab_redis_*]
 user gitlab-redis
 #env.redis_socket /var/opt/gitlab/redis/redis.socket  # optional, defaults to GitLab omnibus redis instance
+
+[gitlab_total_registry_size]
+user registry
 ```
 2. Change your directory to ```/etc/munin/plugins```. Create symlinks for each plugin (```ln -s```) which you want to
 activate. Please take a look at the plugin specific documentation.

gitlab_total_registry_size

@@ -0,0 +1,27 @@
+#!/usr/bin/env python
+from lib import dirsize
+from lib import get_gitlab_instance
+import sys
+
+
+gitlab = get_gitlab_instance()
+registry_root = gitlab.get_registry_dir()
+
+if len(sys.argv) >= 2 and sys.argv[1] == 'config':
+    print('graph_title GitLab docker registry disk usage')
+    print('graph_vlabel docker registry disk usage')
+    print('graph_args -l 0 --base 1024')
+    print('graph_category gitlab')
+    print('registry_size.label docker registry')
+    print('registry_size.draw AREA')
+    sys.exit(0)
+
+registry_size = 0
+
+try:
+    registry_size = dirsize(registry_root)
+except OSError as e:
+    print(e)
+    sys.exit(1)
+
+print('registry_size.value ' + str(registry_size))

lib/gitlab.py

@@ -21,6 +21,9 @@ def get_artifacts_dir(self):
     def get_cache_dir(self):
         return os.path.join(self.shared_dir, 'cache')
 
+    def get_registry_dir(self):
+        return os.path.join(self.shared_dir, 'registry')
+
     def get_repository_dir(self):
         return os.path.join(self.get_data_dir(), 'repositories')