MyBatis--工具类模板

第一种:

public class JDBCUtils {
    private static ThreadLocal<SqlSession> threadLocal = new ThreadLocal<SqlSession>();
    private static SqlSessionFactory sqlSessionFactory=null;
    static{
        Reader resourceAsReader = null;
        try {
            resourceAsReader = Resources.getResourceAsReader("Configuration.xml");
        } catch (IOException e) {
            e.printStackTrace();
            throw new RuntimeException("读取配置文件异常");
        }
        sqlSessionFactory = new SqlSessionFactoryBuilder().build(resourceAsReader);
        if(resourceAsReader!=null) {
            try {
                resourceAsReader.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    private JDBCUtils() {
    }

    public static SqlSession getSqlSession(){
       SqlSession sqlSession = threadLocal.get();
       if(sqlSession==null){
           sqlSession = sqlSessionFactory.openSession(false);
           threadLocal.set(sqlSession);
       }
        return sqlSession;
    }

    /**
     * 产生动态代理的sqlSession对象,修改原来的commit与rollback方法
     * @return
     */
    public static SqlSession getProxySqlsession(){
        final SqlSession sqlSession = getSqlSession();
        SqlSession proxySqlSession = (SqlSession)Proxy.newProxyInstance(sqlSession.getClass().getClassLoader(), new Class[]{SqlSession.class}, new InvocationHandler() {
            public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
                if(method.getName().equals("commit")){
                    Object invoke = method.invoke(sqlSession, args);
                    close(sqlSession);
                    return invoke;
                }
                else if(method.getName().equals("rollback")){
                    Object invoke = method.invoke(sqlSession, args);
                    close(sqlSession);
                    return invoke;
                }
                else if(method.getName().equals("close")){
                    close(sqlSession);
                    return null;
                }
                else{
                    return method.invoke(sqlSession,args);
                }
            }
        });
        return proxySqlSession;
    }

    public static void close(SqlSession sqlSession) {
        if(sqlSession!=null) {
            sqlSession.close();
            threadLocal.remove();
            if(threadLocal!=null){
                threadLocal.set(null);
            }
        }
    }
}

第二种:
import java.io.*; 
import org.apache.ibatis.io.Resources; 
import org.apache.ibatis.session.*; 
public class MyBatisSqlSessionFactory{ 
   private static  SqlSessionFactory sqlSessionFactory; 
   public static SqlSessionFactory getSqlSessionFactory(){ 
        if(sqlSessionFactory == null){ 
            InputStream inputStream; 
            try{ 
                inputStream = Resources.getResourceAsStream("mybatis-config.xml"); 
                sqlSessionFactory = new SqlSessionFactoryBuilder().build(inputStream); 
            }catch (IOException e){ 
                throw newRuntimeException(e.getCause()); 
            } 
        } 
        return sqlSessionFactory; 
    } 
   public static SqlSession openSession(){ 
       return getSqlSessionFactory().openSession(); 
    } 

猜你喜欢

转载自blog.csdn.net/sinat_37064286/article/details/86772732