com.logsniffer.util.value.support.PropertiesBasedSource.java Source code

Introduction

Here is the source code for com.logsniffer.util.value.support.PropertiesBasedSource.java

Source

/*******************************************************************************
 * logsniffer, open source tool for viewing, monitoring and analysing log data.
 * Copyright (c) 2015 Scaleborn UG, www.scaleborn.com
 *
 * logsniffer is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * logsniffer is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *******************************************************************************/
package com.logsniffer.util.value.support;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.Properties;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;

import com.logsniffer.app.CoreAppConfig;
import com.logsniffer.app.LogSnifferHome;
import com.logsniffer.util.value.ConfigValueStore;

/**
 * Retrieves properties from
 * {@link CoreAppConfig#logSnifferProperties(org.springframework.context.ApplicationContext)}
 * . To fulfill the {@link ConfigValueStore} interface the properties are stored
 * to permanently to {@link CoreAppConfig#LOGSNIFFER_PROPERTIES_FILE} and
 * replace in the bean
 * {@link CoreAppConfig#logSnifferProperties(org.springframework.context.ApplicationContext)}
 * during current runtime.
 * 
 * @author mbok
 * 
 */
public class PropertiesBasedSource implements ConfigValueStore {
    private static final Logger LOGGER = LoggerFactory.getLogger(PropertiesBasedSource.class);
    @Autowired
    private LogSnifferHome homeDir;

    @Autowired
    @Qualifier(CoreAppConfig.BEAN_LOGSNIFFER_PROPS)
    private Properties logsnifferProperties;

    @Override
    public String getValue(final String key) {
        return logsnifferProperties.getProperty(key);
    }

    @Override
    public void store(final String key, final String value) throws IOException {
        if (value != null) {
            logsnifferProperties.setProperty(key, value);
        } else {
            logsnifferProperties.remove(key);
        }
        File file = new File(homeDir.getHomeDir(), CoreAppConfig.LOGSNIFFER_PROPERTIES_FILE);
        LOGGER.info("Saving config value for key '{}' to file: {}", key, file.getAbsolutePath());
        Properties properties = new Properties();
        try {
            properties.load(new FileInputStream(file));
        } catch (IOException e) {
            LOGGER.warn("Failed to load current properties from file, continue with empty properties: "
                    + file.getAbsolutePath(), e);
        }
        if (value != null) {
            properties.setProperty(key, value);
        } else {
            properties.remove(key);
        }
        properties.store(new FileOutputStream(file), null);
    }

}