Added parameter to set await termination timeout

Author: snorwin

src/main/java/com/splunk/logging/HttpEventCollectorLog4jAppender.java

@@ -157,6 +157,7 @@ public static HttpEventCollectorLog4jAppender createAppender(
             @PluginAttribute(value = "call_timeout", defaultLong = HttpEventCollectorSender.TimeoutSettings.DEFAULT_CALL_TIMEOUT) final long callTimeout,
             @PluginAttribute(value = "read_timeout", defaultLong = HttpEventCollectorSender.TimeoutSettings.DEFAULT_READ_TIMEOUT) final long readTimeout,
             @PluginAttribute(value = "write_timeout", defaultLong = HttpEventCollectorSender.TimeoutSettings.DEFAULT_WRITE_TIMEOUT) final long writeTimeout,
+            @PluginAttribute(value = "await_termination_timeout", defaultLong = HttpEventCollectorSender.TimeoutSettings.DEFAULT_AWAIT_TERMINATION_TIMEOUT) final long awaitTerminationTimeout,
             @PluginElement("Layout") Layout<? extends Serializable> layout,
             @PluginElement("Filter") final Filter filter
     )
@@ -219,7 +220,7 @@ public static HttpEventCollectorLog4jAppender createAppender(
                 disableCertificateValidation,
                 eventBodySerializer,
                 eventHeaderSerializer,
-                new HttpEventCollectorSender.TimeoutSettings(connectTimeout, callTimeout, readTimeout, writeTimeout)
+                new HttpEventCollectorSender.TimeoutSettings(connectTimeout, callTimeout, readTimeout, writeTimeout, awaitTerminationTimeout)
         );
     }
 

src/main/java/com/splunk/logging/HttpEventCollectorLogbackAppender.java

@@ -383,6 +383,14 @@ public long getWriteTimeout(long milliseconds) {
         return this.timeoutSettings.writeTimeout = milliseconds;
     }
 
+    public void setAwaitTerminationTimeout(long milliseconds) {
+        this.timeoutSettings.awaitTerminationTimeout = milliseconds;
+    }
+
+    public long getAwaitTerminationTimeout(long milliseconds) {
+        return this.timeoutSettings.awaitTerminationTimeout = milliseconds;
+    }
+
     private static long parseLong(String string, int defaultValue) {
         try {
             return Long.parseLong(string);

src/main/java/com/splunk/logging/HttpEventCollectorLoggingHandler.java

@@ -114,6 +114,7 @@ public final class HttpEventCollectorLoggingHandler extends Handler {
     private final String CallTimeoutConfTag = "call_timeout";
     private final String ReadTimeoutConfTag = "read_timeout";
     private final String WriteTimeoutConfTag = "write_timeout";
+    private final String AwaitTerminationTimeoutConfTag = "await_termination_timeout";
 
     /** HttpEventCollectorLoggingHandler c-or */
     public HttpEventCollectorLoggingHandler() {
@@ -165,7 +166,8 @@ public HttpEventCollectorLoggingHandler() {
             getConfigurationNumericProperty(ConnectTimeoutConfTag, HttpEventCollectorSender.TimeoutSettings.DEFAULT_CONNECT_TIMEOUT),
             getConfigurationNumericProperty(CallTimeoutConfTag, HttpEventCollectorSender.TimeoutSettings.DEFAULT_CALL_TIMEOUT),
             getConfigurationNumericProperty(ReadTimeoutConfTag, HttpEventCollectorSender.TimeoutSettings.DEFAULT_READ_TIMEOUT),
-            getConfigurationNumericProperty(WriteTimeoutConfTag, HttpEventCollectorSender.TimeoutSettings.DEFAULT_WRITE_TIMEOUT)
+            getConfigurationNumericProperty(WriteTimeoutConfTag, HttpEventCollectorSender.TimeoutSettings.DEFAULT_WRITE_TIMEOUT),
+            getConfigurationNumericProperty(AwaitTerminationTimeoutConfTag, HttpEventCollectorSender.TimeoutSettings.DEFAULT_AWAIT_TERMINATION_TIMEOUT)
         );
 
         if ("raw".equalsIgnoreCase(type)) {

src/main/java/com/splunk/logging/HttpEventCollectorSender.java

@@ -29,6 +29,7 @@
 import java.io.Serializable;
 import java.security.cert.CertificateException;
 import java.util.*;
+import java.util.concurrent.TimeUnit;
 import java.util.stream.Collectors;
 
 
@@ -263,6 +264,13 @@ public static void putIfPresent(JsonObject collection, String tag, Object value)
     private void stopHttpClient() {
         if (httpClient != null) {
             httpClient.dispatcher().executorService().shutdown();
+
+            if (timeoutSettings.awaitTerminationTimeout > 0) {
+                try {
+                    httpClient.dispatcher().executorService().awaitTermination(timeoutSettings.awaitTerminationTimeout, TimeUnit.MILLISECONDS);
+                } catch (InterruptedException ignored) { /* nop */ }
+            }
+
             httpClient = null;
         }
     }
@@ -392,19 +400,22 @@ public static class TimeoutSettings {
         public static final long DEFAULT_WRITE_TIMEOUT = 0; // 0 means no timeout
         public static final long DEFAULT_CALL_TIMEOUT = 0;
         public static final long DEFAULT_READ_TIMEOUT = 0;
+        public static final long DEFAULT_AWAIT_TERMINATION_TIMEOUT = 0;
 
         public long connectTimeout = DEFAULT_CONNECT_TIMEOUT;
         public long callTimeout = DEFAULT_CALL_TIMEOUT;
         public long readTimeout = DEFAULT_READ_TIMEOUT;
         public long writeTimeout = DEFAULT_WRITE_TIMEOUT;
+        public long awaitTerminationTimeout = DEFAULT_AWAIT_TERMINATION_TIMEOUT;
 
         public TimeoutSettings() {}
 
-        public TimeoutSettings(long connectTimeout, long callTimeout, long readTimeout, long writeTimeout) {
+        public TimeoutSettings(long connectTimeout, long callTimeout, long readTimeout, long writeTimeout, long awaitTerminationTimeout) {
             this.connectTimeout = connectTimeout;
             this.callTimeout = callTimeout;
             this.readTimeout = readTimeout;
             this.writeTimeout = writeTimeout;
+            this.awaitTerminationTimeout = awaitTerminationTimeout;
         }
     }
 }

src/test/resources/log4j2.xml

@@ -45,6 +45,7 @@ under the License.
               batch_size_count="0"
               batch_interval="0"
               connect_timeout="5000"
+              await_termination_timeout="1000"
               disableCertificateValidation="true">
             <PatternLayout pattern="%m"/>
         </SplunkHttp>

src/test/resources/logback.xml

@@ -60,6 +60,7 @@ under the License.
         <messageFormat>text</messageFormat>
         <middleware>HttpEventCollectorUnitTestMiddleware</middleware>
         <connectTimeout>5000</connectTimeout>
+        <awaitTerminationTieout>2000</awaitTerminationTieout>
         <layout class="ch.qos.logback.classic.PatternLayout">
             <pattern>%msg</pattern>
         </layout>