rohangarg commented on code in PR #16382: URL: https://github.com/apache/druid/pull/16382#discussion_r1593637779
########## sql/src/test/java/org/apache/druid/sql/calcite/SqlTestFrameworkConfig.java: ########## @@ -36,77 +39,200 @@ import java.lang.annotation.RetentionPolicy; import java.lang.annotation.Target; import java.lang.reflect.Constructor; +import java.lang.reflect.InvocationTargetException; import java.lang.reflect.Method; +import java.util.ArrayList; +import java.util.Arrays; import java.util.HashMap; +import java.util.HashSet; +import java.util.List; import java.util.Map; import java.util.Objects; +import java.util.Set; import java.util.function.Function; /** - * Annotation to specify desired framework settings. + * Specifies current framework settings. * - * This class provides junit rule facilities to build the framework accordingly - * to the annotation. These rules also cache the previously created frameworks. + * Intended usage from tests is via the annotations: + * @SqlTestFrameworkConfig.MinTopNThreshold(33) + * + * In case of annotations used; it picks up all annotations from: + * * the method + * * its enclosing class and its parents + * if none contains a specific setting the default is being taken. + * + * All configurable setting should have: + * * an annotation with `value` with the desired type + * * the annotation itself should be annotated with itslef to set the default value + * * a field should be added to the main config class */ -@Retention(RetentionPolicy.RUNTIME) -@Target({ElementType.METHOD}) -public @interface SqlTestFrameworkConfig +public class SqlTestFrameworkConfig { - int numMergeBuffers() default 0; + @Retention(RetentionPolicy.RUNTIME) + @Target({ElementType.METHOD, ElementType.TYPE}) + @NumMergeBuffers(0) + public @interface NumMergeBuffers + { + int value(); + } - int minTopNThreshold() default TopNQueryConfig.DEFAULT_MIN_TOPN_THRESHOLD; + @Retention(RetentionPolicy.RUNTIME) + @Target({ElementType.METHOD, ElementType.TYPE}) + @MinTopNThreshold(TopNQueryConfig.DEFAULT_MIN_TOPN_THRESHOLD) + public @interface MinTopNThreshold + { + int value(); + } - ResultCacheMode resultCache() default ResultCacheMode.DISABLED; + @Retention(RetentionPolicy.RUNTIME) + @Target({ElementType.METHOD, ElementType.TYPE}) + @ResultCache(ResultCacheMode.DISABLED) + public @interface ResultCache + { + ResultCacheMode value(); + } /** - * Non-annotation version of {@link SqlTestFrameworkConfig}. - * - * Makes it less convoluted to work with configurations created at runtime. + * Declares which {@link QueryComponentSupplier} must be used for the class. */ - class SqlTestFrameworkConfigInstance + @Retention(RetentionPolicy.RUNTIME) + @Target({ElementType.METHOD, ElementType.TYPE}) + @Supplier(StandardComponentSupplier.class) + public @interface Supplier { - public final int numMergeBuffers; - public final int minTopNThreshold; - public final ResultCacheMode resultCache; + Class<? extends QueryComponentSupplier> value(); + } - public SqlTestFrameworkConfigInstance(SqlTestFrameworkConfig annotation) - { - numMergeBuffers = annotation.numMergeBuffers(); - minTopNThreshold = annotation.minTopNThreshold(); - resultCache = annotation.resultCache(); + public final int numMergeBuffers; + public final int minTopNThreshold; + public final ResultCacheMode resultCache; + public final Class<? extends QueryComponentSupplier> supplier; + + public SqlTestFrameworkConfig(List<Annotation> annotations) + { + try { + numMergeBuffers = getValueFromAnnotation(annotations, NumMergeBuffers.class); + minTopNThreshold = getValueFromAnnotation(annotations, MinTopNThreshold.class); + resultCache = getValueFromAnnotation(annotations, ResultCache.class); + supplier = getValueFromAnnotation(annotations, Supplier.class); + } + catch (NoSuchMethodException | SecurityException | IllegalAccessException | IllegalArgumentException + | InvocationTargetException e) { + throw new RuntimeException(e); } + } - @Override - public int hashCode() - { - return Objects.hash(minTopNThreshold, numMergeBuffers, resultCache); + public SqlTestFrameworkConfig(Map<String, String> queryParams) + { + try { + numMergeBuffers = getValueFromMap(queryParams, NumMergeBuffers.class); + minTopNThreshold = getValueFromMap(queryParams, MinTopNThreshold.class); + resultCache = getValueFromMap(queryParams, ResultCache.class); + supplier = getValueFromMap(queryParams, Supplier.class); } + catch (NoSuchMethodException | SecurityException | IllegalAccessException | IllegalArgumentException + | InvocationTargetException e) { + throw new RuntimeException(e); + } + } - @Override - public boolean equals(Object obj) - { - if (obj == null || getClass() != obj.getClass()) { - return false; + @SuppressWarnings("unchecked") + private <T> T getValueFromMap(Map<String, String> map, Class<? extends Annotation> annotationClass) + throws IllegalAccessException, InvocationTargetException, NoSuchMethodException, SecurityException + { + String value = map.get(annotationClass.getSimpleName()); + if (value == null) { + return defaultValue(annotationClass); + } + Class<?> type = annotationClass.getMethod("value").getReturnType(); + + if (type == int.class) { Review Comment: Thanks for the new set of changes, they look good to me :+1: -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: commits-unsubscr...@druid.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: commits-unsubscr...@druid.apache.org For additional commands, e-mail: commits-h...@druid.apache.org