Example usage for com.google.common.collect ImmutableSet.Builder addAll

List of usage examples for com.google.common.collect ImmutableSet.Builder addAll

Introduction

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

Prototype

boolean addAll(Collection<? extends E> c);

Source Link

Document

Adds all of the elements in the specified collection to this set if they're not already present (optional operation).

Usage

From source file:com.spectralogic.ds3autogen.java.generators.requestmodels.StreamRequestPayloadGenerator.java

/**
 * Gets all the required imports that the Request will need in order to properly
 * generate the Java request code, including the Stream and Channel imports
 *//*from www .  jav a2  s . c o m*/
@Override
public ImmutableList<String> getAllImports(final Ds3Request ds3Request, final String packageName) {
    final ImmutableSet.Builder<String> builder = ImmutableSet.builder();

    builder.add(getParentImport(ds3Request));
    builder.addAll(getImportsFromParamList(ds3Request.getRequiredQueryParams()));
    builder.addAll(getImportsFromParamList(ds3Request.getOptionalQueryParams()));

    if (isResourceAnArg(ds3Request.getResource(), ds3Request.includeIdInPath())) {
        if (RequestConverterUtil.isResourceId(ds3Request.getResource())) {
            builder.add("java.util.UUID");
        }
        builder.add("com.google.common.net.UrlEscapers");
    }
    builder.add("com.spectralogic.ds3client.utils.SeekableByteChannelInputStream");
    builder.add("java.nio.channels.SeekableByteChannel");
    builder.add("java.io.InputStream");

    return builder.build().asList();
}

From source file:com.facebook.presto.tests.tpch.TpchIndexResolver.java

@Override
public ConnectorResolvedIndex resolveIndex(ConnectorSession session, ConnectorTableHandle tableHandle,
        Set<ColumnHandle> indexableColumns, TupleDomain<ColumnHandle> tupleDomain) {
    TpchTableHandle tpchTableHandle = checkType(tableHandle, TpchTableHandle.class, "tableHandle");

    // Keep the fixed values that don't overlap with the indexableColumns
    // Note: technically we could more efficiently utilize the overlapped columns, but this way is simpler for now

    Map<ColumnHandle, NullableValue> fixedValues = TupleDomain.extractFixedValues(tupleDomain)
            .orElse(ImmutableMap.of()).entrySet().stream()
            .filter(entry -> !indexableColumns.contains(entry.getKey()))
            .filter(entry -> !entry.getValue().isNull()) // strip nulls since meaningless in index join lookups
            .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));

    // determine all columns available for index lookup
    ImmutableSet.Builder<String> builder = ImmutableSet.builder();
    builder.addAll(transform(indexableColumns, columnNameGetter()));
    builder.addAll(transform(fixedValues.keySet(), columnNameGetter()));
    Set<String> lookupColumnNames = builder.build();

    // do we have an index?
    if (!indexedData.getIndexedTable(tpchTableHandle.getTableName(), tpchTableHandle.getScaleFactor(),
            lookupColumnNames).isPresent()) {
        return null;
    }//from  w w  w  .  j  a  v a2s.  c  o m

    TupleDomain<ColumnHandle> filteredTupleDomain = tupleDomain;
    if (!tupleDomain.isNone()) {
        filteredTupleDomain = TupleDomain.withColumnDomains(
                Maps.filterKeys(tupleDomain.getDomains().get(), not(in(fixedValues.keySet()))));
    }
    return new ConnectorResolvedIndex(new TpchIndexHandle(connectorId, tpchTableHandle.getTableName(),
            tpchTableHandle.getScaleFactor(), lookupColumnNames, TupleDomain.fromFixedValues(fixedValues)),
            filteredTupleDomain);
}

From source file:io.airlift.discovery.InMemoryDynamicStore.java

@Override
public synchronized Set<Service> getAll() {
    removeExpired();/*from ww  w.  j a  v  a2  s . co m*/

    ImmutableSet.Builder<Service> builder = ImmutableSet.builder();
    for (Entry entry : descriptors.values()) {
        builder.addAll(entry.getServices());
    }
    return builder.build();
}

From source file:com.spectralogic.ds3autogen.java.generators.responsemodels.BaseResponseGenerator.java

/**
 * Gets all the imports associated with response types that the response will
 * need in order to properly generate the Java request code
 *///w  w  w.ja v a  2  s  .c  o m
@Override
public ImmutableSet<String> getAllImports(final Ds3Request ds3Request) {
    if (isEmpty(ds3Request.getDs3ResponseCodes())) {
        return ImmutableSet.of(getParentImport());
    }
    final ImmutableList<Ds3ResponseCode> responseCodes = ds3Request.getDs3ResponseCodes().stream()
            .filter(i -> i.getCode() < ERROR_CODE_THRESHOLD).collect(GuavaCollectors.immutableList());

    final ImmutableSet.Builder<String> builder = ImmutableSet.builder();
    builder.addAll(getImportListFromResponseCodes(responseCodes));
    builder.add("com.spectralogic.ds3client.models.ChecksumType");
    builder.add(getParentImport());
    return builder.build();
}

