/*
* JBoss, Home of Professional Open Source
* Copyright 2005-2008, Red Hat Middleware LLC, and individual contributors
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* This 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.1 of
* the License, or (at your option) any later version.
*
* This software 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 software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.messaging.tests.util;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.Hashtable;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import javax.naming.Binding;
import javax.naming.Context;
import javax.naming.Name;
import javax.naming.NameAlreadyBoundException;
import javax.naming.NameNotFoundException;
import javax.naming.NameParser;
import javax.naming.NamingEnumeration;
import javax.naming.NamingException;
import javax.naming.RefAddr;
import javax.naming.Reference;
import org.jboss.messaging.core.logging.Logger;
/**
* @author <a href="mailto:ovidiu@feodorov.com">Ovidiu Feodorov</a>
* @version <tt>$Revision: 2868 $</tt>
*
* $Id: InVMContext.java 2868 2007-07-10 20:22:16Z timfox $
*/
public class InVMContext implements Context, Serializable
{
// Constants -----------------------------------------------------
private static final long serialVersionUID = 385743957345L;
private static final Logger log = Logger.getLogger(InVMContext.class);
// Static --------------------------------------------------------
// Attributes ----------------------------------------------------
protected Map map;
protected NameParser parser = new InVMNameParser();
private String nameInNamespace = "";
// Constructors --------------------------------------------------
public InVMContext()
{
map = Collections.synchronizedMap(new HashMap());
}
public InVMContext(String nameInNamespace)
{
this();
this.nameInNamespace = nameInNamespace;
}
// Context implementation ----------------------------------------
public Object lookup(Name name) throws NamingException
{
throw new UnsupportedOperationException();
}
public Object lookup(String name) throws NamingException
{
name = trimSlashes(name);
int i = name.indexOf("/");
String tok = i == -1 ? name : name.substring(0, i);
Object value = map.get(tok);
if (value == null)
{
throw new NameNotFoundException("Name not found: " + tok);
}
if (value instanceof InVMContext && i != -1)
{
return ((InVMContext)value).lookup(name.substring(i));
}
if (value instanceof Reference)
{
Reference ref = (Reference)value;
RefAddr refAddr = ref.get("nns");
// we only deal with references create by NonSerializableFactory
String key = (String)refAddr.getContent();
return NonSerializableFactory.lookup(key);
}
else
{
return value;
}
}
public void bind(Name name, Object obj) throws NamingException
{
throw new UnsupportedOperationException();
}
public void bind(String name, Object obj) throws NamingException
{
internalBind(name, obj, false);
}
public void rebind(Name name, Object obj) throws NamingException
{
throw new UnsupportedOperationException();
}
public void rebind(String name, Object obj) throws NamingException
{
internalBind(name, obj, true);
}
public void unbind(Name name) throws NamingException
{
unbind(name.toString());
}
public void unbind(String name) throws NamingException
{
name = trimSlashes(name);
int i = name.indexOf("/");
boolean terminal = i == -1;
if (terminal)
{
map.remove(name);
}
else
{
String tok = name.substring(0, i);
InVMContext c = (InVMContext)map.get(tok);
if (c == null)
{
throw new NameNotFoundException("Context not found: " + tok);
}
c.unbind(name.substring(i));
}
}
public void rename(Name oldName, Name newName) throws NamingException
{
throw new UnsupportedOperationException();
}
public void rename(String oldName, String newName) throws NamingException
{
throw new UnsupportedOperationException();
}
public NamingEnumeration list(Name name) throws NamingException
{
throw new UnsupportedOperationException();
}
public NamingEnumeration list(String name) throws NamingException
{
throw new UnsupportedOperationException();
}
public NamingEnumeration listBindings(Name name) throws NamingException
{
throw new UnsupportedOperationException();
}
public NamingEnumeration listBindings(String contextName) throws NamingException
{
contextName = trimSlashes(contextName);
if (!"".equals(contextName) && !".".equals(contextName))
{
try
{
return ((InVMContext)lookup(contextName)).listBindings("");
}
catch(Throwable t)
{
throw new NamingException(t.getMessage());
}
}
List l = new ArrayList();
for(Iterator i = map.keySet().iterator(); i.hasNext(); )
{
String name = (String)i.next();
Object object = map.get(name);
l.add(new Binding(name, object));
}
return new NamingEnumerationImpl(l.iterator());
}
public void destroySubcontext(Name name) throws NamingException
{
destroySubcontext(name.toString());
}
public void destroySubcontext(String name) throws NamingException
{
map.remove(trimSlashes(name));
}
public Context createSubcontext(Name name) throws NamingException
{
throw new UnsupportedOperationException();
}
public Context createSubcontext(String name) throws NamingException
{
name = trimSlashes(name);
if (map.get(name) != null)
{
throw new NameAlreadyBoundException(name);
}
InVMContext c = new InVMContext(getNameInNamespace());
map.put(name, c);
return c;
}
public Object lookupLink(Name name) throws NamingException
{
throw new UnsupportedOperationException();
}
public Object lookupLink(String name) throws NamingException
{
throw new UnsupportedOperationException();
}
public NameParser getNameParser(Name name) throws NamingException
{
return getNameParser(name.toString());
}
public NameParser getNameParser(String name) throws NamingException
{
return parser;
}
public Name composeName(Name name, Name prefix) throws NamingException
{
throw new UnsupportedOperationException();
}
public String composeName(String name, String prefix) throws NamingException
{
throw new UnsupportedOperationException();
}
public Object addToEnvironment(String propName, Object propVal) throws NamingException
{
throw new UnsupportedOperationException();
}
public Object removeFromEnvironment(String propName) throws NamingException
{
throw new UnsupportedOperationException();
}
public Hashtable getEnvironment() throws NamingException
{
Hashtable env = new Hashtable();
env.put("java.naming.factory.initial",
"org.jboss.test.messaging.tools.container.InVMInitialContextFactory");
env.put("java.naming.provider.url", "org.jboss.naming:org.jnp.interface");
return env;
}
public void close() throws NamingException
{
}
public String getNameInNamespace() throws NamingException
{
return nameInNamespace;
}
// Public --------------------------------------------------------
// Package protected ---------------------------------------------
// Protected -----------------------------------------------------
// Private -------------------------------------------------------
private String trimSlashes(String s)
{
int i = 0;
while(true)
{
if (i == s.length() || s.charAt(i) != '/')
{
break;
}
i++;
}
s = s.substring(i);
i = s.length() - 1;
while(true)
{
if (i == -1 || s.charAt(i) != '/')
{
break;
}
i--;
}
return s.substring(0, i + 1);
}
private void internalBind(String name, Object obj, boolean rebind) throws NamingException
{
log.debug("Binding " + name + " obj " + obj + " rebind " + rebind);
name = trimSlashes(name);
int i = name.lastIndexOf("/");
InVMContext c = this;
if (i != -1)
{
String path = name.substring(0, i);
c = (InVMContext)lookup(path);
}
name = name.substring(i + 1);
if (!rebind && c.map.get(name) != null)
{
throw new NameAlreadyBoundException(name);
}
c.map.put(name, obj);
}
// Inner classes -------------------------------------------------
private class NamingEnumerationImpl implements NamingEnumeration
{
private Iterator iterator;
NamingEnumerationImpl(Iterator bindingIterator)
{
this.iterator = bindingIterator;
}
public void close() throws NamingException
{
throw new UnsupportedOperationException();
}
public boolean hasMore() throws NamingException
{
return iterator.hasNext();
}
public Object next() throws NamingException
{
return iterator.next();
}
public boolean hasMoreElements()
{
return iterator.hasNext();
}
public Object nextElement()
{
return iterator.next();
}
}
}