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

Go to download
Show more of this group Show more artifacts with this name
Show all versions of weld-servlet-shaded Show documentation
This jar bundles all the bits of Weld and CDI required for running in a Servlet container.
There is a newer version: 6.0.0.Final
/*
 * 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 java.lang.annotation.Annotation;
import java.lang.reflect.AnnotatedElement;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.HashSet;
import java.util.Set;
import java.util.function.Function;

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

import org.jboss.weld.bootstrap.api.helpers.AbstractBootstrapService;
import org.jboss.weld.metadata.TypeStore;
import org.jboss.weld.util.Annotations;
import org.jboss.weld.util.cache.ComputingCache;
import org.jboss.weld.util.cache.ComputingCacheBuilder;
import org.jboss.weld.util.collections.ImmutableSet;

public class DefaultReflectionCache extends AbstractBootstrapService implements ReflectionCache {

    private final TypeStore store;
    private final Function> ANNOTATIONS_FUNCTION = input -> ImmutableSet.of(internalGetAnnotations(input));
    private final Function> DECLARED_ANNOTATIONS_FUNCTION = input -> ImmutableSet.of(internalGetDeclaredAnnotations(input));

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

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

    private final ComputingCache> annotations;
    private final ComputingCache> declaredAnnotations;
    private final ComputingCache, Set> backedAnnotatedTypeAnnotations;
    private final ComputingCache, AnnotationClass> annotationClasses;

    public DefaultReflectionCache(TypeStore store) {
        this.store = store;
        ComputingCacheBuilder cacheBuilder = ComputingCacheBuilder.newBuilder();
        this.annotations = cacheBuilder.build(ANNOTATIONS_FUNCTION);
        this.declaredAnnotations = cacheBuilder.build(DECLARED_ANNOTATIONS_FUNCTION);
        this.backedAnnotatedTypeAnnotations = cacheBuilder.build(new BackedAnnotatedTypeAnnotationsFunction());
        this.annotationClasses =  cacheBuilder.build(new AnnotationClassFunction());
    }

    @Override
    public void cleanupAfterBoot() {
        annotations.clear();
        declaredAnnotations.clear();
        backedAnnotatedTypeAnnotations.clear();
        annotationClasses.clear();
    }

    @Override
    public Set getAnnotations(AnnotatedElement element) {
        return annotations.getValue(element);
    }

    @Override
    public Set getDeclaredAnnotations(AnnotatedElement element) {
        return declaredAnnotations.getValue(element);
    }

    @Override
    public Set getBackedAnnotatedTypeAnnotationSet(Class javaClass) {
        return backedAnnotatedTypeAnnotations.getValue(javaClass);
    }

    private class BackedAnnotatedTypeAnnotationsFunction implements Function, Set> {

        @Override
        public Set apply(Class javaClass) {
            Set annotations = getAnnotations(javaClass);
            boolean scopeFound = false;
            for (Annotation annotation : annotations) {
                boolean isScope = getAnnotationClass(annotation.annotationType()).isScope();
                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) {
            ImmutableSet.Builder result = ImmutableSet.builder();
            for (Annotation annotation : annotations) {
                if (!getAnnotationClass(annotation.annotationType()).isScope()) {
                    result.add(annotation);
                }
            }
            result.addAll(findTopLevelScopeDefinitions(javaClass));
            return result.build();
        }

        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 (getAnnotationClass(annotation.annotationType()).isScope()) {
                        scopes.add(annotation);
                    }
                }
                if (scopes.size() > 0) {
                    return scopes;
                }
            }
            throw new IllegalStateException();
        }
    }

    private class AnnotationClassFunction implements Function, AnnotationClass> {
        @Override
        public AnnotationClass apply(Class input) {
            boolean scope = input.isAnnotationPresent(NormalScope.class) || input.isAnnotationPresent(Scope.class) || store.isExtraScope(input);
            Method repeatableAnnotationAccessor = Annotations.getRepeatableAnnotationAccessor(input);
            Set metaAnnotations = ImmutableSet.of(internalGetAnnotations(input));
            return new AnnotationClassImpl<>(scope, repeatableAnnotationAccessor, metaAnnotations);
        }
    }

    private static class AnnotationClassImpl implements AnnotationClass {

        private final boolean scope;
        private final Method repeatableAnnotationAccessor;
        private final Set metaAnnotations;

        public AnnotationClassImpl(boolean scope, Method repeatableAnnotationAccessor, Set metaAnnotations) {
            this.scope = scope;
            this.repeatableAnnotationAccessor = repeatableAnnotationAccessor;
            this.metaAnnotations = metaAnnotations;
        }

        @Override
        public Set getMetaAnnotations() {
            return metaAnnotations;
        }

        @Override
        public boolean isScope() {
            return scope;
        }

        @Override
        public boolean isRepeatableAnnotationContainer() {
            return repeatableAnnotationAccessor != null;
        }

        @Override
        public Annotation[] getRepeatableAnnotations(Annotation annotation) {
            if (!isRepeatableAnnotationContainer()) {
                throw new IllegalStateException("Not a repeatable annotation container " + annotation);
            }
            try {
                return (Annotation[]) repeatableAnnotationAccessor.invoke(annotation);
            } catch (IllegalAccessException | IllegalArgumentException | InvocationTargetException e) {
                throw new RuntimeException("Error reading repeatable annotations on " + annotation.annotationType(), e);
            }
        }
    }

    @Override
    public  AnnotationClass getAnnotationClass(Class clazz) {
        return annotationClasses.getCastValue(clazz);
    }
}