确保从另一个方法调用方法？

Question

我有一个层，称为Service，另一个称为Perache。我想知道我是否可以强制执行一条规则，上面写着：

在参数列表包含名为foo的参数的服务层中的任何公共类的任何公共方法中，断言它从权限层调用方法（理想情况下，确保在服务层中的任何其他方法之前调用该方法）。

ArchUnit可以这样做吗？

Answer 1

它不是现成的，但您可以通过以下解决方案实现该目标：

DescribedPredicate<JavaClass> isPermissionClass = JavaClass.Predicates
    .resideInAnyPackage("..permission..");

classes()
    .that().resideInAnyPackage("..service..")
    .should(new ContainOnlyMethodsCallingPermissionClass(isPermissionClass));

ContainOnlyMethodsCallingPermissionClass可以定义如下。它首先收集所有声明的方法，然后检查对权限类的传出方法调用是否涵盖所有收集的方法。

public static class ContainOnlyMethodsCallingPermissionClass extends ArchCondition<JavaClass> {

    private final DescribedPredicate<JavaClass> isPermissionClass;

    public ContainOnlyMethodsCallingPermissionClass(DescribedPredicate<JavaClass> isPermissionClass) {
        super("only contain methods calling a permission class");
        this.isPermissionClass = isPermissionClass;
    }

    @Override
    public void check(JavaClass javaClass, ConditionEvents events) {
        Set<String> methodIdentifiers = getMethodIdentifiersOfClass(javaClass);
        methodIdentifiers.removeAll(collectMethodsCallingPermissionClass(javaClass));

        for (String methodId : methodIdentifiers) {
            events.add(new SimpleConditionEvent(javaClass, false, methodId));
        }
    }

    private Set<String> collectMethodsCallingPermissionClass(JavaClass javaClass) {
        Set<String> methodIdentifiers = new HashSet<>();

        for (JavaAccess<?> access : javaClass.getAccessesFromSelf()) {
            if (!isCallFromMethod(access)) {
                continue;
            }

            if (isMethodCallToPermissionClass(access)) {
                JavaMethod callingMethod = (JavaMethod) access.getOwner();
                methodIdentifiers.remove(callingMethod.getFullName());
            }
        }

        return methodIdentifiers;
    }

    private Set<String> getMethodIdentifiersOfClass(JavaClass javaClass) {
        return javaClass.getMethods().stream().filter(method -> !method.isConstructor())
                .map(method -> method.getFullName()).collect(Collectors.toSet());
    }

    private boolean isCallFromMethod(JavaAccess<?> access) {
        return access.getOrigin() instanceof JavaMethod;
    }

    private boolean isMethodCallToPermissionClass(JavaAccess<?> access) {
        return isPermissionClass.apply(access.getTargetOwner());
    }
}

您还可以将在JavaClass上操作的谓词更改为在JavaMethodCall上操作的谓词，以便更准确地定位相关的权限方法。这还允许过滤掉服务层中不相关的方法（例如，toString（）方法）。