com.frameworkset.common.poolman.management.PoolManConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bboss-persistent Show documentation
bboss is a j2ee framework include aop/ioc,mvc,persistent,taglib,rpc,event ,bean-xml serializable and so on.http://www.bbossgroups.com
The newest version!
/*
 *  Copyright 2008 biaoping.yin
 *
 *  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" bboss persistent,
 *  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 com.frameworkset.common.poolman.management;

import com.frameworkset.common.poolman.PoolManConstants;
import com.frameworkset.common.poolman.sql.ParserException;
import com.frameworkset.common.poolman.util.DatasourceConfig;
import com.frameworkset.util.SimpleStringUtil;
import org.frameworkset.soa.BBossStringWriter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.xml.parsers.SAXParser;
import javax.xml.parsers.SAXParserFactory;
import java.io.InputStream;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.net.URL;
import java.util.*;

//import java.util.Hashtable;

/**
 * This service configures ObjectPools by first looking for 'poolman.xml' in the
 * CLASSPATH. If it finds it, it parses the database entries and generic Class
 * entries and creates MBeans and pools for them, adding each one to an
 * appropriate PoolManager and the MBeanServer. If it cannot find this xml file,
 * it looks for 'poolman.props' and loads those properties in order to create
 * the pools (with no MBeans). If it cannot find poolman.props, it looks for the
 * deprecated pool.props file and treats it as if it were named 'poolman.props'.
 * If it cannot find any of these files, it throws an Exception.
 */
public class PoolManConfiguration   {
	private static Logger log = LoggerFactory.getLogger(PoolManConfiguration.class);

	private String configFile;
	private ConfigParser handler;
    private DatasourceConfig datasourceConfig;




	private ArrayList datasources;
    private Properties connectionProperties;
	private ArrayList genericObjects;

	public static String getSqlMappingDir() {
		return sqlMappingDir;
	}

	public static void setSqlMappingDir(String sqlMappingDir) {
		PoolManConfiguration.sqlMappingDir = sqlMappingDir;
	}

	private static  String sqlMappingDir;

	public static long getRefresh_interval() {
		return refresh_interval;
	}

	public static void setRefresh_interval(Long refresh_interval) {
		if(refresh_interval != null)
			PoolManConfiguration.refresh_interval = refresh_interval;
	}

	private static long refresh_interval = 5000L;

	public static boolean isColumnNameMapping() {
		return columnNameMapping;
	}

	public static void setColumnNameMapping(boolean columnNameMapping) {
		PoolManConfiguration.columnNameMapping = columnNameMapping;
	}

	private static boolean columnNameMapping = true;
	private boolean useJMX = PoolManConstants.DEFAULT_USE_JMX;
	public static boolean needtableinfo = false;
	public static String jndi_principal = null;
	public static String jndi_credentials = null;
	private String[] filterdbname = null;
	private String dbnamespace;
//	public static String pooltemplates;

	/**
	 * 用户自定义的适配器
	 */
	public Map adaptors = new HashMap();

	public PoolManConfiguration(String configFile, String dbname) {

		// this.configFile = configFile;
		// this.datasources = new ArrayList();
		// this.genericObjects = new ArrayList();
		// this.filterdbname = dbname;
		this(configFile, null, dbname == null ? null : new String[] { dbname });

	}

	public PoolManConfiguration(String configFile, String dbnamespace,
			String[] dbnames) {

		this.configFile = configFile;
		this.datasources = new ArrayList();
		this.genericObjects = new ArrayList();
		this.filterdbname = dbnames;
		this.dbnamespace = dbnamespace;

	}

	public PoolManConfiguration(String configFile) {

		this(configFile, null);
	}



	public boolean isUsingJMX() {
		return handler.isManagementJMX();
	}

