/**
* 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.camel.impl;
import java.util.Hashtable;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.Set;
import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.NameClassPair;
import javax.naming.NameNotFoundException;
import javax.naming.NamingEnumeration;
import javax.naming.NamingException;
import org.apache.camel.NoSuchBeanException;
import org.apache.camel.RuntimeCamelException;
import org.apache.camel.spi.Registry;
import org.apache.camel.util.jndi.CamelInitialContextFactory;
/**
* A {@link Registry} implementation which looks up the objects in JNDI
*/
public class JndiRegistry implements Registry {
private Context context;
private Map environment;
private final boolean standalone;
public JndiRegistry() {
this.standalone = false;
}
public JndiRegistry(Map environment) {
this.environment = environment;
this.standalone = false;
}
public JndiRegistry(Context context) {
this.context = context;
this.standalone = false;
}
/**
* Whether to use standalone mode, where the JNDI initial context factory is using
* {@link CamelInitialContextFactory}.
*/
public JndiRegistry(boolean standalone) {
this.standalone = true;
}
public <T> T lookupByNameAndType(String name, Class<T> type) {
Object answer = lookupByName(name);
// just to be safe
if (answer == null) {
return null;
}
try {
return type.cast(answer);
} catch (Throwable e) {
String msg = "Found bean: " + name + " in JNDI Context: " + context
+ " of type: " + answer.getClass().getName() + " expected type was: " + type;
throw new NoSuchBeanException(name, msg, e);
}
}
public Object lookupByName(String name) {
try {
return getContext().lookup(name);
} catch (NameNotFoundException e) {
return null;
} catch (NamingException e) {
return null;
}
}
public <T> Map<String, T> findByTypeWithName(Class<T> type) {
Map<String, T> answer = new LinkedHashMap<String, T>();
try {
NamingEnumeration<NameClassPair> list = getContext().list("");
while (list.hasMore()) {
NameClassPair pair = list.next();
Object instance = context.lookup(pair.getName());
if (type.isInstance(instance)) {
answer.put(pair.getName(), type.cast(instance));
}
}
} catch (NamingException e) {
// ignore
}
return answer;
}
public <T> Set<T> findByType(Class<T> type) {
Set<T> answer = new LinkedHashSet<T>();
try {
NamingEnumeration<NameClassPair> list = getContext().list("");
while (list.hasMore()) {
NameClassPair pair = list.next();
Object instance = context.lookup(pair.getName());
if (type.isInstance(instance)) {
answer.add(type.cast(instance));
}
}
} catch (NamingException e) {
// ignore
}
return answer;
}
public Object lookup(String name) {
return lookupByName(name);
}
public <T> T lookup(String name, Class<T> type) {
return lookupByNameAndType(name, type);
}
public <T> Map<String, T> lookupByType(Class<T> type) {
return findByTypeWithName(type);
}
public void bind(String name, Object object) {
try {
getContext().bind(name, object);
} catch (NamingException e) {
throw new RuntimeCamelException(e);
}
}
public void close() throws NamingException {
if (context != null) {
context.close();
}
}
public Context getContext() throws NamingException {
if (context == null) {
context = createContext();
}
return context;
}
public void setContext(Context context) {
this.context = context;
}
protected Context createContext() throws NamingException {
Hashtable<Object, Object> properties = new Hashtable<Object, Object>(System.getProperties());
if (environment != null) {
properties.putAll(environment);
}
// must include a factory if none provided in standalone mode
if (standalone && !properties.containsKey("java.naming.factory.initial")) {
properties.put("java.naming.factory.initial", "org.apache.camel.util.jndi.CamelInitialContextFactory");
}
return new InitialContext(properties);
}
}