Example usage for com.google.common.collect ImmutableSet builder

Introduction

In this page you can find the example usage for com.google.common.collect ImmutableSet builder.

Prototype

public static <E> Builder<E> builder()

Source Link

Usage

From source file:com.googlesource.gerrit.plugins.github.group.GitHubGroupMembership.java

@Override
public Set<UUID> intersection(Iterable<UUID> groupIds) {
    ImmutableSet.Builder<UUID> groups = new ImmutableSet.Builder<>();
    for (UUID uuid : groupIds) {
        if (contains(uuid)) {
            groups.add(uuid);// www  .  j a  v  a2s . c o m
        }
    }
    return groups.build();
}

From source file:org.apache.gobblin.metrics.context.filter.AllContextFilter.java

private void addContextsRecursively(ImmutableSet.Builder<MetricContext> builder, MetricContext metricContext) {
    builder.add(metricContext);/* w  w  w  . ja  va2  s .  c o  m*/
    for (MetricContext context : metricContext.getChildContextsAsMap().values()) {
        addContextsRecursively(builder, context);
    }
}

From source file:org.b1.pack.standard.writer.CompressedFormatDetector.java

private static void readResource(final URL url, ImmutableSet.Builder<String> builder) throws IOException {
    StringTokenizer tokenizer = new StringTokenizer(CharStreams.toString(new InputSupplier<Reader>() {
        @Override//from   w  w w.  j  a v a  2s  .  c  o  m
        public Reader getInput() throws IOException {
            return new InputStreamReader(url.openStream(), Charsets.UTF_8);
        }
    }));
    while (tokenizer.hasMoreTokens()) {
        builder.add(tokenizer.nextToken().toLowerCase());
    }
}

From source file:org.graylog2.lookup.DtoLoader.java

@Inject
DtoLoader(@Assisted Collection<LookupTableDto> lookupTableDtos, MongoLutCacheService cacheService,
        MongoLutDataAdapterService dataAdapterService) {
    final ImmutableSet.Builder<String> cacheIds = ImmutableSet.builder();
    final ImmutableSet.Builder<String> adapterIds = ImmutableSet.builder();

    lookupTableDtos.forEach(dto -> {//  ww  w  .  j ava2  s.c o  m
        cacheIds.add(dto.cacheId());
        adapterIds.add(dto.dataAdapterId());
    });

    this.caches = Maps.uniqueIndex(cacheService.findByIds(cacheIds.build()), CacheDto::id);
    this.dataAdapters = Maps.uniqueIndex(dataAdapterService.findByIds(adapterIds.build()), DataAdapterDto::id);
}

From source file:org.apache.aurora.benchmark.JobUpdates.java

/**
 * Saves job updates into provided storage.
 *
 * @param storage {@link Storage} instance.
 * @param updates updates to save./*from ww w. ja v a  2 s  .  c  om*/
 * @return update keys.
 */
static Set<IJobUpdateKey> saveUpdates(Storage storage, Iterable<IJobUpdateDetails> updates) {
    ImmutableSet.Builder<IJobUpdateKey> keyBuilder = ImmutableSet.builder();
    storage.write((Storage.MutateWork.NoResult.Quiet) store -> {
        JobUpdateStore.Mutable updateStore = store.getJobUpdateStore();
        updateStore.deleteAllUpdatesAndEvents();
        for (IJobUpdateDetails details : updates) {
            IJobUpdateKey key = details.getUpdate().getSummary().getKey();
            keyBuilder.add(key);
            String lockToken = UUID.randomUUID().toString();
            store.getLockStore().saveLock(ILock.build(new Lock().setKey(LockKey.job(key.getJob().newBuilder()))
                    .setToken(lockToken).setUser(Builder.USER).setTimestampMs(0L)));

            updateStore.saveJobUpdate(details.getUpdate(), Optional.of(lockToken));

            for (IJobUpdateEvent updateEvent : details.getUpdateEvents()) {
                updateStore.saveJobUpdateEvent(key, updateEvent);
            }

            for (IJobInstanceUpdateEvent instanceEvent : details.getInstanceEvents()) {
                updateStore.saveJobInstanceUpdateEvent(key, instanceEvent);
            }
        }
    });
    return keyBuilder.build();
}

From source file:io.druid.server.lookup.cache.polling.OnHeapPollingCache.java

