栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Java

自定义mybatis框架

Java 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

自定义mybatis框架

一、回顾原生框架工作流程

    1.xml方式

通常我们在service层注入mapper,在maper中写接口,接口名和xml中的statement标签中的id对应,当调用对应接口的时候,就会把statement中的sql语句解析出来,然后连接数据库执行。

2.注解方式

与xml形式不同的是,sql语句直接放在接口上的注解中,解析出sql了之后,同样连接数据库执行。

二、自定义框架的思想步骤

1、mapper只是接口,没有实现,怎么执行一系列操作的?

2.接口中的参数怎么和解析出来的sql的参数对应的?

@MyInsert("INSERT INTO pe_user (id, username, password) values (#{id},#{username}, #{password}) ")
    int insertUser(@MyParam("id") String id,@MyParam("username") String username,@MyParam("password") String password);

3.sql是怎么样和数据库连接并执行的?

三、代码实现

1、只有方法被实现,才能处理业务,所以我们想办法让其有实现

  1.1 cglib动态代理,字节码技术生成虚拟子类

  1.2 jdk动态代理,代理方式解决

  1.3 匿名内部类方式去实现

我们用jdk动态代理实现,mybatis框架也是用这个实现的。

public class SqlSession {
    public static  T getMapper(Class cls){
        Object proxyInstance = Proxy.newProxyInstance(cls.getClassLoader(), new Class[]{cls}, new MyInovationHandler(cls));
        return (T) proxyInstance;
    }
}
public class MyInovationHandler implements InvocationHandler {
    private Object object;


    public MyInovationHandler(Object object) {
        this.object = object;

    }

    // proxy 代理对象 method 执行方法 args方法参数
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
    

        return 0;
    }

PS:invoke方法一定不要return null,否则代理出来的对象,去调用方法时会报错。但是如果构造方法传入的不是class对象时就不会报错。

2.接口中的参数怎么和解析出来的sql的参数对应的

3.sql是怎么样和数据库连接并执行的

 private Object doInsert(Method method, Object[] args, MyInsert myInsert) {
        //2.如果存在注解则获取注解的值
        String insertSql = myInsert.value();
        //3.将传入的参数和sql上的参数对应起来
        Parameter[] parameters = method.getParameters();
        //3.1 将传入的参数放到固定的容器中
        ConcurrentHashMap map = getStringObjectConcurrentHashMap(args, parameters);
        //4.将sql的值替换为?
        //4.1 创建绑定值的容器
        List list =new ArrayList<>();
        //4.2 获取sql的参数名
        String[] strings = getStrings(insertSql, map, list);
        String newInsertSql = SQLUtils.parameQuestion(insertSql, strings);
        //5.调用jdbc去查询
        return JDBCUtils.insert(newInsertSql,false,list);
    }

    private String[] getStrings(String insertSql, ConcurrentHashMap map, List list) {
        String[] strings = SQLUtils.sqlInsertParameter(insertSql);
        for (String paramName:strings) {
            //4.3 从容器中获取参数的值
            Object paramValue = map.get(paramName.trim());
            list.add(paramValue);
        }
        return strings;
    }

    private ConcurrentHashMap getStringObjectConcurrentHashMap(Object[] args, Parameter[] parameters) {
        ConcurrentHashMap map = new ConcurrentHashMap();
        //3.2 判断参数上是否有MyParam注解
        for (int i = 0; i < parameters.length; i++) {
            MyParam myParam = parameters[i].getDeclaredAnnotation(MyParam.class);
            if (myParam!=null){
                //3.3 有注解的话 则将注解得值作为key:paramName 传入的参数值作为value:paramValue存储
                String paramName = myParam.value();
                Object paramValue = args[i];
                map.put(paramName,paramValue);
            }
        }
        return map;
    } 

4.工具类

