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

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<MapperInfo> 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;
    }
}

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;
}

package spring.mybatis;
 
import java.lang.reflect.Proxy;
/**
 * @ClassName MySqlSession
 * @Description
 * @Data 2018/7/8
 * @Author xiao liang
 */
public class MySqlSession {
    public <T> T selectOne(MapperInfo mapperInfo ,Object[] paremeters){
        MyExecutor myexecutor = new MyExecutor();
        return myexecutor.query(mapperInfo,paremeters);
    }
  
    public <T> T getMapper(Class<?> aClass,String mybatisXmlName){
        return (T) Proxy.newProxyInstance(aClass.getClassLoader(),new Class[]{aClass},new MyMapperProxy(this,mybatisXmlName));
    }
}

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;
    }
}

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> 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;
    }
}