net.dv8tion.jda.internal.utils.cache.UnifiedCacheViewImpl Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of JDA Show documentation
Show all versions of JDA Show documentation
Java wrapper for the popular chat & VOIP service: Discord https://discord.com
/*
* Copyright 2015 Austin Keener, Michael Ritter, Florian Spieß, and the JDA contributors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package net.dv8tion.jda.internal.utils.cache;
import net.dv8tion.jda.api.entities.ISnowflake;
import net.dv8tion.jda.api.entities.Member;
import net.dv8tion.jda.api.entities.Role;
import net.dv8tion.jda.api.utils.ClosableIterator;
import net.dv8tion.jda.api.utils.cache.CacheView;
import net.dv8tion.jda.api.utils.cache.MemberCacheView;
import net.dv8tion.jda.api.utils.cache.SnowflakeCacheView;
import net.dv8tion.jda.api.utils.cache.UnifiedMemberCacheView;
import net.dv8tion.jda.internal.utils.ChainedClosableIterator;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.*;
import java.util.function.Consumer;
import java.util.function.Supplier;
import java.util.stream.Collectors;
import java.util.stream.Stream;
public class UnifiedCacheViewImpl> implements CacheView
{
protected final Supplier> generator;
public UnifiedCacheViewImpl(Supplier> generator)
{
this.generator = generator;
}
@Override
public long size()
{
return distinctStream().mapToLong(CacheView::size).sum();
}
@Override
public boolean isEmpty()
{
return distinctStream().allMatch(CacheView::isEmpty);
}
@Override
public void forEach(Consumer action)
{
Objects.requireNonNull(action);
try (ClosableIterator it = lockedIterator())
{
while (it.hasNext())
action.accept(it.next());
}
}
@Nonnull
@Override
public List asList()
{
List list = new LinkedList<>();
forEach(list::add);
return Collections.unmodifiableList(list);
}
@Nonnull
@Override
public Set asSet()
{
try (ChainedClosableIterator it = lockedIterator())
{
//because the iterator needs to retain elements to avoid duplicates,
// we can use the resulting HashSet as our return value!
while (it.hasNext()) it.next();
return Collections.unmodifiableSet(it.getItems());
}
}
@Nonnull
@Override
public ChainedClosableIterator lockedIterator()
{
Iterator gen = generator.get().iterator();
return new ChainedClosableIterator<>(gen);
}
@Nonnull
@Override
public List getElementsByName(@Nonnull String name, boolean ignoreCase)
{
return Collections.unmodifiableList(distinctStream()
.flatMap(view -> view.getElementsByName(name, ignoreCase).stream())
.distinct()
.collect(Collectors.toList()));
}
@Nonnull
@Override
public Stream stream()
{
return distinctStream().flatMap(CacheView::stream).distinct();
}
@Nonnull
@Override
public Stream parallelStream()
{
return distinctStream().flatMap(CacheView::parallelStream).distinct();
}
@Nonnull
@Override
public Iterator iterator()
{
return stream().iterator();
}
protected Stream distinctStream()
{
return generator.get().distinct();
}
public static class UnifiedSnowflakeCacheView
extends UnifiedCacheViewImpl> implements SnowflakeCacheView
{
public UnifiedSnowflakeCacheView(Supplier>> generator)
{
super(generator);
}
@Override
public T getElementById(long id)
{
return generator.get()
.map(view -> view.getElementById(id))
.filter(Objects::nonNull)
.findFirst().orElse(null);
}
}
public static class UnifiedMemberCacheViewImpl
extends UnifiedCacheViewImpl implements UnifiedMemberCacheView
{
public UnifiedMemberCacheViewImpl(Supplier> generator)
{
super(generator);
}
@Nonnull
@Override
public List getElementsById(long id)
{
return Collections.unmodifiableList(distinctStream()
.map(view -> view.getElementById(id))
.filter(Objects::nonNull)
.collect(Collectors.toList()));
}
@Nonnull
@Override
public List getElementsByUsername(@Nonnull String name, boolean ignoreCase)
{
return Collections.unmodifiableList(distinctStream()
.flatMap(view -> view.getElementsByUsername(name, ignoreCase).stream())
.collect(Collectors.toList()));
}
@Nonnull
@Override
public List getElementsByNickname(@Nullable String name, boolean ignoreCase)
{
return Collections.unmodifiableList(distinctStream()
.flatMap(view -> view.getElementsByNickname(name, ignoreCase).stream())
.collect(Collectors.toList()));
}
@Nonnull
@Override
public List getElementsWithRoles(@Nonnull Role... roles)
{
return Collections.unmodifiableList(distinctStream()
.flatMap(view -> view.getElementsWithRoles(roles).stream())
.collect(Collectors.toList()));
}
@Nonnull
@Override
public List getElementsWithRoles(@Nonnull Collection roles)
{
return Collections.unmodifiableList(distinctStream()
.flatMap(view -> view.getElementsWithRoles(roles).stream())
.collect(Collectors.toList()));
}
}
}