edu.northwestern.bioinformatics.studycalendar.test.restfulapi.ConfigurationInitializer.java Source code

Introduction

Here is the source code for edu.northwestern.bioinformatics.studycalendar.test.restfulapi.ConfigurationInitializer.java

Source

/*L
 * Copyright Northwestern University.
 *
 * Distributed under the OSI-approved BSD 3-Clause License.
 * See http://ncip.github.io/psc/LICENSE.txt for details.
 */

package edu.northwestern.bioinformatics.studycalendar.test.restfulapi;

import edu.northwestern.bioinformatics.studycalendar.test.integrated.ConnectionSource;
import edu.northwestern.bioinformatics.studycalendar.test.integrated.EmptySchemaInitializer;
import org.apache.commons.lang.StringUtils;
import org.jvyaml.YAML;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.core.io.Resource;

import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * Ensures (in {@link #oneTimeSetup} that the configuration tables mirror the values stored in
 * an external YAML file (and provided to this class deserialized).
 *
 * @author Rhett Sutphin
 */
public class ConfigurationInitializer extends EmptySchemaInitializer implements InitializingBean {
    private final Logger log = LoggerFactory.getLogger(getClass());

    private Map<String, Object> configurations;
    private Resource yamlResource;

    @Override
    public void oneTimeSetup(ConnectionSource connectionSource) {
        for (String table : configurations.keySet()) {
            oneTimeSetup(connectionSource, table);
        }
    }

    private void oneTimeSetup(ConnectionSource connectionSource, String table) {
        Map<String, String> desiredConfigProps = (Map<String, String>) configurations.get(table);

        List<Map<String, Object>> currentPropRows = connectionSource.currentJdbcTemplate()
                .queryForList("SELECT prop FROM " + table);
        List<String> existingProps = new ArrayList<String>(currentPropRows.size());
        for (Map<String, Object> row : currentPropRows)
            existingProps.add(row.values().iterator().next().toString());

        for (String prop : desiredConfigProps.keySet()) {
            String desiredValue = desiredConfigProps.get(prop);
            if (existingProps.contains(prop)) {
                log.debug("Already a record for {} in {}; updating", prop, table);
                connectionSource.currentJdbcTemplate()
                        .update(String.format("UPDATE %s SET value=? WHERE prop=?", table), desiredValue, prop);
                existingProps.remove(prop);
            } else {
                log.debug("No record for {} in {}; inserting", prop, table);
                connectionSource.currentJdbcTemplate().update(
                        String.format("INSERT INTO %s (prop, value) VALUES (?, ?)", table), prop, desiredValue);
            }
        }
        // remaining contents of existingProps are not referenced in the config data, so should be deleted
        deleteConfigurationProperties(connectionSource, table, existingProps);
    }

    private void deleteConfigurationProperties(ConnectionSource connectionSource, String table,
            List<String> propsToDelete) {
        if (!propsToDelete.isEmpty()) {
            log.debug("Deleting extra props {} from {}", propsToDelete, table);
            String[] deleteConds = new String[propsToDelete.size()];
            for (int i = 0; i < deleteConds.length; i++) {
                deleteConds[i] = "prop=?";
            }
            connectionSource.currentJdbcTemplate().update(
                    String.format("DELETE FROM %s WHERE %s", table, StringUtils.join(deleteConds, " OR ")),
                    propsToDelete.toArray(new Object[propsToDelete.size()]));
        }
    }

    ////// CONFIGURATION

    public void setYamlResource(Resource yamlResource) {
        this.yamlResource = yamlResource;
    }

    @SuppressWarnings({ "unchecked" })
    public void afterPropertiesSet() throws Exception {
        configurations = (Map<String, Object>) YAML.load(new InputStreamReader(yamlResource.getInputStream()));
    }
}