微MyBatis自己手写

什么是 MyBatis？

MyBatis 是一款优秀的持久层框架，它支持定制化 SQL、存储过程以及高级映射。MyBatis 避免了几乎所有的 JDBC 代码和手动设置参数以及获取结果集。MyBatis 可以使用简单的 XML 或注解来配置和映射原生类型、接口和 Java 的 POJO（Plain Old Java Objects，普通老式 Java 对象）为数据库中的记录。

Execute执行器

接口

public interface MyExecute {
    <T> T query(String sql);
}

实现

public class MyExecuteImpl implements MyExecute {
    private final String URL = "jdbc:mysql://127.0.0.1:3306/mini_mybatis";
    private final String USER = "root";
    private final String PWD = "root";

    @Override
    public <T> T query(String sql) {
        Connection connection = null;
        PreparedStatement preparedStatement = null;
        ResultSet resultSet = null;
        try {
            connection = DriverManager.getConnection(URL, USER, PWD);
            preparedStatement = connection.prepareStatement(sql);
            resultSet = preparedStatement.executeQuery();
            return (T) handleResult(resultSet);
        } catch (Exception e) {

        }
        return null;
    }

    private Object handleResult(ResultSet resultSet) throws Exception {
        //此处就简单写死了,Member只需简单创建一个实体类即可
        Member member = null;
        if (resultSet.next()) {
            member = new Member();
            member.setWechat_nickname(resultSet.getString("wechat_nickname"));
            return member;
        }
        return null;
    }
}

Mapper

public interface MemberMapper {
    Member findById(Integer id);
}

mapper xml

public class MemberMapperXml {
    public static String nameSpace = "mini.mapper.MemberMapper";

    public static Map<String, String> sqlMap = new HashMap<String, String>();

    static {
        sqlMap.put("findById", "select * from member as m where m.member_id=%s");
    }

    public static String getSqLMap(String method) {
        return sqlMap.get(method);
    }
}

SqlSession

public interface MySqlSession {
    <T> T selectOne(String sql);

    <T> T getMapper(Class<T> t);
}

public class MySqlSessionImpl implements MySqlSession {

    private MyExecute myExecute = new MyExecuteImpl();
    @Override
    public <T> T selectOne(String sql) {
        return myExecute.query(sql);
    }
    @Override
    public <T> T getMapper(Class<T> t) {
        return (T) Proxy.newProxyInstance(t.getClassLoader(), new Class[]{t}, new MyMapperProxy(this));
    }
}

Proxy

public class MyMapperProxy implements InvocationHandler {
    private MySqlSession mySqlSession;

    public MyMapperProxy() {
    }

    public MyMapperProxy(MySqlSession mySqlSession) {
        this.mySqlSession = mySqlSession;
    }

    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        String nameSpace = method.getDeclaringClass().getName();
        if (MemberMapperXml.nameSpace.equals(nameSpace)) {
            String methodName = method.getName();
            String sql = String.format(MemberMapperXml.getSqLMap(methodName), String.valueOf(args[0]));
            return mySqlSession.selectOne(sql);
        }
        return null;
    }
}

测试

public static void main(String[] args) {
        MySqlSession mySqlSession = new MySqlSessionImpl();
        MemberMapper memberMapper=  mySqlSession.getMapper(MemberMapper.class);
        Member member =memberMapper.findById(2);
        System.out.println(member.getWechat_nickname());
    }

目录结构