rohangarg commented on code in PR #16382: URL: https://github.com/apache/druid/pull/16382#discussion_r1593631761
########## 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 Review Comment: Thanks for the clarification and changes! And yes, I think adding a validation for config keys in future might be good - since these configs are only used in tests, hence I think it makes sense for us to be strict in accepting them. -- 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