基於springboot實現註解形式的mybatis
阿新 • • 發佈：2018-12-06
1.首先是用到的兩個工具類：
package com.wangyu.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; /** 宣告一個 Connection型別的靜態屬性，用來快取一個已經存在的連線物件 */ private static Connection conn; static { config(); } /** * 開頭配置自己的資料庫資訊 */ private static void config() { /* * 獲取驅動 */ driverClassName = "com.mysql.jdbc.Driver"; /* * 獲取URL */ URL = "jdbc:mysql://localhost:3306/rbac_db?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<Object> 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; /* 表示 執行 DDL 或 DML 操作是否成功的一個標識變數 */ /* 獲得 被執行的 SQL 語句的 字首 */ 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; } /** * * @param SQL 需要執行的 INSERT 語句 * * @param autoGeneratedKeys 指示是否需要返回由資料庫產生的鍵 * * @param params 將要執行的SQL語句中包含的引數佔位符的 引數值 * * @return 如果指定 autoGeneratedKeys 為 true 則返回由資料庫產生的鍵； 如果指定 autoGeneratedKeys * 為 false 則返回受當前SQL影響的記錄數目 */ public static int insert(String SQL, boolean autoGeneratedKeys, List<Object> 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(); } } } } } ///////////////////////////////////////////////////////////////////////////////////////// package com.wangyu.utils; import java.util.ArrayList; import java.util.List; /** * SQL拼接<br> * 作者: 每特教育-餘勝軍<br> * 聯絡方式:QQ644064779|WWW.itmayiedu.com<br> */ public class SQLUtils { /** * * 獲取Insert語句後面values 引數資訊<br> * 作者: 每特教育-餘勝軍<br> * 聯絡方式:QQ644064779|WWW.itmayiedu.com<br> * * @param sql * @return */ 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; } /** * * 獲取select 後面where語句 作者: 每特教育-餘勝軍<br> * 聯絡方式:QQ644064779|WWW.itmayiedu.com<br> * * @param sql * @return */ public static List<String> sqlSelectParameter(String sql) { int startIndex = sql.indexOf("where"); int endIndex = sql.length(); String substring = sql.substring(startIndex + 5, endIndex); String[] split = substring.split("and"); List<String> listArr = new ArrayList(); for (String string : split) { String[] sp2 = string.split("="); listArr.add(sp2[0].trim()); } return listArr; } /** * 將SQL語句的引數替換變為?<br> * 作者: 每特教育-餘勝軍<br> * 聯絡方式:QQ644064779|WWW.itmayiedu.com<br> * * @param sql * @param parameterName * @return */ public static String parameQuestion(String sql, String[] parameterName) { for (int i = 0; i < parameterName.length; i++) { String string = parameterName[i]; sql = sql.replace("#{" + string + "}", "?"); } return sql; } public static String parameQuestion(String sql, List<String> parameterName) { for (int i = 0; i < parameterName.size(); i++) { String string = parameterName.get(i); sql = sql.replace("#{" + string + "}", "?"); } return sql; } public static void main(String[] args) { // String[] sqlParameter = sqlInsertParameter(sql); // for (String string : sqlParameter) { // System.out.println(string); // } } }
2.自定義的註解和sqlsession
//新增 @Documented @Retention(RetentionPolicy.RUNTIME) @Target({ElementType.METHOD}) public @interface ExtInsert { String value(); } //引數 @Documented @Retention(RetentionPolicy.RUNTIME) @Target(ElementType.PARAMETER) public @interface ExtParam { String value(); } //查詢 @Documented @Retention(RetentionPolicy.RUNTIME) @Target({ElementType.METHOD}) public @interface ExtSelect { String value(); } /////////////////////////////////////////////////////////////////////////////////////// public class SqlSession { public static <T> T getMapper(Class classz) { return (T)Proxy.newProxyInstance(classz.getClassLoader(), new Class[]{classz}, new MyInvocationHandlerMbatis(classz)); } }
3.最重要的aop攔截
package com.wangyu.aop; import com.wangyu.annotation.ExtInsert; import com.wangyu.annotation.ExtParam; import com.wangyu.annotation.ExtSelect; import com.wangyu.utils.JDBCUtils; import com.wangyu.utils.SQLUtils; import org.apache.ibatis.binding.MapperMethod; import java.lang.reflect.*; import java.sql.ResultSet; import java.util.ArrayList; import java.util.List; import java.util.Set; import java.util.concurrent.ConcurrentHashMap; import static javax.xml.soap.SOAPFactory.newInstance; /** * 類說明此demo只是實現了新增和查詢，且查詢還需改造方可用於實戰（返回型別） * * @author wangyu * @version 1.0 * @date $2018/11/20$ $15:35$ */ public class MyInvocationHandlerMbatis implements InvocationHandler { public Object obj; public MyInvocationHandlerMbatis(Object obj) { this.obj = obj; } @Override public Object invoke(Object proxy, Method method, Object[] args) throws Throwable { System.out.println("使用動態代理攔截dao介面---"); ExtInsert extInsert = method.getDeclaredAnnotation(ExtInsert.class); //1.判斷方法是否加了insert註解 if (extInsert != null) { return extInsert(extInsert, method, args); } ExtSelect extSelect = method.getDeclaredAnnotation(ExtSelect.class); //2.查詢 if (extSelect != null) { //註解存在獲取sql String querySql = extSelect.value(); //獲取引數繫結 ConcurrentHashMap<Object, Object> paramMap = getConcurrentHashMap(method, args); List<String> sqlSelectParameter = SQLUtils.sqlSelectParameter(querySql); List sqlParam = new ArrayList(); for (String string : sqlSelectParameter) { Object paramValue = paramMap.get(string); sqlParam.add(paramValue); } String newSql = SQLUtils.parameQuestion(querySql, sqlSelectParameter); //呼叫jdbc ResultSet resultSet = JDBCUtils.query(newSql, sqlParam); if (!resultSet.next()) { return null; } resultSet.previous();//下標上移一位 Class<?> returnType = method.getReturnType(); Type type = method.getReturnType(); System.out.println("returnType:"+returnType); Object object = returnType.newInstance(); while (resultSet.next()) { //反射機制獲取例項化物件 Field[] declaredFields = returnType.getDeclaredFields(); for (Field field : declaredFields) { String fieldName = field.getName(); Object filedValue = resultSet.getObject(fieldName); field.setAccessible(true);//設定私有屬性可以訪問 field.set(object, filedValue); } /* for (String paramName : sqlSelectParameter) { Object resultValue = resultSet.getObject(paramName); Field field = returnType.getDeclaredField(paramName); field.setAccessible(true); field.set(object, resultValue); }*/ } return object; } return null; } //insert private Object extInsert(ExtInsert extInsert, Method method, Object[] args) { //2.存在註解，獲取sql 擷取sql String insertSql = extInsert.value(); System.out.println("insertSql:" + insertSql); //3.獲取引數 定義map集合接收引數 ConcurrentHashMap<Object, Object> paramMap = getConcurrentHashMap(method, args); //4.替換引數 ? String[] sqlInsertParameter = SQLUtils.sqlInsertParameter(insertSql); List<Object> sqlParams = getSqlParam(sqlInsertParameter, paramMap); String newSql = SQLUtils.parameQuestion(insertSql, sqlInsertParameter); System.out.println("拼接後的sql：" + newSql + ",引數有：" + sqlParams.toString()); //5.呼叫jdbc JDBCUtils.insert(newSql, false, sqlParams); return 1; } //獲取方法的引數 private ConcurrentHashMap<Object, Object> getConcurrentHashMap(Method method, Object[] args) { ConcurrentHashMap<Object, Object> paramMap = new ConcurrentHashMap<>(); Parameter[] parameters = method.getParameters(); for (int i = 0; i < parameters.length; i++) { Parameter parameter = parameters[i]; ExtParam extParam = parameter.getAnnotation(ExtParam.class); if (extParam != null) { String paramName = extParam.value(); Object paramValue = args[i]; System.out.println(paramName + "---" + paramValue); /* if(extParam.value()==parameter.getName()){}*/ paramMap.put(paramName, paramValue); } } return paramMap; } //替換引數 private List getSqlParam(String[] sqlInsertParameter, ConcurrentHashMap<Object, Object> paramMap) { List<Object> sqlParams = new ArrayList<>(); for (String paramName : sqlInsertParameter) { Object paramValue = paramMap.get(paramName); sqlParams.add(paramValue); } return sqlParams; } }
4.介面和實體類
public interface UserMapper {
/* @Insert(value = "insert into rbac_db.user (name, age)values(#{name}, #{age)")
void addUser(User user);*/
@ExtInsert(value = "insert into rbac_db.user (name, age)values(#{name},#{age})")
int addUser(@ExtParam("name") String name, @ExtParam("age") int age);
@ExtSelect("select * from rbac_db.user where name=#{name}")
User query(@ExtParam("name") String name);
}
//////////////////////////////////////////////////////////////////////
package com.wangyu.entity;
/**
* 類說明
*
* @author wangyu
* @version 1.0
* @date $2018/11/20$ $14:52$
*/
public class User {
private String name;
private int age;
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;
}
}
5.測試
package com.wangyu;
import com.wangyu.SqlSeeion.SqlSession;
import com.wangyu.aop.MyInvocationHandlerMbatis;
import com.wangyu.entity.User;
import com.wangyu.mapper.UserMapper;
import java.util.List;
/**
* 類說明
*
* @author wangyu
* @version 1.0
* @date $2018/11/20$ $15:10$
*/
public class Test001 {
public static void main(String[] args) {
UserMapper userMapper = SqlSession.getMapper(UserMapper.class);
// int addResult = userMapper.addUser("A2", 12);
User user = userMapper.query("項羽");
System.out.println(user.getAge() + "----" + user.getName());
//System.out.println(list);
}
}