/**
* 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.openejb.server.ejbd;
import org.junit.Assert;
import junit.framework.TestCase;
import org.apache.openejb.OpenEJB;
import org.apache.openejb.assembler.classic.Assembler;
import org.apache.openejb.config.ConfigurationFactory;
import org.apache.openejb.core.ServerFederation;
import org.apache.openejb.jee.EjbJar;
import org.apache.openejb.jee.StatelessBean;
import org.apache.openejb.loader.SystemInstance;
import org.apache.openejb.server.ServiceDaemon;
import org.apache.openejb.server.ServicePool;
import javax.ejb.Remote;
import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import java.util.Properties;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
/**
* @version $Rev$ $Date$
*/
@SuppressWarnings("UseOfSystemOutOrSystemErr")
public class KeepAilveTest extends TestCase {
@SuppressWarnings("unchecked")
public void _testPool() throws Exception {
final int threads = 2;
final ThreadPoolExecutor pool = new ThreadPoolExecutor(threads, threads, 120, TimeUnit.SECONDS, new LinkedBlockingQueue<Runnable>(threads));
final Runnable runnable = new Runnable() {
@Override
public void run() {
waitOneSecond();
}
};
print(pool);
for (int i = 0; i < 10; i++) {
System.out.println("" + i);
pool.execute(runnable);
print(pool);
}
waitOneSecond();
for (int i = 0; i < 10; i++) {
print(pool);
waitOneSecond();
}
print(pool);
fail();
}
private void waitOneSecond() {
try {
Thread.sleep(1000);
} catch (final InterruptedException e) {
e.printStackTrace();
}
}
private void print(final ThreadPoolExecutor pool) {
System.out.println("==========================================");
final int activeCount = pool.getActiveCount();
System.out.println("activeCount = " + activeCount);
final int corePoolSize = pool.getCorePoolSize();
System.out.println("corePoolSize = " + corePoolSize);
final int largestPoolSize = pool.getLargestPoolSize();
System.out.println("largestPoolSize = " + largestPoolSize);
final int maximumPoolSize = pool.getMaximumPoolSize();
System.out.println("maximumPoolSize = " + maximumPoolSize);
final int poolSize = pool.getPoolSize();
System.out.println("poolSize = " + poolSize);
final int queueSize = pool.getQueue().size();
System.out.println("queueSize = " + queueSize);
final long taskCount = pool.getTaskCount();
System.out.println("taskCount = " + taskCount);
System.out.println("==========================================");
}
public void _test() throws Exception {
}
public void test() throws Exception {
final EjbServer ejbServer = new EjbServer();
final KeepAliveServer keepAliveServer = new KeepAliveServer(ejbServer, false);
final Properties initProps = new Properties();
initProps.setProperty("openejb.deployments.classpath.include", "");
initProps.setProperty("openejb.deployments.classpath.filter.descriptors", "true");
OpenEJB.init(initProps, new ServerFederation());
ejbServer.init(new Properties());
final ServicePool pool = new ServicePool(keepAliveServer, 10, 5000, true);
final ServiceDaemon serviceDaemon = new ServiceDaemon(pool, 0, "localhost");
serviceDaemon.start();
try {
final int port = serviceDaemon.getPort();
final Assembler assembler = SystemInstance.get().getComponent(Assembler.class);
final ConfigurationFactory config = new ConfigurationFactory();
final EjbJar ejbJar = new EjbJar();
ejbJar.addEnterpriseBean(new StatelessBean(EchoBean.class));
assembler.createApplication(config.configureApplication(ejbJar));
// good creds
final int threads = 1;
final CountDownLatch latch = new CountDownLatch(threads);
for (int i = 0; i < threads; i++) {
final Client client = new Client(latch, i, port);
thread(client, false);
}
final boolean await = latch.await(60, TimeUnit.SECONDS);
assertTrue(await);
} finally {
serviceDaemon.stop();
OpenEJB.destroy();
}
}
public static void thread(final Runnable runnable, final boolean daemon) {
final Thread thread = new Thread(runnable);
thread.setDaemon(daemon);
thread.start();
}
public static class Client implements Runnable {
private final Echo echo;
private final CountDownLatch latch;
private final int id;
public Client(final CountDownLatch latch, final int i, final int port) throws NamingException {
this.latch = latch;
this.id = i;
final Properties props = new Properties();
props.put("java.naming.factory.initial", "org.apache.openejb.client.RemoteInitialContextFactory");
props.put("java.naming.provider.url", "ejbd://127.0.0.1:" + port + "?" + id);
final Context context = new InitialContext(props);
this.echo = (Echo) context.lookup("EchoBeanRemote");
}
@Override
public void run() {
try {
int count = 10;
for (; count >= 0; count--) {
final String message = count + " bottles of beer on the wall";
// Thread.currentThread().setName("client-"+id+": "+count);
final String response = echo.echo(message);
Assert.assertEquals(message, reverse(response));
try {
Thread.sleep(5000);
} catch (final InterruptedException e) {
e.printStackTrace();
}
}
} finally {
latch.countDown();
}
}
private Object reverse(final String s) {
return new StringBuilder(s).reverse().toString();
}
}
public static class EchoBean implements Echo {
@Override
public String echo(final String s) {
// System.out.println(s);
return new StringBuilder(s).reverse().toString();
}
}
@Remote
public static interface Echo {
public String echo(String s);
}
}