From f5a10d5ae1290d8ceefee143815555d59be6ff9a Mon Sep 17 00:00:00 2001 From: Administrator <zhubaomin> Date: 星期四, 23 五月 2024 08:50:58 +0800 Subject: [PATCH] Merge branch 'master' of http://8.140.179.55:20000/r/pipIrr-SV --- pipIrr-platform/pipIrr-common/src/main/java/com/dy/common/mybatis/AutoGenerateIdInterceptor.java | 28 ++++++++++++++++++++-------- 1 files changed, 20 insertions(+), 8 deletions(-) diff --git a/pipIrr-platform/pipIrr-common/src/main/java/com/dy/common/mybatis/AutoGenerateIdInterceptor.java b/pipIrr-platform/pipIrr-common/src/main/java/com/dy/common/mybatis/AutoGenerateIdInterceptor.java index 6265e27..f518814 100644 --- a/pipIrr-platform/pipIrr-common/src/main/java/com/dy/common/mybatis/AutoGenerateIdInterceptor.java +++ b/pipIrr-platform/pipIrr-common/src/main/java/com/dy/common/mybatis/AutoGenerateIdInterceptor.java @@ -25,8 +25,9 @@ static int MAPPED_STATEMENT_INDEX = 0; static int PARAMETER_INDEX = 1; - static String BASE_FIELD_SET_PRIMARY_KEY_FUNTION_NAME = "setId"; - static String BASE_FIELD_SET_PRIMARY_KEY_FUNTION_NAME2 = "setUserId"; + static String BASE_FIELD_SET_PRIMARY_KEY_FUNTION_SETID = "setId"; + static String BASE_FIELD_SET_PRIMARY_KEY_FUNTION_SETUSERID = "setUserId"; + static String BASE_FIELD_SET_PRIMARY_KEY_FUNTION_SETROLEID = "setRoleId"; /** * 鎷︽埅閫昏緫瀹炵幇 @@ -42,24 +43,35 @@ if (entity instanceof BaseEntity) { //Class<? extends Object> entityClass = entity.getClass(); Class<?> entityClass = entity.getClass(); - Method setMt = null ; + Method setId = null ; + Method setUserId = null ; + Method setRoleId = null ; try{ + /* // 娣诲姞鐢ㄦ埛鏃秛serId涓鸿嚜鍔ㄧ敓鎴愪富閿�俽oleId瀛樺湪鏃舵槸涓虹敤鎴风粦瑙掕壊 String jsonString = JSONObject.toJSONString(entity, JSONWriter.Feature.WriteMapNullValue); JSONObject jsonObject = JSONObject.parseObject(jsonString); if(jsonObject.containsKey("userId") && !jsonObject.containsKey("roleId")) { - setMt = entityClass.getMethod(BASE_FIELD_SET_PRIMARY_KEY_FUNTION_NAME2, Long.class) ; + setUserId = entityClass.getMethod(BASE_FIELD_SET_PRIMARY_KEY_FUNTION_SETUSERID, Long.class) ; + setRoleId = entityClass.getMethod(BASE_FIELD_SET_PRIMARY_KEY_FUNTION_SETROLEID, Long.class) ; }else { //鏈変竴浜涘疄浣撴病鏈塱d锛屼緥濡備腑闂磋〃 - setMt = entityClass.getMethod(BASE_FIELD_SET_PRIMARY_KEY_FUNTION_NAME, Long.class) ; + setId = entityClass.getMethod(BASE_FIELD_SET_PRIMARY_KEY_FUNTION_SETID, Long.class) ; } + */ + //鏈変竴浜涘疄浣撴病鏈塱d锛屼緥濡備腑闂磋〃 + setId = entityClass.getMethod(BASE_FIELD_SET_PRIMARY_KEY_FUNTION_SETID, Long.class) ; + // BaUser瀹炰綋涓槸setUserId, BaUserRole瀹炰綋涓寘鍚玸etUserId,鍜宻etRoleId + setUserId = entityClass.getMethod(BASE_FIELD_SET_PRIMARY_KEY_FUNTION_SETUSERID, Long.class) ; + setRoleId = entityClass.getMethod(BASE_FIELD_SET_PRIMARY_KEY_FUNTION_SETROLEID, Long.class) ; }catch (Exception e){ //褰揺ntityClass娌℃湁setId鏂规硶鏃讹紝浼氭姏鍑哄紓甯� } - if(setMt != null){ - setMt.invoke(entity, new IDLongGenerator().generate()); + if(setUserId != null && setRoleId == null){ + setUserId.invoke(entity, new IDLongGenerator().generate()); + }else if(setId != null){ + setId.invoke(entity, new IDLongGenerator().generate()); } - invocation.getArgs()[PARAMETER_INDEX] = entity; } } -- Gitblit v1.8.0