Merge pull request #3 from shakeelmohamed/master

damiendallimore · damiendallimore · commit 18b1b289ac37 · 2015-06-20T17:25:50.000+07:00
Tcp appender: add methods for changing the socket buffer size
diff --git a/src/com/splunk/logging/SplunkRawTCPInput.java b/src/com/splunk/logging/SplunkRawTCPInput.java
@@ -14,6 +14,7 @@
  */
 
 public class SplunkRawTCPInput extends SplunkInput{
+	private static int SOCKET_BUFFER_SIZE = 8 * 1024; // Default to 8192
 
 	// connection props
 	private String host = "";
@@ -24,6 +25,14 @@ public class SplunkRawTCPInput extends SplunkInput{
 	private OutputStream ostream;
 	private Writer writerOut = null;
 
+	public static int getSocketBufferSize() {
+		return SOCKET_BUFFER_SIZE;
+	}
+
+	public static void setSocketBufferSize(int bufferSize) {
+		SOCKET_BUFFER_SIZE = bufferSize;
+	}
+
 	/**
 	 * Create a SplunkRawTCPInput object to send events to Splunk via Raw TCP
 	 * 
@@ -50,6 +59,8 @@ private void openStream() throws Exception {
 
 		streamSocket = new Socket(host, port);
 		if (streamSocket.isConnected()) {
+			streamSocket.setSendBufferSize(getSocketBufferSize());
+			streamSocket.setReceiveBufferSize(getSocketBufferSize());
 			ostream = streamSocket.getOutputStream();
 			writerOut = new OutputStreamWriter(ostream, "UTF8");
 		}
