PropertyFileConfigureProvider.java

package com.ctrip.platform.dal.dao.configure;

import java.io.File;
import java.io.FileReader;
import java.net.URL;
import java.util.Map;
import java.util.Properties;
import java.util.Set;

import com.ctrip.platform.dal.dao.DalClientFactory;

public class PropertyFileConfigureProvider implements DataSourceConfigureProvider {
	private final String CONFIG_NAME = "database.properties";
	private final String PATH = "path";
	
	private final String USER_NAME = ".userName";
	private final String PASSWORD = ".password";
	private final String CONNECTION_URL = ".connectionUrl";
	private final String DRIVER = ".driverClassName";
	
	private final String[] MUST_HAVES = new String[]{USER_NAME, PASSWORD, CONNECTION_URL, DRIVER};
	
	private String location;
	private Properties databaseConfig = new Properties();

	@Override
	public void initialize(Map<String, String> settings) throws Exception {
		location = settings.get(PATH);
		if (location != null) {
			databaseConfig.load(new FileReader(location));
			return;
		}
		
		// check classpath
		ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
		if (classLoader == null) {
			classLoader = DalClientFactory.class.getClassLoader();
		}
		
		URL configUrl = classLoader.getResource(CONFIG_NAME);
		
		if(configUrl == null)
			throw new IllegalStateException(
					"Can not find " + CONFIG_NAME + " to initilize database configure provider");

		databaseConfig.load(new FileReader(new File(configUrl.toURI())));
	}

	@Override
	public DataSourceConfigure getDataSourceConfigure(String dbName) {
		DataSourceConfigure conf = new DataSourceConfigure();
		
		conf.setUserName(databaseConfig.getProperty(dbName + USER_NAME));
		conf.setPassword(databaseConfig.getProperty(dbName + PASSWORD));
		conf.setConnectionUrl(databaseConfig.getProperty(dbName + CONNECTION_URL));
		conf.setDriverClass(databaseConfig.getProperty(dbName + DRIVER));
		
		return conf;
	}

	@Override
	public void setup(Set<String> dbNames) {
		for(String name: dbNames) {
			for(String item: MUST_HAVES) {
				if(databaseConfig.getProperty(name+item) == null)
					throw new IllegalStateException(
							"Can not find " + name+item + " to initilize database configure provider");
			}
		}
	}
}