Example usage for com.google.common.collect Sets newTreeSet

List of usage examples for com.google.common.collect Sets newTreeSet

Introduction

In this page you can find the example usage for com.google.common.collect Sets newTreeSet.

Prototype

public static <E> TreeSet<E> newTreeSet(Comparator<? super E> comparator) 

Source Link

Document

Creates a mutable, empty TreeSet instance with the given comparator.

Usage

From source file:org.gradle.api.reporting.internal.DefaultReportContainer.java

@Override
public SortedSet<String> getEnabledReportNames() {
    return Sets.newTreeSet(Iterables.transform(getEnabled(), REPORT_NAME));
}

From source file:org.apache.bookkeeper.bookie.ScanAndCompareGarbageCollector.java

@Override
public void gc(GarbageCleaner garbageCleaner) {
    try {/*from  w w w .  j a v a  2  s . com*/
        // Get a set of all ledgers on the bookie
        NavigableSet<Long> bkActiveLedgers = Sets
                .newTreeSet(ledgerStorage.getActiveLedgersInRange(0, Long.MAX_VALUE));

        // Iterate over all the ledger on the metadata store
        LedgerRangeIterator ledgerRangeIterator = ledgerManager.getLedgerRanges();

        if (!ledgerRangeIterator.hasNext()) {
            // Empty global active ledgers, need to remove all local active ledgers.
            for (long ledgerId : bkActiveLedgers) {
                garbageCleaner.clean(ledgerId);
            }
        }

        long lastEnd = -1;

        long curTime = MathUtils.now();
        boolean checkOverreplicatedLedgers = (enableGcOverReplicatedLedger
                && curTime - lastOverReplicatedLedgerGcTimeMillis > gcOverReplicatedLedgerIntervalMillis);
        if (checkOverreplicatedLedgers) {
            zk = ZooKeeperClient.newBuilder().connectString(conf.getZkServers())
                    .sessionTimeoutMs(conf.getZkTimeout()).build();
            // remove all the overreplicated ledgers from the local bookie
            Set<Long> overReplicatedLedgers = removeOverReplicatedledgers(bkActiveLedgers, garbageCleaner);
            if (overReplicatedLedgers.isEmpty()) {
                LOG.info("No over-replicated ledgers found.");
            } else {
                LOG.info("Removed over-replicated ledgers: {}", overReplicatedLedgers);
            }
            lastOverReplicatedLedgerGcTimeMillis = MathUtils.now();
        }

        while (ledgerRangeIterator.hasNext()) {
            LedgerRange lRange = ledgerRangeIterator.next();

            Long start = lastEnd + 1;
            Long end = lRange.end();
            if (!ledgerRangeIterator.hasNext()) {
                end = Long.MAX_VALUE;
            }

            Iterable<Long> subBkActiveLedgers = bkActiveLedgers.subSet(start, true, end, true);

            Set<Long> ledgersInMetadata = lRange.getLedgers();
            if (LOG.isDebugEnabled()) {
                LOG.debug("Active in metadata {}, Active in bookie {}", ledgersInMetadata, subBkActiveLedgers);
            }
            for (Long bkLid : subBkActiveLedgers) {
                if (!ledgersInMetadata.contains(bkLid)) {
                    garbageCleaner.clean(bkLid);
                }
            }
            lastEnd = end;
        }
    } catch (Throwable t) {
        // ignore exception, collecting garbage next time
        LOG.warn("Exception when iterating over the metadata {}", t);
    } finally {
        if (zk != null) {
            try {
                zk.close();
            } catch (InterruptedException e) {
                LOG.error("Error closing zk session", e);
            }
            zk = null;
        }
    }
}

From source file:org.apache.druid.segment.IndexMerger.java

static <T extends Comparable<? super T>> ArrayList<T> mergeIndexed(List<Iterable<T>> indexedLists) {
    Set<T> retVal = Sets.newTreeSet(Comparators.naturalNullsFirst());

    for (Iterable<T> indexedList : indexedLists) {
        for (T val : indexedList) {
            retVal.add(val);
        }/*from w  ww . java  2  s. c  o  m*/
    }

    return Lists.newArrayList(retVal);
}

From source file:org.apache.calcite.sql.validate.AggregatingSelectScope.java

