如何将只能作用一次的注解修改为作用多次

一、介绍

当一个注解只能作用于一次类上，如何修改代码，使其能够作用于多次

就以下面的这个注解为例子

代码语言：javascript复制

package com.banmoon.test.spv.annotation;

import com.banmoon.test.spv.listener.SystemPropertyTestExecutionListener;
import org.springframework.test.context.TestExecutionListeners;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 * @author banmoon
 * @date 2024/07/16 15:16:05
 */
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@TestExecutionListeners(listeners = SystemPropertyTestExecutionListener.class, mergeMode = TestExecutionListeners.MergeMode.MERGE_WITH_DEFAULTS)
public @interface SystemPropeerty {

    /**
     * 系统变量key
     */
    String key();

    /**
     * 系统变量value
     */
    String value();

}

只能作用于类上，且只能一次

二、代码

思路，原本的注解只能作用一次，我们直接新写一个注解，将旧的注解当做一个数组进行，间接达到多次注解的效果

代码如下

代码语言：javascript复制

package com.banmoon.test.spv.annotation;

import com.banmoon.test.spv.listener.SystemPropertyTestExecutionListener;
import org.springframework.test.context.TestExecutionListeners;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 * @author banmoon
 * @date 2024/07/16 15:16:05
 */
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@TestExecutionListeners(listeners = SystemPropertyTestExecutionListener.class, mergeMode = TestExecutionListeners.MergeMode.MERGE_WITH_DEFAULTS)
public @interface SystemPropeerties {

    SystemPropeerty[] value();

}

相应的，我们的SystemPropertyTestExecutionListener.java需要改动，两个注解用的都是同一个

如此一来，我们需要做兼容

代码语言：javascript复制

package com.banmoon.test.spv.listener;

import com.banmoon.test.spv.annotation.SystemPropeerties;
import com.banmoon.test.spv.annotation.SystemPropeerty;
import org.springframework.test.context.TestContext;
import org.springframework.test.context.support.AbstractTestExecutionListener;

import java.util.Arrays;
import java.util.Objects;

/**
 * @author banmoon
 * @date 2024/07/16 14:02:42
 */
public class SystemPropertyTestExecutionListener extends AbstractTestExecutionListener {

    @Override
    public void beforeTestClass(TestContext testContext) {
        SystemPropeerties systemPropeerties = testContext.getTestClass().getAnnotation(SystemPropeerties.class);
        if (Objects.nonNull(systemPropeerties)) {
            Arrays.stream(systemPropeerties.value()).forEach(item -> {
                String key = item.key();
                String value = item.value();
                System.setProperty(key, value);
            });
        }
        SystemPropeerty systemPropeerty = testContext.getTestClass().getAnnotation(SystemPropeerty.class);
        if (Objects.nonNull(systemPropeerty)) {
            String key = systemPropeerty.key();
            String value = systemPropeerty.value();
            System.setProperty(key, value);
        }
    }
}

三、最后

简单的来说，就是新写一个注解，里面有个原来注解数组的属性，再简单改造一下注解处理类，就能达到作用多次的效果了。

java annotations

0 人点赞