	/** Load DataSource info from XML and create a Service for each entry set. */
	public boolean loadConfiguration(Map context) throws Exception {

		// first try XML
		try {
			parseXML(context);
			return true;
		} catch (NullPointerException ne) {
//			ne.printStackTrace();
			// then try deprecated properties
			Exception e = new ParserException("Load datasource Configuration from default bboss persistent config file,please config and start datasource follow document：https://doc.bbossgroups.com/#/persistent/PersistenceLayer1",ne);
			log.error("",e);
			// don't try the props files anymore, it's been over a year
			// System.out.println("** WARNING: Attempting to use deprecated properties files\n");
			// this.datasources =
			// parseProperties(PoolManConstants.PROPS_CONFIG_FILE);

		} catch (Exception e) {
//			log.warn("Load datasource Configuration from default bboss persistent config file： "
//					+ configFile );
//			log.warn("Load datasource Configuration failed from "
//					+ configFile ,e);
			Exception ex = new ParserException("Load datasource Configuration from default bboss persistent config file, please config and start datasource follow document：https://doc.bbossgroups.com/#/persistent/PersistenceLayer1",e);
			log.error("",ex);
		}
		return false;

	}

	@Override
	public String toString(){
		try {
			StringBuilder builder = new StringBuilder();
			BBossStringWriter writer = new BBossStringWriter(builder);
			writer.write("adaptors:\r\n");
			SimpleStringUtil.object2json(this.adaptors, writer);
			writer.write("\r\ndatasources:\r\n");
			SimpleStringUtil.object2json(this.datasources, writer);
			writer.write("\r\nenericObjects:\r\n");
			SimpleStringUtil.object2json(this.genericObjects, writer);
			writer.flush();
			return builder.toString();
		}
		catch (Exception e){
			e.printStackTrace();
			return "";
		}
	}

	public ArrayList getDataSources() {
		return this.datasources;
	}

	public ArrayList getGenericPools() {
		return this.genericObjects;
	}

	public Properties getAdminProperties() {
		return handler.getAdminProps();
	}

	private static Object lock = new Object();

    /**
     * DBConf tempConf = new DBConf();
     *                         tempConf.setPoolname(statusDbname);
     *                         tempConf.setDriver(statusDBConfig.getDbDriver());
     *
     *                         tempConf.setJndiName(dbJNDIName);
     *
     *                         tempConf.setShowsql(statusDBConfig.isShowSql());
     *                         tempConf.setDbAdaptor(statusDBConfig.getDbAdaptor());
     *                         tempConf.setDbtype(statusDBConfig.getDbtype());
     *                         tempConf.setColumnLableUpperCase(statusDBConfig.isColumnLableUpperCase());
     * @param context
     * @return
     */

    public boolean initDatasourceConfig(Map context){
        /**
         * 用户自定义的适配器
         */
        Map adaptors = new LinkedHashMap();
        String dbAdaptor = (String)context.get("dbAdaptor");
        String driver = (String)context.get("driver");
        if(dbAdaptor != null && !dbAdaptor.equals("")){
            adaptors.put(driver,dbAdaptor);
            String dbtype = (String)context.get("dbtype");
            if(dbtype != null && !dbtype.equals("")){
                adaptors.put(dbtype,dbAdaptor);
            }

        }
        this.adaptors = adaptors;
        ArrayList datasources = new ArrayList();
        Properties datasource = new Properties();
        String external = (String)context.get("external");
        datasource.put("external",external);
        String dbname = (String)context.get("dbname");
        datasource.put("dbname",dbname);
        datasource.put("loadmetadata","false");

        String dbname_datasource_jndiname = (String)context.get("dbname_datasource_jndiname");
        datasource.put("jndiName".toLowerCase(),dbname_datasource_jndiname);
        datasource.put("autoprimarykey".toLowerCase(),"false");
 
        String cachequerymetadata = (String)context.get("cachequerymetadata");
        datasource.put("cachequerymetadata".toLowerCase(),cachequerymetadata);
       
        String enablejta = (String)context.get("enablejta");
        datasource.put("enablejta".toLowerCase(),enablejta);

      

        datasource.put("nativeResults".toLowerCase(),"true");

        datasource.put("poolPreparedStatements".toLowerCase(),"false");
        

        datasource.put("keygenerate".toLowerCase(),"composite");

       
        String showsql = (String)context.get("showsql");
        datasource.put("showsql".toLowerCase(),showsql);
 
      
        datasource.put("RETURN_GENERATED_KEYS".toLowerCase(),"true");

        Integer queryfetchsize = (Integer) context.get("queryfetchsize");
        if(queryfetchsize != null && queryfetchsize != 0)
            datasource.put("queryfetchsize".toLowerCase(),queryfetchsize+"");
 
        Boolean columnLableUpperCase = (Boolean) context.get("columnLableUpperCase");
        if(columnLableUpperCase != null)
            datasource.put("columnLableUpperCase".toLowerCase(),columnLableUpperCase + "");
        else
            datasource.put("columnLableUpperCase".toLowerCase(), "true");

       
        datasources.add(datasource);
        this.datasources = datasources;
        this.connectionProperties = (Properties) context.get("connectionProperties");
        return true;

    }

	public boolean initConfig(Map context){
		/**
		 * 用户自定义的适配器
		 */
		Map adaptors = new LinkedHashMap();
		String dbAdaptor = (String)context.get("dbAdaptor");
		String driver = (String)context.get("driver");
		if(dbAdaptor != null && !dbAdaptor.equals("")){
			adaptors.put(driver,dbAdaptor);
			String dbtype = (String)context.get("dbtype");
			if(dbtype != null && !dbtype.equals("")){
				adaptors.put(dbtype,dbAdaptor);
			}

		}
		this.adaptors = adaptors;
		ArrayList datasources = new ArrayList();
		Properties datasource = new Properties();
		String external = (String)context.get("external");
		datasource.put("external",external);
		String dbname = (String)context.get("dbname");
		datasource.put("dbname",dbname);
		datasource.put("loadmetadata","false");
        
		String dbname_datasource_jndiname = (String)context.get("dbname_datasource_jndiname");
		datasource.put("jndiName".toLowerCase(),dbname_datasource_jndiname);
		datasource.put("autoprimarykey".toLowerCase(),"false");

		String encryptdbinfo = (String)context.get("encryptdbinfo");
		datasource.put("encryptdbinfo".toLowerCase(),encryptdbinfo);
		String cachequerymetadata = (String)context.get("cachequerymetadata");
		datasource.put("cachequerymetadata".toLowerCase(),cachequerymetadata);
        if(driver != null)
		    datasource.put("driver".toLowerCase(),driver);
		String enablejta = (String)context.get("enablejta");
		datasource.put("enablejta".toLowerCase(),enablejta);

		String jdbcurl = (String)context.get("jdbcurl");
        if(jdbcurl != null)
            datasource.put("url".toLowerCase(),jdbcurl);

		String username = (String)context.get("username");
		datasource.put("username".toLowerCase(),username);

		String password = (String)context.get("password");
		datasource.put("password".toLowerCase(),password);

		String txIsolationLevel = (String)context.get("txIsolationLevel");
		datasource.put("txIsolationLevel".toLowerCase(),txIsolationLevel);

		datasource.put("nativeResults".toLowerCase(),"true");

		datasource.put("poolPreparedStatements".toLowerCase(),"false");
		String initialConnections = (String)context.get("initialConnections");
		datasource.put("initialConnections".toLowerCase(),initialConnections);
		String minimumSize = (String)context.get("minimumSize");
		datasource.put("minimumSize".toLowerCase(),minimumSize);

		String maximumSize = (String)context.get("maximumSize");
		datasource.put("maximumSize".toLowerCase(),maximumSize);

		String removeAbandoned = (String)context.get("removeAbandoned");
		datasource.put("removeAbandoned".toLowerCase(),removeAbandoned);

		String userTimeout = (String)context.get("userTimeout");
		datasource.put("userTimeout".toLowerCase(),userTimeout);

		String logAbandoned = (String)context.get("logAbandoned");
		datasource.put("logAbandoned".toLowerCase(),logAbandoned);

		String readOnly = (String)context.get("readOnly");
		datasource.put("readOnly".toLowerCase(),readOnly);

		String skimmerFrequency = (String)context.get("skimmerFrequency");
		datasource.put("skimmerFrequency".toLowerCase(),skimmerFrequency);

		String connectionTimeout = (String)context.get("connectionTimeout");
		datasource.put("connectionTimeout".toLowerCase(),connectionTimeout);

		String shrinkBy = (String)context.get("shrinkBy");
		datasource.put("shrinkBy".toLowerCase(),shrinkBy);

		String testWhileidle = (String)context.get("testWhileidle");
		datasource.put("testWhileidle".toLowerCase(),testWhileidle);

		datasource.put("keygenerate".toLowerCase(),"composite");

		String maxWait = (String)context.get("maxWait");
		datasource.put("maxWait".toLowerCase(),maxWait);

		String validationQuery = (String)context.get("validationQuery");
		datasource.put("validationQuery".toLowerCase(),validationQuery);

		String showsql = (String)context.get("showsql");
		datasource.put("showsql".toLowerCase(),showsql);

		String externaljndiName = (String)context.get("externaljndiName");
		if(externaljndiName != null && !externaljndiName.equals(""))
			datasource.put("externaljndiName".toLowerCase(),externaljndiName);

		String usepool = (String)context.get("usepool");
		datasource.put("usepool".toLowerCase(),usepool);

		datasource.put("RETURN_GENERATED_KEYS".toLowerCase(),"true");

		Integer queryfetchsize = (Integer) context.get("queryfetchsize");
		if(queryfetchsize != null && queryfetchsize != 0)
			datasource.put("queryfetchsize".toLowerCase(),queryfetchsize+"");

		String dbInfoEncryptClass = (String)context.get("dbInfoEncryptClass");
		if(SimpleStringUtil.isNotEmpty(dbInfoEncryptClass))
			datasource.put("dbInfoEncryptClass".toLowerCase(),dbInfoEncryptClass);

		Boolean columnLableUpperCase = (Boolean) context.get("columnLableUpperCase");
		if(columnLableUpperCase != null)
			datasource.put("columnLableUpperCase".toLowerCase(),columnLableUpperCase + "");
		else
			datasource.put("columnLableUpperCase".toLowerCase(), "true");

        String balance = (String)context.get("balance");
        if(balance != null)
            datasource.put("balance",context.get("balance"));

        Object enableBalance = context.get("enableBalance");
        if(enableBalance != null) {
            datasource.put("enableBalance".toLowerCase(), enableBalance + "");
        }
		datasources.add(datasource);
		this.datasources = datasources;
        this.connectionProperties = (Properties) context.get("connectionProperties");
        return true;

	}

