de.gematik.test.tiger.testenvmgr.servers.log.TigerServerLogManager Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2024 gematik GmbH
 *
 * 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 de.gematik.test.tiger.testenvmgr.servers.log;

import ch.qos.logback.classic.Level;
import ch.qos.logback.classic.LoggerContext;
import ch.qos.logback.classic.encoder.PatternLayoutEncoder;
import ch.qos.logback.classic.spi.ILoggingEvent;
import ch.qos.logback.core.ConsoleAppender;
import ch.qos.logback.core.FileAppender;
import de.gematik.test.tiger.testenvmgr.TigerTestEnvMgr;
import de.gematik.test.tiger.testenvmgr.servers.AbstractTigerServer;
import de.gematik.test.tiger.testenvmgr.servers.TigerProxyServer;
import java.io.File;
import lombok.AccessLevel;
import lombok.NoArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

@Slf4j
@NoArgsConstructor(access = AccessLevel.PRIVATE)
public class TigerServerLogManager {

  private static final String DEFAULT_LOGFILE_LOCATION = "./target/serverLogs/";
  private static final String LOGFILE_EXTENSION = ".log";
  private static final String DEFAULT_PATTERN_LAYOUT =
      "%date %level [%thread] %logger{10} [%file:%line] %msg%n";

  public static void addAppenders(AbstractTigerServer server) {
    createAndAddAppenders(server, (ch.qos.logback.classic.Logger) server.getLog());
  }

  public static void setLoggingLevel(String loggerName, String levelString) {
    log.debug("Changing loglevel for '{}' to ({})", loggerName, levelString);
    ch.qos.logback.classic.Logger logger =
        (ch.qos.logback.classic.Logger) LoggerFactory.getLogger(loggerName);
    logger.setLevel(ch.qos.logback.classic.Level.toLevel(levelString.toUpperCase(), Level.INFO));
  }

  private static void createAndAddAppenders(
      AbstractTigerServer server, ch.qos.logback.classic.Logger log) {
    LoggerContext loggerContext = (LoggerContext) LoggerFactory.getILoggerFactory();

    PatternLayoutEncoder patternLayoutEncoder = new PatternLayoutEncoder();
    patternLayoutEncoder.setPattern(DEFAULT_PATTERN_LAYOUT);
    patternLayoutEncoder.setContext(loggerContext);
    patternLayoutEncoder.start();

    if (server.getConfiguration().getExternalJarOptions() == null
        || (server.getConfiguration().getExternalJarOptions().isActivateLogs())) {
      log.addAppender(
          createFileAppender(
              server.getServerId(),
              server.getConfiguration().getLogFile(),
              loggerContext,
              patternLayoutEncoder));
      log.addAppender(createConsoleAppender(loggerContext, patternLayoutEncoder));
      log.addAppender(createCustomerAppender(server, loggerContext));
    }
    log.setAdditive(false);
  }

  private static FileAppender createFileAppender(
      String serverId,
      String logFilePath,
      LoggerContext loggerContext,
      PatternLayoutEncoder patternLayoutEncoder) {
    String fileName = createFileName(serverId, logFilePath);
    FileAppender fileAppender = new FileAppender<>();
    fileAppender.setFile(fileName);
    fileAppender.setAppend(false);
    fileAppender.setEncoder(patternLayoutEncoder);
    fileAppender.setContext(loggerContext);
    fileAppender.start();

    return fileAppender;
  }

  private static ConsoleAppender createConsoleAppender(
      LoggerContext loggerContext, PatternLayoutEncoder patternLayoutEncoder) {
    ConsoleAppender consoleAppender = new ConsoleAppender<>();
    consoleAppender.setEncoder(patternLayoutEncoder);
    consoleAppender.setContext(loggerContext);
    consoleAppender.start();

    return consoleAppender;
  }

  private static String createFileName(String serverId, String logFilePath) {
    if (logFilePath != null) {
      File file = new File(logFilePath);
      if (file.isDirectory()) {
        return file.getAbsolutePath() + serverId + LOGFILE_EXTENSION;
      } else {
        return file.getAbsolutePath();
      }
    }
    return DEFAULT_LOGFILE_LOCATION + serverId + LOGFILE_EXTENSION;
  }

  private static CustomerAppender createCustomerAppender(
      AbstractTigerServer server, LoggerContext loggerContext) {
    CustomerAppender customerAppender = new CustomerAppender(server);
    customerAppender.setContext(loggerContext);
    customerAppender.start();

    return customerAppender;
  }

  public static void addProxyCustomerAppender(TigerProxyServer tigerProxyServer) {
    ch.qos.logback.classic.Logger logbackLogger =
        ((ch.qos.logback.classic.Logger) tigerProxyServer.getTigerProxy().getLog());
    LoggerContext loggerContext = (LoggerContext) LoggerFactory.getILoggerFactory();
    CustomerProxyServerAppender customerAppender =
        new CustomerProxyServerAppender(tigerProxyServer);
    customerAppender.setContext(loggerContext);
    customerAppender.start();
    logbackLogger.addAppender(customerAppender);
  }

  public static void addProxyCustomerAppender(TigerTestEnvMgr tigerTestEnvMgr, Logger log) {
    LoggerContext loggerContext = (LoggerContext) LoggerFactory.getILoggerFactory();
    CustomerLocalProxyAppender customerAppender = new CustomerLocalProxyAppender(tigerTestEnvMgr);
    customerAppender.setContext(loggerContext);
    customerAppender.start();
    ((ch.qos.logback.classic.Logger) log).addAppender(customerAppender);
  }
}