diff options
Diffstat (limited to 'daemon/src/main/java/org/openslx/taskmanager/network/NetworkHandlerTcp.java')
-rw-r--r-- | daemon/src/main/java/org/openslx/taskmanager/network/NetworkHandlerTcp.java | 40 |
1 files changed, 30 insertions, 10 deletions
diff --git a/daemon/src/main/java/org/openslx/taskmanager/network/NetworkHandlerTcp.java b/daemon/src/main/java/org/openslx/taskmanager/network/NetworkHandlerTcp.java index 540ff42..6492f2e 100644 --- a/daemon/src/main/java/org/openslx/taskmanager/network/NetworkHandlerTcp.java +++ b/daemon/src/main/java/org/openslx/taskmanager/network/NetworkHandlerTcp.java @@ -9,10 +9,13 @@ import java.net.Socket; import java.nio.charset.StandardCharsets; import java.util.concurrent.RejectedExecutionException; import java.util.concurrent.SynchronousQueue; +import java.util.concurrent.ThreadFactory; import java.util.concurrent.ThreadPoolExecutor; import java.util.concurrent.TimeUnit; +import java.util.concurrent.atomic.AtomicInteger; -import org.apache.log4j.Logger; +import org.apache.logging.log4j.LogManager; +import org.apache.logging.log4j.Logger; import org.openslx.taskmanager.Global; /** @@ -22,14 +25,22 @@ import org.openslx.taskmanager.Global; public class NetworkHandlerTcp extends NetworkHandlerBase { - private static final Logger log = Logger.getLogger( NetworkHandlerTcp.class ); + private static final Logger log = LogManager.getLogger( NetworkHandlerTcp.class ); /** * UDP socket for sending and receiving. */ private final ServerSocket socket; - private final ThreadPoolExecutor threadPool = new ThreadPoolExecutor( 1, 8, 1, TimeUnit.MINUTES, new SynchronousQueue<Runnable>() ); + private final ThreadPoolExecutor threadPool = new ThreadPoolExecutor( 0, 64, 1, TimeUnit.MINUTES, new SynchronousQueue<Runnable>(), new ThreadFactory() { + AtomicInteger id = new AtomicInteger(); + + @Override + public Thread newThread( Runnable r ) + { + return new Thread( r, "TCP-" + id.incrementAndGet() ); + } + } ); /** * Initialize the NetworkHandler by starting threads and opening the socket. @@ -77,7 +88,7 @@ public class NetworkHandlerTcp extends NetworkHandlerBase } finally { Thread.currentThread().interrupt(); Global.doShutdown = true; - log.info( "UDP receiver finished." ); + log.info( "TCP receiver finished." ); } } @@ -86,7 +97,7 @@ public class NetworkHandlerTcp extends NetworkHandlerBase private final Socket socket; private DataOutputStream writer = null; private DataInputStream reader = null; - + public ClientTask( Socket client ) { this.socket = client; @@ -97,7 +108,6 @@ public class NetworkHandlerTcp extends NetworkHandlerBase { try { try { - socket.setSoTimeout( (int)TimeUnit.MINUTES.toMillis( 15 ) ); reader = new DataInputStream( socket.getInputStream() ); writer = new DataOutputStream( socket.getOutputStream() ); } catch ( IOException e ) { @@ -145,7 +155,7 @@ public class NetworkHandlerTcp extends NetworkHandlerBase } } } - + private void sendMsg( String reply ) throws SendException { sendMsg( reply.getBytes( StandardCharsets.UTF_8 ) ); @@ -154,6 +164,7 @@ public class NetworkHandlerTcp extends NetworkHandlerBase private void sendMsg( byte[] reply ) throws SendException { try { + socket.setSoTimeout( (int)TimeUnit.SECONDS.toMillis( 15 ) ); writer.writeInt( reply.length ); writer.write( reply ); } catch ( IOException e ) { @@ -165,6 +176,7 @@ public class NetworkHandlerTcp extends NetworkHandlerBase { int bytes; try { + socket.setSoTimeout( (int)TimeUnit.MINUTES.toMillis( 15 ) ); bytes = reader.readInt(); } catch ( IOException e ) { // This should be fine... Client went away @@ -178,16 +190,24 @@ public class NetworkHandlerTcp extends NetworkHandlerBase return ""; // Nothing to read byte[] buffer = new byte[ bytes ]; try { + socket.setSoTimeout( (int)TimeUnit.SECONDS.toMillis( 2 ) ); reader.readFully( buffer ); } catch ( IOException e ) { log.warn( "Client went away when trying to read payload" ); return null; } return new String( buffer, StandardCharsets.UTF_8 ); - } + } + + } + + private static class SendException extends Exception + { + /** + * Version for serialization. + */ + private static final long serialVersionUID = 7591668934699490535L; } - - private static class SendException extends Exception {} } |