Mybatis手写Mybatis框架之获取Mapper映射并完成整个框架的实现-----Mybatis框架

超、凢脫俗 2024-03-24 21:53 68阅读 0赞

>     package org.god.ibatis.core;
>     
>     public class MappedStatement
>     {
>         //这是一个普通的java类,封装了一个SQL标签
>         //一个MappedStatement对象对应一个SQL标签
>         //一个sql标签中所有的信息封装到该对象中
>         //体现了面向对象编程实现
>         //要封装的结果集类型
>         //当select语句的时候,会对应一个resultType的值
>         public MappedStatement(){}
>         private String sql;
>         private String resultType;
>     
>         @Override
>         public String toString() {
>             return "MappedStatement{" +
>                     "sql='" + sql + '\'' +
>                     ", resultType='" + resultType + '\'' +
>                     '}';
>         }
>     
>         public String getSql() {
>             return sql;
>         }
>     
>         public void setSql(String sql) {
>             this.sql = sql;
>         }
>     
>         public String getResultType() {
>             return resultType;
>         }
>     
>         public void setResultType(String resultType) {
>             this.resultType = resultType;
>         }
>     
>         public MappedStatement(String sql, String resultType) {
>             this.sql = sql;
>             this.resultType = resultType;
>         }
>     }
> 
>     package org.god.ibatis.core;
>     
>     public class MappedStatement
>     {
>         //这是一个普通的java类,封装了一个SQL标签
>         //一个MappedStatement对象对应一个SQL标签
>         //一个sql标签中所有的信息封装到该对象中
>         //体现了面向对象编程实现
>         //要封装的结果集类型
>         //当select语句的时候,会对应一个resultType的值
>         public MappedStatement(){}
>         private String sql;
>         private String resultType;
>     
>         @Override
>         public String toString() {
>             return "MappedStatement{" +
>                     "sql='" + sql + '\'' +
>                     ", resultType='" + resultType + '\'' +
>                     '}';
>         }
>     
>         public String getSql() {
>             return sql;
>         }
>     
>         public void setSql(String sql) {
>             this.sql = sql;
>         }
>     
>         public String getResultType() {
>             return resultType;
>         }
>     
>         public void setResultType(String resultType) {
>             this.resultType = resultType;
>         }
>     
>         public MappedStatement(String sql, String resultType) {
>             this.sql = sql;
>             this.resultType = resultType;
>         }
>     }

