@BeforeClass public static void setUpConnection() throws SQLException { JdbcWithServerTestBase.setUpConnection(); try(Statement stmt = getConnection().createStatement()) { stmt.execute(String.format("alter session set \"%s\" = true", PlannerSettings.ENABLE_DECIMAL_DATA_TYPE_KEY)); stmt.execute(String.format("alter session set \"%s\" = false", ExecConstants.ENABLE_REATTEMPTS.getOptionName())); } }
@BeforeClass public static void setUpConnection() throws SQLException { JdbcWithServerTestBase.setUpConnection(); try(Statement stmt = getConnection().createStatement()) { stmt.execute(String.format("alter session set \"%s\" = false", ExecConstants.ENABLE_REATTEMPTS.getOptionName())); } }
@Test public void testMultiFilesWithDifferentSchema() throws Exception { test("ALTER SYSTEM SET \"" + ExecConstants.ENABLE_REATTEMPTS.getOptionName() + "\" = true"); try { final String query = String.format("select * from dfs_root.\"%s/schemachange/multi/\" order by id", TEST_RES_PATH); test(query); } finally { test("ALTER SYSTEM RESET \"" + ExecConstants.ENABLE_REATTEMPTS.getOptionName() + "\""); } }
public boolean isReduceCalcExpressionsEnabled() { return options.getOption(ENABLE_REDUCE_CALC.getOptionName()).getBoolVal(); }
public boolean isReduceFilterExpressionsEnabled() { return options.getOption(ENABLE_REDUCE_FILTER.getOptionName()).getBoolVal(); }
public boolean isHashJoinSwapEnabled() { return options.getOption(HASH_JOIN_SWAP.getOptionName()).getBoolVal(); }
public boolean isReduceProjectExpressionsEnabled() { return options.getOption(ENABLE_REDUCE_PROJECT.getOptionName()).getBoolVal(); }
@Before public void setup() throws Exception { if (created) { return; } created = true; // We don’t care about the reflection manager taking too long to respond in these tests so we disable wakeup events to avoid timing issues. l(ContextService.class).get().getOptionManager().setOption( OptionValue.createBoolean(SYSTEM, REFLECTION_PERIODIC_WAKEUP_ONLY.getOptionName(), true)); addCPSource(); addEmployeesJson(); uploadHomeFile(); Dataset dataset = createDataset(); datasetId = dataset.getId(); }
public boolean isNlJoinForScalarOnly() { return options.getOption(NLJOIN_FOR_SCALAR.getOptionName()).getBoolVal(); }
protected static void setEnableReAttempts(boolean enabled) throws Exception { runSQL("ALTER SYSTEM SET " + SqlUtils.QUOTE + ExecConstants.ENABLE_REATTEMPTS.getOptionName() + SqlUtils.QUOTE + " = " + enabled); }
protected void onlyAllowPeriodicWakeup(boolean periodicOnly) { getSabotContext().getOptionManager() .setOption( OptionValue.createBoolean(SYSTEM, REFLECTION_PERIODIC_WAKEUP_ONLY.getOptionName(), periodicOnly)); }
@BeforeClass public static void enableReattempts() throws Exception { test("ALTER SYSTEM SET \"" + ExecConstants.ENABLE_REATTEMPTS.getOptionName() + "\" = true"); }
@Override public boolean getOption(BooleanValidator validator) { return getOption(validator.getOptionName()).getBoolVal(); }
@Override public void setSubstitutionEnabled(boolean enable) { getOptionManager().setOption(OptionValue.createBoolean(SYSTEM, REFLECTION_ENABLE_SUBSTITUTION.getOptionName(), enable)); }
@BeforeClass public static void enableReAttempts() { setSessionOption(ExecConstants.ENABLE_REATTEMPTS.getOptionName(), "true"); }
protected static void setMaterializationCacheSettings(boolean enabled, long refreshDelayInSeconds) { l(ContextService.class).get().getOptionManager().setOption( OptionValue.createBoolean(SYSTEM, MATERIALIZATION_CACHE_ENABLED.getOptionName(), enabled)); l(ContextService.class).get().getOptionManager().setOption( OptionValue.createLong(SYSTEM, ReflectionOptions.MATERIALIZATION_CACHE_REFRESH_DELAY_MILLIS.getOptionName(), refreshDelayInSeconds*1000)); }
@Override public void close() throws Exception { testContext.getOptions().setOption(OptionValue.createBoolean(OptionType.SYSTEM, validator.getOptionName(), oldValue)); }}; }
public AutoCloseable with(final BooleanValidator validator, final boolean value){ final boolean oldValue = testContext.getOptions().getOption(validator); testContext.getOptions().setOption(OptionValue.createBoolean(OptionType.SYSTEM, validator.getOptionName(), value)); return new AutoCloseable(){ @Override public void close() throws Exception { testContext.getOptions().setOption(OptionValue.createBoolean(OptionType.SYSTEM, validator.getOptionName(), oldValue)); }}; }
@Override public void applyOptions(OptionManager manager) { if (optionProvider != null) { optionProvider.applyOptions(manager); } // TODO(DX-5912): disable hash join after merge join is implemented // manager.setOption(OptionValue.createBoolean(OptionValue.OptionType.QUERY, // PlannerSettings.HASHJOIN.getOptionName(), false)); manager.setOption(OptionValue.createBoolean(OptionValue.OptionType.QUERY, PlannerSettings.HASHAGG.getOptionName(), false)); } }
@Override public PartitionExplorer getPartitionExplorer() { throw UserException.unsupportedError().message("The partition explorer interface can only be used " + "in functions that can be evaluated at planning time. Make sure that the %s configuration " + "option is set to true.", PlannerSettings.CONSTANT_FOLDING.getOptionName()).build(logger); }