Performance: fixed server's big memory usage in long games and in big stack sizes (related to #11285, fixes #9302)

This commit is contained in:
Oleg Agafonov 2023-10-14 15:54:10 +04:00
parent 36ccfb0a2a
commit d57a3c100d
7 changed files with 107 additions and 61 deletions

View file

@ -0,0 +1,46 @@
package org.mage.test.AI.basic;
import mage.constants.PhaseStep;
import mage.constants.Zone;
import org.junit.Test;
import org.mage.test.serverside.base.CardTestPlayerBaseWithAIHelps;
/**
* @author JayDi85
*/
public class GameStatePerformanceTest extends CardTestPlayerBaseWithAIHelps {
@Test
public void test_StackObjects_BookmarkMustCleanDataAfterTriggerResolve() {
// memory overflow problem on too much stack objects
// related bug: https://github.com/magefree/mage/issues/9302
// go to "save(GameState gameState)" and enable size logs to test real usage
final int MAX_STACK_OBJECTS = 300;
// Whenever an artifact creature you control deals combat damage to a player,
// you may create a 1/1 blue Thopter artifact creature token with flying.
addCard(Zone.BATTLEFIELD, playerA, "Sharding Sphinx", MAX_STACK_OBJECTS);
//
// You cant lose the game and your opponents cant win the game.
addCard(Zone.BATTLEFIELD, playerB, "Platinum Angel", 1);
//
// Creatures cant block this turn.
addCard(Zone.HAND, playerA, "Order // Chaos", 1); // instant, {2}{R}
addCard(Zone.BATTLEFIELD, playerA, "Mountain", 3);
// prepare not loose
castSpell(1, PhaseStep.PRECOMBAT_MAIN, playerA, "Chaos");
waitStackResolved(1, PhaseStep.PRECOMBAT_MAIN);
// attack and use triggers
attack(1, playerA, "Sharding Sphinx");
setChoice(playerA, "Whenever an artifact creature you control", MAX_STACK_OBJECTS - 1); // triggers order
setChoice(playerA, true, MAX_STACK_OBJECTS); // triggers use
setStrictChooseMode(true);
setStopAt(1, PhaseStep.END_TURN);
execute();
assertPermanentCount(playerA, "Thopter Token", MAX_STACK_OBJECTS);
}
}

View file

