Server updated to have a single Server class which instantiates all the server components and can start the 2 server threads.

Author: jjenkov

src/main/java/com/jenkov/nioserver/IMessageReader.java

@@ -9,6 +9,8 @@
  */
 public interface IMessageReader {
 
+    public void init(MessageBuffer readMessageBuffer);
+
     public void read(Socket socket, ByteBuffer byteBuffer) throws IOException;
 
     public List<Message> getMessages();

src/main/java/com/jenkov/nioserver/IMessageReaderFactory.java

@@ -6,4 +6,5 @@
 public interface IMessageReaderFactory {
 
     public IMessageReader createMessageReader();
+
 }

src/main/java/com/jenkov/nioserver/Server.java

@@ -1,54 +1,46 @@
 package com.jenkov.nioserver;
 
 import java.io.IOException;
-import java.net.InetSocketAddress;
-import java.net.ServerSocket;
-import java.nio.channels.ServerSocketChannel;
-import java.nio.channels.SocketChannel;
 import java.util.Queue;
+import java.util.concurrent.ArrayBlockingQueue;
+import java.util.concurrent.BlockingQueue;
 
 /**
- * Created by jjenkov on 19-10-2015.
+ * Created by jjenkov on 24-10-2015.
  */
-public class Server implements Runnable{
+public class Server {
 
-    private int tcpPort = 0;
-    private ServerSocketChannel serverSocket = null;
+    private SocketAccepter  socketAccepter  = null;
+    private SocketProcessor socketProcessor = null;
 
-    private Queue socketQueue = null;
+    private int tcpPort = 0;
+    private IMessageReaderFactory messageReaderFactory = null;
+    private IMessageProcessor     messageProcessor = null;
 
-    public Server(int tcpPort, Queue socketQueue)  {
-        this.tcpPort     = tcpPort;
-        this.socketQueue = socketQueue;
+    public Server(int tcpPort, IMessageReaderFactory messageReaderFactory, IMessageProcessor messageProcessor) {
+        this.tcpPort = tcpPort;
+        this.messageReaderFactory = messageReaderFactory;
+        this.messageProcessor = messageProcessor;
     }
 
+    public void start() throws IOException {
 
+        Queue socketQueue = new ArrayBlockingQueue(1024); //move 1024 to ServerConfig
 
-    public void run() {
-        try{
-            this.serverSocket = ServerSocketChannel.open();
-            this.serverSocket.bind(new InetSocketAddress(tcpPort));
-        } catch(IOException e){
-            e.printStackTrace();
-            return;
-        }
-
+        this.socketAccepter  = new SocketAccepter(tcpPort, socketQueue);
 
-        while(true){
-            try{
-                SocketChannel socketChannel = this.serverSocket.accept();
 
-                System.out.println("Socket accepted");
+        MessageBuffer readBuffer  = new MessageBuffer();
+        MessageBuffer writeBuffer = new MessageBuffer();
 
-                //todo check if the queue can even accept more sockets.
-                this.socketQueue.add(new Socket(socketChannel));
+        this.socketProcessor = new SocketProcessor(socketQueue, readBuffer, writeBuffer,  this.messageReaderFactory, this.messageProcessor);
 
+        Thread accepterThread  = new Thread(this.socketAccepter);
+        Thread processorThread = new Thread(this.socketProcessor);
 
-            } catch(IOException e){
-                e.printStackTrace();
-            }
+        accepterThread.start();
+        processorThread.start();
+    }
 
-        }
 
-    }
 }

src/main/java/com/jenkov/nioserver/Socket.java

@@ -17,7 +17,6 @@ public class Socket {
 
     public boolean endOfStreamReached = false;
 
-
     public Socket() {
     }
 
@@ -40,7 +39,6 @@ public int read(ByteBuffer byteBuffer) throws IOException {
         return totalBytesRead;
     }
 
-
     public int write(ByteBuffer byteBuffer) throws IOException{
         int bytesWritten      = this.socketChannel.write(byteBuffer);
         int totalBytesWritten = bytesWritten;

src/main/java/com/jenkov/nioserver/SocketAccepter.java

@@ -0,0 +1,53 @@
+package com.jenkov.nioserver;
+
+import java.io.IOException;
+import java.net.InetSocketAddress;
+import java.net.ServerSocket;
+import java.nio.channels.ServerSocketChannel;
+import java.nio.channels.SocketChannel;
+import java.util.Queue;
+
+/**
+ * Created by jjenkov on 19-10-2015.
+ */
+public class SocketAccepter implements Runnable{
+
+    private int tcpPort = 0;
+    private ServerSocketChannel serverSocket = null;
+
+    private Queue socketQueue = null;
+
+    public SocketAccepter(int tcpPort, Queue socketQueue)  {
+        this.tcpPort     = tcpPort;
+        this.socketQueue = socketQueue;
+    }
+
+
+
+    public void run() {
+        try{
+            this.serverSocket = ServerSocketChannel.open();
+            this.serverSocket.bind(new InetSocketAddress(tcpPort));
+        } catch(IOException e){
+            e.printStackTrace();
+            return;
+        }
+
+
+        while(true){
+            try{
+                SocketChannel socketChannel = this.serverSocket.accept();
+
+                System.out.println("Socket accepted: " + socketChannel);
+
+                //todo check if the queue can even accept more sockets.
+                this.socketQueue.add(new Socket(socketChannel));
+
+            } catch(IOException e){
+                e.printStackTrace();
+            }
+
+        }
+
+    }
+}

src/main/java/com/jenkov/nioserver/ServerCore.java renamed to src/main/java/com/jenkov/nioserver/SocketProcessor.java

@@ -10,7 +10,7 @@
 /**
  * Created by jjenkov on 16-10-2015.
  */
-public class ServerCore implements Runnable {
+public class SocketProcessor implements Runnable {
 
     private Queue<Socket>  inboundSocketQueue   = null;
 
@@ -37,7 +37,7 @@ public class ServerCore implements Runnable {
     private Set<Socket> nonEmptyToEmptySockets = new HashSet<>();
 
 
-    public ServerCore(Queue<Socket> inboundSocketQueue, MessageBuffer readMessageBuffer, MessageBuffer writeMessageBuffer, IMessageReaderFactory messageReaderFactory, IMessageProcessor messageProcessor) throws IOException {
+    public SocketProcessor(Queue<Socket> inboundSocketQueue, MessageBuffer readMessageBuffer, MessageBuffer writeMessageBuffer, IMessageReaderFactory messageReaderFactory, IMessageProcessor messageProcessor) throws IOException {
         this.inboundSocketQueue = inboundSocketQueue;
 
         this.readMessageBuffer    = readMessageBuffer;
@@ -84,6 +84,8 @@ public void takeNewSockets() throws IOException {
             newSocket.socketChannel.configureBlocking(false);
 
             newSocket.messageReader = this.messageReaderFactory.createMessageReader();
+            newSocket.messageReader.init(this.readMessageBuffer);
+
             newSocket.messageWriter = new MessageWriter();
 
             this.socketMap.put(newSocket.socketId, newSocket);

src/main/java/com/jenkov/nioserver/example/Main.java

@@ -13,16 +13,6 @@
 public class Main {
 
     public static void main(String[] args) throws IOException {
-        BlockingQueue inboundSocketQueue = new ArrayBlockingQueue(1024);
-
-        Server server = new Server(9999, inboundSocketQueue);
-
-        Thread serverThread = new Thread(server);
-        serverThread.start();
-
-        MessageBuffer readMessageBuffer  = new MessageBuffer();
-        MessageBuffer writeMessageBuffer = new MessageBuffer();
-        IMessageReaderFactory messageReaderFactory = new HttpMessageReaderFactory(readMessageBuffer);
 
         String httpResponse = "HTTP/1.1 200 OK\r\n" +
                 "Content-Length: 38\r\n" +
@@ -42,9 +32,10 @@ public static void main(String[] args) throws IOException {
             writeProxy.enqueue(response);
         };
 
-        ServerCore serverCore       = new ServerCore(inboundSocketQueue, readMessageBuffer, writeMessageBuffer, messageReaderFactory, messageProcessor);
-        Thread     serverCoreThread = new Thread(serverCore);
-        serverCoreThread.start();
+        Server server = new Server(9999, new HttpMessageReaderFactory(), messageProcessor);
+
+        server.start();
+
     }
 
 

src/main/java/com/jenkov/nioserver/http/HttpMessageReader.java

@@ -20,8 +20,12 @@ public class HttpMessageReader implements IMessageReader {
     private List<Message> completeMessages = new ArrayList<Message>();
     private Message       nextMessage      = null;
 
-    public HttpMessageReader(MessageBuffer messageBuffer) {
-        this.messageBuffer        = messageBuffer;
+    public HttpMessageReader() {
+    }
+
+    @Override
+    public void init(MessageBuffer readMessageBuffer) {
+        this.messageBuffer        = readMessageBuffer;
         this.nextMessage          = messageBuffer.getMessage();
         this.nextMessage.metaData = new HttpHeaders();
     }

src/main/java/com/jenkov/nioserver/http/HttpMessageReaderFactory.java

@@ -9,14 +9,11 @@
  */
 public class HttpMessageReaderFactory implements IMessageReaderFactory {
 
-    private MessageBuffer messageBuffer = null;
-
-    public HttpMessageReaderFactory(MessageBuffer messageBuffer) {
-        this.messageBuffer = messageBuffer;
+    public HttpMessageReaderFactory() {
     }
 
     @Override
     public IMessageReader createMessageReader() {
-        return new HttpMessageReader(this.messageBuffer);
+        return new HttpMessageReader();
     }
 }