org.multiverse.instrumentation.metadata.ClassMetadata Maven / Gradle / Ivy

Go to download
package org.multiverse.instrumentation.metadata;

import org.objectweb.asm.Opcodes;

import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

import static java.lang.String.format;

/**
 * Contains the metadata for a class.
 *
 * @author Peter Veentjer.
 */
public final class ClassMetadata {

    private final String className;

    private final Map fields = new HashMap();
    private final Map methods = new HashMap();
    private final List interfaces = new LinkedList();

    private boolean isTransactionalObject;
    private ClassMetadata superClassMetadata;
    private boolean ignoredClass;
    private int access;

    public ClassMetadata(String className) {
        this.className = className;
    }

    public MethodMetadata createMethodMetadata(String methodName, String desc) {
        if (methodName == null || desc == null) {
            throw new NullPointerException();
        }

        Multikey methodKey = new Multikey(methodName, desc);
        if (methods.containsKey(methodKey)) {
            String msg = format("There already is method metadata for %s.%s%s", className, methodName, desc);
            throw new IllegalArgumentException(msg);
        }

        MethodMetadata method = new MethodMetadata(this, methodName, desc);
        methods.put(methodKey, method);
        return method;
    }

    public List getInterfaces() {
        return interfaces;
    }

    public boolean isIgnoredClass() {
        return ignoredClass;
    }

    public void setIgnoredClass(boolean ignoredClass) {
        this.ignoredClass = ignoredClass;
    }

    public ClassMetadata getSuperClassMetadata() {
        return superClassMetadata;
    }

    public void setSuperClassMetadata(ClassMetadata superClassMetadata) {
        this.superClassMetadata = superClassMetadata;
    }

    //todo: problem is with interfaces are not included in search

    public MethodMetadata getMethodMetadata(String name, String desc) {
        MethodMetadata methodMetadata = methods.get(new Multikey(name, desc));

        if (methodMetadata == null && superClassMetadata != null) {
            methodMetadata = superClassMetadata.getMethodMetadata(name, desc);
        }

        return methodMetadata;
    }

    public FieldMetadata getFieldMetadata(String fieldName) {
        FieldMetadata fieldMetadata = fields.get(fieldName);
        if (fieldMetadata != null) {
            return fieldMetadata;
        }

        if (superClassMetadata != null) {
            fieldMetadata = superClassMetadata.getFieldMetadata(fieldName);
            if (fieldMetadata != null) {
                return fieldMetadata;
            }
        }

        for (ClassMetadata interfaceMetadata : interfaces) {
            fieldMetadata = interfaceMetadata.getFieldMetadata(fieldName);
            if (fieldMetadata != null) {
                return fieldMetadata;
            }
        }

        return null;
    }

    public FieldMetadata createFieldMetadata(String fieldname) {
        if (fieldname == null) {
            throw new NullPointerException();
        }

        if (fields.containsKey(fieldname)) {
            String msg = format("There already is field metadata for %s.%s", className, fieldname);
            throw new IllegalArgumentException(msg);
        }

        FieldMetadata fieldMetadata = new FieldMetadata(this, fieldname);
        fields.put(fieldname, fieldMetadata);
        return fieldMetadata;
    }

    public boolean hasManagedFieldsWithFieldGranularity() {
        for (FieldMetadata field : fields.values()) {
            if (field.isManagedField() && field.hasFieldGranularity()) {
                return true;
            }
        }

        return false;
    }

    public boolean hasManagedFieldsWithObjectGranularity() {
        for (FieldMetadata field : fields.values()) {
            if (field.isManagedField() && !field.hasFieldGranularity()) {
                return true;
            }
        }

        return false;
    }


    public boolean hasManagedFields() {
        for (FieldMetadata field : fields.values()) {
            if (field.isManagedField()) {
                return true;
            }
        }

        return false;
    }

    public String getName() {
        return className;
    }

    public boolean isTransactionalObject() {
        return isTransactionalObject;
    }

    public void setIsTransactionalObject(boolean isTransactionalObject) {
        this.isTransactionalObject = isTransactionalObject;
    }

    public boolean isTransactionalObjectWithObjectGranularFields() {
        if (isParentTransactionalObjectWithObjectGranularFields()) {
            return true;
        }

        if (isTransactionalObject()) {
            for (FieldMetadata fieldMetadata : fields.values()) {
                if (fieldMetadata.isManagedField() && !fieldMetadata.hasFieldGranularity()) {
                    return true;
                }
            }
        }

        return false;
    }

    private boolean isParentTransactionalObjectWithObjectGranularFields() {
        return superClassMetadata != null
                && superClassMetadata.isTransactionalObjectWithObjectGranularFields();
    }

    public boolean hasTransactionalMethods() {
        for (MethodMetadata methodMetadata : methods.values()) {
            if (methodMetadata.isTransactional()) {
                return true;
            }
        }

        return false;
    }

    public boolean isFirstGenerationTransactionalObjectWithObjectGranularFields() {
        if (!isTransactionalObject) {
            return false;
        }

        if (!hasManagedFieldsWithObjectGranularity()) {
            return false;
        }

        if (superClassMetadata == null) {
            return true;
        }

        return !superClassMetadata.isTransactionalObjectWithObjectGranularFields();
    }

    public String getTranlocalName() {
        return className + "__Tranlocal";
    }

    public String getTranlocalSnapshotName() {
        return className + "__TranlocalSnapshot";
    }

    public int getAccess() {
        return access;
    }

    public void setAccess(int access) {
        this.access = access;
    }

    public final boolean isFinal() {
        return (access & Opcodes.ACC_FINAL) != 0;
    }

    public boolean isInterface() {
        return (access & Opcodes.ACC_INTERFACE) != 0;
    }

    public String toString() {
        return className;
    }
}