package org.jrest4guice.persistence.ibatis;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import org.jrest4guice.transaction.IbatisTransaction;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NodeList;
import com.google.inject.Singleton;
import com.ibatis.common.resources.Resources;
import com.ibatis.sqlmap.client.SqlMapClient;
/**
*
* @author <a href="mailto:zhangyouqun@gmail.com">cnoss (QQ:86895156)</a>
*
*/
@Singleton
public class SqlMapClientHolder {
private static SqlMapClient sqlMapClient;
private static List<Class<?>> daos = new ArrayList<Class<?>>(0);
private final ThreadLocal<IbatisTransaction> transaction = new ThreadLocal<IbatisTransaction>();
public static void addIbatisDao(Class<?> clazz) {
daos.add(clazz);
}
/**
* 初始化sqlMapClient
*/
public static void initSqlMapClient(){
FileInputStream input = null;
try {
//本地的配置文件路径
File sqlMapConfigFile = Resources.getResourceAsFile("SqlMapConfig.xml");
//生成sqmMapping的临时xml文件
StringBuffer sb = new StringBuffer();
sb.append("<?xml version=\"1.0\" encoding=\"UTF-8\" ?>");
sb.append("\n<!DOCTYPE sqlMap ");
sb.append("\n PUBLIC \"-//ibatis.apache.org//DTD SQL Map 2.0//EN\" ");
sb.append("\n\"http://ibatis.apache.org/dtd/sql-map-2.dtd\">");
sb.append("\n<sqlMap>");
//从IbatisDao类中读取sqlMap的配置信息
SqlMapping sqlMapping;
for (Class<?> clazz : daos) {
sqlMapping = SqlMapClientXmlHelper.generateXmlConfig(clazz);
sb.append("\n"+sqlMapping.getCacheModel());
sb.append("\n"+sqlMapping.getParameterMap());
sb.append("\n"+sqlMapping.getResultMap());
sb.append("\n"+sqlMapping.getStatement());
}
sb.append("\n</sqlMap>");
//输出sqmMapping文件
FileOutputStream fout = new FileOutputStream(sqlMapConfigFile.getParent()+File.separator+"sqlMap.xml");
String sqlMappingStr = sb.toString();
// System.out.println(sqlMappingStr);
fout.write(sqlMappingStr.getBytes());
fout.flush();
fout.close();
//修改配置的配置文件，在其中增加sqlMap元素
DocumentBuilderFactory factory = DocumentBuilderFactory
.newInstance();
DocumentBuilder ibatisConfigBuilder = factory.newDocumentBuilder();
InputStream inputStream = Resources.getResourceAsStream("SqlMapConfig.xml");
Document doc = ibatisConfigBuilder.parse(inputStream);
NodeList sqlMapElem = doc.getElementsByTagName("sqlMapConfig");
Element elem = doc.createElement("sqlMap");
elem.setAttribute("resource", "sqlMap.xml");
sqlMapElem.item(0).appendChild(elem);
Transformer trans = TransformerFactory.newInstance().newTransformer();
File tempFile = File.createTempFile("sqlMapConfig", ".xml");
tempFile.deleteOnExit();
trans.transform(new DOMSource(doc),new StreamResult(tempFile));
//从临时的配置文件中初始化sqlMapClient
input = new FileInputStream(tempFile);
SqlMapClientHolder.sqlMapClient = SqlMapClientBuilder
.buildSqlMapClient(input);
} catch (Throwable e) {
throw new RuntimeException("初始化Ibatis上下文失败！", e);
}finally{
if(input != null)
try {
input.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
public SqlMapClient getSqlMapClient() {
transaction.set(new IbatisTransaction(SqlMapClientHolder.sqlMapClient));
return SqlMapClientHolder.sqlMapClient;
}
public IbatisTransaction getIbatisTransaction(){
return this.transaction.get();
}
public void closeSqlMapClient(){
IbatisTransaction ibatisTransaction = this.getIbatisTransaction();
if(ibatisTransaction != null){
try {
ibatisTransaction.end();
} catch (Exception e) {
}
this.transaction.remove();
}
}
}