public OnHeapPollingCache(Iterable<Map.Entry<K, V>> entries) {

    if (entries == null) {
        immutableMap = ImmutableMap.of();
        immutableReverseMap = ImmutableMap.of();
    } else {/*from   w ww.  j  ava  2s  .com*/
        ImmutableSet.Builder<V> setOfValuesBuilder = ImmutableSet.builder();
        ImmutableMap.Builder<K, V> mapBuilder = ImmutableMap.builder();
        for (Map.Entry<K, V> entry : entries) {
            setOfValuesBuilder.add(entry.getValue());
            mapBuilder.put(entry.getKey(), entry.getValue());
        }
        final Set<V> setOfValues = setOfValuesBuilder.build();
        immutableMap = mapBuilder.build();
        immutableReverseMap = ImmutableMap.copyOf(Maps.asMap(setOfValues, new Function<V, List<K>>() {
            @Override
            public List<K> apply(final V input) {
                return Lists.newArrayList(Maps.filterKeys(immutableMap, new Predicate<K>() {
                    @Override
                    public boolean apply(K key) {
                        V retVal = immutableMap.get(key);
                        if (retVal == null) {
                            return false;
                        }
                        return retVal.equals(input);
                    }
                }).keySet());
            }
        }));
    }

}

From source file:com.netflix.genie.common.dto.ClusterCriteria.java

/**
 * Create a cluster criteria object with the included tags.
 *
 * @param tags The tags to add. Not null or empty and must have at least one non-empty tag.
 *//*from w  ww.ja  v  a 2s  .c o  m*/
@JsonCreator
public ClusterCriteria(@JsonProperty("tags") final Set<String> tags) {
    final ImmutableSet.Builder<String> builder = ImmutableSet.builder();
    tags.forEach(tag -> {
        if (StringUtils.isNotBlank(tag)) {
            builder.add(tag);
        }
    });
    this.tags = builder.build();
}

From source file:com.twitter.aurora.scheduler.configuration.ParsedConfiguration.java

/**
 * Constructs a ParsedConfiguration object and populates the set of {@link TaskConfig}s for
 * the provided config./*from   ww w.  j  a  v a  2  s  . c  om*/
 *
 * @param parsed A parsed {@link JobConfiguration}.
 */
@VisibleForTesting
public ParsedConfiguration(JobConfiguration parsed) {
    this.parsed = parsed;
    ImmutableSet.Builder<TaskConfig> builder = ImmutableSet.builder();
    for (int i = 0; i < parsed.getShardCount(); i++) {
        builder.add(parsed.getTaskConfig().deepCopy().setShardId(i));
    }
    this.tasks = builder.build();
}

From source file:com.google.caliper.runner.RunnerModule.java

@Provides
ImmutableSet<VirtualMachine> provideVirtualMachines(CaliperOptions options, CaliperConfig config)
        throws InvalidConfigurationException {
    ImmutableSet<String> vmNames = options.vmNames();
    ImmutableSet.Builder<VirtualMachine> builder = ImmutableSet.builder();
    if (vmNames.isEmpty()) {
        builder.add(new VirtualMachine("default", config.getDefaultVmConfig()));
    } else {/*from  ww w . j ava 2  s . c o m*/
        for (String vmName : vmNames) {
            VmConfig vmConfig = config.getVmConfig(vmName);
            builder.add(new VirtualMachine(vmName, vmConfig));
        }
    }
    return builder.build();
}

From source file:zipkin2.storage.cassandra.v1.CompositeIndexer.java

CompositeIndexer(CassandraStorage storage, CacheBuilderSpec spec, int indexTtl) {
    this.sharedState = CacheBuilder.from(spec).<PartitionKeyToTraceId, Pair>build().asMap();
    Indexer.Factory factory = new Indexer.Factory(storage.session(), indexTtl, sharedState);
    ImmutableSet.Builder<Indexer> indexers = ImmutableSet.builder();
    indexers.add(factory.create(new InsertTraceIdByServiceName(storage.bucketCount)));
    if (storage.metadata().hasRemoteService) {
        indexers.add(factory.create(new InsertTraceIdByRemoteServiceName()));
    }/* ww  w.  ja v  a  2 s . c o m*/
    indexers.add(factory.create(new InsertTraceIdBySpanName()));
    indexers.add(factory.create(new InsertTraceIdByAnnotation(storage.bucketCount)));
    this.indexers = indexers.build();
}