net.dv8tion.jda.internal.utils.cache.ShardCacheViewImpl 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 gnu.trove.map.TIntObjectMap;
import gnu.trove.map.hash.TIntObjectHashMap;
import gnu.trove.set.TIntSet;
import gnu.trove.set.hash.TIntHashSet;
import net.dv8tion.jda.api.JDA;
import net.dv8tion.jda.api.utils.ClosableIterator;
import net.dv8tion.jda.api.utils.LockIterator;
import net.dv8tion.jda.api.utils.MiscUtil;
import net.dv8tion.jda.api.utils.cache.CacheView;
import net.dv8tion.jda.api.utils.cache.ShardCacheView;
import net.dv8tion.jda.internal.utils.ChainedClosableIterator;
import net.dv8tion.jda.internal.utils.Checks;
import net.dv8tion.jda.internal.utils.Helpers;
import net.dv8tion.jda.internal.utils.UnlockHook;
import org.apache.commons.collections4.iterators.ObjectArrayIterator;
import javax.annotation.Nonnull;
import java.util.*;
import java.util.concurrent.locks.ReentrantReadWriteLock;
import java.util.function.Consumer;
import java.util.function.Supplier;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;
public class ShardCacheViewImpl extends ReadWriteLockCache implements ShardCacheView
{
protected static final JDA[] EMPTY_ARRAY = new JDA[0];
protected final TIntObjectMap elements;
public ShardCacheViewImpl()
{
this.elements = new TIntObjectHashMap<>();
}
public ShardCacheViewImpl(int initialCapacity)
{
this.elements = new TIntObjectHashMap<>(initialCapacity);
}
public void clear()
{
try (UnlockHook hook = writeLock())
{
elements.clear();
}
}
public JDA remove(int shardId)
{
try (UnlockHook hook = writeLock())
{
return elements.remove(shardId);
}
}
public TIntObjectMap getMap()
{
if (!lock.writeLock().isHeldByCurrentThread())
throw new IllegalStateException("Cannot access map without holding write lock!");
return elements;
}
public TIntSet keySet()
{
try (UnlockHook hook = readLock())
{
return new TIntHashSet(elements.keySet());
}
}
@Override
public void forEach(Consumer action)
{
Objects.requireNonNull(action);
try (UnlockHook hook = readLock())
{
for (JDA shard : elements.valueCollection())
{
action.accept(shard);
}
}
}
@Nonnull
@Override
public List asList()
{
if (isEmpty())
return Collections.emptyList();
try (UnlockHook hook = readLock())
{
List list = getCachedList();
if (list != null)
return list;
return cache(new ArrayList<>(elements.valueCollection()));
}
}
@Nonnull
@Override
public Set asSet()
{
if (isEmpty())
return Collections.emptySet();
try (UnlockHook hook = readLock())
{
Set set = getCachedSet();
if (set != null)
return set;
return cache(new HashSet<>(elements.valueCollection()));
}
}
@Nonnull
@Override
public LockIterator lockedIterator()
{
ReentrantReadWriteLock.ReadLock readLock = lock.readLock();
MiscUtil.tryLock(readLock);
try
{
Iterator directIterator = elements.valueCollection().iterator();
return new LockIterator<>(directIterator, readLock);
}
catch (Throwable t)
{
readLock.unlock();
throw t;
}
}
@Override
public long size()
{
return elements.size();
}
@Override
public boolean isEmpty()
{
return elements.isEmpty();
}
@Nonnull
@Override
public List getElementsByName(@Nonnull String name, boolean ignoreCase)
{
Checks.notEmpty(name, "Name");
if (elements.isEmpty())
return Collections.emptyList();
try (UnlockHook hook = readLock())
{
List list = new LinkedList<>();
for (JDA elem : elements.valueCollection())
{
String elementName = elem.getShardInfo().getShardString();
if (elementName != null)
{
if (ignoreCase)
{
if (elementName.equalsIgnoreCase(name))
list.add(elem);
}
else
{
if (elementName.equals(name))
list.add(elem);
}
}
}
return list;
}
}
@Override
public Spliterator spliterator()
{
try (UnlockHook hook = readLock())
{
return Spliterators.spliterator(iterator(), size(), Spliterator.IMMUTABLE | Spliterator.NONNULL);
}
}
@Nonnull
@Override
public Stream stream()
{
return StreamSupport.stream(spliterator(), false);
}
@Nonnull
@Override
public Stream parallelStream()
{
return StreamSupport.stream(spliterator(), true);
}
@Nonnull
@Override
public Iterator iterator()
{
try (UnlockHook hook = readLock())
{
JDA[] arr = elements.values(EMPTY_ARRAY);
return new ObjectArrayIterator<>(arr);
}
}
@Override
public JDA getElementById(int id)
{
try (UnlockHook hook = readLock())
{
return this.elements.get(id);
}
}
@Override
public int hashCode()
{
try (UnlockHook hook = readLock())
{
return elements.hashCode();
}
}
@Override
public boolean equals(Object obj)
{
if (obj == this)
return true;
if (!(obj instanceof ShardCacheViewImpl))
return false;
ShardCacheViewImpl view = (ShardCacheViewImpl) obj;
try (UnlockHook hook = readLock(); UnlockHook otherHook = view.readLock())
{
return this.elements.equals(view.elements);
}
}
@Override
public String toString()
{
return asList().toString();
}
public static class UnifiedShardCacheViewImpl implements ShardCacheView
{
protected final Supplier> generator;
public UnifiedShardCacheViewImpl(Supplier> generator)
{
this.generator = generator;
}
@Override
public long size()
{
return distinctStream().mapToLong(CacheView::size).sum();
}
@Override
public boolean isEmpty()
{
return generator.get().allMatch(CacheView::isEmpty);
}
@Nonnull
@Override
public List asList()
{
List list = new ArrayList<>();
stream().forEach(list::add);
return Collections.unmodifiableList(list);
}
@Nonnull
@Override
public Set asSet()
{
Set set = new HashSet<>();
generator.get().flatMap(CacheView::stream).forEach(set::add);
return Collections.unmodifiableSet(set);
}
@Nonnull
@Override
public ClosableIterator lockedIterator()
{
Iterator gen = this.generator.get().iterator();
return new ChainedClosableIterator<>(gen);
}
@Nonnull
@Override
public List getElementsByName(@Nonnull String name, boolean ignoreCase)
{
return distinctStream()
.flatMap(view -> view.getElementsByName(name, ignoreCase).stream())
.collect(Helpers.toUnmodifiableList());
}
@Override
public JDA getElementById(int id)
{
return generator.get()
.map(view -> view.getElementById(id))
.filter(Objects::nonNull)
.findFirst().orElse(null);
}
@Nonnull
@Override
public Stream stream()
{
return generator.get().flatMap(CacheView::stream).distinct();
}
@Nonnull
@Override
public Stream parallelStream()
{
return generator.get().flatMap(CacheView::parallelStream).distinct();
}
@Nonnull
@Override
public Iterator iterator()
{
return stream().iterator();
}
protected Stream distinctStream()
{
return generator.get().distinct();
}
}
}