private Resolved resolve() {
    final ImmutableList.Builder<ImmutableList<ImmutableBitSet>> builder = ImmutableList.builder();
    List<SqlNode> extraExprs = ImmutableList.of();
    Map<Integer, Integer> groupExprProjection = ImmutableMap.of();
    if (select.getGroup() != null) {
        final SqlNodeList groupList = select.getGroup();
        final SqlValidatorUtil.GroupAnalyzer groupAnalyzer = new SqlValidatorUtil.GroupAnalyzer(
                temporaryGroupExprList);
        for (SqlNode groupExpr : groupList) {
            SqlValidatorUtil.analyzeGroupItem(this, groupAnalyzer, builder, groupExpr);
        }//from  w ww.  j a v  a 2 s . c o  m
        extraExprs = groupAnalyzer.extraExprs;
        groupExprProjection = groupAnalyzer.groupExprProjection;
    }

    final Set<ImmutableBitSet> flatGroupSets = Sets.newTreeSet(ImmutableBitSet.COMPARATOR);
    for (List<ImmutableBitSet> groupSet : Linq4j.product(builder.build())) {
        flatGroupSets.add(ImmutableBitSet.union(groupSet));
    }

    // For GROUP BY (), we need a singleton grouping set.
    if (flatGroupSets.isEmpty()) {
        flatGroupSets.add(ImmutableBitSet.of());
    }

    return new Resolved(extraExprs, temporaryGroupExprList, flatGroupSets, groupExprProjection);
}

From source file:ru.codeinside.adm.parser.EmployeeFixtureParser.java

private Set<Role> parseRoles(Splitter groupSplitter, int lineNumber, String rolesString) {
    Set<String> roleNames = Sets.newTreeSet(groupSplitter.split(rolesString));
    Set<Role> roles = EnumSet.noneOf(Role.class);
    for (String roleName : roleNames) {
        try {/*from ww  w . jav  a 2s  .c  o m*/
            roles.add(Role.valueOf(roleName));
        } catch (Exception e) {
            throw new IllegalStateException(
                    "???  ( ?:" + lineNumber + "):" + roleName);
        }
    }

    return roles;
}

From source file:edu.washington.cs.cupid.internal.CapabilityRegistry.java

@Override
public SortedSet<ICapability> getCapabilities(Predicate<ICapability> filter) {
    SortedSet<ICapability> result = Sets.newTreeSet(CapabilityUtil.COMPARE_NAME);
    for (ICapability capability : capabilities) {
        if (filter.apply(capability)) {
            result.add(capability);//from  www . j a  va  2s .c  o  m
        }
    }
    return result;
}

From source file:com.eucalyptus.reporting.domain.ReportingComputeDomainModel.java

public static Set<String> getZones() {
    return Sets.newTreeSet(zoneModels.keySet());
}

From source file:de.faustedition.search.SearchResource.java

private static SortedSet<String> toSortedValues(String[] metadata) {
    return Sets.newTreeSet(Arrays.asList(Objects.firstNonNull(metadata, new String[0])));
}

From source file:uk.co.unclealex.executable.generator.CodeGeneratorImpl.java

/**
 * {@inheritDoc}// w w  w  .j a  v  a2  s .  c om
 */
@Override
public void generate(ClassLoader classLoader, Path sourceDirectory, Path targetDirectory)
        throws IOException, ExecutableScanException {
    ClassLoader generatedClassLoader = new URLClassLoader(new URL[] { sourceDirectory.toUri().toURL() },
            classLoader);
    List<String> allClassNames = getAllClassNamesCollector().listAllClassNames(sourceDirectory);
    List<Class<?>> allClasses = Lists
            .newArrayList(Iterables.transform(allClassNames, new ClassFunction(generatedClassLoader)));
    List<ExecutableAnnotationInformation> executableAnnotationInformations = getExecutableAnnotationInformationFinder()
            .findExecutableAnnotationInformation(allClasses);
    log.info("Found the following scripts: " + Joiner.on(", ").join(
            Sets.newTreeSet(Iterables.transform(executableAnnotationInformations, new ScriptNameFunction()))));
    GeneratedClassWriter generatedClassWriter = getGeneratedClassWriter();
    Map<String, String> commandRunnerClassNamesByScriptName = Maps.newTreeMap();
    for (ExecutableAnnotationInformation executableAnnotationInformation : executableAnnotationInformations) {
        String commandRunnerClassName = generatedClassWriter
                .writeClass(executableAnnotationInformation, generatedClassLoader, targetDirectory)
                .getClassName();
        String scriptName = executableAnnotationInformation.getScriptName();
        commandRunnerClassNamesByScriptName.put(scriptName, commandRunnerClassName);
        log.info("Written class " + commandRunnerClassName + " for script " + scriptName);
    }
    generatedClassWriter.writeEntryPointClass(targetDirectory, commandRunnerClassNamesByScriptName);
}

From source file:nl.socrates.dom.communicationchannel.CommunicationChannels.java

@Programmatic
public SortedSet<CommunicationChannel> findByOwner(final CommunicationChannelOwner owner) {
    return Sets.newTreeSet(allMatches("findByOwner", "owner", owner));
}