PonyMirrors/Unicopia
1
0
Fork 0
mirror of https://github.com/Sollace/Unicopia.git synced 2025-02-12 16:14:24 +01:00
Code Issues Projects Releases Packages Wiki Activity

Add generics to SpellContainer.stream

Browse source
This commit is contained in:
Sollace 2022-10-01 18:21:08 +02:00
parent 2b00977e7f
commit 8c65adeea2
2 changed files with 14 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/main/java/com/minelittlepony/unicopia/ability/magic/SpellContainer.java
View file

@ -62,9 +62,16 @@ public interface SpellContainer {
*/ */
boolean forEach(Function<Spell, Operation> action, boolean update); boolean forEach(Function<Spell, Operation> action, boolean update);
/**
* Gets all active effects for this caster updating it if needed.
*/
Stream<Spell> stream(boolean update); Stream<Spell> stream(boolean update);
Stream<Spell> stream(@Nullable SpellPredicate<?> type, boolean update); /**
* Gets all active effects for this caster that match the given type updating it if needed.
*/
<T extends Spell> Stream<T> stream(@Nullable SpellPredicate<T> type, boolean update);
/** /**
* Removes all effects currently active in this slot. * Removes all effects currently active in this slot.

12
src/main/java/com/minelittlepony/unicopia/network/datasync/EffectSync.java
View file

@ -51,9 +51,8 @@ public class EffectSync implements SpellContainer {
} }
@Override @Override
@SuppressWarnings("unchecked")
public <T extends Spell> Optional<T> get(@Nullable SpellPredicate<T> type, boolean update) { public <T extends Spell> Optional<T> get(@Nullable SpellPredicate<T> type, boolean update) {
return (Optional<T>)(read(type, update, true).findFirst()); return read(type, update, true).findFirst();
} }
@Override @Override
@ -95,7 +94,7 @@ public class EffectSync implements SpellContainer {
} }
@Override @Override
public Stream<Spell> stream(@Nullable SpellPredicate<?> type, boolean update) { public <T extends Spell> Stream<T> stream(@Nullable SpellPredicate<T> type, boolean update) {
return read(type, update, true); return read(type, update, true);
} }
@ -111,15 +110,16 @@ public class EffectSync implements SpellContainer {
return false; return false;
} }
private Stream<Spell> read(@Nullable SpellPredicate<?> type, boolean synchronize, boolean sendUpdate) { @SuppressWarnings("unchecked")
private <T extends Spell> Stream<T> read(@Nullable SpellPredicate<T> type, boolean synchronize, boolean sendUpdate) {
if (synchronize && spells.fromNbt(owner.getEntity().getDataTracker().get(param)) && sendUpdate) { if (synchronize && spells.fromNbt(owner.getEntity().getDataTracker().get(param)) && sendUpdate) {
owner.getEntity().getDataTracker().set(param, spells.toNbt()); owner.getEntity().getDataTracker().set(param, spells.toNbt());
} }
if (type == null) { if (type == null) {
return spells.getReferences(); return (Stream<T>)spells.getReferences();
} }
return spells.getReferences().flatMap(s -> s.findMatches(type)); return (Stream<T>)spells.getReferences().flatMap(s -> s.findMatches(type));
} }
public boolean reduce(Alteration alteration) { public boolean reduce(Alteration alteration) {