/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*
*/
package org.apache.mina.core.service;
import junit.framework.Assert;
import org.apache.mina.core.future.CloseFuture;
import org.apache.mina.core.future.ConnectFuture;
import org.apache.mina.core.future.IoFuture;
import org.apache.mina.core.future.IoFutureListener;
import org.apache.mina.core.session.IdleStatus;
import org.apache.mina.core.session.IoSession;
import org.apache.mina.filter.codec.ProtocolCodecFilter;
import org.apache.mina.filter.codec.textline.TextLineCodecFactory;
import org.apache.mina.filter.logging.LoggingFilter;
import org.apache.mina.transport.socket.nio.NioSocketAcceptor;
import org.apache.mina.transport.socket.nio.NioSocketConnector;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CountDownLatch;
/**
* test the AbstractIoService
*
* @author <a href="http://mina.apache.org">Apache MINA Project</a>
*/
public class AbstractIoServiceTest {
private static final int PORT = 9123;
@Test
public void testDispose() throws IOException, InterruptedException {
List threadsBefore = getThreadNames();
final IoAcceptor acceptor = new NioSocketAcceptor();
acceptor.getFilterChain().addLast("logger", new LoggingFilter());
acceptor.getFilterChain().addLast("codec",
new ProtocolCodecFilter(new TextLineCodecFactory(Charset.forName("UTF-8"))));
acceptor.setHandler(new ServerHandler());
acceptor.getSessionConfig().setReadBufferSize(2048);
acceptor.getSessionConfig().setIdleTime(IdleStatus.BOTH_IDLE, 10);
acceptor.bind(new InetSocketAddress(PORT));
System.out.println("Server running ...");
final NioSocketConnector connector = new NioSocketConnector();
// Set connect timeout.
connector.setConnectTimeoutMillis(30 * 1000L);
connector.setHandler(new ClientHandler());
connector.getFilterChain().addLast("logger", new LoggingFilter());
connector.getFilterChain().addLast("codec",
new ProtocolCodecFilter(new TextLineCodecFactory(Charset.forName("UTF-8"))));
// Start communication.
ConnectFuture cf = connector.connect(new InetSocketAddress("localhost", 9123));
cf.awaitUninterruptibly();
IoSession session = cf.getSession();
// send a message
session.write("Hello World!\r");
// wait until response is received
CountDownLatch latch = (CountDownLatch) session.getAttribute("latch");
latch.await();
// close the session
CloseFuture closeFuture = session.close(false);
System.out.println("session.close called");
//Thread.sleep(5);
// wait for session close and then dispose the connector
closeFuture.addListener(new IoFutureListener<IoFuture>() {
public void operationComplete(IoFuture future) {
System.out.println("managed session count=" + connector.getManagedSessionCount());
System.out.println("Disposing connector ...");
connector.dispose(true);
System.out.println("Disposing connector ... *finished*");
}
});
closeFuture.awaitUninterruptibly();
acceptor.dispose(true);
List threadsAfter = getThreadNames();
System.out.println("threadsBefore = " + threadsBefore);
System.out.println("threadsAfter = " + threadsAfter);
// Assert.assertEquals(threadsBefore, threadsAfter);
}
public static class ClientHandler extends IoHandlerAdapter {
private static final Logger LOGGER = LoggerFactory.getLogger("CLIENT");
@Override
public void sessionCreated(IoSession session) throws Exception {
session.setAttribute("latch", new CountDownLatch(1));
}
@Override
public void messageReceived(IoSession session, Object message) throws Exception {
LOGGER.info("client: messageReceived(" + session + ", " + message + ")");
CountDownLatch latch = (CountDownLatch) session.getAttribute("latch");
latch.countDown();
}
@Override
public void exceptionCaught(IoSession session, Throwable cause) throws Exception {
LOGGER.warn("exceptionCaught:", cause);
}
}
public static class ServerHandler extends IoHandlerAdapter {
private static final Logger LOGGER = LoggerFactory.getLogger("SERVER");
@Override
public void messageReceived(IoSession session, Object message) throws Exception {
LOGGER.info("server: messageReceived(" + session + ", " + message + ")");
session.write(message.toString());
}
@Override
public void exceptionCaught(IoSession session, Throwable cause) throws Exception {
LOGGER.warn("exceptionCaught:", cause);
}
}
public static void main(String[] args) throws IOException, InterruptedException {
new AbstractIoServiceTest().testDispose();
}
private List<String> getThreadNames() {
List<String> list = new ArrayList<String>();
int active = Thread.activeCount();
Thread[] threads = new Thread[active];
Thread.enumerate(threads);
for (Thread thread : threads) {
try {
String name = thread.getName();
list.add(name);
} catch (NullPointerException ignore) {
}
}
return list;
}
}