email: 849586227@qq.com30 | *
blog: http://blog.csdn.net/wangyangzhizhou31 | *
This Acceptor provides a NIO mode to accept client sockets.
32 | */ 33 | public final class Acceptor extends Thread { 34 | 35 | private static final Logger LOGGER = LoggerFactory 36 | .getLogger(Acceptor.class); 37 | private final int port; 38 | private final Selector selector; 39 | private final ServerSocketChannel serverChannel; 40 | private long acceptCount; 41 | private static final AcceptIdGenerator IdGenerator = new AcceptIdGenerator(); 42 | private ReactorPool reactorPool; 43 | private Listemail: 849586227@qq.com16 | *
blog: http://blog.csdn.net/wangyangzhizhou17 | *
A buffer pool. This pool provide a buffer queue for the operation of offer and poll.
18 | */ 19 | public final class BufferPool { 20 | private static final Logger LOGGER = LoggerFactory 21 | .getLogger(BufferPool.class); 22 | private final int chunkSize; 23 | private final ConcurrentLinkedQueueemail: 849586227@qq.com22 | *
blog: http://blog.csdn.net/wangyangzhizhou23 | *
Reactor reacts all sockets.
24 | */ 25 | public final class Reactor extends Thread { 26 | private static final Logger LOGGER = LoggerFactory.getLogger(Reactor.class); 27 | private final Selector selector; 28 | private final ConcurrentLinkedQueueemail: 849586227@qq.com13 | *
blog: http://blog.csdn.net/wangyangzhizhou14 | *
Reactor pool. Socket connections are polling to the reactor of this pool.
15 | */ 16 | public class ReactorPool { 17 | private final Reactor[] reactors; 18 | private volatile int nextReactor; 19 | private String name = "reactor"; 20 | private BufferPool bufferPool; 21 | 22 | public ReactorPool(int poolSize, Handler handler) throws IOException { 23 | reactors = new Reactor[poolSize]; 24 | for (int i = 0; i < poolSize; i++) { 25 | Reactor reactor = new Reactor(name + "-" + i, handler, this); 26 | reactors[i] = reactor; 27 | reactor.start(); 28 | } 29 | this.bufferPool = new BufferPool(1024 * 1024 * 52, 10 * 1024); 30 | } 31 | 32 | public Reactor getNextReactor() { 33 | if (++nextReactor == reactors.length) { 34 | nextReactor = 0; 35 | } 36 | return reactors[nextReactor]; 37 | } 38 | 39 | public BufferPool getBufferPool() { 40 | return bufferPool; 41 | } 42 | } 43 | -------------------------------------------------------------------------------- /src/main/java/com/seaboat/net/reactor/connection/Connection.java: -------------------------------------------------------------------------------- 1 | package com.seaboat.net.reactor.connection; 2 | 3 | import java.io.EOFException; 4 | import java.io.IOException; 5 | import java.nio.ByteBuffer; 6 | import java.nio.channels.SelectionKey; 7 | import java.nio.channels.Selector; 8 | import java.nio.channels.SocketChannel; 9 | import java.util.LinkedList; 10 | import java.util.List; 11 | import java.util.concurrent.ConcurrentLinkedQueue; 12 | 13 | import org.slf4j.Logger; 14 | import org.slf4j.LoggerFactory; 15 | 16 | import com.seaboat.net.reactor.Reactor; 17 | 18 | /** 19 | * 20 | * @author seaboat 21 | * @date 2016-08-25 22 | * @version 1.0 23 | *email: 849586227@qq.com24 | *
blog: http://blog.csdn.net/wangyangzhizhou25 | *
This is a abstraction of client connection.
26 | */ 27 | public class Connection { 28 | private static final Logger LOGGER = LoggerFactory 29 | .getLogger(Connection.class); 30 | private long id; 31 | private SocketChannel channel; 32 | private SelectionKey selectionKey; 33 | private ByteBuffer readBuffer; 34 | private ConcurrentLinkedQueueconnection event handler interface.6 | * @author 7 | *
seaboat8 | *
email: 849586227@qq.com9 | *
blog: http://blog.csdn.net/wangyangzhizhou10 | * @version 1.0 11 | */ 12 | public interface ConnectionEventHandler { 13 | 14 | public void event(Connection connection, int event); 15 | 16 | } 17 | -------------------------------------------------------------------------------- /src/main/java/com/seaboat/net/reactor/connection/ConnectionEvents.java: -------------------------------------------------------------------------------- 1 | package com.seaboat.net.reactor.connection; 2 | 3 | /** 4 | * 5 | *
connection event.6 | * @author 7 | *
seaboat8 | *
email: 849586227@qq.com9 | *
blog: http://blog.csdn.net/wangyangzhizhou10 | * @version 1.0 11 | */ 12 | public class ConnectionEvents { 13 | 14 | public static int ACCEPT = 1; 15 | 16 | public static int REGISTE = 2; 17 | 18 | public static int CLOSE = 4; 19 | 20 | } 21 | -------------------------------------------------------------------------------- /src/main/java/com/seaboat/net/reactor/connection/ConnectionFactory.java: -------------------------------------------------------------------------------- 1 | package com.seaboat.net.reactor.connection; 2 | 3 | import java.nio.channels.SocketChannel; 4 | 5 | import com.seaboat.net.reactor.Reactor; 6 | 7 | /** 8 | * 9 | *
a connection factory.10 | * @author 11 | *
seaboat12 | *
email: 849586227@qq.com13 | *
blog: http://blog.csdn.net/wangyangzhizhou14 | * @version 1.0 15 | */ 16 | public interface ConnectionFactory { 17 | 18 | public Connection createConnection(SocketChannel channel, long id, 19 | Reactor reactor); 20 | 21 | } 22 | -------------------------------------------------------------------------------- /src/main/java/com/seaboat/net/reactor/connection/DefaultConnectionFactory.java: -------------------------------------------------------------------------------- 1 | package com.seaboat.net.reactor.connection; 2 | 3 | import java.nio.channels.SocketChannel; 4 | 5 | import com.seaboat.net.reactor.Reactor; 6 | 7 | /** 8 | * 9 | *
the default connection factory.10 | * @author 11 | *
seaboat12 | *
email: 849586227@qq.com13 | *
blog: http://blog.csdn.net/wangyangzhizhou14 | * @version 1.0 15 | */ 16 | public class DefaultConnectionFactory implements ConnectionFactory { 17 | 18 | public Connection createConnection(SocketChannel channel, long id, 19 | Reactor reactor) { 20 | return new Connection(channel, id, reactor); 21 | } 22 | 23 | } 24 | -------------------------------------------------------------------------------- /src/main/java/com/seaboat/net/reactor/handler/Handler.java: -------------------------------------------------------------------------------- 1 | package com.seaboat.net.reactor.handler; 2 | 3 | import java.io.IOException; 4 | 5 | import com.seaboat.net.reactor.connection.Connection; 6 | 7 | /** 8 | * 9 | * @author seaboat 10 | * @date 2016-08-25 11 | * @version 1.0 12 | *
email: 849586227@qq.com13 | *
blog: http://blog.csdn.net/wangyangzhizhou14 | *
This Handler will be call when there is a data having be ready.
15 | */ 16 | public interface Handler { 17 | 18 | public void handle(Connection connection) throws IOException; 19 | 20 | } 21 | -------------------------------------------------------------------------------- /src/test/java/com/seaboat/net/reactor/test/Bootstrap.java: -------------------------------------------------------------------------------- 1 | package com.seaboat.net.reactor.test; 2 | 3 | import org.slf4j.Logger; 4 | import org.slf4j.LoggerFactory; 5 | 6 | import com.seaboat.net.reactor.Acceptor; 7 | import com.seaboat.net.reactor.ReactorPool; 8 | import com.seaboat.net.reactor.handler.Handler; 9 | 10 | /** 11 | * 12 | * @author seaboat 13 | * @date 2016-08-25 14 | * @version 1.0 15 | *email: 849586227@qq.com16 | *
blog: http://blog.csdn.net/wangyangzhizhou17 | *
The reactor bootstrap.
18 | */ 19 | public class Bootstrap { 20 | private static final Logger LOGGER = LoggerFactory 21 | .getLogger(Bootstrap.class); 22 | private static String acceptorName = "acceptor-thread"; 23 | private static String host = "localhost"; 24 | private static int port = 6789; 25 | 26 | public static void main(String[] args) { 27 | try { 28 | LOGGER.info("starting up ......"); 29 | Handler handler = new MyHandler(); 30 | ReactorPool reactorPool = new ReactorPool(Runtime.getRuntime().availableProcessors(), handler); 31 | new Acceptor(reactorPool, acceptorName, host, port).start(); 32 | LOGGER.info("started up successfully."); 33 | while (true) { 34 | Thread.sleep(300 * 1000); 35 | } 36 | } catch (Throwable e) { 37 | LOGGER.error(" launch error", e); 38 | System.exit(-1); 39 | } 40 | } 41 | } 42 | -------------------------------------------------------------------------------- /src/test/java/com/seaboat/net/reactor/test/BufferPoolTest.java: -------------------------------------------------------------------------------- 1 | package com.seaboat.net.reactor.test; 2 | 3 | import java.nio.ByteBuffer; 4 | import java.util.ArrayList; 5 | 6 | import com.seaboat.net.reactor.BufferPool; 7 | 8 | public class BufferPoolTest { 9 | 10 | @org.junit.Test 11 | public void Test() { 12 | BufferPool pool = new BufferPool(1024 * 5, 1024); 13 | int i = pool.capacity(); 14 | ArrayListemail: 849586227@qq.com19 | *
blog: http://blog.csdn.net/wangyangzhizhou20 | *
Demo.
21 | */ 22 | public class MyHandler implements Handler { 23 | 24 | private static final Logger LOGGER = LoggerFactory 25 | .getLogger(MyHandler.class); 26 | private long readSize; 27 | 28 | /** 29 | * The logic to deal with the received data. 30 | * 31 | * It means that reactor will trigger this function once the data is received. 32 | * @throws IOException 33 | */ 34 | public void handle(Connection connection) throws IOException { 35 | Buffer buff = connection.getReadBuffer(); 36 | readSize = +readSize + buff.position(); 37 | LOGGER.info(connection.getId() + " connection has receive " + readSize); 38 | if (readSize % 5 == 0) { 39 | ByteBuffer sendBuffer = ByteBuffer.allocate(10); 40 | ; 41 | ByteBuffer.wrap("hello".getBytes(connection.getCharset())); 42 | connection.WriteToQueue(sendBuffer); 43 | connection.write(); 44 | } 45 | 46 | } 47 | 48 | } 49 | -------------------------------------------------------------------------------- /src/test/java/com/seaboat/net/reactor/test/SocketClient.java: -------------------------------------------------------------------------------- 1 | package com.seaboat.net.reactor.test; 2 | 3 | import java.net.Socket; 4 | 5 | import org.junit.Test; 6 | 7 | public class SocketClient { 8 | 9 | @Test 10 | public void visit() { 11 | for(int i = 0 ; i <= 100; i++){ 12 | new Thread(){ 13 | public void run(){ 14 | try { 15 | Socket socket = new Socket("132.121.95.184", 6789); 16 | for (;;) { 17 | socket.getOutputStream().write(new byte[] { 1, 2, 3, 4, 5 }); 18 | Thread.currentThread().sleep(100); 19 | } 20 | } catch (Exception e) { 21 | e.printStackTrace(); 22 | } 23 | }; 24 | }.start();; 25 | } 26 | while(true){ 27 | try { 28 | Thread.currentThread().sleep(3000); 29 | } catch (InterruptedException e) { 30 | // TODO Auto-generated catch block 31 | e.printStackTrace(); 32 | } 33 | } 34 | } 35 | } 36 | --------------------------------------------------------------------------------