From source file:com.facebook.buck.rules.AbstractNodeBuilder.java

protected <C extends Comparable<?>> ImmutableSet<C> amendSet(ImmutableSet<C> existing, C instance) {
    ImmutableSet.Builder<C> toReturn = ImmutableSet.builder();
    toReturn.addAll(existing);
    toReturn.add(instance);// w w  w .j ava2s .co m
    return toReturn.build();
}

From source file:com.lithium.flow.config.BaseConfig.java

@Override
@Nonnull/*from w  w  w.  j  a  va 2 s  . c  om*/
public Set<String> keySet() {
    ImmutableSet.Builder<String> builder = ImmutableSet.builder();
    if (defaults != null) {
        builder.addAll(defaults.keySet());
    }
    builder.addAll(store.getKeys());
    return builder.build();
}

From source file:org.lanternpowered.server.asset.AbstractMultiAssetRepository.java

@Override
public Collection<Asset> getLoadedAssets() {
    final ImmutableSet.Builder<Asset> builder = ImmutableSet.builder();
    for (AssetRepository repository : getRepositories()) {
        builder.addAll(repository.getLoadedAssets());
    }/*from  w w w  .  java  2s .c  o m*/
    return builder.build();
}

From source file:org.gradle.api.internal.file.CachingTaskInputFileCollection.java

@Override
public void visitContents(FileCollectionResolveContext context) {
    if (canCache) {
        if (cachedValue == null) {
            DefaultFileCollectionResolveContext nested = new DefaultFileCollectionResolveContext(fileResolver);
            super.visitContents(nested);
            ImmutableSet.Builder<File> files = ImmutableSet.builder();
            for (FileCollectionInternal fileCollection : nested.resolveAsFileCollections()) {
                files.addAll(fileCollection);
            }//from  w  w w  .j a va 2s . com
            this.cachedValue = files.build();
        }
        context.add(cachedValue);
    } else {
        super.visitContents(context);
    }
}

From source file:com.facebook.presto.tpch.TpchIndexResolver.java

@Override
public ConnectorResolvedIndex resolveIndex(ConnectorTableHandle tableHandle,
        Set<ConnectorColumnHandle> indexableColumns, TupleDomain<ConnectorColumnHandle> tupleDomain) {
    checkArgument(tableHandle instanceof TpchTableHandle,
            "tableHandle is not an instance of TpchTableHandle: %s", tableHandle);
    TpchTableHandle tpchTableHandle = (TpchTableHandle) tableHandle;

    // Keep the fixed values that don't overlap with the indexableColumns
    // Note: technically we could more efficiently utilize the overlapped columns, but this way is simpler for now
    Map<ConnectorColumnHandle, Comparable<?>> fixedValues = Maps.filterKeys(tupleDomain.extractFixedValues(),
            not(in(indexableColumns)));/*  w ww . j a v  a  2  s  . c  om*/

    // determine all columns available for index lookup
    ImmutableSet.Builder<String> builder = ImmutableSet.builder();
    builder.addAll(transform(indexableColumns, columnNameGetter()));
    builder.addAll(transform(fixedValues.keySet(), columnNameGetter()));
    Set<String> lookupColumnNames = builder.build();

    // do we have an index?
    if (!indexedData.getIndexedTable(tpchTableHandle.getTableName(), tpchTableHandle.getScaleFactor(),
            lookupColumnNames).isPresent()) {
        return null;
    }

    TupleDomain<ConnectorColumnHandle> filteredTupleDomain = tupleDomain;
    if (!tupleDomain.isNone()) {
        filteredTupleDomain = TupleDomain
                .withColumnDomains(Maps.filterKeys(tupleDomain.getDomains(), not(in(fixedValues.keySet()))));
    }
    return new ConnectorResolvedIndex(new TpchIndexHandle(connectorId, tpchTableHandle.getTableName(),
            tpchTableHandle.getScaleFactor(), lookupColumnNames, TupleDomain.withFixedValues(fixedValues)),
            filteredTupleDomain);
}

From source file:org.sosy_lab.cpachecker.cpa.pointer2.util.ExplicitLocationSet.java

@Override
public LocationSet addElement(String pLocation) {
    if (explicitSet.contains(pLocation)) {
        return this;
    }/*from   ww  w.  j av  a  2 s.  c  om*/
    ImmutableSet.Builder<String> builder = ImmutableSet.builder();
    builder.addAll(explicitSet).add(pLocation);
    return new ExplicitLocationSet(builder.build());
}