From 1ab173bb3536182a1eb40f49e59c44bc030d649b Mon Sep 17 00:00:00 2001
From: Jean-Noël Rouvignac <jean-noel.rouvignac@forgerock.com>
Date: Tue, 06 Oct 2015 08:08:53 +0000
Subject: [PATCH] bugfix: dsconfig on loggers trigger thread leaks

---
 opendj-server-legacy/src/main/java/org/opends/server/loggers/TextAccessLogPublisher.java |   87 ++++++++++++++++++++++++++++++++++++-------
 1 files changed, 72 insertions(+), 15 deletions(-)

diff --git a/opendj-server-legacy/src/main/java/org/opends/server/loggers/TextAccessLogPublisher.java b/opendj-server-legacy/src/main/java/org/opends/server/loggers/TextAccessLogPublisher.java
index d95a719..046bc25 100644
--- a/opendj-server-legacy/src/main/java/org/opends/server/loggers/TextAccessLogPublisher.java
+++ b/opendj-server-legacy/src/main/java/org/opends/server/loggers/TextAccessLogPublisher.java
@@ -118,30 +118,72 @@
 
         if (config.isAsynchronous())
         {
-          // The asynchronous setting is being turned on.
-          if (!(writer instanceof AsynchronousTextWriter))
+          if (useAsyncWriter(config))
           {
-            writer = newAsyncWriter(mfWriter, config);
+            if (writer instanceof AsynchronousTextWriter)
+            {
+              if (hasAsyncConfigChanged(config))
+              {
+                // reinstantiate
+                final AsynchronousTextWriter previousWriter = (AsynchronousTextWriter) writer;
+                writer = newAsyncWriter(mfWriter, config);
+                previousWriter.shutdown(false);
+              }
+            }
+            else if (writer instanceof ParallelTextWriter)
+            {
+              // convert parallel to async
+              final ParallelTextWriter previousWriter = (ParallelTextWriter) writer;
+              writer = newAsyncWriter(mfWriter, config);
+              previousWriter.shutdown(false);
+            }
+            else
+            {
+              // turn async text writer on
+              writer = newAsyncWriter(mfWriter, config);
+            }
           }
-          if (!(writer instanceof ParallelTextWriter))
+          else
           {
-            writer = newParallelWriter(mfWriter, config);
+            if (writer instanceof AsynchronousTextWriter)
+            {
+              // convert async to parallel
+              final AsynchronousTextWriter previousWriter = (AsynchronousTextWriter) writer;
+              writer = newParallelWriter(mfWriter, config);
+              previousWriter.shutdown(false);
+            }
+            else if (writer instanceof ParallelTextWriter)
+            {
+              if (hasParallelConfigChanged(config))
+              {
+                // reinstantiate
+                final ParallelTextWriter previousWriter = (ParallelTextWriter) writer;
+                writer = newParallelWriter(mfWriter, config);
+                previousWriter.shutdown(false);
+              }
+            }
+            else
+            {
+              // turn parallel text writer on
+              writer = newParallelWriter(mfWriter, config);
+            }
           }
         }
         else
         {
-          // The asynchronous setting is being turned off.
-          if (writer instanceof AsynchronousTextWriter)
-          {
-            final AsynchronousTextWriter asyncWriter = (AsynchronousTextWriter) writer;
-            writer = mfWriter;
-            asyncWriter.shutdown(false);
-          }
           if (writer instanceof ParallelTextWriter)
           {
-            final ParallelTextWriter asyncWriter = (ParallelTextWriter) writer;
+            // asynchronous is being turned off, remove parallel text writers.
+            final ParallelTextWriter previousWriter = (ParallelTextWriter) writer;
             writer = mfWriter;
-            asyncWriter.shutdown(false);
+            previousWriter.shutdown(false);
+          }
+          else if (writer instanceof AsynchronousTextWriter)
+          {
+            // asynchronous is being turned off, remove async text writers.
+            final AsynchronousTextWriter previousWriter = (AsynchronousTextWriter) writer;
+            writer = mfWriter;
+            previousWriter.shutdown(false);
           }
         }
 
@@ -234,7 +276,7 @@
 
       if (cfg.isAsynchronous())
       {
-        if (cfg.getQueueSize() > 0)
+        if (useAsyncWriter(cfg))
         {
           this.writer = newAsyncWriter(theWriter, cfg);
         }
@@ -269,6 +311,21 @@
     cfg.addFileBasedAccessChangeListener(this);
   }
 
+  private boolean useAsyncWriter(FileBasedAccessLogPublisherCfg config)
+  {
+    return config.getQueueSize() > 0;
+  }
+
+  private boolean hasAsyncConfigChanged(FileBasedAccessLogPublisherCfg newConfig)
+  {
+    return hasParallelConfigChanged(newConfig) && cfg.getQueueSize() != newConfig.getQueueSize();
+  }
+
+  private boolean hasParallelConfigChanged(FileBasedAccessLogPublisherCfg newConfig)
+  {
+    return !cfg.dn().equals(newConfig.dn()) && cfg.isAutoFlush() != newConfig.isAutoFlush();
+  }
+
   private AsynchronousTextWriter newAsyncWriter(MultifileTextWriter mfWriter, FileBasedAccessLogPublisherCfg config)
   {
     String name = "Asynchronous Text Writer for " + config.dn();

--
Gitblit v1.10.0