自定义框架模拟mybatis的运行过程

自定义框架模拟mybatis的运行过程

核心类的依赖关系以及功能介绍

配置文件

SqlMapConfig.xml：主配置文件

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE configuration
        PUBLIC "-//mybatis.org//DTD Config 3.0//EN"
        "http://mybatis.org/dtd/mybatis-3-config.dtd">
<configuration>
    <!--默认环境为mysql-->
    <environments default="mysql">
        <!--配置mysql环境-->
        <environment id="mysql">
            <!--事务管理器控制,配置JDBC的事务管理-->
            <transactionManager type="JDBC"></transactionManager>
            <!--配置mybatis内置连接池参数-->
            <dataSource type="pooled">
                <!--配置驱动-->
                <property name="driver" value="com.mysql.jdbc.Driver"></property>
                <!--配置链接的数据库-->
                <property name="url" value="jdbc:mysql://localhost:8080/mybatis?characterEncoding=utf-8"></property>
                <!--用户名-->
                <property name="username" value="root"></property>
                <!--密码-->
                <property name="password" value="awsed2zx"></property>
            </dataSource>
        </environment>
    </environments>
    
    <mappers>
        <mapper resource="com/zmysna/dao/IUserDao.xml"></mapper>
    </mappers>
</configuration>

IUserDao：dao层接口

package com.zmysna.dao;

import com.zmysna.entity.User;

import java.util.List;

public interface IUserDao {
    List<User> findAll();
}

IUserDao.xml：接口映射配置文件

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper
        PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
        "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<!--namespace名称空间,代表当前xml是哪个接口的映射-->
<mapper namespace="com.zmysna.dao.IUserDao">
    <!--d 对应的是 IUserDao 接口的方法名称-->
    <!--resultType 方法的返回值类型，如果方法返回的是集合，只要指定集合中元素的类型-->
    <select id="" resultType="com.zmysna.entity.User">
        select * from USER
    </select>
</mapper>

User：实体类，与数据库字段一一对应

package com.zmysna.entity;

import java.sql.Date;

public class User {
    private int id;
    private String username;
    private Date birthday;
    private String sex;
    private String address;
}

核心类实现

SqlSessionFactory: SqlSession工厂类

package com.zmysna.mybatis.factory;

import com.zmysna.mybatis.sqlSession.SqlSession;
import com.zmysna.mybatis.sqlSession.impl.DefaultSqlSession;

/**
 * SqlSession工厂类：负责生产SqlSession
 */
public class SqlSessionFactory {
    //生产一个SqlSession
    public SqlSession openSession(){
        return new DefaultSqlSession();
    }
}

DefaultSqlSession: 默认SqlSession接口实现

package com.zmysna.mybatis.sqlSession.impl;

import com.zmysna.mybatis.proxy.MapperProxy;
import com.zmysna.mybatis.sqlSession.SqlSession;

import java.lang.reflect.Proxy;

public class DefaultSqlSession implements SqlSession {

    @Override
    public <T> T getMapper(Class<T> daoClass) {
        return (T) Proxy.newProxyInstance(this.getClass().getClassLoader(),
                new Class[]{daoClass}, new MapperProxy());
    }
}

MapperProxy：代理事件处理程序

package com.zmysna.mybatis.proxy;

import com.mchange.v2.c3p0.ComboPooledDataSource;
import com.mchange.v2.c3p0.jboss.C3P0PooledDataSource;
import com.zmysna.mybatis.config.Configuration;
import com.zmysna.mybatis.config.Mapper;
import com.zmysna.utils.Executor;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.util.List;
import java.util.Map;

public class MapperProxy implements InvocationHandler{

    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        //获取全类名和方法名作为获取xml数据集合的键
        String className = method.getDeclaringClass().getName();
        String methodName = method.getName();
        //创建Configuation对象，config对象解析xml配置文件获取需要的数据
        Configuration config = Configuration.getInstance();
        // 获取 Mapper
        Map<String, Mapper> mappers = config.getMappers();
        // 获取数据库连接池
        ComboPooledDataSource dataSource = config.getDataSource();
        String key = className +"."+ methodName;
        Mapper mapper = mappers.get(key);
        // 执行数据查询，从数据库中查询数据
        List list = Executor.execute(mapper, dataSource);
        return list;
    }
}

Mapper：封装接口映射文件的信息

public class Mapper {
    private String namespace;
    private String id;
    private String resultType;
    private String queryString;
}

Executor：执行SQL的工具类

package com.zmysna.utils;

import com.mchange.v2.c3p0.ComboPooledDataSource;
import com.zmysna.mybatis.config.Mapper;

import java.beans.PropertyDescriptor;
import java.lang.reflect.Method;
import java.sql.*;
import java.util.ArrayList;
import java.util.List;

public class Executor {


    public static List execute(Mapper mapper, ComboPooledDataSource dataSource) {
        try {
            //获取连接
            Connection connection = dataSource.getConnection();
            //注入SQL语句得到预编译对象
            PreparedStatement ps = connection.prepareStatement(mapper.getQueryString());
            //执行得到结果集
            ResultSet rs = ps.executeQuery();
            //创建一个List集合作为返回结果
            List list = new ArrayList();
            //得到返回的结果类型
            String resultType = mapper.getResultType();
            //得到结果集元数据
            ResultSetMetaData metaData = rs.getMetaData();
            //获得结果类型的字节码对象
            Class<?> clazz = Class.forName(resultType);
            //得到结果集的列的个数
            int count = metaData.getColumnCount();

            //遍历结果集
            while (rs.next()) {
                //创建一个结果类型的对象
                Object obj = clazz.getConstructor().newInstance();

                //遍历结果集每一列将数据封装到对象中
                for (int i = 1; i <= count; i++) {
                    String columnName = metaData.getColumnName(i);
                    Object value = rs.getObject(columnName);
                    PropertyDescriptor pd = new PropertyDescriptor(columnName, clazz);
                    Method method = pd.getWriteMethod();
                    method.invoke(obj, value);
                }
                //将封装好对象添加到list集合
                list.add(obj);
            }
            return list;
        } catch (SQLException e) {
            e.printStackTrace();
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }
}

测试

UserTest ：测试类

package com.zmysna.dao;
import com.zmysna.entity.User;
import org.apache.ibatis.io.Resources;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;
import org.junit.Test;

import java.io.IOException;
import java.io.InputStream;
import java.util.List;

public class UserTest {

    @Test
    public void testFindAll() throws IOException {
        // 加载配置文件
        InputStream in = Resources.getResourceAsStream("SqlMapConfig.xml");
        // 获得sqlSession
        SqlSession session = new SqlSessionFactoryBuilder().build(in).openSession();
        // 获得IUserao接口的代理对象,代理对象由mybatis根据配置文件对接口进行实现
        IUserDao mapper = session.getMapper(IUserDao.class);
        // 打印实现类的信息,可以看出它是一个动态代理对象
        System.out.println(mapper);
        // 调用实现类的方法
        List<User> users = mapper.findAll();
    }
}

结果