public void revokeAllRolesFor(String username) { final RoleUser roleUser = new RoleUser(username); synchronized (roleToUsersMappings) { Set<PluginRoleConfig> pluginRoles = new HashSet<>(roleToUsersMappings.keySet()); for (PluginRoleConfig pluginRole : pluginRoles) { roleToUsersMappings.get(pluginRole).remove(roleUser); } } }
public void setComponentRemoved(int networkId, Class<? extends Component> component) { if (netRelevant.contains(networkId) && !netInitial.contains(networkId)) { if (!addedComponents.remove(networkId, component)) { removedComponents.put(networkId, component); if (!dirtyComponents.remove(networkId, component)) { netDirty.add(networkId); } } } }
final String indexName = types.indexName(); types.fields().forEach(fieldType -> { final String fieldName = fieldType.fieldName(); final String physicalType = fieldType.physicalType(); if (indexNames.isEmpty() || indexNames.contains(indexName)) { if (!fields.containsKey(fieldName)) { fields.put(fieldName, HashMultimap.create()); fields.get(fieldName).put(physicalType, indexName); final Map<String, Collection<String>> physicalTypes = fieldNameEntry.getValue().asMap(); .map(entry -> { final String physicalType = entry.getKey(); final Set<String> indices = ImmutableSet.copyOf(entry.getValue()); .filter(Optional::isPresent) .map(Optional::get) .collect(Collectors.toSet());
private Map<HivePartitionName, PartitionStatistics> loadPartitionColumnStatistics(Iterable<? extends HivePartitionName> keys) { SetMultimap<HiveTableName, HivePartitionName> tablePartitions = stream(keys) .collect(toImmutableSetMultimap(HivePartitionName::getHiveTableName, key -> key)); ImmutableMap.Builder<HivePartitionName, PartitionStatistics> result = ImmutableMap.builder(); tablePartitions.keySet().forEach(table -> { Set<String> partitionNames = tablePartitions.get(table).stream() .map(partitionName -> partitionName.getPartitionName().get()) .collect(toImmutableSet()); Map<String, PartitionStatistics> partitionStatistics = delegate.getPartitionStatistics(table.getDatabaseName(), table.getTableName(), partitionNames); for (String partitionName : partitionNames) { if (!partitionStatistics.containsKey(partitionName)) { throw new PrestoException(HIVE_PARTITION_DROPPED_DURING_QUERY, "Statistics result does not contain entry for partition: " + partitionName); } result.put(HivePartitionName.hivePartitionName(table, partitionName), partitionStatistics.get(partitionName)); } }); return result.build(); }
@Override public JsonElement serialize(BindsConfig bindsConfig, Type typeOfBindsConfig, JsonSerializationContext context) { JsonObject result = new JsonObject(); SetMultimap<Name, SimpleUri> bindingByModuleName = listBindingsByModuleName(bindsConfig); List<Name> sortedModuleNames = bindingByModuleName.keySet().stream().sorted().collect(toList()); for (Name moduleName : sortedModuleNames) { SetMultimap<String, Input> moduleBinds = HashMultimap.create(); for (SimpleUri bindingUri : bindingByModuleName.get(moduleName)) { moduleBinds.putAll(bindingUri.getObjectName().toString(), bindsConfig.uriBoundInputs.get(bindingUri)); } JsonElement map = context.serialize(moduleBinds, SetMultimap.class); result.add(moduleName.toString(), map); } return result; }
nodeMap.getNodesByHostAndPort().get(host).stream() .filter(node -> includeCoordinator || !coordinatorIds.contains(node.getNodeIdentifier())) .forEach(chosen::add); nodeMap.getNodesByHost().get(address).stream() .filter(node -> includeCoordinator || !coordinatorIds.contains(node.getNodeIdentifier())) .forEach(chosen::add); nodeMap.getNodesByHostAndPort().get(host).stream() .forEach(chosen::add); nodeMap.getNodesByHost().get(address).stream() .forEach(chosen::add);
constructors.stream().forEach((c) -> builder.putAll(c, initThusFar)); Symbol.ClassSymbol classSymbol = ASTHelpers.getSymbol(enclosingClass); FieldInitEntities entities = class2Entities.get(classSymbol); MethodTree initMethod = entities.instanceInitializerMethods().iterator().next(); Set<Symbol> constructorUninitSymbols = class2ConstructorUninit.get(classSymbol);
@Override SetMultimap<String, Integer> filter(SetMultimap<String, Integer> multimap) { ImmutableSetMultimap<String, Integer> badEntries = ImmutableSetMultimap.of("foo", 314, "one", 159, "two", 265, "bar", 358); multimap.putAll(badEntries); multimap = Multimaps.filterEntries( multimap, Predicates.not( Predicates.in(ImmutableMap.of("one", 159, "two", 265).entrySet()))); return Multimaps.filterKeys( multimap, Predicates.not(Predicates.in(ImmutableSet.of("foo", "bar")))); } })
public void testFilteredKeysSetMultimapGetBadValue() { SetMultimap<String, Integer> multimap = LinkedHashMultimap.create(); multimap.put("foo", 1); multimap.put("bar", 2); multimap.put("baz", 3); multimap.put("bar", 4); SetMultimap<String, Integer> filtered = Multimaps.filterKeys(multimap, Predicates.in(ImmutableSet.of("foo", "bar"))); Set<Integer> bazSet = filtered.get("baz"); assertThat(bazSet).isEmpty(); try { bazSet.add(5); fail("Expected IllegalArgumentException"); } catch (IllegalArgumentException expected) { } try { bazSet.addAll(ImmutableSet.of(6, 7)); fail("Expected IllegalArgumentException"); } catch (IllegalArgumentException expected) { } }
public void testFilteredKeysSetMultimapReplaceValues() { SetMultimap<String, Integer> multimap = LinkedHashMultimap.create(); multimap.put("foo", 1); multimap.put("bar", 2); multimap.put("baz", 3); multimap.put("bar", 4); SetMultimap<String, Integer> filtered = Multimaps.filterKeys(multimap, Predicates.in(ImmutableSet.of("foo", "bar"))); assertEquals(ImmutableSet.of(), filtered.replaceValues("baz", ImmutableSet.<Integer>of())); try { filtered.replaceValues("baz", ImmutableSet.of(5)); fail("Expected IllegalArgumentException"); } catch (IllegalArgumentException expected) { } }
@MapFeature.Require(SUPPORTS_PUT) public void testPutDuplicateValue() { List<Entry<K, V>> entries = copyToList(multimap().entries()); for (Entry<K, V> entry : entries) { resetContainer(); K k = entry.getKey(); V v = entry.getValue(); Set<V> values = multimap().get(k); Set<V> expectedValues = copyToSet(values); assertFalse(multimap().put(k, v)); assertEquals(expectedValues, values); assertGet(k, expectedValues); } }