手写spring+springmvc+mybatis框架篇【Mybatis】

2021-12-03 12:54:57

整合Mybatis是本项目中的一个难点。
实现功能：
1 动态绑定用户输入参数
2 Mybatis的resultType动态绑定返回实体类。
3 在spring中的接口注入
4 xml版本的mapper注入。
关于Mybatis的优秀文章给大家推荐两个:
1、手写简化版mybatis
https://my.oschina.net/liughDevelop/blog/1631006
2、Mybatis源码解读-设计模式总结
http://www.crazyant.net/2022.html
手写板大致思路如下：
这里的Myconfiguration和我的JDBCUtils类似。
实现思路:
首先用XmlBuilderMapper类读取mapper.xml(我是在initBean中指定要读取的配置文件，并没有写成在xml中配置动态的读取xml。)文件，获取MapperInfo对象保存信息。
用户用MysqlSession的getMapper方法，返回一个代理对象，用这个代理对象来执行MySqlSession定义的selectOne方法来查询，Mybaits中的SqlSession中定义了大量的方法，我这里简化只有一个selectOne方法。
然后这个方法调用executor执行器来解析sql，传入参数，执行数据库操作。最后返回结果。将返回结果封装成对象。
动态代理一般有两种，一种是jdk一种是cglib动态代理，本项目采用是jdk动态代理实现。
XmlBuilderMapper.class
package spring.mybatis;
import lombok.extern.slf4j.Slf4j;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;
import spring.constants.Constants;
import java.io.File;
import java.util.ArrayList;
import java.util.List;
/**

@ClassName XmlBuilder
@Description
@Data 2018/7/8
@Author xiao liang

*/
@Slf4j
public class XmlBuilderMapper {
public List buildMapper(String xmlMapperPath){

   //实例化mapperInfo的链表，每一条sql对应一个MapperInfo对象
   List<MapperInfo> mapperInfoList = new ArrayList<>();
   MapperInfo mapperInfo = new MapperInfo();
   // 创建saxReader对象
   SAXReader reader = new SAXReader();
   // 通过read方法读取一个文件 转换成Document对象
   Document document = null;
   String pathName = Constants.PATH + xmlMapperPath;
   try {
       document = reader.read(new File(pathName));
   } catch (DocumentException e) {
       log.error("文件没有找到,{}", pathName);
   }
   //获取根节点元素
   Element node = document.getRootElement();
   mapperInfo.setInterfaceName(node.attributeValue("namespace"));
   //获取所有的bean
   List<Element> elementsList = node.elements();
   for (Element element :
           elementsList) {
       if ("select".equals(element.getName())){
           mapperInfo.setMethodName(element.attributeValue("id"));
           mapperInfo.setResultClassName(element.attributeValue("resultType"));
           mapperInfo.setSqlContent(element.getText());
       }
       mapperInfoList.add(mapperInfo);
   }
   return mapperInfoList;

}
}
然后介绍一下MapperInfo对象
package spring.mybatis;
import lombok.Data;
/**

@ClassName MapperInfo
@Description 用来封装读取mapper.xml文件后的信息
@Data 2018/7/8
@Author xiao liang

*/
@Data
public class MapperInfo {
//namespace命名空间
private String interfaceName;
//sql内容
private String sqlContent;
//对应的方法名
private String methodName;
//返回值的class名
private String resultClassName;
}
JDBCUtils工具类我在开篇就介绍了，也没什么新内容，不贴在这里了
其实关键点就是动态代理和执行器
MySqlSession（动态代理部分）
package spring.mybatis;
import java.lang.reflect.Proxy;
/**

@ClassName MySqlSession
@Description
@Data 2018/7/8
@Author xiao liang

*/
public class MySqlSession {
public T selectOne(MapperInfo mapperInfo ,Object[] paremeters){

   MyExecutor myexecutor = new MyExecutor();
   return myexecutor.query(mapperInfo,paremeters);

}

public T getMapper(Class<?> aClass,String mybatisXmlName){

   return (T) Proxy.newProxyInstance(aClass.getClassLoader(),new Class[]{aClass},new MyMapperProxy(this,mybatisXmlName));

}
}
MyMapperProxy(jdk动态代理的实现) 代理之后执行的还是sqlSession中的方法
package spring.mybatis;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.util.List;
/**

@ClassName MyMapperProxy
@Description
@Data 2018/7/8
@Author xiao liang

*/
public class MyMapperProxy implements InvocationHandler {
private MySqlSession mySqlSession;
private String mybatisXmlName;
//mybatisXmlName传入的要读取的xml文件名
public MyMapperProxy(MySqlSession mySqlSession , String mybatisXmlName){

  this.mySqlSession = mySqlSession;
  this.mybatisXmlName = mybatisXmlName;

}
@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {

   XmlBuilderMapper xmlBuilderMapper = new XmlBuilderMapper();
   List<MapperInfo> mapperInfoList = xmlBuilderMapper.buildMapper(mybatisXmlName);
   //如果存在sql，开始执行方法
   if (mapperInfoList != null && mapperInfoList.size() != 0){
       for (MapperInfo mapperInfo :
               mapperInfoList) {
           if (!method.getDeclaringClass().getName().equals(mapperInfo.getInterfaceName())){
               return null;
           }
           if (method.getName().equals(mapperInfo.getMethodName())){
               //其实最后执行的mySqlSession中的方法，args是用户传递的参数数组
               return mySqlSession.selectOne(mapperInfo,args);
           }
       }
   }
   return null;

}
}
返回到MySqlSession后，就轮到执行器MyExcutor出场了。
MyExcutor：也是一个难点，先说一下程序的逻辑。读取到mapperinfo对象，获得定义的sql，返回类的名称等信息。
用正则解析sql，根据#{}，判断sql中有几处？，然后将用户传递的参数按照顺序依次写入到sql的？中。
最后读取结果集，用set方法注入到返回的对象中，这样就实现了返回时候的实体类绑定了。
package spring.mybatis;
import lombok.extern.slf4j.Slf4j;
import spring.Utils.GetMethodName;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
/**

@ClassName MyExecutor
@Description 执行器
@Data 2018/7/8
@Author xiao liang

*/
@Slf4j
public class MyExecutor {
public T query(MapperInfo mapperInfo, Object[] paremeters) {

   //获取mapper.xml文件中的sql语句
   String preSql = mapperInfo.getSqlContent();
   //正则匹配规则，匹配有几个#{}
   String rgex = "#\\{.*?}";
   String sql = null;
   String resultClassName = mapperInfo.getResultClassName();
   Class<?> aClass = null;
   Field[] fields = null;
   Method[] methods = null;
   Object object = null;
   Pattern pattern = Pattern.compile(rgex);
   Matcher m = pattern.matcher(preSql);
   Connection connection = null;
   ResultSet rs = null;
   PreparedStatement preparedStatement = null;
   //Preparement注入参数的个数
   int orderPre = 0;
   //每匹配一次，加一
   while (m.find()) {
       orderPre++;
   }
   //匹配完成之后，将#{}用？代替，preparement执行sql
   sql = m.replaceAll("?");
   try {
       aClass = Class.forName(resultClassName);
       fields = aClass.getDeclaredFields();
   } catch (ClassNotFoundException e) {
       e.printStackTrace();
   }
   try {
       JDBCUtils jdbcUtils = new JDBCUtils();
        connection = jdbcUtils.getConnection();
       preparedStatement = connection.prepareStatement(sql);
       //将用户传递过来的参数按照顺序依次赋值给prepareStatement的sql
       for (int i = 1; i <= orderPre; i++) {
           preparedStatement.setObject(i, paremeters[i - 1]);
       }
       rs = preparedStatement.executeQuery();
       object = aClass.newInstance();
       while (rs.next()) {
           int i = 1;
           for (Field field :
                   fields) {
               //遍历每个属性，然后将结果集中的数据用set方法注入到返回的对象中
               String setMethodNameByField = GetMethodName.getSetMethodNameByField(field.getName());
               Method method2 = aClass.getMethod(setMethodNameByField, field.getType());
               if (field.getType().getSimpleName().equals("String")) {
                   method2.invoke(object, rs.getString(i));
               } else if (field.getType().getSimpleName().equals("Integer")) {
                   method2.invoke(object, rs.getInt(i));
               }
               i++;
           }
       }
       return (T) object;
   } catch (SQLException e) {
       log.error("sql语句异常，请检查sql{}", sql);
       e.printStackTrace();
   } catch (InstantiationException e) {
       e.printStackTrace();
   } catch (IllegalAccessException e) {
       e.printStackTrace();
   } catch (InvocationTargetException e) {
       e.printStackTrace();
   } catch (NoSuchMethodException e) {
       e.printStackTrace();
   }
   finally {
       JDBCUtils.colseResource(connection,preparedStatement,rs);
   }
   return null;

}
}
我将此项目上传到了github，需要的童鞋可以自行下载。
https://github.com/836219171/MySSM

码农公寓

相关文章