All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.jboss.weld.resources.DefaultReflectionCache Maven / Gradle / Ivy

There is a newer version: 3.0.0.Alpha1
Show newest version
/*
 * JBoss, Home of Professional Open Source
 * Copyright 2012, Red Hat, Inc., and individual contributors
 * by the @authors tag. See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 *
 * Licensed 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.jboss.weld.resources;

import static org.jboss.weld.util.cache.LoadingCacheUtils.getCacheValue;

import java.lang.annotation.Annotation;
import java.lang.reflect.AnnotatedElement;
import java.lang.reflect.Constructor;
import java.lang.reflect.Member;
import java.lang.reflect.Method;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;

import javax.enterprise.context.NormalScope;
import javax.inject.Scope;

import org.jboss.weld.annotated.slim.backed.BackedAnnotatedParameter;
import org.jboss.weld.bootstrap.api.helpers.AbstractBootstrapService;
import org.jboss.weld.metadata.TypeStore;
import org.jboss.weld.util.collections.Arrays2;

import com.google.common.cache.CacheBuilder;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.LoadingCache;
import com.google.common.collect.ImmutableSet;

public class DefaultReflectionCache extends AbstractBootstrapService implements ReflectionCache {

    private final TypeStore store;

    protected Annotation[] internalGetAnnotations(AnnotatedElement element) {
        return element.getAnnotations();
    }

    protected Annotation[] internalGetDeclaredAnnotations(AnnotatedElement element) {
        return element.getDeclaredAnnotations();
    }

    private static class Annotations {
        private final Annotation[] annotations;
        private final Set annotationSet;

        public Annotations(Annotation[] annotations) {
            if (annotations.length == 0) {
                this.annotations = Arrays2.EMPTY_ANNOTATION_ARRAY;
                this.annotationSet = Collections.emptySet();
            } else {
                this.annotations = annotations;
                this.annotationSet = ImmutableSet.copyOf(annotations);
            }
        }
    }

    private final LoadingCache annotations;
    private final LoadingCache declaredAnnotations;
    private final LoadingCache, Annotation[][]> constructorParameterAnnotations;
    private final LoadingCache methodParameterAnnotations;
    private final LoadingCache, Set> parameterAnnotationSet;
    private final LoadingCache, Set> backedAnnotatedTypeAnnotations;
    private final LoadingCache, Boolean> isScopeAnnotation;

    public DefaultReflectionCache(TypeStore store) {
        this.store = store;
        CacheBuilder cacheBuilder = CacheBuilder.newBuilder();
        this.annotations = cacheBuilder.build(new CacheLoader() {
            @Override
            public Annotations load(AnnotatedElement input) {
                return new Annotations(internalGetAnnotations(input));
            }
        });
        this.declaredAnnotations = cacheBuilder.build(new CacheLoader() {
            @Override
            public Annotations load(AnnotatedElement input) {
                return new Annotations(internalGetDeclaredAnnotations(input));
            }
        });
        this.constructorParameterAnnotations = cacheBuilder.build(new CacheLoader, Annotation[][]>() {
            @Override
            public Annotation[][] load(Constructor input) {
                return input.getParameterAnnotations();
            }
        });
        this.methodParameterAnnotations = cacheBuilder.build(new CacheLoader() {
            @Override
            public Annotation[][] load(Method input) {
                return input.getParameterAnnotations();
            }
        });
        this.parameterAnnotationSet = cacheBuilder.build(new CacheLoader, Set>() {
            @Override
            public Set load(BackedAnnotatedParameter parameter) throws Exception {
                final Member member = parameter.getDeclaringCallable().getJavaMember();
                if (member instanceof Method) {
                    return ImmutableSet.copyOf( getParameterAnnotations((Method) member, parameter.getPosition()));
                } else {
                    return ImmutableSet.copyOf( getParameterAnnotations((Constructor) member, parameter.getPosition()));
                }
            }

        });
        this.backedAnnotatedTypeAnnotations = cacheBuilder.build(new BackedAnnotatedTypeAnnotationsFunction());
        this.isScopeAnnotation = cacheBuilder.build(new IsScopeAnnotationFunction());
    }

    @Override
    public Annotation[] getAnnotations(AnnotatedElement element) {
        return getCacheValue(annotations, element).annotations;
    }

    @Override
    public Annotation[] getDeclaredAnnotations(AnnotatedElement element) {
        return getCacheValue(declaredAnnotations, element).annotations;
    }

    @Override
    public Annotation[] getParameterAnnotations(Constructor constructor, int parameterPosition) {
        return getCacheValue(constructorParameterAnnotations, constructor)[parameterPosition];
    }

    @Override
    public Annotation[] getParameterAnnotations(Method method, int parameterPosition) {
        return getCacheValue(methodParameterAnnotations, method)[parameterPosition];
    }

    @Override
    public void cleanupAfterBoot() {
        annotations.invalidateAll();
        declaredAnnotations.invalidateAll();
        constructorParameterAnnotations.invalidateAll();
        methodParameterAnnotations.invalidateAll();
        backedAnnotatedTypeAnnotations.invalidateAll();
        isScopeAnnotation.invalidateAll();
        parameterAnnotationSet.invalidateAll();
    }

    @Override
    public Set getAnnotationSet(AnnotatedElement element) {
        return getCacheValue(annotations, element).annotationSet;
    }

    @Override
    public Set getDeclaredAnnotationSet(AnnotatedElement element) {
        return getCacheValue(declaredAnnotations, element).annotationSet;
    }

    @Override
    public Set getParameterAnnotationSet(BackedAnnotatedParameter parameter) {
        return getCacheValue(parameterAnnotationSet, parameter);
    }

    @Override
    public Set getBackedAnnotatedTypeAnnotationSet(Class javaClass) {
        return getCacheValue(backedAnnotatedTypeAnnotations, javaClass);
    }

    private class BackedAnnotatedTypeAnnotationsFunction extends CacheLoader, Set> {

        @Override
        public Set load(Class javaClass) {
            Set annotations = getAnnotationSet(javaClass);
            boolean scopeFound = false;
            for (Annotation annotation : annotations) {
                boolean isScope = getCacheValue(isScopeAnnotation, annotation.annotationType());
                if (isScope && scopeFound) {
                    // there are at least two scopes, we need to choose one using scope inheritance rules (4.1)
                    return applyScopeInheritanceRules(annotations, javaClass);
                }
                if (isScope) {
                    scopeFound = true;
                }
            }
            return annotations;
        }

        public Set applyScopeInheritanceRules(Set annotations, Class javaClass) {
            Set result = new HashSet();
            for (Annotation annotation : annotations) {
                if (!getCacheValue(isScopeAnnotation, annotation.annotationType())) {
                    result.add(annotation);
                }
            }
            result.addAll(findTopLevelScopeDefinitions(javaClass));
            return ImmutableSet.copyOf(result);
        }

        public Set findTopLevelScopeDefinitions(Class javaClass) {
            for (Class clazz = javaClass; clazz != null && clazz != Object.class; clazz = clazz.getSuperclass()) {
                Set scopes = new HashSet();
                for (Annotation annotation : getDeclaredAnnotations(clazz)) {
                    if (getCacheValue(isScopeAnnotation, annotation.annotationType())) {
                        scopes.add(annotation);
                    }
                }
                if (scopes.size() > 0) {
                    return scopes;
                }
            }
            throw new IllegalStateException();
        }
    }

    private class IsScopeAnnotationFunction extends CacheLoader, Boolean> {

        @Override
        public Boolean load(Class input) {
            if (input.isAnnotationPresent(NormalScope.class)) {
                return true;
            }
            if (input.isAnnotationPresent(Scope.class)) {
                return true;
            }
            return store.isExtraScope(input);
        }
    }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy