package com.pingan.emall.biz.communication.handler;
import com.pingan.emall.biz.communication.TcpSessionLock;
import com.pingan.emall.biz.communication.TcpSessionManager;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;
import org.apache.log4j.Logger;
import org.apache.mina.core.service.IoHandlerAdapter;
import org.apache.mina.core.session.IdleStatus;
import org.apache.mina.core.session.IoSession;
import org.apache.mina.transport.socket.SocketSessionConfig;
/**
* 核心I/O线程处理类, 接收filterChain处理后的MidResponseDTO, 并释放当前session 锁,
* 通知等待线程返回
*
* @author LICHAO844
*
*/
public class TcpCommunicationHandler extends IoHandlerAdapter {
private TcpSessionManager sessionManager;
private Map<Object, TcpSessionLock> locks;
private AtomicInteger sessionCount = new AtomicInteger();
private static final Logger LOG = Logger.getLogger(TcpCommunicationHandler.class);
public TcpCommunicationHandler(TcpSessionManager sessionManager, Map<Object, TcpSessionLock> locks) {
this.sessionManager = sessionManager;
this.locks = locks;
}
@Override
public void exceptionCaught(IoSession session, Throwable cause)
throws Exception {
Object sessionId = session.getId();
// Close IoSession and remove it from session manager
sessionManager.closeSession(sessionId);
StringBuilder builder = new StringBuilder();
builder.append("sessionId : ").append(sessionId)
.append(", Exception happens in tcp communication");
LOG.error(builder.toString(), cause);
}
@Override
public void messageReceived(IoSession session, Object message)
throws Exception {
Object sessionId = session.getId();
if (sessionId == null) {
LOG.error("No sessionKey bind with mina IoSession" + session);
return;
}
TcpSessionLock lock = locks.get(sessionId);
if (lock != null) {
synchronized (lock) {
//将信息set 唤起线程 实现 异步转同步
lock.setMessage(message);
lock.notify();
}
}
}
@Override
public void sessionClosed(IoSession session) throws Exception {
int totalSessionCount = sessionCount.decrementAndGet();
if (LOG.isDebugEnabled()) {
StringBuilder builder = new StringBuilder();
builder.append("IoSession closed, total session count is ")
.append(totalSessionCount).append(", ").append(session).append(", sessionId : ")
.append(session.getId());
LOG.debug(builder.toString());
}
}
@Override
public void sessionCreated(IoSession session) throws Exception {
int totalSessionCount = sessionCount.incrementAndGet();
if (session.getConfig() instanceof SocketSessionConfig) {
SocketSessionConfig config = (SocketSessionConfig) session.getConfig();
config.setKeepAlive(true);
config.setTcpNoDelay(true);
config.setSoLinger(0);
}
if (LOG.isDebugEnabled()) {
StringBuilder builder = new StringBuilder();
builder.append("IoSession created, total session count is ")
.append(totalSessionCount).append(", ").append(session)
.append(", sessionId : ").append(session.getId());
LOG.debug(builder.toString());
}
}
@Override
public void sessionIdle(IoSession session, IdleStatus status)
throws Exception {
if (LOG.isDebugEnabled()) {
StringBuilder builder = new StringBuilder();
builder.append("IoSession idle, status is ").append(status)
.append(", ").append(session).append(", TxnId : ")
.append(session.getId());
LOG.debug(builder.toString());
}
// both idle for 30s, close session
if (status == IdleStatus.BOTH_IDLE) {
Object sessionId = session.getId();
if (sessionId != null) {
sessionManager.closeSession(sessionId);
} else {
session.close(true);
}
}
}
@Override
public void sessionOpened(IoSession session) throws Exception {
if (LOG.isDebugEnabled()) {
StringBuilder builder = new StringBuilder();
builder.append("IoSession opened : ").append(session).append(", sessionId : ")
.append(session.getId());
LOG.debug(builder.toString());
}
}
}
import com.pingan.emall.biz.communication.TcpSessionLock;
import com.pingan.emall.biz.communication.TcpSessionManager;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;
import org.apache.log4j.Logger;
import org.apache.mina.core.service.IoHandlerAdapter;
import org.apache.mina.core.session.IdleStatus;
import org.apache.mina.core.session.IoSession;
import org.apache.mina.transport.socket.SocketSessionConfig;
/**
* 核心I/O线程处理类, 接收filterChain处理后的MidResponseDTO, 并释放当前session 锁,
* 通知等待线程返回
*
* @author LICHAO844
*
*/
public class TcpCommunicationHandler extends IoHandlerAdapter {
private TcpSessionManager sessionManager;
private Map<Object, TcpSessionLock> locks;
private AtomicInteger sessionCount = new AtomicInteger();
private static final Logger LOG = Logger.getLogger(TcpCommunicationHandler.class);
public TcpCommunicationHandler(TcpSessionManager sessionManager, Map<Object, TcpSessionLock> locks) {
this.sessionManager = sessionManager;
this.locks = locks;
}
@Override
public void exceptionCaught(IoSession session, Throwable cause)
throws Exception {
Object sessionId = session.getId();
// Close IoSession and remove it from session manager
sessionManager.closeSession(sessionId);
StringBuilder builder = new StringBuilder();
builder.append("sessionId : ").append(sessionId)
.append(", Exception happens in tcp communication");
LOG.error(builder.toString(), cause);
}
@Override
public void messageReceived(IoSession session, Object message)
throws Exception {
Object sessionId = session.getId();
if (sessionId == null) {
LOG.error("No sessionKey bind with mina IoSession" + session);
return;
}
TcpSessionLock lock = locks.get(sessionId);
if (lock != null) {
synchronized (lock) {
//将信息set 唤起线程 实现 异步转同步
lock.setMessage(message);
lock.notify();
}
}
}
@Override
public void sessionClosed(IoSession session) throws Exception {
int totalSessionCount = sessionCount.decrementAndGet();
if (LOG.isDebugEnabled()) {
StringBuilder builder = new StringBuilder();
builder.append("IoSession closed, total session count is ")
.append(totalSessionCount).append(", ").append(session).append(", sessionId : ")
.append(session.getId());
LOG.debug(builder.toString());
}
}
@Override
public void sessionCreated(IoSession session) throws Exception {
int totalSessionCount = sessionCount.incrementAndGet();
if (session.getConfig() instanceof SocketSessionConfig) {
SocketSessionConfig config = (SocketSessionConfig) session.getConfig();
config.setKeepAlive(true);
config.setTcpNoDelay(true);
config.setSoLinger(0);
}
if (LOG.isDebugEnabled()) {
StringBuilder builder = new StringBuilder();
builder.append("IoSession created, total session count is ")
.append(totalSessionCount).append(", ").append(session)
.append(", sessionId : ").append(session.getId());
LOG.debug(builder.toString());
}
}
@Override
public void sessionIdle(IoSession session, IdleStatus status)
throws Exception {
if (LOG.isDebugEnabled()) {
StringBuilder builder = new StringBuilder();
builder.append("IoSession idle, status is ").append(status)
.append(", ").append(session).append(", TxnId : ")
.append(session.getId());
LOG.debug(builder.toString());
}
// both idle for 30s, close session
if (status == IdleStatus.BOTH_IDLE) {
Object sessionId = session.getId();
if (sessionId != null) {
sessionManager.closeSession(sessionId);
} else {
session.close(true);
}
}
}
@Override
public void sessionOpened(IoSession session) throws Exception {
if (LOG.isDebugEnabled()) {
StringBuilder builder = new StringBuilder();
builder.append("IoSession opened : ").append(session).append(", sessionId : ")
.append(session.getId());
LOG.debug(builder.toString());
}
}
}