ORM框架实现的深度剖析

我对写笔记什么的没什么太大兴趣，用篇博文记录下自己的学习，还是挺不错的。

这篇讲的是ORM框架的深度剖析，不废话了。

直接放图

这是数据库类型和java数据类型的映射关系

　　我们用代码来说明一下

这是代码的bean类，也就是跟数据库表作对应的

这里用到了两个自定义的注解Table和Column，主要是模拟

package com.orm.demo.domain;

import org.orm.test.annotation.Column;
import org.orm.test.annotation.Table;

@Table("t_user")
public class User {
    private int id;
    @Column("name")
    private String name;
    private int age;

    public int getId() {
        return id;
    }

    public void setId(int id) {
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }

    public User(String name, int age) {
        this.name = name;
        this.age = age;
    }
}

 我们再来看dao层的代码

这里的session是自定义的类

package com.orm.demo.mapper;

import com.orm.demo.domain.User;

import java.util.List;

public interface UserMapper {
    List<User> findAll();
    void insert(User user);
}

package com.orm.demo.mapper.impl;

import com.orm.demo.domain.User;
import com.orm.demo.mapper.UserMapper;
import org.orm.test.Session;
import org.orm.test.impl.SessionImpl;

import java.util.List;

public class UserMapperImpl implements UserMapper {
    private Session session = new SessionImpl();

    public List<User> findAll() {
        List<User> users = session.findAll();
        return users;
    }

    public void insert(User user) {
        session.insert(user);
    }
}

package org.orm.test;

import java.util.List;

public interface Session {
    <T> List<T> findAll();
    <T> void insert(T obj);
}

package org.orm.test.impl;

import org.orm.test.Session;
import org.orm.test.annotation.Column;
import org.orm.test.annotation.Table;

import java.lang.reflect.Field;
import java.sql.*;
import java.util.ArrayList;
import java.util.List;

public class SessionImpl implements Session {
    static {
        //加载驱动
        try {
            Class.forName("com.mysql.jdbc.Driver");
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }
    }

    /**
     * 获取数据源
     * @return
     */
    private Connection getDataSource(){
        try {
            return DriverManager.getConnection(
                    "jdbc:mysql:///springboot",
                    "root",
                    "123"
            );
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return null;
    }


    public <T> List<T> findAll() {
        Connection conn = getDataSource();
        String sql = "";
        try {
            PreparedStatement pst = conn.prepareStatement(sql);
            ResultSet res = pst.executeQuery();
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return null;
    }

    public <T> void insert(T obj) {
        //实例化一个sb，用以存放sql
        StringBuilder sb = new StringBuilder();
        //存放参数的列表
        List<Object> paramList = new ArrayList<Object>();
        //调用方法拼接sql和拿到参数列表
        buildCondition(obj,sb,paramList);
        //拿到sql的字符串
        String sql = sb.toString();
        //拿到数据库连接
        Connection conn = getDataSource();
        try {
            PreparedStatement pst = conn.prepareStatement(sql);
            for(int i=0;i<paramList.size();i++){
                pst.setObject(i+1,paramList.get(i));
            }
            pst.executeUpdate();
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    public void buildCondition(Object obj,StringBuilder sb,List<Object> paramList){
        //获得当前对象的类
        Class<?> cls = obj.getClass();
        //拿到当前对象的类名作为表名
        String tableName = cls.getName().toUpperCase();
        //判断当前类是否有table注解
        if (cls.isAnnotationPresent(Table.class)){
            //如果有的话，把映射的表名取出
            tableName = cls.getAnnotation(Table.class).value().toUpperCase();
        }
        //添加sql前面的语句
        sb.append("insert into ").append(tableName).append("(");
        //获得当前类的所有属性
        Field[] fields = cls.getDeclaredFields();
        //遍历属性
        for(Field field : fields){
            //拿到当前属性的名字
            String col = field.getName().toUpperCase();
            //判断属性是否有column注解
            if(field.isAnnotationPresent(Column.class)){
                //如果有，就把列名赋值为映射的名字
                col = field.getAnnotation(Column.class).value().toUpperCase();
            }
            //添加sql
            sb.append(col+",");
            //获取字段值，设置可获取
            field.setAccessible(true);
            try {
                //拿到当前对象相应属性的值
                Object val = field.get(obj);
                //在参数列表中加入值
                paramList.add(val);
            } catch (IllegalAccessException e) {
                e.printStackTrace();
            }
        }
        //删除sb中最后一个逗号（多余）
        sb.deleteCharAt(sb.length() - 1).append(") values(");
        //遍历参数列表
        for(int i=0;i<paramList.size();i++){
            //拼接sql
            sb.append("?,");
        }
        //删除sb中最后一个逗号（多余）
        sb.deleteCharAt(sb.length() - 1).append(")");
        System.out.print("sql=========>");
        //输出sb
        System.out.println(sb);
    }
}

这里奉上session的代码，注释中有详细说明

这里是两个注解的代码

package org.orm.test.annotation;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
public @interface Table {
    String value() default "";
}

package org.orm.test.annotation;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.FIELD)
public @interface Column {
    String value();
}

测试代码

import com.orm.demo.domain.User;
import com.orm.demo.mapper.UserMapper;
import com.orm.demo.mapper.impl.UserMapperImpl;
import org.junit.Test;

import java.util.List;

public class TestDao {
    private UserMapper userMapper = new UserMapperImpl();
    @Test
    public void test(){
        User user = new User("小刘",18);
        userMapper.insert(user);
        System.out.println("插入成功");
//        List<User> users = userMapper.findAll();
//        System.out.println(users);
    }
}

学习本无底，前进莫徬徨