一.服务端
1.耗能大
package com.learning.notblock;
import lombok.extern.slf4j.Slf4j;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.ArrayList;
import java.util.List;
import static com.learning.bytebuffer.ByteBufferUtil.debugRead;
@Slf4j
public class Server {
public static void main(String[] args) throws IOException {
ByteBuffer byteBuffer = ByteBuffer.allocate(16);
ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
serverSocketChannel.configureBlocking(false);
serverSocketChannel.bind(new InetSocketAddress(8080));
List<SocketChannel> socketChannelList = new ArrayList<>();
while(true){
SocketChannel socketChannel = serverSocketChannel.accept();
if(socketChannel != null){
log.info("connected...{}", socketChannel);
socketChannel.configureBlocking(false);
socketChannelList.add(socketChannel);
}
for(SocketChannel channel : socketChannelList){
int read = channel.read(byteBuffer);
if(read > 0){
byteBuffer.flip();
debugRead(byteBuffer);
byteBuffer.clear();
log.debug("after read...{}", channel);
}
}
}
}
}
2.selector处理accept
package com.learning.notblock;
import lombok.extern.slf4j.Slf4j;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.channels.*;
import java.util.Iterator;
@Slf4j
public class SelectorServer {
public static void main(String[] args) throws IOException {
Selector selector = Selector.open();
ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
serverSocketChannel.configureBlocking(false);
SelectionKey serverSocketChannelSelectionKey = serverSocketChannel.register(selector, 0, null);
log.info("register key:{}", serverSocketChannelSelectionKey);
serverSocketChannelSelectionKey.interestOps(SelectionKey.OP_ACCEPT);
serverSocketChannel.bind(new InetSocketAddress(8080));
while(true){
selector.select();
Iterator<SelectionKey> iterator = selector.selectedKeys().iterator();
while(iterator.hasNext()){
SelectionKey key = iterator.next();
log.info("key:{}", key);
ServerSocketChannel channel = (ServerSocketChannel)key.channel();
SocketChannel socketChannel = channel.accept();
log.info("{}", socketChannel);
}
}
}
}
3.selector处理cancel
package com.learning.notblock;
import lombok.extern.slf4j.Slf4j;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.channels.*;
import java.util.Iterator;
@Slf4j
public class SelectorServer {
public static void main(String[] args) throws IOException {
Selector selector = Selector.open();
ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
serverSocketChannel.configureBlocking(false);
SelectionKey serverSocketChannelSelectionKey = serverSocketChannel.register(selector, 0, null);
log.info("register key:{}", serverSocketChannelSelectionKey);
serverSocketChannelSelectionKey.interestOps(SelectionKey.OP_ACCEPT);
serverSocketChannel.bind(new InetSocketAddress(8080));
while(true){
selector.select();
Iterator<SelectionKey> iterator = selector.selectedKeys().iterator();
while(iterator.hasNext()){
SelectionKey key = iterator.next();
log.info("key:{}", key);
key.cancel();
}
}
}
}
3.客户端正常或异常关闭
package com.learning.notblock;
import lombok.extern.slf4j.Slf4j;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.*;
import java.util.Iterator;
import static com.learning.bytebuffer.ByteBufferUtil.debugRead;
@Slf4j
public class SelectorServer {
public static void main(String[] args) throws IOException {
Selector selector = Selector.open();
ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
serverSocketChannel.configureBlocking(false);
SelectionKey serverSocketChannelSelectionKey = serverSocketChannel.register(selector, 0, null);
log.info("register key:{}", serverSocketChannelSelectionKey);
serverSocketChannelSelectionKey.interestOps(SelectionKey.OP_ACCEPT);
serverSocketChannel.bind(new InetSocketAddress(8080));
while(true){
selector.select();
Iterator<SelectionKey> iterator = selector.selectedKeys().iterator();
while(iterator.hasNext()){
SelectionKey key = iterator.next();
iterator.remove();
log.info("key:{}", key);
if (key.isAcceptable()) {
ServerSocketChannel channel = (ServerSocketChannel)key.channel();
SocketChannel socketChannel = channel.accept();
socketChannel.configureBlocking(false);
SelectionKey socketChannelSelectionKey = socketChannel.register(selector,0, null);
socketChannelSelectionKey.interestOps(SelectionKey.OP_READ);
log.info("{}", socketChannel);
} else if(key.isReadable()){
try {
SocketChannel channel = (SocketChannel) key.channel();
ByteBuffer buffer = ByteBuffer.allocate(16);
int read = channel.read(buffer);
if(read == -1){
key.cancel();
}else{
buffer.flip();
debugRead(buffer);
}
}catch(IOException e){
e.printStackTrace();
key.cancel();
}
}
}
}
}
}
4.消息边界超出扩容模式
package com.learning.notblock;
import lombok.extern.slf4j.Slf4j;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.*;
import java.util.Iterator;
import static com.learning.bytebuffer.ByteBufferUtil.debugAll;
@Slf4j
public class SelectorServer {
private static void split(ByteBuffer source) {
source.flip();
for(int i=0;i<source.limit();i++){
if (source.get(i)=='\n') {
int length = i + 1 -source.position();
ByteBuffer target = ByteBuffer.allocate(length);
for(int j=0;j<length;j++){
target.put(source.get());
}
debugAll(target);
}
}
source.compact();
}
public static void main(String[] args) throws IOException {
Selector selector = Selector.open();
ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
serverSocketChannel.configureBlocking(false);
SelectionKey serverSocketChannelSelectionKey = serverSocketChannel.register(selector, 0, null);
log.info("register key:{}", serverSocketChannelSelectionKey);
serverSocketChannelSelectionKey.interestOps(SelectionKey.OP_ACCEPT);
serverSocketChannel.bind(new InetSocketAddress(8080));
while(true){
selector.select();
Iterator<SelectionKey> iterator = selector.selectedKeys().iterator();
while(iterator.hasNext()){
SelectionKey key = iterator.next();
iterator.remove();
log.info("key:{}", key);
if (key.isAcceptable()) {
ServerSocketChannel channel = (ServerSocketChannel)key.channel();
SocketChannel socketChannel = channel.accept();
socketChannel.configureBlocking(false);
ByteBuffer buffer = ByteBuffer.allocate(16);
SelectionKey socketChannelSelectionKey = socketChannel.register(selector,0, buffer);
socketChannelSelectionKey.interestOps(SelectionKey.OP_READ);
log.info("{}", socketChannel);
} else if(key.isReadable()){
try {
SocketChannel channel = (SocketChannel) key.channel();
ByteBuffer buffer = (ByteBuffer) key.attachment();
int read = channel.read(buffer);
if(read == -1){
key.cancel();
}else{
split(buffer);
if(buffer.position() == buffer.limit()) {
ByteBuffer newBuffer = ByteBuffer.allocate(buffer.capacity() * 2);
buffer.flip();
newBuffer.put(buffer);
key.attach(newBuffer);
}
}
}catch(IOException e){
e.printStackTrace();
key.cancel();
}
}
}
}
}
}
二.客户端
package com.learning.notblock;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.channels.SocketChannel;
public class Client {
public static void main(String[] args) throws IOException {
SocketChannel sc = SocketChannel.open();
sc.connect(new InetSocketAddress("localhost", 8080));
System.out.println("waiting...");
}
}