org.jboss.mod_cluster.JBossWeb Maven / Gradle / Ivy
/*
* mod_cluster
*
* Copyright(c) 2008 Red Hat Middleware, LLC,
* and individual contributors as indicated by the @authors tag.
* See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library in the file COPYING.LIB;
* if not, write to the Free Software Foundation, Inc.,
* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
*
* @author Jean-Frederic Clere
* @version $Revision$
*/
package org.jboss.mod_cluster;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import org.apache.tomcat.InstanceManager;
import java.lang.reflect.InvocationTargetException;
import javax.naming.NamingException;
import org.apache.catalina.startup.ContextConfig;
import org.apache.catalina.*;
import org.apache.catalina.connector.Connector;
import org.apache.catalina.core.*;
import org.apache.catalina.startup.HostConfig;
import org.apache.tomcat.util.IntrospectionUtils;
public class JBossWeb extends StandardService {
private String route = null;
private void copyFile(File in, File out) throws IOException {
FileInputStream fis = new FileInputStream(in);
FileOutputStream fos = new FileOutputStream(out);
byte[] buf = new byte[1024];
int i = 0;
while((i=fis.read(buf))!=-1) {
fos.write(buf, 0, i);
}
fis.close();
fos.close();
}
private static void copyFiles(File src, File dest) throws IOException {
if (src.isDirectory()) {
if (!dest.exists())
dest.mkdirs();
String list[] = src.list();
for (int i = 0; i < list.length; i++) {
File dest1 = new File(dest, list[i]);
File src1 = new File(src, list[i]);
copyFiles(src1 , dest1);
}
} else {
//This was not a directory, so lets just copy the file
FileInputStream fin = null;
FileOutputStream fout = null;
byte[] buffer = new byte[4096]; //Buffer 4K at a time (you can change this).
int bytesRead;
//open the files for input and output
fin = new FileInputStream(src);
fout = new FileOutputStream (dest);
//while bytesRead indicates a successful read, lets write...
while ((bytesRead = fin.read(buffer)) >= 0) {
fout.write(buffer,0,bytesRead);
}
fout.close();
fin.close();
}
}
private void copyNativeDir(String route) throws IOException {
File in = new File("bin/");
if (!in.exists()) {
return;
}
File ou = new File("node1/bin");
if (!ou.exists()) {
ou.mkdirs();
}
copyFiles(in, ou);
}
/**
* Returns a JBossWeb service to use in our embedded tests.
*
* @param route the JVMRoute of the corresponding node.
* @param host the name of the virtualhost
* @param nat indicated to use native or not.
* @param webapp name of webapp (ROOT maps as / and bla as /bla).
* @param Aliasias an array of String containing the Aliases names.
* @return the created jbossweb service.
*/
public JBossWeb(String route, String host, boolean nat, String webapp, String[] Aliases) throws IOException {
// Copy native tree...
if (nat) {
copyNativeDir(route);
}
System.setProperty( "catalina.base", route);
System.setProperty( "catalina.home", route);
this.route = route;
//Create an Engine
Engine baseEngine = new StandardEngine();
baseEngine.setName(host + "Engine" + route);
baseEngine.setDefaultHost(host);
baseEngine.setJvmRoute(route);
baseEngine.setRealm(null);
// Create node1/webapps/ROOT and index.html
File fd = new File ( route + "/webapps/" + webapp);
fd.mkdirs();
String docBase = fd.getAbsolutePath();
String appBase = fd.getParent();
fd = new File (route + "/webapps/" + webapp, "index.html");
FileWriter out = new FileWriter(fd);
out.write(route + ":This is a test\n");
out.close();
// Copy a small servlets for testing.
fd = new File ( route + "/webapps/" + webapp + "/WEB-INF/classes");
fd.mkdirs();
// Session logic tests...
fd = new File (route + "/webapps/" + webapp + "/WEB-INF/classes" , "MyCount.class");
File fdin = new File ("MyCount.class");
if (!fdin.exists())
fdin = new File ("target/classes/MyCount.class");
copyFile(fdin, fd);
// Simple tests...
fd = new File (route + "/webapps/" + webapp + "/WEB-INF/classes" , "MyTest.class");
fdin = new File ("MyTest.class");
if (!fdin.exists())
fdin = new File ("target/classes/MyTest.class");
copyFile(fdin, fd);
//Create Host
StandardHost baseHost = new StandardHost();
baseHost.setAppBase(appBase);
baseHost.setName(host);
// createHost( host, appBase);
// baseHost.setDeployOnStartup(true);
baseHost.setBackgroundProcessorDelay(1);
//StandardHost stdhost = (StandardHost)baseHost;
// stdhost.setDeployXML(true);
baseHost.setConfigClass("org.apache.catalina.startup.ContextConfig");
// stdhost.setUnpackWARs(true);
if (Aliases != null && Aliases.length>0) {
for (int j = 0; j < Aliases.length; j++) {
baseHost.addAlias(Aliases[j]);
}
}
HostConfig hostConfig = new HostConfig();
baseHost.addLifecycleListener(hostConfig);
baseEngine.addChild(baseHost);
//Create default context
Context rootContext = new StandardContext();
rootContext.setDocBase(docBase);
if (webapp.equals("ROOT"))
rootContext.setPath("");
else
rootContext.setPath("/" + webapp);
ContextConfig config = new ContextConfig();
((Lifecycle) rootContext).addLifecycleListener(config);
rootContext.setIgnoreAnnotations(true);
rootContext.setPrivileged(true);
rootContext.setInstanceManager(new LocalInstanceManager());
Wrapper testwrapper = rootContext.createWrapper();
testwrapper.setName("MyCount");
testwrapper.setServletClass("MyCount");
testwrapper.setLoadOnStartup(0);
rootContext.addChild(testwrapper);
rootContext.addServletMapping("/MyCount", "MyCount");
Wrapper wrapper = rootContext.createWrapper();
wrapper.setName("MyTest");
wrapper.setServletClass("MyTest");
wrapper.setLoadOnStartup(0);
rootContext.addChild(wrapper);
rootContext.addServletMapping("/MyTest", "MyTest");
baseHost.addChild( rootContext );
// addEngine( baseEngine );
this.container = baseEngine;
baseEngine.setService(this);
this.setName(host + "Engine" + route);
// setRedirectStreams(false);
}
public JBossWeb(String route, String host) throws IOException {
this(route, host, false);
}
public JBossWeb(String route, String host, boolean nat) throws IOException {
this(route, host, nat, "ROOT");
}
public JBossWeb(String route, String host, String webapp) throws IOException {
this(route, host, false, webapp);
}
public JBossWeb(String route, String host, boolean nat, String webapp) throws IOException {
this(route, host, nat, webapp, null);
}
/** Add a host
* @param host name of the host
* @param Aliasias an array of String containing the Aliases names.
*/
void AddHost(String host, String[] Aliases) {
File fd = new File ( host + "/webapps/");
fd.mkdirs();
String appBase = fd.getPath();
//Create Host
StandardHost baseHost = new StandardHost();
baseHost.setAppBase(appBase);
baseHost.setName(host);
baseHost.setBackgroundProcessorDelay(1);
baseHost.setConfigClass("org.apache.catalina.startup.ContextConfig");
if (Aliases != null && Aliases.length>0) {
for (int j = 0; j < Aliases.length; j++) {
baseHost.addAlias(Aliases[j]);
}
}
HostConfig hostConfig = new HostConfig();
baseHost.addLifecycleListener(hostConfig);
Engine engine = (Engine) getContainer();
engine.addChild(baseHost);
}
/**
* Add a context to the StandardService (to all the virtual host defined in the Engine).
* @param path for bla is will deploy as /bla.
* @param docBase path to use
* @param servletname name of the server.
* @param wait tell if the wait is 10000 in the parameters of the servlet.
* @param host name of the virtual host where to add the context (null : every virtual hosts).
* @throws IOException
*/
void AddContext(String path, String base, String servletname, boolean wait, String hostname) throws IOException {
File fd = new File ( route + "/webapps/" + base);
fd.mkdirs();
String docBase = fd.getAbsolutePath();
fd = new File (route + "/webapps/" + base + "/WEB-INF/classes");
fd.mkdirs();
// Session logic tests...
fd = new File (route + "/webapps/" + base + "/WEB-INF/classes" , "MyCount.class");
File fdin = new File ("MyCount.class");
if (!fdin.exists())
fdin = new File ("target/classes/MyCount.class");
copyFile(fdin, fd);
// Simple tests...
fd = new File (route + "/webapps/" + base + "/WEB-INF/classes" , "MyTest.class");
fdin = new File ("MyTest.class");
if (!fdin.exists())
fdin = new File ("target/classes/MyTest.class");
copyFile(fdin, fd);
System.out.println("AddContext: " + path + "/" + servletname + " on " + docBase);
Context context = new StandardContext();
context.setDocBase(docBase);
context.setPath(path);
context.setInstanceManager(new LocalInstanceManager());
ContextConfig config = new ContextConfig();
((Lifecycle) context).addLifecycleListener(config);
context.setIgnoreAnnotations(true);
context.setPrivileged(true);
if (servletname != null) {
Wrapper wrapper = context.createWrapper();
wrapper.setName(servletname);
wrapper.setServletClass(servletname);
if (wait) {
wrapper.addInitParameter("wait", "10000");
wrapper.setLoadOnStartup(1);
} else {
wrapper.setLoadOnStartup(0);
}
context.addChild(wrapper);
context.addServletMapping("/" + servletname, servletname);
}
Engine engine = (Engine) getContainer();
Container[] containers = engine.findChildren();
for (int j = 0; j < containers.length; j++) {
if (containers[j] instanceof Host) {
Host host = (Host) containers[j];
if (hostname != null && !host.getName().equals(hostname))
continue;
System.out.println("AddContext: " + path + " added on " + host);
host.addChild(context);
}
}
}
void AddContext(String path, String docBase, String servletname, boolean wait) throws IOException {
AddContext(path, docBase, servletname, wait, null);
}
void AddContext(String path, String docBase) throws IOException {
AddContext(path, docBase, "MyCount", false, null);
}
public void addWAR(String file, String route) throws IOException {
File fd = new File ( route + "/" + route + "/webapps");
fd.mkdirs();
String sep = System.getProperty("file.separator");
String [] paths = file.split(sep);
fd = new File (route + "/" + route + "/webapps", paths[paths.length-1]);
File fdin = new File (file);
copyFile(fdin, fd);
}
public Connector addConnector(int port) throws Exception {
return addConnector(port, "ajp");
}
public Connector addConnector(int port, String scheme) throws Exception {
return addConnector(port, scheme, null);
}
public Connector addConnector(int port, String protocol, String address) throws Exception {
Connector connector = null;
if (protocol.equals("ajp")) {
connector= new Connector("org.apache.coyote.ajp.AjpProtocol");
} else if (protocol.equals("http")) {
connector= new Connector(protocol);
}
if (address != null) {
IntrospectionUtils.setProperty(connector, "address",
"" + address);
}
IntrospectionUtils.setProperty(connector, "port", "" + port);
IntrospectionUtils.setProperty(connector, "backlog", "400");
// Look in StandardService to see why it works ;-)
addConnector( connector );
return connector;
}
/*
* remove the context from the virtualhost.
*/
public void removeContext(String path, String hostname) {
Engine engine = (Engine) getContainer();
Container[] containers = engine.findChildren();
for (int j = 0; j < containers.length; j++) {
if (containers[j] instanceof StandardHost) {
StandardHost host = (StandardHost) containers[j];
if (hostname != null && !host.getName().equals(hostname))
continue;
Context context = (Context) host.findChild(path);
if (context != null)
containers[j].removeChild(context);
}
}
}
public void removeContext(String path) {
removeContext(path, null);
}
private static class LocalInstanceManager implements InstanceManager {
@Override
public Object newInstance(String className) throws IllegalAccessException, InvocationTargetException, NamingException, InstantiationException, ClassNotFoundException {
return Class.forName(className).newInstance();
}
@Override
public Object newInstance(String fqcn, ClassLoader classLoader) throws IllegalAccessException, InvocationTargetException, NamingException, InstantiationException, ClassNotFoundException {
return Class.forName(fqcn, false, classLoader).newInstance();
}
@Override
public Object newInstance(Class c) throws IllegalAccessException, InvocationTargetException, NamingException, InstantiationException {
return c.newInstance();
}
@Override
public void newInstance(Object o) throws IllegalAccessException, InvocationTargetException, NamingException {
throw new IllegalStateException();
}
@Override
public void destroyInstance(Object o) throws IllegalAccessException, InvocationTargetException {
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy