1、JDBCUtils
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;
/** 声明一个 Connection类型的静态属性,用来缓存一个已经存在的连接对象 */
private static Connection conn;
static {
config();
}
/**
* 开头配置自己的数据库信息
*/
private static void config() {
/*
* 获取驱动
*/
driverClassName = "com.mysql.jdbc.Driver";
/*
* 获取URL
*/
URL = "jdbc:mysql://localhost:3306/test?useUnicode=true&characterEncoding=utf8";
/*
* 获取用户名
*/
username = "root";
/*
* 获取密码
*/
password = "root";
/*
* 设置是否自动提交,一般为false不用改
*/
autoCommit = false;
}
/**
* 载入数据库驱动类
*/
private static boolean load() {
try {
Class.forName(driverClassName);
return true;
} catch (ClassNotFoundException e) {
System.out.println("驱动类 " + driverClassName + " 加载失败");
}
return false;
}
/**
* 专门检查缓存的连接是否不可以被使用 ,不可以被使用的话,就返回 true
*/
private static boolean invalid() {
if (conn != null) {
try {
if (conn.isClosed() || !conn.isValid(3)) {
return true;
/*
* isValid方法是判断Connection是否有效,如果连接尚未关闭并且仍然有效,则返回 true
*/
}
} catch (SQLException e) {
e.printStackTrace();
}
/*
* conn 既不是 null 且也没有关闭 ,且 isValid 返回 true,说明是可以使用的 ( 返回 false )
*/
return false;
} else {
return true;
}
}
/**
* 建立数据库连接
*/
public static Connection connect() {
if (invalid()) { /* invalid为true时,说明连接是失败的 */
/* 加载驱动 */
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());
}
}
/**
* 创建 Statement 对象
*/
public static Statement statement() {
Statement st = null;
connect();
/* 如果连接是无效的就重新连接 */
transaction();
/* 设置事务的提交方式 */
try {
st = conn.createStatement();
} catch (SQLException e) {
System.out.println("创建 Statement 对象失败: " + e.getMessage());
}
return st;
}
/**
* 根据给定的带参数占位符的SQL语句,创建 PreparedStatement 对象
*
* @param SQL
* 带参数占位符的SQL语句
* @return 返回相应的 PreparedStatement 对象
*/
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
2、定义MyInvocationHandlerMbatis
/**
* 功能说明:手写mybatis框架注解版本
* 1.使用动态代理技术,获取接口方法上的sql语句
* 2.根据不同的SQL语句来执行
*/
public class MyInvocationHandlerMbatis implements InvocationHandler {
/**
* 这个就是我们要代理的真实对象
*/
private Object subject;
/**
* 构造方法,给我们要代理的真实对象赋初值
*
* @param subject
*/
public MyInvocationHandlerMbatis(Object subject) {
this.subject = subject;
}
/**
* 该方法负责集中处理动态代理类上的所有方法调用。 调用处理器根据这三个参数进行预处理或分派到委托类实例上反射执行
*
* @param proxy
* 代理类实例
* @param method
* 被调用的方法对象
* @param args
* 调用参数
* @return
* @throws Throwable
*/
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
// 判断方法上是否有ExtInsert注解
ExtInsert extInsert = method.getAnnotation(ExtInsert.class);
if (extInsert != null) {
return insertSQL(extInsert, method, args);
}
// 判断方法上注解类型
ExtSelect extSelect = method.getAnnotation(ExtSelect.class);
if (extSelect != null) {
return selectMybatis(extSelect, method, args);
}
return null;
}
public int insertSQL(ExtInsert extInsert, Method method, Object[] args) {
// 获取注解上的sql
String insertSql = extInsert.value();
System.out.println("sql:" + insertSql);
// 获取方法上的参数
Parameter[] parameters = method.getParameters();
// 将方法上的参数存放在Map集合中
ConcurrentHashMap parameterMap = getExtParams(parameters, args);
// 获取SQL语句上需要传递的参数
String[] sqlParameter = SQLUtils.sqlInsertParameter(insertSql);
List parameValues = new ArrayList<>();
for (int i = 0; i < sqlParameter.length; i++) {
String str = sqlParameter[i];
Object object = parameterMap.get(str);
parameValues.add(object);
}
// 将SQL语句替换为?号
String newSql = SQLUtils.parameQuestion(insertSql, sqlParameter);
System.out.println("newSql:" + newSql);
// 调用jdbc代码执行
int insertResult = JDBCUtils.insert(newSql, false, parameValues);
return insertResult;
}
public Object selectMybatis(ExtSelect extInsert, Method method, Object[] args) throws SQLException {
try {
// 获取查询SQL语句
String selectSQL = extInsert.value();
// 将方法上的参数存放在Map集合中
Parameter[] parameters = method.getParameters();
// 获取方法上参数集合
ConcurrentHashMap parameterMap = getExtParams(parameters, args);
// 获取SQL传递参数
List sqlSelectParameter = SQLUtils.sqlSelectParameter(selectSQL);
// 排序参数
List parameValues = new ArrayList<>();
for (int i = 0; i < sqlSelectParameter.size(); i++) {
String parameterName = sqlSelectParameter.get(i);
Object object = parameterMap.get(parameterName);
parameValues.add(object.toString());
}
// 变为?号
String newSql = SQLUtils.parameQuestion(selectSQL, sqlSelectParameter);
System.out.println("执行SQL:" + newSql + "参数信息:" + parameValues.toString());
// 调用JDBC代码查询
ResultSet rs = JDBCUtils.query(newSql, parameValues);
// 获取返回类型
Class> returnType = method.getReturnType();
if (!rs.next()) {
// 没有查找数据
return null;
}
// 向上移动
rs.previous();
// 实例化对象
Object newInstance = returnType.newInstance();
while (rs.next()) {
for (String parameterName : sqlSelectParameter) {
// 获取集合中数据
Object value = rs.getObject(parameterName);
// 查找对应属性
Field field = returnType.getDeclaredField(parameterName);
// 设置允许私有访问
field.setAccessible(true);
// 赋值参数
field.set(newInstance, value);
}
}
return newInstance;
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
private ConcurrentHashMap getExtParams(Parameter[] parameters, Object[] args) {
// 获取方法上参数集合
ConcurrentHashMap parameterMap = new ConcurrentHashMap<>();
for (int i = 0; i < parameters.length; i++) {
// 参数信息
Parameter parameter = parameters[i];
ExtParam extParam = parameter.getDeclaredAnnotation(ExtParam.class);
// 参数名称
String paramValue = extParam.value();
// 参数值
Object oj = args[i];
parameterMap.put(paramValue, oj);
}
return parameterMap;
}
}
3、定义SqlSession
/**
* 获取SqlSession对象
*/
public class SqlSession {
// 获取getMapper
public static T getMapper(Class clas)
throws IllegalArgumentException, InstantiationException, IllegalAccessException {
return (T) Proxy.newProxyInstance(clas.getClassLoader(), new Class[] { clas },
new MyInvocationHandlerMbatis(clas));
}
}
4、运行
UserDao mapper = SqlSession.getMapper(UserDao.class);
int insertUser = mapper.insertUser(18, "马云");
System.out.println("影响行数:" + insertUser);
User user = mapper.selectUser("Xiaoxin", 21);
System.out.println("查询结果:" + user.getUserName() + "," + user.getUserAge());