这篇来看看AbstractPollingIoConnector抽象类,它用于用于实现客户端连接的轮询策略。处理逻辑基本上和上一篇文章说的AbstractPollingIoAcceptor类似,它继承自AbstractIoConnector,两个泛型参数分别是所处理的会话和客户端socket连接。底层的sockets会被不断检测,并当有任何一个socket需要被处理时就会被唤醒去处理。这个类封装了客户端socket的bind,connect和dispose等动作,其成员变量Executor用于发起连接请求,另一个AbstractPollingIoProcessor用于处理已经连接客户端的I/O操作请求,如读写和关闭连接。
其最重要的几个成员变量是:
private final Queue<ConnectionRequest> connectQueue = new ConcurrentLinkedQueue<ConnectionRequest>();//连接队列
private final Queue<ConnectionRequest> cancelQueue = new ConcurrentLinkedQueue<ConnectionRequest>();// 取消连接队列
先来看看当服务端调用connect后的处理过程:
protected final ConnectFuture connect0(
SocketAddress remoteAddress, SocketAddress localAddress,
IoSessionInitializer<? extends ConnectFuture> sessionInitializer) {
H handle = null;
boolean success = false;
try {
handle = newHandle(localAddress);
if (connect(handle, remoteAddress)) {//若已经连接服务器成功
ConnectFuture future = new DefaultConnectFuture();
T session = newSession(processor, handle);//创建新会话
finishSessionInitialization(session, future, sessionInitializer);//结束会话初始化
session.getProcessor().add(session);//将剩下的处理交给IoProcessor
success = true;
return future;
}
success = true;
} catch (Exception e) {
return DefaultConnectFuture.newFailedFuture(e);
} finally {
if (!success && handle != null) {
try {
close(handle);
} catch (Exception e) {
ExceptionMonitor.getInstance().exceptionCaught(e);
}
}
}
ConnectionRequest request = new ConnectionRequest(handle, sessionInitializer);
connectQueue.add(request);//连接请求加入连接队列中
startupWorker();//开启工作线程处理连接请求
wakeup();//中断select操作
return request;
}
真正的负责处理客户端请求的工作都是Worker线程完成的,
private class Worker implements Runnable {
public void run() {
int nHandles = 0;
while (selectable) {
try {
int timeout = (int)Math.min(getConnectTimeoutMillis(), 1000L);//等待超时时间
boolean selected = select(timeout);//在超时时限内查看是否有可以被处理的选择键(状态
nHandles += registerNew();//取出连接队列队头的连接请求,将其注册一个用于连接的新的客户端socket, 并把它加入连接轮询池中
if (selected) {
nHandles-= processSessions(selectedHandles());//处理连接请求
}
processTimedOutSessions(allHandles());//处理超时连接请求
nHandles -= cancelKeys();
if (nHandles == 0) {
synchronized (lock) {
if (connectQueue.isEmpty()) {
worker = null;
break;
}
}
}
} catch (Throwable e) {
ExceptionMonitor.getInstance().exceptionCaught(e);
try {
Thread.sleep(1000);
} catch (InterruptedException e1) {
ExceptionMonitor.getInstance().exceptionCaught(e1);
}
}
}
if (selectable && isDisposing()) {
selectable = false;
try {
if (createdProcessor) {
processor.dispose();
}
} finally {
try {
synchronized (disposalLock) {
if (isDisposing()) {
destroy();
}
}
} catch (Exception e) {
ExceptionMonitor.getInstance().exceptionCaught(e);
} finally {
disposalFuture.setDone();
}
}
}
}
}
private int registerNew() {
int nHandles = 0;
for (; ;) {
ConnectionRequest req = connectQueue.poll();//取连接队列队头请求
if (req == null) {
break;
}
H handle = req.handle;
try {
register(handle, req);//注册一个用于连接的新的客户端socket, 并把它加入连接轮询池中
nHandles ++;
} catch (Exception e) {
req.setException(e);
try {
close(handle);
} catch (Exception e2) {
ExceptionMonitor.getInstance().exceptionCaught(e2);
}
}
}
return nHandles;
}
private int processSessions(Iterator<H> handlers) {//处理连接请求
int nHandles = 0;
while (handlers.hasNext()) {
H handle = handlers.next();
handlers.remove();
ConnectionRequest entry = connectionRequest(handle);
boolean success = false;
try {
if (finishConnect(handle)) {//连接请求成功完成,创建一个新会话
T session = newSession(processor, handle);
finishSessionInitialization(session, entry, entry.getSessionInitializer());//结束会话初始化
session.getProcessor().add(session);//将剩下的工作交给IoProcessor去处理
nHandles ++;
}
success = true;
} catch (Throwable e) {
entry.setException(e);
} finally {
if (!success) {//若连接失败,则将此连接请求放到取消连接队列中
cancelQueue.offer(entry);
}
}
}
return nHandles;
}
private void processTimedOutSessions(Iterator<H> handles) {//处理超时的连接请求
long currentTime = System.currentTimeMillis();//当前时间
while (handles.hasNext()) {
H handle = handles.next();
ConnectionRequest entry = connectionRequest(handle);
if (currentTime >= entry.deadline) {//当前时间已经超出了连接请求的底限
entry.setException(
new ConnectException("Connection timed out."));
cancelQueue.offer(entry);//将此连接请求放入取消连接队列中
}
}
}
private int cancelKeys() {//把取消队列中的连接请求给cancel掉
int nHandles = 0;
for (; ;) {
ConnectionRequest req = cancelQueue.poll();
if (req == null) {
break;
}
H handle = req.handle;
try {
close(handle);//关闭对应的客户端socket
} catch (Exception e) {
ExceptionMonitor.getInstance().exceptionCaught(e);
} finally {
nHandles ++;
}
}
return nHandles;
}
作者:phinecos(洞庭散人)
出处:http://phinecos.cnblogs.com/
本文版权归作者和博客园共有,欢迎转载,但请保留此段声明,并在文章页面明显位置给出原文连接。
作者:洞庭散人
出处:http://phinecos.cnblogs.com/
本博客遵从Creative Commons Attribution 3.0 License,若用于非商业目的,您可以自由转载,但请保留原作者信息和文章链接URL。