@Override public OptionList getOptionList() { final OptionList list = new OptionList(); for (final OptionValue value : getLocalOptions()) { list.add(value); } return list; } }
public void merge(OptionList list){ this.addAll(list); }
public OptionList getSystemOptions() { final OptionList options = new OptionList(); Iterators.addAll(options, Iterators.filter(iterator(), IS_SYSTEM_OPTION)); return options; }
/** * @return all system options that have been set to a non-default value */ public OptionList getNonDefaultOptions() { Iterator<OptionValue> persistedOptions = Iterators.transform(this.options.getAll(), EXTRACT_OPTIONS); OptionList nonDefaultOptions = new OptionList(); Iterators.addAll(nonDefaultOptions, persistedOptions); return nonDefaultOptions; }
final OptionList list; if (!fragment.hasOptionsJson() || fragment.getOptionsJson().isEmpty()) { list = new OptionList(); } else { try { for (OptionValue option : list.getSystemOptions()) { optionManager.setOption(option); final OptionManager fragmentOptions = new FragmentOptionManager(optionManager, list.getNonSystemOptions());
/** * only merge options from list that are not already set locally */ public void mergeIfNotPresent(OptionList list) { Set<String> options = Sets.newHashSet(Iterables.transform(this, EXTRACT_OPTION_NAME)); for (OptionValue optionValue : list) { if (!options.contains(optionValue.getName())) { this.add(optionValue); } } }
@Override public OptionList getOptionList() { OptionList list = super.getOptionList(); list.merge(fallback.getOptionList()); return list; }
/** * Return the list of all non-default options including QUERY, SESSION and SYSTEM level * @return */ public OptionList getNonDefaultOptions() { final OptionList nonDefaultOptions = queryOptions.getOptionList(); nonDefaultOptions.mergeIfNotPresent(sabotContext.getOptionManager().getNonDefaultOptions()); return nonDefaultOptions; }
@Before public void setUp() { MockitoAnnotations.initMocks(this); // Boilerplate AccelerationManager accelerationManager = mock(AccelerationManager.class); AccelerationDetailsPopulator populator = mock(AccelerationDetailsPopulator.class); when(populator.computeAcceleration()).thenReturn(ByteString.EMPTY_BYTE_ARRAY); when(accelerationManager.newPopulator()).thenReturn(populator); when(context.getAccelerationManager()).thenReturn(accelerationManager); when(context.getQueryUserName()).thenReturn("myuser"); when(context.getSession()).thenReturn(UserSession.Builder.newBuilder().build()); when(context.getNonDefaultOptions()).thenReturn(new OptionList()); when(catalog.getMetadataStatsCollector()).thenReturn(new MetadataStatsCollector()); }
private void print(String fragmentFile, int bitCount, int expectedFragmentCount) throws Exception{ System.out.println(String.format("=================Building plan fragments for [%s]. Allowing %d total Nodes.==================", fragmentFile, bitCount)); PhysicalPlanReader ppr = PhysicalPlanReaderTestFactory.defaultPhysicalPlanReader(DEFAULT_SABOT_CONFIG, CLASSPATH_SCAN_RESULT); Fragment fragmentRoot = getRootFragment(ppr, fragmentFile); SimpleParallelizer par = new SimpleParallelizer(1000*1000, 5, 10, 1.2, AbstractAttemptObserver.NOOP, true, 1.5d); List<NodeEndpoint> endpoints = Lists.newArrayList(); NodeEndpoint localBit = null; for(int i =0; i < bitCount; i++) { NodeEndpoint b1 = NodeEndpoint.newBuilder().setAddress("localhost").setFabricPort(1234+i).build(); if (i == 0) { localBit = b1; } endpoints.add(b1); } final QueryContextInformation queryContextInfo = Utilities.createQueryContextInfo("dummySchemaName"); List<PlanFragment> qwu = par.getFragments(new OptionList(), localBit, QueryId.getDefaultInstance(), endpoints, ppr, fragmentRoot, UserSession.Builder.newBuilder().withCredentials(UserBitShared.UserCredentials.newBuilder().setUserName("foo").build()).build(), queryContextInfo, null); assertEquals(expectedFragmentCount, qwu.size() + 1 /* root fragment is not part of the getFragments() list*/); }
public OptionList getNonSystemOptions() { final OptionList options = new OptionList(); Iterators.addAll(options, Iterators.filter(iterator(), Predicates.not(IS_SYSTEM_OPTION))); return options; } }
List<PlanFragment> fragments = PARALLELIZER.getFragments(new OptionList(), sabotContext.getEndpoint(), QueryId.getDefaultInstance(), sabotContext.getExecutors(), planReader, rootFragment, USER_SESSION, queryContextInfo, Mockito.mock(FunctionLookupContext.class));