如何在junit 5扩展中获取重复计数

import org.junit.jupiter.api.extension.ExtensionContext;
import org.junit.jupiter.api.extension.TestTemplateInvocationContext;
import org.junit.jupiter.api.extension.TestTemplateInvocationContextProvider;

import java.util.Collections;
import java.util.List;
import java.util.stream.Stream;

public class RepeatCountExtension implements TestTemplateInvocationContextProvider {

    @Override
    public boolean supportsTestTemplate(ExtensionContext context) {
        return true;
    }

    @Override
    public Stream<TestTemplateInvocationContext> provideTestTemplateInvocationContexts(ExtensionContext context) {
        int repeatCount = getRepeatCountFromAnnotation(context);
        return Stream.generate(() -> createInvocationContext(repeatCount)).limit(repeatCount);
    }

    private int getRepeatCountFromAnnotation(ExtensionContext context) {
        return context.getRequiredTestMethod().getAnnotation(RepeatCount.class).value();
    }

    private TestTemplateInvocationContext createInvocationContext(int repeatCount) {
        return new TestTemplateInvocationContext() {
            @Override
            public String getDisplayName(int invocationIndex) {
                return "Repetition " + (invocationIndex + 1);
            }

            @Override
            public List<Extension> getAdditionalExtensions() {
                return Collections.singletonList(new RepetitionInfoParameterResolver(invocationIndex));
            }
        };
    }
}

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

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface RepeatCount {
    int value() default 1;
}

import org.junit.jupiter.api.extension.ExtensionContext;
import org.junit.jupiter.api.extension.ParameterContext;
import org.junit.jupiter.api.extension.ParameterResolver;
import org.junit.jupiter.api.extension.TestTemplateInvocationContext;

public class RepetitionInfoParameterResolver implements ParameterResolver {

    private final int invocationIndex;

    public RepetitionInfoParameterResolver(int invocationIndex) {
        this.invocationIndex = invocationIndex;
    }

    @Override
    public boolean supportsParameter(ParameterContext parameterContext, ExtensionContext extensionContext) {
        return parameterContext.getParameter().getType() == RepetitionInfo.class;
    }

    @Override
    public Object resolveParameter(ParameterContext parameterContext, ExtensionContext extensionContext) {
        return new RepetitionInfo(invocationIndex);
    }
}

public class RepetitionInfo {

    private final int currentRepetition;

    public RepetitionInfo(int currentRepetition) {
        this.currentRepetition = currentRepetition;
    }

    public int getCurrentRepetition() {
        return currentRepetition;
    }
}

import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.TestTemplate;
import org.junit.jupiter.api.extension.ExtendWith;

import static org.junit.jupiter.api.Assertions.assertEquals;

@ExtendWith(RepeatCountExtension.class)
public class MyTest {

    @TestTemplate
    @RepeatCount(3)
    @DisplayName("示例测试")
    void test(RepetitionInfo repetitionInfo) {
        int repetition = repetitionInfo.getCurrentRepetition();
        assertEquals(1, repetition);
        // 执行测试逻辑
    }
}