com.hangum.tadpole.commons.sql.map.SQLMap.java Source code

Introduction

Here is the source code for com.hangum.tadpole.commons.sql.map.SQLMap.java

Source

/*******************************************************************************
 * Copyright (c) 2012 Cho Hyun Jong.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *     Cho Hyun Jong - initial API and implementation
 ******************************************************************************/
package com.hangum.tadpole.commons.sql.map;

import java.io.InputStream;
import java.io.StringReader;

import org.apache.commons.lang.StringEscapeUtils;
import org.apache.log4j.Logger;

import com.hangum.tadpole.commons.sql.define.DBDefine;
import com.hangum.tadpole.dao.system.UserDBDAO;
import com.ibatis.sqlmap.client.SqlMapClient;
import com.ibatis.sqlmap.client.SqlMapClientBuilder;

/**
 * SQLMapMap?    ? ?.
 * 
 * @author hangum
 */
public class SQLMap {
    private final static String URL = "${JDBC.ConnectionURL}";
    private final static String USERNAME = "${JDBC.Username}";
    private final static String PASSWORD = "${JDBC.Password}";

    /**
     * Logger for this class
     */
    private static final Logger logger = Logger.getLogger(SQLMap.class);

    private SQLMap() {
    }

    public static SqlMapClient getInstance(UserDBDAO dbInfo) throws Exception {
        String config = getConfig(dbInfo);

        return SqlMapClientBuilder.buildSqlMapClient(new StringReader(config));
    }

    /**
     * DB  ??   .
     * @param dbInfo 
     * @return
     * @throws Exception
     */
    private static String getConfig(UserDBDAO dbInfo) throws Exception {
        String config = getFileToString(DBDefine.getDBDefine(dbInfo.getTypes()).getLocation());

        // url chnage
        config = config.replace(URL, StringEscapeUtils.escapeXml(dbInfo.getUrl()));
        // id change
        config = config.replace(USERNAME, StringEscapeUtils.escapeXml(dbInfo.getUsers()));
        // pass change
        config = config.replace(PASSWORD, StringEscapeUtils.escapeXml(dbInfo.getPasswd()));

        return config;
    }

    private static String getFileToString(String url) throws Exception {
        ClassLoader loader = SQLMap.class.getClassLoader();
        InputStream is = loader == null ? ClassLoader.getSystemResourceAsStream(url)
                : loader.getResourceAsStream(url);

        int size = is.available();
        byte[] dataByte = new byte[size];
        is.read(dataByte, 0, size);
        is.close();

        return new String(dataByte);
    }

}