    public Properties getConnectionProperties() {
        return connectionProperties;
    }

    private void parseXML(Map context) throws Exception {

		/* CHANGED TO USE JAXP */

		if (!this.configFile.equals(PoolManConstants.XML_CONFIG_FILE_TEMPLATE)) {//非模版解析，直接从配置文件加载持久层数据源
			URL confURL = PoolManConfiguration.class.getClassLoader()
					.getResource(configFile);
			if (confURL == null)
				PoolManConfiguration.class.getClassLoader().getResource(
						"/" + configFile);

			if (confURL == null)
				getTCL().getResource(configFile);
			if (confURL == null)
				getTCL().getResource("/" + configFile);
			if (confURL == null)
				confURL = ClassLoader.getSystemResource(configFile);
			if (confURL == null)
				confURL = ClassLoader.getSystemResource("/" + configFile);
			String url = "";
			if (confURL == null) {
				url = System.getProperty("user.dir");

				url += "/" + configFile;
				log.debug("Pool Config file:" + System.getProperty("user.dir")
						+ "/" + configFile);
			} else {
				url = confURL.toString();
			}
			this.handler = new ConfigParser(url, this.dbnamespace,
					this.filterdbname);

			SAXParserFactory factory = SAXParserFactory.newInstance();
			factory.setNamespaceAware(false);
			factory.setValidating(false);
			SAXParser parser = factory.newSAXParser();
			parser.parse(url, handler);
			this.adaptors = handler.getAdaptors();
			if(handler.getSqlMappingDir() != null)
				this.sqlMappingDir = handler.getSqlMappingDir();
			this.datasources = handler.getDataSourceProperties();
			this.genericObjects = handler.getGenericProperties();
		} else {
			this.initConfig(context);
		}

	}

	private static ClassLoader getTCL() throws IllegalAccessException,
			InvocationTargetException {
		Method method = null;
		try {
			method = (java.lang.Thread.class).getMethod(
					"getContextClassLoader");
		} catch (NoSuchMethodException e) {
			return null;
		}
		return (ClassLoader) method.invoke(Thread.currentThread());
	}

