Examples with TObjectProcedure gnu.trove.procedure.TObjectProcedure used on opensource projects

Example 1 with TObjectProcedure

use of gnu.trove.procedure.TObjectProcedure in project FredBoat by Frederikam.

the class JDAUtil method countUniqueUsers.

/**
 * A count of unique users over the provided shards. This is an expensive operation given FredBoats scale.
 * <p>
 * Optionally pass in a value of value of previous counts / expected size to that we can initialize the set used
 * to count the unique values with an approriate size reducing expensive resizing operations.
 */
@CheckReturnValue
public static int countUniqueUsers(@Nonnull Collection<JDA> shards, @Nullable AtomicInteger expectedUserCount) {
    if (shards.size() == 1) {
        // a single shard provides a cheap call for getting user cardinality
        return Math.toIntExact(shards.iterator().next().getUserCache().size());
    }
    int expected = expectedUserCount != null && expectedUserCount.get() > 0 ? expectedUserCount.get() : LongOpenHashSet.DEFAULT_INITIAL_SIZE;
    // add 10k for good measure
    LongOpenHashSet uniqueUsers = new LongOpenHashSet(expected + 10000);
    TObjectProcedure<User> adder = user -> {
        uniqueUsers.add(user.getIdLong());
        return true;
    };
    Collections.unmodifiableCollection(shards).forEach(// this means however, that for the (small) duration, the map cannot be used by other threads (if there are any)
    shard -> ((JDAImpl) shard).getUserMap().forEachValue(adder));
    return uniqueUsers.size();
}