org.apache.geronimo.openejb.cdi.GeronimoResourceInjectionService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of geronimo-openejb Show documentation
The newest version!
/*
 * 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.geronimo.openejb.cdi;

import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectOutput;
import java.lang.annotation.Annotation;
import java.lang.reflect.Field;
import java.lang.reflect.Modifier;

import javax.enterprise.inject.Produces;
import javax.enterprise.inject.spi.Bean;
import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.NamingException;

import org.apache.webbeans.component.ResourceBean;
import org.apache.webbeans.config.WebBeansContext;
import org.apache.webbeans.exception.WebBeansException;
import org.apache.webbeans.spi.ResourceInjectionService;
import org.apache.webbeans.spi.SecurityService;
import org.apache.webbeans.spi.api.ResourceReference;
import org.apache.webbeans.util.AnnotationUtil;

/**
 * @version $Rev: 1145436 $ $Date: 2011-07-12 10:26:54 +0800 (Tue, 12 Jul 2011) $
 */
public class GeronimoResourceInjectionService implements ResourceInjectionService {

    private final Context context;
    
    private WebBeansContext webBeansContext;

    public GeronimoResourceInjectionService(WebBeansContext webBeansContext) {
        try {
            this.context = new InitialContext();
            this.webBeansContext = webBeansContext;
        } catch (NamingException e) {
            throw new WebBeansException("could not set up naming context", e);
        }
    }

    //from StandaloneResourceInjectionService
    @Override
    public void injectJavaEEResources(Object managedBeanInstance) {
        Class currentClass = managedBeanInstance.getClass();
        
        SecurityService securityService = webBeansContext.getSecurityService();
        while (currentClass != null && Object.class != currentClass) {
            Field[] fields = securityService.doPrivilegedGetDeclaredFields(currentClass);

            for (Field field : fields) {
                if (Modifier.isStatic(field.getModifiers())) continue;

                Annotation ann = AnnotationUtil.hasOwbInjectableResource(field.getDeclaredAnnotations());

                if (ann == null) continue;

                @SuppressWarnings("unchecked")
                ResourceReference resourceRef = new ResourceReference(field.getDeclaringClass(), field.getName(), field.getType(), ann);
                boolean acess = field.isAccessible();
                try {
                    securityService.doPrivilegedSetAccessible(field, true);
                    field.set(managedBeanInstance, getResourceReference(resourceRef));
                } catch (Exception e) {
                    throw new WebBeansException("Unable to inject field" + field, e);
                } finally {
                    securityService.doPrivilegedSetAccessible(field, acess);
                }
            }
            currentClass = currentClass.getSuperclass();
        }
    }

    @Override
    public  X getResourceReference(ResourceReference resourceReference) {
        String jndiName = resourceReference.getJndiName();
        Class type = resourceReference.getResourceType();

        try {
            return type.cast(context.lookup(jndiName));
        } catch (NamingException e) {
            throw new WebBeansException("Could not get resource of type " + type + " at jndi name " + jndiName, e);
        }
    }

    @Override
    public void clear() {
    }

    /**
     * delegation of serialization behavior
     */
    public  void writeExternal(Bean bean, T actualResource, ObjectOutput out) throws IOException {
        //do nothing
    }

    /**
     * delegation of serialization behavior
     */
    public  T readExternal(Bean bean, ObjectInput out) throws IOException,
            ClassNotFoundException {
        return (T) ((ResourceBean)bean).getActualInstance();
    }


}