About us about archunit HOT 4 CLOSED

import static com.tngtech.archunit.lang.syntax.ArchRuleDefinition.no;

@ArchTest
public static final ArchRule entitiesShouldntFetchEager =
        no(fields()).should(beAnnotatedWithOneToManyFetchingEager());

private static ClassesTransformer<JavaField> fields() {
    return new AbstractClassesTransformer<JavaField>("fields") {
        @Override
        public Iterable<JavaField> doTransform(JavaClasses collection) {
            Set<JavaField> result = new HashSet<>();
            for (JavaClass javaClass : collection) {
                result.addAll(javaClass.getFields());
            }
            return result;
        }
    };
}

private static ArchCondition<JavaField> beAnnotatedWithOneToManyFetchingEager() {
    return new ArchCondition<JavaField>("be annotated with @OneToMany(fetch = EAGER)") {
        @Override
        public void check(JavaField field, ConditionEvents events) {
            Optional<OneToMany> oneToMany = field.tryGetAnnotationOfType(OneToMany.class);
            boolean satisfied = oneToMany.isPresent() && oneToMany.get().fetch() == FetchType.EAGER;
            String message = String.format("Field %s.%s is%s annotated with @OneToMany(fetch = EAGER)",
                    field.getOwner().getName(), field.getName(), satisfied ? "" : " not");
            events.add(new SimpleConditionEvent(field, satisfied, message));
        }
    };
}