org.jbb.lib.logging.LoggingBootstrapper.java Source code

Introduction

Here is the source code for org.jbb.lib.logging.LoggingBootstrapper.java

Source

/*
 * Copyright (C) 2018 the original author or authors.
 *
 * This file is part of jBB Application Project.
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  You may obtain a copy of the License at
 *        http://www.apache.org/licenses/LICENSE-2.0
 */

package org.jbb.lib.logging;

import org.apache.commons.io.FileUtils;
import org.jbb.lib.commons.JbbMetaData;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.core.io.ClassPathResource;
import org.springframework.util.ResourceUtils;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.net.URL;

import javax.annotation.PostConstruct;

import ch.qos.logback.classic.LoggerContext;
import ch.qos.logback.classic.util.ContextInitializer;
import ch.qos.logback.core.joran.spi.JoranException;
import ch.qos.logback.ext.spring.ApplicationContextHolder;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;

@Slf4j
@RequiredArgsConstructor
public class LoggingBootstrapper implements ApplicationContextAware {
    private static final String LOG_DIR_NAME = "log";
    private static final String LOG_CONF_FILE_NAME = "logback-webapp.xml";
    private static final String CLASSPATH_DEFAULT_LOG_CONF_FILE_NAME = "default-logback.xml";

    private static String jbbLogPathValue;

    private final JbbMetaData jbbMetaData;
    private final LoggerContext loggerContext;
    private final ApplicationContextHolder applicationContextHolder;

    private ApplicationContext applicationContext;

    public static String getLogPath() {
        return jbbLogPathValue;
    }

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) {
        this.applicationContext = applicationContext;
    }

    @PostConstruct
    public void configure() {
        prepareLogDirectory();
        String location = prepareLogConfigurationFile();
        initLogging(location);
        applicationContextHolder.onApplicationEvent(new ContextRefreshedEvent(applicationContext));
        applicationContextHolder.setApplicationContext(applicationContext);
        log.info("Reconfiguration of logger finished");
        DelegatingLogbackAppenderHolder.getInstance().stop();
    }

    public String getLogConfFilePath() {
        return jbbMetaData.jbbConfigDirectory() + File.separator + LOG_CONF_FILE_NAME;
    }

    private void prepareLogDirectory() {
        String logDirPath = jbbMetaData.jbbHomePath() + File.separator + LOG_DIR_NAME;

        File logDir = new File(logDirPath);
        if (!logDir.exists()) {
            logDir.mkdir();
        }

        jbbLogPathValue = logDir.getAbsolutePath(); //NOSONAR
    }

    private String prepareLogConfigurationFile() {
        String logConfFilePath = getLogConfFilePath();
        File logbackConfigurationFile = new File(logConfFilePath);
        if (!logbackConfigurationFile.exists()) {
            copyDefaultConfigurationToFile(logbackConfigurationFile);
        }
        return logConfFilePath;
    }

    private void copyDefaultConfigurationToFile(File logbackConfigurationFile) {
        ClassPathResource classPathResource = new ClassPathResource(CLASSPATH_DEFAULT_LOG_CONF_FILE_NAME);
        try {
            FileUtils.copyURLToFile(classPathResource.getURL(), logbackConfigurationFile);
        } catch (IOException e) {
            throw new IllegalStateException("Unexpected error during reading default logging configuration file",
                    e);
        }
    }

    private void initLogging(String logConfFile) {
        try {
            URL url = ResourceUtils.getURL(logConfFile);
            new ContextInitializer(loggerContext).configureByResource(url);
        } catch (FileNotFoundException | JoranException e) {
            throw new IllegalStateException("Unexpected error during init logging configuration", e);
        }
    }

}