org.honton.chas.testpojo.PojoScanner Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of testpojo-maven-plugin Show documentation
Unit test pojos
The newest version!
package org.honton.chas.testpojo;

import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.List;

import org.reflections.adapters.MetadataAdapter;
import org.reflections.scanners.AbstractScanner;

import lombok.RequiredArgsConstructor;

/**
 * Reflections scanner that finds pojos.  A pojo is defined here as any class with 
 * an equals and hashCode method. 
 */
@RequiredArgsConstructor
public class PojoScanner extends AbstractScanner {
    final private Collection collector;

    @SuppressWarnings("unchecked")
    @Override
    public void scan(Object cls) {
        @SuppressWarnings("rawtypes")
        MetadataAdapter meta = getConfiguration().getMetadataAdapter();
        if (meta.isPublic(cls) && hasEqualsAndHashCode(meta, cls)) {
            collector.add(meta.getClassName(cls));
        }
    }

    private static final List NO_ARGS = Collections.emptyList();
    private static final List SINGLE_OBJECT_ARG = Arrays.asList("java.lang.Object");

    private static  boolean hasEqualsAndHashCode(MetadataAdapter meta, C cls) {
        boolean equals = false;
        boolean hashCode = false;
        boolean publicConstructor = false;
        for (M method : meta.getMethods(cls)) {
            equals |= isMethod(meta, method, "boolean", "equals", SINGLE_OBJECT_ARG);
            hashCode |= isMethod(meta, method, "int", "hashCode", NO_ARGS);
            publicConstructor |= isConstructor(meta, method, "int", "", NO_ARGS);
            if (equals && hashCode && publicConstructor) {
                return true;
            }
        }
        return false;
    }

    private static  boolean isMethod(MetadataAdapter meta, M method, 
            String returnType, String methodName, List parameterNames) {
        return returnType.equals(meta.getReturnTypeName(method)) 
                && methodName.equals(meta.getMethodName(method))
                && parameterNames.equals(meta.getParameterNames(method));
    }

    private static  boolean isConstructor(MetadataAdapter meta, M method, String returnType, String methodName,
            List parameterNames) {
        return "".equals(meta.getMethodName(method)) && meta.isPublic(method);
    }
}