>     package org.god.ibatis.core;
>     
>     import org.dom4j.Document;
>     import org.dom4j.DocumentException;
>     import org.dom4j.Element;
>     import org.dom4j.Node;
>     import org.dom4j.io.SAXReader;
>     import org.god.ibatis.utils.Resources;
>     
>     import javax.sql.DataSource;
>     import java.io.InputStream;
>     import java.util.ArrayList;
>     import java.util.HashMap;
>     import java.util.List;
>     import java.util.Map;
>     
>     public class SqlSessionFactoryBuilder
>     {
>         //通过godBatis-config文件,创建SqlSessionFactory对象
>         public SqlSessionFactoryBuilder(){}
>         public SqlSessionFactory build(InputStream is)
>         {
>             //解析godBatis-config文件创建SqlSessionFactory对象
>             //build操作的核心就是要解析XML文件
>             //解析完成要构建SqlSessionFactory对象
>             //解析GodBatis-config文件
>             SAXReader reader = new SAXReader();
>             Document document = null;
>             SqlSessionFactory sqlSessionFactory = null;
>             try
>             {
>                 document = reader.read(is);
>                 Element rootElement = (Element)document.selectSingleNode("/configuration/environments");
>                 String defaultId = rootElement.attributeValue("default");
>                 Element environment = (Element) document.selectSingleNode("/configuration/environments/environment[@id='" + defaultId +"']");
>                 Element transactionManagerEle = environment.element("transactionManager");
>                 Element dataSourceEle = environment.element("dataSource");
>                 DataSource dataSource = getDataSource(dataSourceEle);
>                 //获取事务管理器
>                 Transaction transaction = getTransaction(transactionManagerEle,dataSource);
>                 //获取MappedStatements
>                 List<Node> mappers = document.selectNodes("//mapper");
>                 List<String> mapperList = new ArrayList<>();
>                 mappers.forEach(node ->
>                 {
>                     Element mapper = (Element)node;
>                     String resource = mapper.attributeValue("resource");
>                     mapperList.add(resource);
>                 });
>                 //获取整个文件中所有的mapper标签
>                 Map<String,MappedStatement> mappedStatements = getMappedStatements(mapperList);
>                 sqlSessionFactory = new SqlSessionFactory(transaction,mappedStatements);
>             }
>             catch (Exception e)
>             {
>                 e.printStackTrace();
>             }
>             return sqlSessionFactory;
>         }
>     
>         private Map<String, MappedStatement> getMappedStatements(List<String> mapperList)
>         {
>             Map<String,MappedStatement> map = new HashMap<>();
>             mapperList.forEach(resource ->
>             {
>                 SAXReader reader = new SAXReader();
>                 InputStream ip = Resources.getResourceAsStream(resource);
>                 Document document = null;
>                 try
>                 {
>                     document = reader.read(ip);
>                 }
>                 catch (DocumentException e)
>                 {
>                     throw new RuntimeException(e);
>                 }
>                 Element mapper = document.getRootElement();
>                 String namespace = mapper.attributeValue("namespace");
>                 List<Element> elements = mapper.elements();
>                 elements.forEach(element ->
>                 {
>                     MappedStatement mappedStatement = new MappedStatement();
>                     String resultType = element.attributeValue("resultType");
>                     String id = element.attributeValue("id");
>                     if(resultType != null)
>                     {
>                         mappedStatement.setResultType(resultType);
>                     }
>                     else
>                     {
>                         mappedStatement.setResultType("null");
>                     }
>                     String sql = element.getTextTrim();
>                     mappedStatement.setSql(sql);
>                     String sqlId = namespace + "." + id;
>                     map.put(sqlId,mappedStatement);
>                 });
>             });
>             return map;
>         }
>     
>         private Transaction getTransaction(Element transactionManagerEle,DataSource dataSource)
>         {
>             //获取事务管理器对象
>             String type = transactionManagerEle.attributeValue("type").trim().toUpperCase();
>             Transaction transaction = null;
>             if(Const.JDBC_TRANSACTION.equals(type))
>             {
>                 //默认是开启事务的,将来需要手动提交的
>                 transaction = new JdbcTransaction(dataSource,false);
>             }
>             if(Const.MANAGED_TRANSACTION.equals(type))
>             {
>                 transaction = new ManagedTransaction();
>             }
>             return transaction;
>         }
>     
>         private DataSource getDataSource(Element dataSourceEle) 
>         {
>             //获取数据源对象
>             String type = dataSourceEle.attributeValue("type").trim().toUpperCase();
>             DataSource dataSource = null;
>             List<Element> elements = dataSourceEle.elements();
>             Map<String,String> map = new HashMap<>();
>             elements.forEach(element -> 
>             {
>                 String name = element.attributeValue("name");
>                 String value = element.attributeValue("value");
>                 map.put(name,value);
>             });
>             String driver = map.get("driver");
>             String url = map.get("url");
>             String username = map.get("username");
>             String password = map.get("password");
>             if(Const.UN_POOLED_DATASOURCE.equals(type))
>             {
>                 dataSource = new UnPooledDataSource(driver,url,username,password);
>             }
>             else if(Const.POOLED_DATASOURCE.equals(type))
>             {
>                 dataSource = new PooledDataSource();
>             }
>             else if(Const.JNDI_DATASOURCE.equals(type))
>             {
>                 dataSource = new JndiDataSource();
>             }
>             return dataSource;
>         }
>     }
> 
>     package org.god.ibatis.core;
>     
>     import org.dom4j.Document;
>     import org.dom4j.DocumentException;
>     import org.dom4j.Element;
>     import org.dom4j.Node;
>     import org.dom4j.io.SAXReader;
>     import org.god.ibatis.utils.Resources;
>     
>     import javax.sql.DataSource;
>     import java.io.InputStream;
>     import java.util.ArrayList;
>     import java.util.HashMap;
>     import java.util.List;
>     import java.util.Map;
>     
>     public class SqlSessionFactoryBuilder
>     {
>         //通过godBatis-config文件,创建SqlSessionFactory对象
>         public SqlSessionFactoryBuilder(){}
>         public SqlSessionFactory build(InputStream is)
>         {
>             //解析godBatis-config文件创建SqlSessionFactory对象
>             //build操作的核心就是要解析XML文件
>             //解析完成要构建SqlSessionFactory对象
>             //解析GodBatis-config文件
>             SAXReader reader = new SAXReader();
>             Document document = null;
>             SqlSessionFactory sqlSessionFactory = null;
>             try
>             {
>                 document = reader.read(is);
>                 Element rootElement = (Element)document.selectSingleNode("/configuration/environments");
>                 String defaultId = rootElement.attributeValue("default");
>                 Element environment = (Element) document.selectSingleNode("/configuration/environments/environment[@id='" + defaultId +"']");
>                 Element transactionManagerEle = environment.element("transactionManager");
>                 Element dataSourceEle = environment.element("dataSource");
>                 DataSource dataSource = getDataSource(dataSourceEle);
>                 //获取事务管理器
>                 Transaction transaction = getTransaction(transactionManagerEle,dataSource);
>                 //获取MappedStatements
>                 List<Node> mappers = document.selectNodes("//mapper");
>                 List<String> mapperList = new ArrayList<>();
>                 mappers.forEach(node ->
>                 {
>                     Element mapper = (Element)node;
>                     String resource = mapper.attributeValue("resource");
>                     mapperList.add(resource);
>                 });
>                 //获取整个文件中所有的mapper标签
>                 Map<String,MappedStatement> mappedStatements = getMappedStatements(mapperList);
>                 sqlSessionFactory = new SqlSessionFactory(transaction,mappedStatements);
>             }
>             catch (Exception e)
>             {
>                 e.printStackTrace();
>             }
>             return sqlSessionFactory;
>         }
>     
>         private Map<String, MappedStatement> getMappedStatements(List<String> mapperList)
>         {
>             Map<String,MappedStatement> map = new HashMap<>();
>             mapperList.forEach(resource ->
>             {
>                 SAXReader reader = new SAXReader();
>                 InputStream ip = Resources.getResourceAsStream(resource);
>                 Document document = null;
>                 try
>                 {
>                     document = reader.read(ip);
>                 }
>                 catch (DocumentException e)
>                 {
>                     throw new RuntimeException(e);
>                 }
>                 Element mapper = document.getRootElement();
>                 String namespace = mapper.attributeValue("namespace");
>                 List<Element> elements = mapper.elements();
>                 elements.forEach(element ->
>                 {
>                     MappedStatement mappedStatement = new MappedStatement();
>                     String resultType = element.attributeValue("resultType");
>                     String id = element.attributeValue("id");
>                     if(resultType != null)
>                     {
>                         mappedStatement.setResultType(resultType);
>                     }
>                     else
>                     {
>                         mappedStatement.setResultType("null");
>                     }
>                     String sql = element.getTextTrim();
>                     mappedStatement.setSql(sql);
>                     String sqlId = namespace + "." + id;
>                     map.put(sqlId,mappedStatement);
>                 });
>             });
>             return map;
>         }
>     
>         private Transaction getTransaction(Element transactionManagerEle,DataSource dataSource)
>         {
>             //获取事务管理器对象
>             String type = transactionManagerEle.attributeValue("type").trim().toUpperCase();
>             Transaction transaction = null;
>             if(Const.JDBC_TRANSACTION.equals(type))
>             {
>                 //默认是开启事务的,将来需要手动提交的
>                 transaction = new JdbcTransaction(dataSource,false);
>             }
>             if(Const.MANAGED_TRANSACTION.equals(type))
>             {
>                 transaction = new ManagedTransaction();
>             }
>             return transaction;
>         }
>     
>         private DataSource getDataSource(Element dataSourceEle) 
>         {
>             //获取数据源对象
>             String type = dataSourceEle.attributeValue("type").trim().toUpperCase();
>             DataSource dataSource = null;
>             List<Element> elements = dataSourceEle.elements();
>             Map<String,String> map = new HashMap<>();
>             elements.forEach(element -> 
>             {
>                 String name = element.attributeValue("name");
>                 String value = element.attributeValue("value");
>                 map.put(name,value);
>             });
>             String driver = map.get("driver");
>             String url = map.get("url");
>             String username = map.get("username");
>             String password = map.get("password");
>             if(Const.UN_POOLED_DATASOURCE.equals(type))
>             {
>                 dataSource = new UnPooledDataSource(driver,url,username,password);
>             }
>             else if(Const.POOLED_DATASOURCE.equals(type))
>             {
>                 dataSource = new PooledDataSource();
>             }
>             else if(Const.JNDI_DATASOURCE.equals(type))
>             {
>                 dataSource = new JndiDataSource();
>             }
>             return dataSource;
>         }
>     }