io.rxmicro.logger.internal.jul.ConfigCustomizer Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of rxmicro-logger Show documentation
The module for logging important events during the work of microservices that is integrated to the RxMicro framework.
There is a newer version: 0.11
/*
 * Copyright (c) 2020. https://rxmicro.io
 *
 * 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 io.rxmicro.logger.internal.jul;

import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;

import static io.rxmicro.files.PropertiesResources.loadProperties;
import static io.rxmicro.logger.Constants.LOGGER_VARIABLE_PREFIX;

/**
 * @author nedis
 * @since 0.1
 */
final class ConfigCustomizer {

    private static final int LOGGER_VARIABLE_PREFIX_INDEX = LOGGER_VARIABLE_PREFIX.length();

    private static final String JAVA_LOGGING_TEST_PROPERTIES = "jul.test.properties";

    private static final String JAVA_LOGGING_PROPERTIES = "jul.properties";

    Optional customizeConfig(final Map config) {
        Optional> propertiesOptional = getFromSystemPropertiesConfiguration();
        if (propertiesOptional.isPresent()) {
            customize(config, propertiesOptional.get());
            return Optional.of("Java System Properties");
        }
        propertiesOptional = getFromSystemVariablesConfiguration();
        if (propertiesOptional.isPresent()) {
            customize(config, propertiesOptional.get());
            return Optional.of("System Environment Variables");
        }
        for (final String property : List.of(JAVA_LOGGING_TEST_PROPERTIES, JAVA_LOGGING_PROPERTIES)) {
            propertiesOptional = loadProperties(property);
            if (propertiesOptional.isPresent()) {
                customize(config, propertiesOptional.get());
                return Optional.of("classpath:/" + property);
            }
        }
        return Optional.empty();
    }

    private void customize(final Map dest,
                           final Map src) {
        for (final Map.Entry entry : src.entrySet()) {
            dest.put(entry.getKey().trim(), LevelMappings.fixLevelValue(entry.getValue().trim()));
        }
    }

    private Optional> getFromSystemVariablesConfiguration() {
        final Map properties = new LinkedHashMap<>();
        for (final Map.Entry entry : System.getenv().entrySet()) {
            if (entry.getKey().startsWith(LOGGER_VARIABLE_PREFIX)) {
                properties.put(entry.getKey().substring(LOGGER_VARIABLE_PREFIX_INDEX), entry.getValue());
            }
        }
        return properties.isEmpty() ? Optional.empty() : Optional.of(properties);
    }

    private Optional> getFromSystemPropertiesConfiguration() {
        final Map properties = new LinkedHashMap<>();
        for (final Map.Entry entry : System.getProperties().entrySet()) {
            if (((String) entry.getKey()).startsWith(LOGGER_VARIABLE_PREFIX)) {
                properties.put(((String) entry.getKey()).substring(LOGGER_VARIABLE_PREFIX_INDEX), (String) entry.getValue());
            }
        }
        return properties.isEmpty() ? Optional.empty() : Optional.of(properties);
    }
}