package dl.zhang.test.util;

 import java.io.IOException;

 import java.io.Reader;

 import org.apache.ibatis.io.Resources;

 import org.apache.ibatis.session.SqlSession;

 import org.apache.ibatis.session.SqlSessionFactory;

 import org.apache.ibatis.session.SqlSessionFactoryBuilder;

 public class SessionUtil {

     private static ThreadLocal<SqlSession> th_session = new ThreadLocal<SqlSession>();

     private static SqlSessionFactory ssf = null;

     static {

         try {

             //mybatis.cfg.xml是核心配置的名字要根据自己核心配置名字去填写

             Reader reader = Resources.getResourceAsReader("mybatis.cfg.xml");

             ssf = new SqlSessionFactoryBuilder().build(reader);

         } catch (IOException e) {

             e.printStackTrace();

         }

     }

     public static SqlSession getCurrentSession() {

         SqlSession session = th_session.get();

         if (session == null) {

             session = ssf.openSession();

             th_session.set(session);

         }

         return session;

     }

     public static void closeSession() {

         SqlSession session = th_session.get();

         if (session != null) {

             session.close();

             th_session.set(null);

         }

     }

 }