Project

General

Profile

Download (3.13 KB) Statistics
| Branch: | Tag: | Revision:
1
/**
2
* Copyright (C) 2018 EDIT
3
* European Distributed Institute of Taxonomy
4
* http://www.e-taxonomy.eu
5
*
6
* The contents of this file are subject to the Mozilla Public License Version 1.1
7
* See LICENSE.TXT at the top of this package for the full license terms.
8
*/
9
package eu.etaxonomy.cdm.api.config;
10

    
11
import java.io.IOException;
12
import java.io.InputStream;
13
import java.util.HashMap;
14
import java.util.Map;
15
import java.util.Properties;
16

    
17
import org.springframework.beans.factory.annotation.Autowired;
18
import org.springframework.core.env.Environment;
19
import org.springframework.stereotype.Component;
20

    
21
import eu.etaxonomy.cdm.config.ConfigFileUtil;
22

    
23
/**
24
 * @author a.kohlbecker
25
 * @since Feb 16, 2018
26
 *
27
 */
28
@Component
29
public class ApplicationConfiguration {
30

    
31
    /**
32
     * Annotate your integration test class with
33
     * {@code @TestPropertySource(properties = {"testmode = true"})} to
34
     * load the configuration files from the test resources.
35
     * <p>
36
     * <b>NOTE:</b> This will not work with current implementation of the
37
     * {@link eu.etaxonomy.cdm.test.integration.CdmIntegrationTest CdmIntegrationTests}.
38
     * For these tests please use the {@link unitils.AlternativeUnitilsJUnit4TestClassRunner}.
39
     *
40
     */
41
    public static final String TEST_MODE = "testmode";
42

    
43
    @Autowired
44
    Environment env;
45

    
46
    Map<String, Properties> configurations = new HashMap<>();
47

    
48
    public String getProperty(ApplicationConfigurationFile configFile, String key){
49
        Properties props = loadPropertiesFile(configFile);
50
        return props.getProperty(key);
51
    }
52

    
53
    /**
54
     * @param currentDataSourceId
55
     * @throws IOException
56
     */
57
    protected Properties loadPropertiesFile(ApplicationConfigurationFile configFile) {
58

    
59
        String currentDataSourceId = env.getProperty(CdmConfigurationKeys.CDM_DATA_SOURCE_ID);
60
        if(!configurations.containsKey(configFile.getFileName())){
61
            Properties props;
62
            try {
63
                if(env.getProperty(TEST_MODE) == null){
64
                    // PRODUCTION MODE
65
                    props = new ConfigFileUtil()
66
                            .setDefaultContent(configFile.getDefaultContet())
67
                            .getProperties(currentDataSourceId, configFile.getFileName());
68
                } else {
69
                    // TEST MODE
70
                    InputStream is = ApplicationConfiguration.class.getClassLoader().getResourceAsStream(
71
                            configFile.getFileName());
72
                    if (is == null) {
73
                        throw new RuntimeException("Can't find configuration file '" + configFile.getFileName() + ".properties in the test resoures.");
74
                    }
75
                    props = new Properties();
76
                    props.load(is);
77
                }
78
                if(props != null){
79
                    configurations.put(configFile.getFileName(), props);
80
                }
81
            } catch (IOException e) {
82
                throw new RuntimeException("Error reading the configuration file '" + configFile.getFileName() + ".properties'. File corrupted?", e);
83
            }
84
        }
85
        return configurations.get(configFile.getFileName());
86
    }
87

    
88

    
89

    
90

    
91

    
92
}
(1-1/6)