当一个注解只能作用于一次类上,如何修改代码,使其能够作用于多次
就以下面的这个注解为例子
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();
}
只能作用于类上,且只能一次
思路,原本的注解只能作用一次,我们直接新写一个注解,将旧的注解当做一个数组进行,间接达到多次注解的效果
代码如下
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
需要改动,两个注解用的都是同一个
如此一来,我们需要做兼容
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);
}
}
}
简单的来说,就是新写一个注解,里面有个原来注解数组的属性,再简单改造一下注解处理类,就能达到作用多次的效果了。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。