	private ArrayList parseProperties(String propsfilename) throws Exception {

		// backwards compatibility is a b****

		HashMap datasources = new HashMap();
		this.getClass().getClassLoader().getParent().getSystemClassLoader()
				.getResource(null);
		InputStream is = null;
		Properties p = null;

		try {
			is = Thread.currentThread().getContextClassLoader()
					.getResourceAsStream(propsfilename);
			p = new Properties();
			p.load(is);
		} catch (Exception e) {
			if (propsfilename.equals(PoolManConstants.PROPS_CONFIG_FILE)) {
				try {
					is = Thread.currentThread().getContextClassLoader()
							.getResourceAsStream(
									PoolManConstants.OLDPROPS_CONFIG_FILE);
					p = new Properties();
					p.load(is);
				} catch (Exception e2) {
					throw new Exception(
							"Unable to find and read a valid datasource config file. "
									+ "Please ensure that '"
									+ PoolManConstants.XML_CONFIG_FILE
									+ "' is in a directory that is in your CLASSPATH.\n");
				}
				log.warn("\nPLEASE NOTE: You should replace the pool.props file with a valid config file\n");
			} else {
				throw new Exception(
						"ERROR: Unable to find and read a valid PoolMan properties file.\n"
								+ "Please ensure that "
								+ PoolManConstants.XML_CONFIG_FILE
								+ " or at least "
								+ propsfilename
								+ " is in a directory that is in your CLASSPATH.\n");
			}
		}

		// now have props with each set of datasource entries delimited by a
		// number
		// split those into an ArrayList of Properties objects

		Properties theseProps = null;
		String entrySetNumber = null;

		for (Enumeration enum_ = p.keys(); enum_.hasMoreElements();) {

			String key = (String) enum_.nextElement();

			// get rid of the "_"
			String adjustedKey = key;
			if (key.indexOf("_") != -1) {
				StringBuilder sb = new StringBuilder(key.substring(0, key
						.indexOf("_")));
				sb.append(key.substring(key.indexOf("_") + 1, key.length()));
				adjustedKey = sb.toString();
				if ((!adjustedKey.toLowerCase().equals("dbname"))
						&& (adjustedKey.startsWith("db"))) {
					adjustedKey = adjustedKey
							.substring(2, adjustedKey.length());
				}
			}
			if (adjustedKey.toLowerCase().startsWith("cacherefresh")) {
				adjustedKey = "cacherefreshinterval"
						+ adjustedKey.substring(12, adjustedKey.length());
			}

			try {
				entrySetNumber = key.substring(key.indexOf('.') + 1, key
						.length());
			} catch (StringIndexOutOfBoundsException sbe) {
				throw new Exception("Unnumbered property in poolman.props: "
						+ key);
			}

			if (datasources.containsKey(entrySetNumber))
				theseProps = (Properties) datasources.get(entrySetNumber);
			else
				theseProps = new Properties();

			theseProps.setProperty(adjustedKey.substring(0, adjustedKey
					.indexOf('.')), p.getProperty(key));

			datasources.put(entrySetNumber, theseProps);

		}

		ArrayList finalDataSourceList = new ArrayList();
		for (Iterator iter = datasources.keySet().iterator(); iter.hasNext();) {
			Properties testp = (Properties) datasources.get(iter.next());
			finalDataSourceList.add(testp);
		}

		return finalDataSourceList;
	}

	/**
	 * @return the adaptors
	 */
	public Map getAdaptors() {
		return adaptors;
	}

	/**
	 * @param adaptors
	 *            the adaptors to set
	 */
	public void setAdaptors(Map adaptors) {
		this.adaptors = adaptors;
	}

	public boolean isNeedtableinfo() {
		return needtableinfo;
	}

	public void setNeedtableinfo(boolean needtableinfo) {
		this.needtableinfo = needtableinfo;
	}

    public DatasourceConfig getDatasourceConfig() {
        return datasourceConfig;
    }

    public void setDatasourceConfig(DatasourceConfig datasourceConfig) {
        this.datasourceConfig = datasourceConfig;
    }
}