edu.umd.cs.findbugs.ba.JCIPAnnotationDatabase Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spotbugs Show documentation
SpotBugs: Because it's easy!
The newest version!
/*
 * FindBugs - Find Bugs in Java programs
 * Copyright (C) 2005, University of Maryland
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

package edu.umd.cs.findbugs.ba;

import java.util.HashMap;
import java.util.Map;

import javax.annotation.CheckForNull;

import org.apache.bcel.classfile.ElementValue;

import edu.umd.cs.findbugs.internalAnnotations.DottedClassName;

public class JCIPAnnotationDatabase {
    Map> memberAnnotations = new HashMap<>();

    Map> classAnnotations = new HashMap<>();

    @CheckForNull
    public ElementValue getClassAnnotation(@DottedClassName String dottedClassName, String annotationClass) {
        Map map = getEntryForClass(dottedClassName);
        return map == null ? null : map.get(annotationClass);
    }

    public boolean hasClassAnnotation(@DottedClassName String dottedClassName, String annotationClass) {
        assert dottedClassName.indexOf('/') == -1;
        Map map = getEntryForClass(dottedClassName);
        return map != null && map.containsKey(annotationClass);
    }

    @CheckForNull
    public ElementValue getFieldAnnotation(XField field, String annotationClass) {
        Map map = getEntryForClassMember(field);
        return map == null ? null : map.get(annotationClass);
    }

    public boolean hasFieldAnnotation(XField field, String annotationClass) {
        Map map = getEntryForClassMember(field);
        return map != null && map.containsKey(annotationClass);
    }

    @CheckForNull
    public ElementValue getMethodAnnotation(XMethod method, String annotationClass) {
        Map map = getEntryForClassMember(method);
        return map == null ? null : map.get(annotationClass);
    }

    public boolean hasMethodAnnotation(XMethod method, String annotationClass) {
        Map map = getEntryForClassMember(method);
        return map != null && map.containsKey(annotationClass);
    }

    @CheckForNull
    private Map getEntryForClassMember(ClassMember member) {
        return memberAnnotations.get(member);
    }

    public void addEntryForClassMember(ClassMember member,
            String annotationClass, ElementValue value) {
        Map map = memberAnnotations.computeIfAbsent(member, k -> new HashMap<>());
        map.put(annotationClass, value);
    }

    @CheckForNull
    private Map getEntryForClass(@DottedClassName String dottedClassName) {
        assert dottedClassName.indexOf('/') == -1;
        return classAnnotations.get(dottedClassName);
    }

    public void addEntryForClass(@DottedClassName String dottedClassName,
            String annotationClass, ElementValue value) {
        Map map = getEntryForClass(dottedClassName);
        if (map == null) {
            map = new HashMap<>(3);
            classAnnotations.put(dottedClassName, map);
        }
        map.put(annotationClass, value);
    }

}