/*
* Copyright (c) 1998-2011 Caucho Technology -- all rights reserved
*
* This file is part of Resin(R) Open Source
*
* Each copy or derived work must preserve the copyright notice and this
* notice unmodified.
*
* Resin Open Source is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* Resin Open Source 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, or any warranty
* of NON-INFRINGEMENT. See the GNU General Public License for more
* details.
*
* You should have received a copy of the GNU General Public License
* along with Resin Open Source; if not, write to the
*
* Free Software Foundation, Inc.
* 59 Temple Place, Suite 330
* Boston, MA 02111-1307 USA
*
* @author Scott Ferguson
*/
package com.caucho.server.webbeans;
import java.lang.annotation.Annotation;
import java.util.HashSet;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.annotation.ManagedBean;
import javax.ejb.MessageDriven;
import javax.ejb.Singleton;
import javax.ejb.Stateful;
import javax.ejb.Stateless;
import javax.enterprise.event.Observes;
import javax.enterprise.inject.spi.AfterBeanDiscovery;
import javax.enterprise.inject.spi.Annotated;
import javax.enterprise.inject.spi.AnnotatedType;
import javax.enterprise.inject.spi.Bean;
import javax.enterprise.inject.spi.Extension;
import javax.enterprise.inject.spi.ProcessAnnotatedType;
import javax.enterprise.inject.spi.ProcessBean;
import com.caucho.config.ConfigException;
import com.caucho.config.extension.ProcessBeanImpl;
import com.caucho.config.inject.InjectManager;
import com.caucho.config.inject.ManagedBeanImpl;
import com.caucho.ejb.inject.EjbGeneratedBean;
import com.caucho.ejb.manager.EjbManager;
import com.caucho.ejb.manager.EjbModule;
import com.caucho.hemp.broker.HempBroker;
import com.caucho.inject.Jndi;
import com.caucho.inject.LazyExtension;
import com.caucho.inject.MBean;
import com.caucho.inject.Module;
import com.caucho.jms.JmsMessageListener;
import com.caucho.jmx.Jmx;
import com.caucho.loader.ModuleConfig;
import com.caucho.remote.BamService;
import com.caucho.server.admin.AdminService;
import com.caucho.server.cluster.Server;
import com.caucho.util.L10N;
/**
* Standard XML behavior for META-INF/beans.xml
*/
@Module
public class ResinStandardPlugin implements Extension {
private static final L10N L = new L10N(ResinStandardPlugin.class);
private static final Logger log
= Logger.getLogger(ResinStandardPlugin.class.getName());
private static final HashSet<Class<? extends Annotation>> _ejbAnnotations
= new HashSet<Class<? extends Annotation>>();
private InjectManager _cdiManager;
public ResinStandardPlugin(InjectManager manager)
{
_cdiManager = manager;
}
/**
* Callback on initialization, allowing for registration of contexts.
*/
public void processAfterBeanDiscovery(@Observes AfterBeanDiscovery event)
{
// ioc/0p41
// event.addContext(ThreadContext.getContext());
}
/**
* Callback for discovered annotated types. EJBs are dispatched to
* EJB and disabled for normal processing.
*/
@LazyExtension
public void processAnnotatedType(@Observes ProcessAnnotatedType<?> event)
{
processAnnotatedTypeImpl(event);
}
private <T> void processAnnotatedTypeImpl(ProcessAnnotatedType<T> event)
{
AnnotatedType<T> annotatedType = event.getAnnotatedType();
if (annotatedType == null)
return;
// ioc/0j08
boolean isXmlConfig = true;
EjbManager ejbContainer = EjbManager.create(_cdiManager.getClassLoader());
if (isXmlConfig
&& (annotatedType.isAnnotationPresent(Stateful.class)
|| annotatedType.isAnnotationPresent(Stateless.class)
|| annotatedType.isAnnotationPresent(Singleton.class)
|| annotatedType.isAnnotationPresent(MessageDriven.class)
|| annotatedType.isAnnotationPresent(JmsMessageListener.class)
|| ejbContainer != null && ejbContainer.isConfiguredBean(annotatedType.getJavaClass()))) {
if (ejbContainer != null) {
ejbContainer.createBean(annotatedType, null);
event.veto();
}
}
/*
if (isEjbParent(annotatedType.getJavaClass().getSuperclass())) {
event.veto();
}
*/
}
@LazyExtension
public void processBean(@Observes ProcessBean<?> event)
{
processBeanImpl(event);
}
private <T> void processBeanImpl(ProcessBean<T> event)
{
Annotated annotated = event.getAnnotated();
Bean<T> bean = event.getBean();
if (annotated == null) {
return;
}
if (! (bean instanceof EjbGeneratedBean)
&& (annotated.isAnnotationPresent(Stateful.class)
|| annotated.isAnnotationPresent(Stateless.class)
|| annotated.isAnnotationPresent(Singleton.class)
|| annotated.isAnnotationPresent(MessageDriven.class)
|| annotated.isAnnotationPresent(JmsMessageListener.class))) {
EjbManager ejbContainer = EjbManager.create();
if (bean instanceof ManagedBeanImpl<?>) {
// XXX: shouldn't reach processBeanImpl
if (true)
throw new IllegalStateException(String.valueOf(annotated));
ManagedBeanImpl<T> mBean = (ManagedBeanImpl<T>) bean;
AnnotatedType<T> annType = mBean.getAnnotatedType();
ejbContainer.createBean(annType, mBean.getInjectionTarget());
if (event instanceof ProcessBeanImpl<?>)
((ProcessBeanImpl<?>) event).veto();
}
}
if (annotated.isAnnotationPresent(Jndi.class)) {
Jndi jndi = annotated.getAnnotation(Jndi.class);
String jndiName = jndi.value();
if ("".equals(jndiName)) {
jndiName = bean.getBeanClass().getSimpleName();
}
JndiBeanProxy<T> proxy = new JndiBeanProxy<T>(_cdiManager, bean);
if (log.isLoggable(Level.FINE))
log.fine("bind to JNDI '" + jndiName + "' for " + bean);
try {
com.caucho.naming.Jndi.bindDeepShort(jndiName, proxy);
} catch (Exception e) {
log.log(Level.FINE, e.toString(), e);
}
}
if (annotated.isAnnotationPresent(ManagedBean.class)){
registerManagedBean(annotated, bean);
}
if (annotated.isAnnotationPresent(MBean.class)) {
MBean mbean = annotated.getAnnotation(MBean.class);
String mbeanName = mbean.value();
if ("".equals(mbeanName))
mbeanName = "type=" + bean.getBeanClass().getSimpleName();
AnnotatedType<?> annType = (AnnotatedType<?>) annotated;
try {
Jmx.registerContext(new BeanMBean(_cdiManager, bean, annType), mbeanName);
} catch (Exception e) {
log.log(Level.FINE, e.toString(), e);
}
}
if (annotated.isAnnotationPresent(BamService.class)) {
BamService service = annotated.getAnnotation(BamService.class);
HempBroker broker = HempBroker.getCurrent();
broker.addStartupActor(event.getBean(), service.name(),
service.threadMax());
}
if (annotated.isAnnotationPresent(AdminService.class)) {
AdminService service = annotated.getAnnotation(AdminService.class);
Server server = Server.getCurrent();
if (server == null) {
throw new ConfigException(L
.l("@AdminService requires an active Resin Server."));
}
if (!server.isWatchdog()) {
HempBroker broker = (HempBroker) server.getAdminBroker();
broker.addStartupActor(event.getBean(), service.name(),
service.threadMax());
}
}
}
private void registerManagedBean(Annotated annotated,
Bean<?> bean)
{
ManagedBean manage = annotated.getAnnotation(ManagedBean.class);
String jndiName = manage.value();
if ("".equals(jndiName)) {
jndiName = bean.getBeanClass().getSimpleName();
}
JndiBeanProxy<?> proxy = new JndiBeanProxy(_cdiManager, bean);
if (log.isLoggable(Level.FINE))
log.fine("bind to JNDI '" + jndiName + "' for " + bean);
try {
com.caucho.naming.Jndi.bindDeepShort("java:module/" + jndiName, proxy);
com.caucho.naming.Jndi.bindDeepShort(jndiName, proxy);
} catch (Exception e) {
log.log(Level.FINE, e.toString(), e);
}
}
@Override
public String toString()
{
return getClass().getSimpleName() + "[]";
}
static {
_ejbAnnotations.add(Stateful.class);
_ejbAnnotations.add(Stateless.class);
_ejbAnnotations.add(Singleton.class);
_ejbAnnotations.add(MessageDriven.class);
_ejbAnnotations.add(JmsMessageListener.class);
}
}