@ -488,12 +488,29 @@ public interface Game extends MageItem, Serializable, Copyable<Game> {
//game transaction methods //game transaction methods
void saveState(boolean bookmark); void saveState(boolean bookmark);
/**
* Save current game state and return bookmark to it
*
* @return
*/
int bookmarkState(); int bookmarkState();
GameState restoreState(int bookmark, String context); GameState restoreState(int bookmark, String context);
/**
* Remove selected bookmark and all newer bookmarks and game states
* Part of restore/rollback lifecycle
*
* @param bookmark
*/
void removeBookmark(int bookmark); void removeBookmark(int bookmark);
/**
* TODO: remove logic changed, must research each usage of removeBookmark and replace it with new code
* @param bookmark
*/
void removeBookmark_v2(int bookmark);
int getSavedStateSize(); int getSavedStateSize();
boolean isSaveGame(); boolean isSaveGame();

View file

@ -118,7 +118,7 @@ public abstract class GameImpl implements Game {
protected Map<UUID, Counters> enterWithCounters = new HashMap<>(); protected Map<UUID, Counters> enterWithCounters = new HashMap<>();
protected GameState state; protected GameState state;
private transient Stack<Integer> savedStates = new Stack<>(); private transient Stack<Integer> savedStates = new Stack<>(); // bookmarks - 0-base refs to gameStates
protected transient GameStates gameStates = new GameStates(); protected transient GameStates gameStates = new GameStates();
// game states to allow player rollback // game states to allow player rollback
@ -934,8 +934,7 @@ public abstract class GameImpl implements Game {
} }
@Override @Override
public void removeBookmark(int bookmark public void removeBookmark(int bookmark) {
) {
if (!simulation) { if (!simulation) {
if (bookmark != 0) { if (bookmark != 0) {
while (savedStates.size() > bookmark) { while (savedStates.size() > bookmark) {
@ -946,6 +945,20 @@ public abstract class GameImpl implements Game {
} }
} }
@Override
public void removeBookmark_v2(int bookmark) {
if (!simulation) {
if (bookmark != 0) {
while (savedStates.size() >= bookmark) {
int outdatedIndex = savedStates.pop();
while (gameStates.getSize() - 1 >= outdatedIndex) {
gameStates.remove(gameStates.getSize() - 1);
}
}
}
}
}
private void clearAllBookmarks() { private void clearAllBookmarks() {
if (!simulation) { if (!simulation) {
while (!savedStates.isEmpty()) { while (!savedStates.isEmpty()) {

View file

@ -2,7 +2,7 @@
package mage.game; package mage.game;
import java.io.Serializable; import java.io.Serializable;
import java.util.LinkedList; import java.util.ArrayList;
import java.util.List; import java.util.List;
import org.apache.log4j.Logger; import org.apache.log4j.Logger;
@ -16,13 +16,12 @@ public class GameStates implements Serializable {
private final List<GameState> states; private final List<GameState> states;
public GameStates() { public GameStates() {
this.states = new LinkedList<>(); this.states = new ArrayList<>();
} }
public void save(GameState gameState) { public void save(GameState gameState) {
// states.add(new Copier<GameState>().copyCompressed(gameState));
states.add(gameState.copy()); states.add(gameState.copy());
logger.trace("Saved game state: " + states.size()); //logger.warn("states size: " + states.size());
} }
public int getSize() { public int getSize() {
@ -35,7 +34,6 @@ public class GameStates implements Serializable {
states.remove(states.size() - 1); states.remove(states.size() - 1);
} }
logger.trace("Rolling back state: " + index); logger.trace("Rolling back state: " + index);
// return new Copier<GameState>().uncompressCopy(states.get(index));
return states.get(index); return states.get(index);
} }
return null; return null;
@ -52,7 +50,6 @@ public class GameStates implements Serializable {
public GameState get(int index) { public GameState get(int index) {
if (index < states.size()) { if (index < states.size()) {
// return new Copier<GameState>().uncompressCopy(states.get(index));
return states.get(index); return states.get(index);
} }
return null; return null;

View file

@ -38,7 +38,10 @@ import mage.util.Copyable;
import mage.util.MultiAmountMessage; import mage.util.MultiAmountMessage;
import java.io.Serializable; import java.io.Serializable;
import java.util.*; import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.UUID;
import java.util.stream.Collectors; import java.util.stream.Collectors;
/** /**
@ -56,10 +59,7 @@ public interface Player extends MageItem, Copyable<Player> {
* Default is PayLifeCostLevel.allAbilities. * Default is PayLifeCostLevel.allAbilities.
*/ */
enum PayLifeCostLevel { enum PayLifeCostLevel {
allAbilities, allAbilities, nonSpellnonActivatedAbilities, onlyManaAbilities, none
nonSpellnonActivatedAbilities,
onlyManaAbilities,
none
} }
/** /**
@ -209,6 +209,7 @@ public interface Player extends MageItem, Copyable<Player> {
Cards getHand(); Cards getHand();
void incrementLandsPlayed(); void incrementLandsPlayed();
void resetLandsPlayed(); void resetLandsPlayed();
int getLandsPlayed(); int getLandsPlayed();
@ -556,8 +557,18 @@ public interface Player extends MageItem, Copyable<Player> {
int getStoredBookmark(); int getStoredBookmark();
/**
* Save player's bookmark for undo, e.g. enable undo button on mana payment
*
* @param bookmark
*/
void setStoredBookmark(int bookmark); void setStoredBookmark(int bookmark);
/**
* Reset player's bookmark, e.g. disable undo button
*
* @param game
*/
void resetStoredBookmark(Game game); void resetStoredBookmark(Game game);
default GameState restoreState(int bookmark, String text, Game game) { default GameState restoreState(int bookmark, String text, Game game) {
@ -746,10 +757,8 @@ public interface Player extends MageItem, Copyable<Player> {
* @param game Game * @param game Game
* @return List of integers with size equal to messages.size(). The sum of the integers is equal to max. * @return List of integers with size equal to messages.size(). The sum of the integers is equal to max.
*/ */
default List<Integer> getMultiAmount(Outcome outcome, List<String> messages, int min, int max, MultiAmountType type, default List<Integer> getMultiAmount(Outcome outcome, List<String> messages, int min, int max, MultiAmountType type, Game game) {
Game game) { List<MultiAmountMessage> constraints = messages.stream().map(s -> new MultiAmountMessage(s, 0, max)).collect(Collectors.toList());
List<MultiAmountMessage> constraints = messages.stream().map(s -> new MultiAmountMessage(s, 0, max))
.collect(Collectors.toList());
return getMultiAmountWithIndividualConstraints(outcome, constraints, min, max, type, game); return getMultiAmountWithIndividualConstraints(outcome, constraints, min, max, type, game);
} }
@ -765,8 +774,7 @@ public interface Player extends MageItem, Copyable<Player> {
* @param game Game * @param game Game
* @return List of integers with size equal to messages.size(). The sum of the integers is equal to max. * @return List of integers with size equal to messages.size(). The sum of the integers is equal to max.
*/ */
List<Integer> getMultiAmountWithIndividualConstraints(Outcome outcome, List<MultiAmountMessage> messages, int min, List<Integer> getMultiAmountWithIndividualConstraints(Outcome outcome, List<MultiAmountMessage> messages, int min, int max, MultiAmountType type, Game game);
int max, MultiAmountType type, Game game);
void sideboard(Match match, Deck deck); void sideboard(Match match, Deck deck);

View file

@ -1613,7 +1613,7 @@ public abstract class PlayerImpl implements Player, Serializable {
ability.getId(), ability, ability.getControllerId() ability.getId(), ability, ability.getControllerId()
)); ));
} }
game.removeBookmark(bookmark); game.removeBookmark_v2(bookmark);
return true; return true;
} }
} }

View file

@ -51,39 +51,4 @@ public class Copier<T> {
return copy; return copy;
} }
public byte[] copyCompressed(T obj) {
FastByteArrayOutputStream fbos = null;
ObjectOutputStream out = null;
try {
fbos = new FastByteArrayOutputStream();
out = new ObjectOutputStream(new GZIPOutputStream(fbos));
// Write the object out to a byte array
out.writeObject(obj);
out.flush();
byte[] copy = new byte[fbos.getSize()];
System.arraycopy(fbos.getByteArray(), 0, copy, 0, fbos.getSize());
return copy;
}
catch(IOException e) {
e.printStackTrace();
} finally {
StreamUtils.closeQuietly(fbos);
StreamUtils.closeQuietly(out);
}
return null;
}
public T uncompressCopy(byte[] buffer) {
T copy = null;
try (ObjectInputStream in = new CopierObjectInputStream(loader, new GZIPInputStream(new ByteArrayInputStream(buffer)))) {
copy = (T) in.readObject();
}
catch(IOException | ClassNotFoundException e) {
e.printStackTrace();
}
return copy;
}
} }