Skip to content

Commit

Permalink
Refactor: simplify headless chat appender (#12742)
Browse files Browse the repository at this point in the history
  • Loading branch information
DanVanAtta committed Jul 21, 2024
1 parent dcda979 commit 407c355
Show file tree
Hide file tree
Showing 2 changed files with 25 additions and 29 deletions.
Original file line number Diff line number Diff line change
@@ -1,9 +1,5 @@
package org.triplea.game.server;

import ch.qos.logback.classic.Level;
import ch.qos.logback.classic.Logger;
import ch.qos.logback.classic.LoggerContext;
import ch.qos.logback.classic.filter.ThresholdFilter;
import com.google.common.annotations.VisibleForTesting;
import games.strategy.engine.chat.Chat;
import games.strategy.engine.chat.HeadlessChat;
Expand Down Expand Up @@ -34,7 +30,6 @@
import java.util.Optional;
import java.util.Set;
import lombok.extern.slf4j.Slf4j;
import org.slf4j.LoggerFactory;
import org.triplea.game.chat.ChatModel;
import org.triplea.game.server.debug.ChatAppender;
import org.triplea.java.ThreadRunner;
Expand Down Expand Up @@ -132,27 +127,11 @@ public HeadlessAutoSaveFileUtils getAutoSaveFileUtils() {
return new HeadlessAutoSaveFileUtils();
}

private void registerChatAppender(final Chat chat) {
Logger logger = (Logger) LoggerFactory.getLogger(Logger.ROOT_LOGGER_NAME);
ChatAppender chatAppender = new ChatAppender(chat);
// prevent multiple chat appenders causing memory leak
// ideally this should happen in a shutdown operation somewhere though
logger.detachAppender(chatAppender.getName());

ThresholdFilter filter = new ThresholdFilter();
filter.setLevel(Level.WARN.toString());
filter.start();
chatAppender.addFilter(filter);
chatAppender.setContext((LoggerContext) LoggerFactory.getILoggerFactory());
chatAppender.start();
logger.addAppender(chatAppender);
}

@Override
public ChatModel createChatModel(
String chatName, Messengers messengers, ClientNetworkBridge clientNetworkBridge) {
Chat chat = new Chat(new MessengersChatTransmitter(chatName, messengers, clientNetworkBridge));
registerChatAppender(chat);
ChatAppender.attach(chat);
return new HeadlessChat(chat);
}

Expand Down
Original file line number Diff line number Diff line change
@@ -1,10 +1,15 @@
package org.triplea.game.server.debug;

import ch.qos.logback.classic.Level;
import ch.qos.logback.classic.Logger;
import ch.qos.logback.classic.LoggerContext;
import ch.qos.logback.classic.filter.ThresholdFilter;
import ch.qos.logback.classic.spi.ILoggingEvent;
import ch.qos.logback.core.AppenderBase;
import com.google.common.base.Preconditions;
import games.strategy.engine.chat.Chat;
import java.util.List;
import java.util.stream.Stream;
import org.slf4j.LoggerFactory;

/**
* A {@link AppenderBase} that publishes log records to the chat subsystem. This allows a headless
Expand All @@ -13,20 +18,32 @@
public final class ChatAppender extends AppenderBase<ILoggingEvent> {
private final Chat chat;

public ChatAppender(final Chat chat) {
private ChatAppender(final Chat chat) {
setName("chatMessage");
this.chat = Preconditions.checkNotNull(chat);
}

public static void attach(Chat chat) {
Logger logger = (Logger) LoggerFactory.getLogger(Logger.ROOT_LOGGER_NAME);
ChatAppender chatAppender = new ChatAppender(chat);
// prevent multiple chat appenders causing memory leak
// ideally this should happen in a shutdown operation somewhere though
logger.detachAppender(chatAppender.getName());

ThresholdFilter filter = new ThresholdFilter();
filter.setLevel(Level.WARN.toString());
filter.start();
chatAppender.addFilter(filter);
chatAppender.setContext((LoggerContext) LoggerFactory.getILoggerFactory());
chatAppender.start();
logger.addAppender(chatAppender);
}

@Override
protected void append(final ILoggingEvent record) {
// format log message and send it to the chat window
formatChatMessage(record).stream()
Stream.of(record.getFormattedMessage().trim().split("\\n"))
.map(message -> "[" + record.getLevel() + "] " + message)
.forEach(chat::sendMessage);
}

private List<String> formatChatMessage(final ILoggingEvent record) {
return List.of(record.getFormattedMessage().trim().split("\\n"));
}
}

0 comments on commit 407c355

Please sign in to comment.