public LongValidator(String name, long def) { super(name, Kind.LONG, OptionValue.createLong(OptionType.SYSTEM, name, def)); } }
protected static OptionValue newLongValue(long value) { return OptionValue.createLong(OptionValue.OptionType.SYSTEM, "test-option", value); }
public static OptionValue createOption(Kind kind, OptionType type, String name, String val) { switch (kind) { case BOOLEAN: return createBoolean(type, name, Boolean.valueOf(val)); case LONG: return createLong(type, name, Long.valueOf(val)); case STRING: return createString(type, name, val); case DOUBLE: return createDouble(type, name, Double.valueOf(val)); default: return null; } }
@Override public OptionValue next() { final Entry<String, ConfigValue> e = entries.next(); final ConfigValue cv = e.getValue(); final String name = e.getKey(); OptionValue optionValue = null; switch(cv.valueType()) { case BOOLEAN: optionValue = OptionValue.createBoolean(OptionType.BOOT, name, (Boolean) cv.unwrapped()); break; case LIST: case OBJECT: case STRING: optionValue = OptionValue.createString(OptionType.BOOT, name, cv.render()); break; case NUMBER: optionValue = OptionValue.createLong(OptionType.BOOT, name, ((Number) cv.unwrapped()).longValue()); break; case NULL: throw new IllegalStateException("Config value \"" + name + "\" has NULL type"); } return optionValue; }
break; case LONG: this.setOption(OptionValue.createLong(OptionType.SYSTEM, validator.getOptionName(), Long.parseLong(value))); break; case STRING:
@Before public void setUp() { optionManager = mock(OptionManager.class); when(optionManager.getOption("planner.timeout_per_phase_ms")).thenReturn(OptionValue.createLong(OptionType.QUERY, "planner.timeout_per_phase_ms", 100)); settings = new PlannerSettings(DremioTest.DEFAULT_SABOT_CONFIG, optionManager, null); }
final BigDecimal bigDecimal = (BigDecimal) object; if (bigDecimal.scale() == 0) { return OptionValue.createLong(type, name, bigDecimal.longValue()); } else { return OptionValue.createDouble(type, name, bigDecimal.doubleValue()); case BIGINT: case INTEGER: return OptionValue.createLong(type, name, ((BigDecimal) object).longValue());
@PUT @Consumes(MediaType.APPLICATION_JSON) @Produces(MediaType.APPLICATION_JSON) @Path("/acceleration/settings") public void saveSystemSettings(final SystemSettingsApiDescriptor descriptor) { Preconditions.checkArgument(descriptor.getLimit() != null, "limit is required"); Preconditions.checkArgument(descriptor.getLimit() > 0, "limit must be positive"); Preconditions.checkArgument(descriptor.getAccelerateAggregation() != null, "accelerateAggregation is required"); Preconditions.checkArgument(descriptor.getAccelerateRaw() != null, "accelerateRaw is required"); SystemOptionManager optionManager = context.getOptionManager(); optionManager.setOption(OptionValue.createLong(OptionValue.OptionType.SYSTEM, ReflectionOptions.MAX_AUTOMATIC_REFLECTIONS.getOptionName(), descriptor.getLimit())); optionManager.setOption(OptionValue.createBoolean(OptionValue.OptionType.SYSTEM, ReflectionOptions.ENABLE_AUTOMATIC_AGG_REFLECTIONS.getOptionName(), descriptor.getAccelerateAggregation())); optionManager.setOption(OptionValue.createBoolean(OptionValue.OptionType.SYSTEM, ReflectionOptions.ENABLE_AUTOMATIC_RAW_REFLECTIONS.getOptionName(), descriptor.getAccelerateRaw())); if (descriptor.getLayoutRefreshMaxAttempts() != null) { optionManager.setOption(OptionValue.createLong(OptionValue.OptionType.SYSTEM, ExecConstants.LAYOUT_REFRESH_MAX_ATTEMPTS.getOptionName(), descriptor.getLayoutRefreshMaxAttempts())); } } }
@Override public void close() throws Exception { testContext.getOptions().setOption(OptionValue.createLong(OptionType.SYSTEM, validator.getOptionName(), oldValue)); }}; }
private OptionValue toOptionValue(Setting setting){ if (setting instanceof Setting.BooleanSetting) { return OptionValue.createBoolean(OptionType.SYSTEM, setting.getId(), ((Setting.BooleanSetting)setting).getValue()); } else if(setting instanceof Setting.FloatSetting) { return OptionValue.createDouble(OptionType.SYSTEM, setting.getId(), ((Setting.FloatSetting)setting).getValue()); } else if(setting instanceof Setting.IntegerSetting) { return OptionValue.createLong(OptionType.SYSTEM, setting.getId(), ((Setting.IntegerSetting)setting).getValue()); } else if(setting instanceof Setting.TextSetting) { return OptionValue.createString(OptionType.SYSTEM, setting.getId(), ((Setting.TextSetting)setting).getValue()); } else { throw new IllegalStateException("Unable to handle setting " + setting ); } }
public AutoCloseable with(final LongValidator validator, final long value){ final long oldValue = testContext.getOptions().getOption(validator); testContext.getOptions().setOption(OptionValue.createLong(OptionType.SYSTEM, validator.getOptionName(), value)); return new AutoCloseable(){ @Override public void close() throws Exception { testContext.getOptions().setOption(OptionValue.createLong(OptionType.SYSTEM, validator.getOptionName(), oldValue)); }}; }
protected static void setManagerRefreshDelayMs(long delayInMillis) { l(ContextService.class).get().getOptionManager().setOption( OptionValue.createLong(SYSTEM, REFLECTION_MANAGER_REFRESH_DELAY_MILLIS.getOptionName(), delayInMillis)); }
protected static void setDeletionGracePeriod(long periodInSeconds) { l(ContextService.class).get().getOptionManager().setOption( OptionValue.createLong(SYSTEM, REFLECTION_DELETION_GRACE_PERIOD.getOptionName(), periodInSeconds)); }
@Override public void applyOptions(OptionManager manager) { if (enableLeafLimits) { manager.setOption(createBoolean(QUERY, PlannerSettings.ENABLE_LEAF_LIMITS.getOptionName(), true)); manager.setOption(createLong(QUERY, ExecConstants.MAX_WIDTH_GLOBAL_KEY, manager.getOption(PlannerSettings.LEAF_LIMIT_MAX_WIDTH))); manager.setOption(createLong(QUERY, ExecConstants.SLICE_TARGET, 1)); } if (enableOutputLimits) { manager.setOption(createBoolean(QUERY, PlannerSettings.ENABLE_OUTPUT_LIMITS.getOptionName(), true)); } // always store results manager.setOption(createString(QUERY, PlannerSettings.STORE_QUERY_RESULTS.getOptionName(), StoreQueryResultsPolicy.PATH_AND_ATTEMPT_ID.name())); manager.setOption(createString(QUERY, PlannerSettings.QUERY_RESULTS_STORE_TABLE.getOptionName(), queryResultsStorePath)); if (!allowPartitionPruning) { manager.setOption(createBoolean(QUERY, PlannerSettings.ENABLE_PARTITION_PRUNING.getOptionName(), false)); } if (internalSingleThreaded) { manager.setOption(createBoolean(QUERY, ExecConstants.SORT_FILE_BLOCKS.getOptionName(), true)); manager.setOption(createBoolean(QUERY, PlannerSettings.EXCHANGE.getOptionName(), true)); } }
@Test public void testJobCleanup() throws Exception { jobsService = (LocalJobsService) l(JobsService.class); SqlQuery ctas = getQueryFromSQL("SHOW SCHEMAS"); Job job = jobsService.submitJob(JobRequest.newBuilder() .setSqlQuery(ctas) .build(), NoOpJobStatusListener.INSTANCE); job.getData().loadIfNecessary(); SabotContext context = l(SabotContext.class); OptionValue days = OptionValue.createLong(OptionType.SYSTEM, ExecConstants.RESULTS_MAX_AGE_IN_DAYS.getOptionName(), 0); context.getOptionManager().setOption(days); OptionValue millis = OptionValue.createLong(OptionType.SYSTEM, ExecConstants.DEBUG_RESULTS_MAX_AGE_IN_MILLISECONDS.getOptionName(), 10); context.getOptionManager().setOption(millis); Thread.sleep(20); LocalJobsService.CleanupTask cleanupTask = jobsService.new CleanupTask(); cleanupTask.cleanup(); //make sure that the job output directory is gone assertFalse(jobsService.getJobResultsStore().jobOutputDirectoryExists(job.getJobId())); job = jobsService.getJob(job.getJobId()); assertFalse(JobDetailsUI.of(job).getResultsAvailable()); context.getOptionManager().setOption(OptionValue.createLong(OptionType.SYSTEM, ExecConstants.RESULTS_MAX_AGE_IN_DAYS.getOptionName(), 30)); context.getOptionManager().setOption(OptionValue.createLong(OptionType.SYSTEM, ExecConstants.DEBUG_RESULTS_MAX_AGE_IN_MILLISECONDS.getOptionName(), 0)); }
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)); }
OptionValue.createLong(OptionValue.OptionType.SYSTEM, "planner.slice_target", 1) ); context.getOptionManager().setOption( OptionValue.createLong(OptionValue.OptionType.SYSTEM, "planner.width.max_per_node", 10) ); context.getOptionManager().setOption(
OptionValue.createLong(OptionValue.OptionType.SYSTEM, "planner.slice_target", 1) ); context.getOptionManager().setOption( OptionValue.createLong(OptionValue.OptionType.SYSTEM, "planner.width.max_per_node", 10) ); context.getOptionManager().setOption( ); context.getOptionManager().setOption( OptionValue.createLong(OptionValue.OptionType.SYSTEM, "exec.queue.memory.small", 4096) );