From 5935f86e49df1f2a13a9c90daa8a0ea2d2754871 Mon Sep 17 00:00:00 2001
From: Greg Farnum <gfarnum@redhat.com>
Date: Tue, 12 May 2015 19:53:00 -0700
Subject: [PATCH] ceph: enable mds log rotation

Signed-off-by: Greg Farnum <gfarnum@redhat.com>
---
 tasks/ceph.py        | 53 ++++++++++++++++++++++++++++++++++++++++++++
 tasks/logrotate.conf | 11 +++++++++
 2 files changed, 64 insertions(+)
 create mode 100644 tasks/logrotate.conf

diff --git a/tasks/ceph.py b/tasks/ceph.py
index f1981f39cb340..8396567210f7a 100644
--- a/tasks/ceph.py
+++ b/tasks/ceph.py
@@ -11,6 +11,7 @@ import logging
 import os
 import json
 import time
+import gevent
 
 from ceph_manager import CephManager, write_conf, DEFAULT_CONF_PATH
 from tasks.cephfs.filesystem import Filesystem
@@ -71,10 +72,62 @@ def ceph_log(ctx, config):
             )
         )
 
+    class Rotater:
+        stopping = False
+        def invoke_logrotate(self):
+            #1) install ceph-test.conf in /etc/logrotate.d
+            #2) continuously loop over logrotate invocation with ceph-test.conf
+            while not self.stopping:
+                time.sleep(30)
+                run.wait(
+                    ctx.cluster.run(
+                        args=['sudo', 'logrotate', '/etc/logrotate.d/ceph-test.conf'
+                          ],
+                        wait=False,
+                    )
+                )
+
+        def begin(self):
+            self.thread = gevent.spawn(self.invoke_logrotate)
+
+        def end(self):
+            self.stopping = True
+            self.thread.get()
+            
+    def write_rotate_conf(ctx):
+        testdir = teuthology.get_testdir(ctx)
+        rotate_conf_path = os.path.join(os.path.dirname(__file__), 'logrotate.conf')
+        conf = file(rotate_conf_path, 'rb').read() # does this leak an fd or anything?
+        for remote in ctx.cluster.remotes.iterkeys():
+            teuthology.write_file(remote=remote,
+                                  path='{tdir}/logrotate.mds.conf'.format(tdir=testdir),
+                                  data=conf
+                              )
+            remote.run(
+                args=[
+                    'sudo',
+                    'mv',
+                    '{tdir}/logrotate.mds.conf'.format(tdir=testdir),
+                    '/etc/logrotate.d/ceph-test.conf'
+                ]
+            )
+
+    if ctx.config.get('mds-log-rotate'):
+        log.info('Setting up mds logrotate')
+        write_rotate_conf(ctx)
+        logrotater = Rotater()
+        logrotater.begin()
     try:
         yield
 
     finally:
+        if ctx.config.get('mds-log-rotate'):
+            log.info('Shutting down mds logrotate')
+            logrotater.end()
+            ctx.cluster.run(
+                args=['sudo', 'rm', '/etc/logrotate.d/ceph-test.conf'
+                  ]
+            )
         if ctx.archive is not None and \
                 not (ctx.config.get('archive-on-error') and ctx.summary['success']):
             # and logs
diff --git a/tasks/logrotate.conf b/tasks/logrotate.conf
new file mode 100644
index 0000000000000..eeb99afdfafd5
--- /dev/null
+++ b/tasks/logrotate.conf
@@ -0,0 +1,11 @@
+/var/log/ceph/*mds*.log {
+    rotate 100
+    size 1M
+    compress
+    sharedscripts
+    postrotate
+        killall ceph-mds -1 || true
+    endscript
+    missingok
+    notifempty
+}
\ No newline at end of file
-- 
2.39.5