package com.dong.mybatisstudy.utils;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.ResultSetmetaData;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public final class JDBCUtils {

	private static String connect;
	private static String driverClassName;
	private static String URL;
	private static String username;
	private static String password;
	private static boolean autoCommit;

	
	private static Connection conn;

	static {
		config();
	}

	
	private static void config() {
		
		driverClassName = "com.mysql.jdbc.Driver";
		
		URL = "jdbc:mysql://localhost:3306/shiro_db?useUnicode=true&characterEncoding=utf8";
		
		username = "root";
		
		password = "root";
		
		autoCommit = false;

	}

	
	private static boolean load() {
		try {
			Class.forName(driverClassName);
			return true;
		} catch (ClassNotFoundException e) {
			System.out.println("驱动类 " + driverClassName + " 加载失败");
		}

		return false;
	}

	
	private static boolean invalid() {
		if (conn != null) {
			try {
				if (conn.isClosed() || !conn.isValid(3)) {
					return true;
					
				}
			} catch (SQLException e) {
				e.printStackTrace();
			}
			
			return false;
		} else {
			return true;
		}
	}

	
	public static Connection connect() {
		if (invalid()) { 
			
			load();
			try {
				
				conn = DriverManager.getConnection(URL, username, password);
			} catch (SQLException e) {
				System.out.println("建立 " + connect + " 数据库连接失败 , " + e.getMessage());
			}
		}
		return conn;
	}

	
	public static void transaction() {

		try {
			conn.setAutoCommit(autoCommit);
		} catch (SQLException e) {
			System.out.println("设置事务的提交方式为 : " + (autoCommit ? "自动提交" : "手动提交") + " 时失败: " + e.getMessage());
		}

	}

	
	public static Statement statement() {
		Statement st = null;
		connect();
		
		transaction();
		
		try {
			st = conn.createStatement();
		} catch (SQLException e) {
			System.out.println("创建 Statement 对象失败: " + e.getMessage());
		}

		return st;
	}

	
	private static PreparedStatement prepare(String SQL, boolean autoGeneratedKeys) {

		PreparedStatement ps = null;
		connect();
		
		transaction();
		
		try {
			if (autoGeneratedKeys) {
				ps = conn.prepareStatement(SQL, Statement.RETURN_GENERATED_KEYS);
			} else {
				ps = conn.prepareStatement(SQL);
			}
		} catch (SQLException e) {
			System.out.println("创建 PreparedStatement 对象失败: " + e.getMessage());
		}

		return ps;

	}

	public static ResultSet query(String SQL, List params) {

		if (SQL == null || SQL.trim().isEmpty() || !SQL.trim().toLowerCase().startsWith("select")) {
			throw new RuntimeException("你的SQL语句为空或不是查询语句");
		}
		ResultSet rs = null;
		if (params.size() > 0) {
			
			PreparedStatement ps = prepare(SQL, false);
			try {
				for (int i = 0; i < params.size(); i++) {
					ps.setObject(i + 1, params.get(i));
				}
				rs = ps.executeQuery();
			} catch (SQLException e) {
				System.out.println("执行SQL失败: " + e.getMessage());
			}
		} else {
			
			Statement st = statement();
			try {
				rs = st.executeQuery(SQL); // 直接执行不带参数的 SQL 语句
			} catch (SQLException e) {
				System.out.println("执行SQL失败: " + e.getMessage());
			}
		}

		return rs;

	}

	private static Object typeof(Object o) {
		Object r = o;

		if (o instanceof java.sql.Timestamp) {
			return r;
		}
		// 将 java.util.Date 转成 java.sql.Date
		if (o instanceof java.util.Date) {
			java.util.Date d = (java.util.Date) o;
			r = new java.sql.Date(d.getTime());
			return r;
		}
		// 将 Character 或 char 变成 String
		if (o instanceof Character || o.getClass() == char.class) {
			r = String.valueOf(o);
			return r;
		}
		return r;
	}

	public static boolean execute(String SQL, Object... params) {
		if (SQL == null || SQL.trim().isEmpty() || SQL.trim().toLowerCase().startsWith("select")) {
			throw new RuntimeException("你的SQL语句为空或有错");
		}
		boolean r = false;
		

		
		SQL = SQL.trim();
		SQL = SQL.toLowerCase();
		String prefix = SQL.substring(0, SQL.indexOf(" "));
		String operation = ""; // 用来保存操作类型的 变量
		// 根据前缀 确定操作
		switch (prefix) {
		case "create":
			operation = "create table";
			break;
		case "alter":
			operation = "update table";
			break;
		case "drop":
			operation = "drop table";
			break;
		case "truncate":
			operation = "truncate table";
			break;
		case "insert":
			operation = "insert :";
			break;
		case "update":
			operation = "update :";
			break;
		case "delete":
			operation = "delete :";
			break;
		}
		if (params.length > 0) { // 说明有参数
			PreparedStatement ps = prepare(SQL, false);
			Connection c = null;
			try {
				c = ps.getConnection();
			} catch (SQLException e) {
				e.printStackTrace();
			}
			try {
				for (int i = 0; i < params.length; i++) {
					Object p = params[i];
					p = typeof(p);
					ps.setObject(i + 1, p);
				}
				ps.executeUpdate();
				commit(c);
				r = true;
			} catch (SQLException e) {
				System.out.println(operation + " 失败: " + e.getMessage());
				rollback(c);
			}

		} else { // 说明没有参数

			Statement st = statement();
			Connection c = null;
			try {
				c = st.getConnection();
			} catch (SQLException e) {
				e.printStackTrace();
			}
			// 执行 DDL 或 DML 语句,并返回执行结果
			try {
				st.executeUpdate(SQL);
				commit(c); // 提交事务
				r = true;
			} catch (SQLException e) {
				System.out.println(operation + " 失败: " + e.getMessage());
				rollback(c); // 回滚事务
			}
		}
		return r;
	}

	
	public static int insert(String SQL, boolean autoGeneratedKeys, List params) {
		int var = -1;
		if (SQL == null || SQL.trim().isEmpty()) {
			throw new RuntimeException("你没有指定SQL语句,请检查是否指定了需要执行的SQL语句");
		}
		// 如果不是 insert 开头开头的语句
		if (!SQL.trim().toLowerCase().startsWith("insert")) {
			System.out.println(SQL.toLowerCase());
			throw new RuntimeException("你指定的SQL语句不是插入语句,请检查你的SQL语句");
		}
		// 获得 被执行的 SQL 语句的 前缀 ( 第一个单词 )
		SQL = SQL.trim();
		SQL = SQL.toLowerCase();
		if (params.size() > 0) { // 说明有参数
			PreparedStatement ps = prepare(SQL, autoGeneratedKeys);
			Connection c = null;
			try {
				c = ps.getConnection(); // 从 PreparedStatement 对象中获得 它对应的连接对象
			} catch (SQLException e) {
				e.printStackTrace();
			}
			try {
				for (int i = 0; i < params.size(); i++) {
					Object p = params.get(i);
					p = typeof(p);
					ps.setObject(i + 1, p);
				}
				int count = ps.executeUpdate();
				if (autoGeneratedKeys) { // 如果希望获得数据库产生的键
					ResultSet rs = ps.getGeneratedKeys(); // 获得数据库产生的键集
					if (rs.next()) { // 因为是保存的是单条记录,因此至多返回一个键
						var = rs.getInt(1); // 获得值并赋值给 var 变量
					}
				} else {
					var = count; // 如果不需要获得,则将受SQL影像的记录数赋值给 var 变量
				}
				commit(c);
			} catch (SQLException e) {
				System.out.println("数据保存失败: " + e.getMessage());
				rollback(c);
			}
		} else { // 说明没有参数
			Statement st = statement();
			Connection c = null;
			try {
				c = st.getConnection(); // 从 Statement 对象中获得 它对应的连接对象
			} catch (SQLException e) {
				e.printStackTrace();
			}
			// 执行 DDL 或 DML 语句,并返回执行结果
			try {
				int count = st.executeUpdate(SQL);
				if (autoGeneratedKeys) { // 如果企望获得数据库产生的键
					ResultSet rs = st.getGeneratedKeys(); // 获得数据库产生的键集
					if (rs.next()) { // 因为是保存的是单条记录,因此至多返回一个键
						var = rs.getInt(1); // 获得值并赋值给 var 变量
					}
				} else {
					var = count; // 如果不需要获得,则将受SQL影像的记录数赋值给 var 变量
				}
				commit(c); // 提交事务
			} catch (SQLException e) {
				System.out.println("数据保存失败: " + e.getMessage());
				rollback(c); // 回滚事务
			}
		}
		return var;
	}

	
	private static void commit(Connection c) {
		if (c != null && !autoCommit) {
			try {
				c.commit();
			} catch (SQLException e) {
				e.printStackTrace();
			}
		}
	}

	
	private static void rollback(Connection c) {
		if (c != null && !autoCommit) {
			try {
				c.rollback();
			} catch (SQLException e) {
				e.printStackTrace();
			}
		}
	}

	
	public static void release(Object cloaseable) {

		if (cloaseable != null) {

			if (cloaseable instanceof ResultSet) {
				ResultSet rs = (ResultSet) cloaseable;
				try {
					rs.close();
				} catch (SQLException e) {
					e.printStackTrace();
				}
			}

			if (cloaseable instanceof Statement) {
				Statement st = (Statement) cloaseable;
				try {
					st.close();
				} catch (SQLException e) {
					e.printStackTrace();
				}
			}

			if (cloaseable instanceof Connection) {
				Connection c = (Connection) cloaseable;
				try {
					c.close();
				} catch (SQLException e) {
					e.printStackTrace();
				}
			}

		}

	}

} 
public static String[] sqlInsertParameter(String sql) {
		int startIndex = sql.indexOf("values");
		int endIndex = sql.length();
		String substring = sql.substring(startIndex + 6, endIndex).replace("(", "").replace(")", "").replace("#{", "")
				.replace("}", "");
		String[] split = substring.split(",");
		return split;
	}



public static String parameQuestion(String sql, String[] parameterName) {
		for (int i = 0; i < parameterName.length; i++) {
			String string = parameterName[i].trim();
			sql = sql.replace("#{"+string+"}", "?");
		}
		return sql;
	}

转载请注明:文章转载自 www.mshxw.com
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号