Thanks to visit codestin.com
Credit goes to github.com

Skip to content

求助解惑 mybatis 拦截器 #37

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wwaassded wants to merge 1 commit into
base: master
Choose a base branch
from
Open

求助解惑 mybatis 拦截器 #37

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
43 changes: 29 additions & 14 deletions ...e/src/main/java/com/mall4j/cloud/common/database/interceptor/GeneratedKeyInterceptor.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -78,37 +78,52 @@ public Object intercept(Invocation invocation) throws Throwable {
// 找数据库中的对象
Object dbObject = findDbObject(parameter);

//这里如果批量插入只有list传递参数dbObject为null不会提前退出吗 list中的实体类的id还没有生成
if (dbObject == null) {
return invocation.proceed();
}

// 插入
if (mappedStatement.getId().contains(INSERT) || mappedStatement.getId().contains(SAVE)){
generatedKey(dbObject);
if (mappedStatement.getId().contains(INSERT) || mappedStatement.getId().contains(SAVE)) {
if (dbObject instanceof BaseModel) {
generatedKey(dbObject);
}
}
// 批量插入
else if (mappedStatement.getId().contains(BATCH_INSERT) || mappedStatement.getId().contains(BATCH_SAVE)){
// 获取批量查询的参数并生成主键
if (parameter instanceof HashMap){
Object list = ((Map)parameter).get("list");
if (list instanceof ArrayList) {
for (Object o : (ArrayList) list) {
generatedKey(dbObject);
else if (mappedStatement.getId().contains(BATCH_INSERT) || mappedStatement.getId().contains(BATCH_SAVE)) {
if (parameter instanceof HashMap) {
//insert函数可能通过@Param指定不同的名字 通过“list” 可能会找不到list
//可以考虑通过类型来判断获取list类型
// if (list instanceof ArrayList) {
// for (Object o : (ArrayList) list) {
// generatedKey(dbObject); //为什么要对dbObject调用generatedKey呢 不应该对list中的每一个baseModel调用吗
// }
// }
//dbObject不仅可能是baseModel类也可能是List
if (dbObject instanceof ArrayList<?> arrayList) {
if (!arrayList.isEmpty()) {
Object firstObject = arrayList.get(0);
if (findDbObject(firstObject) != null) {
for (Object o : arrayList) {
generatedKey(o);
}
}
}
}
}
}

return invocation.proceed();
}

protected BaseModel findDbObject(Object parameterObj) {
//不止返回baseModel 将可能的list也进行返回而不是通过“list”名称查找map中的潜在list类型参数
//TODO 但是如果存在多个list如何解决 但是远来的代码中也存在这个问题只能找到map中的一个baseModel类型 可能是一个无需考虑的问题
protected Object findDbObject(Object parameterObj) {
if (parameterObj instanceof BaseModel) {
return (BaseModel)parameterObj;
return parameterObj;
} else if (parameterObj instanceof Map) {
for (Object val : ((Map<?, ?>) parameterObj).values()) {
if (val instanceof BaseModel) {
return (BaseModel)val;
if (val instanceof BaseModel || val instanceof ArrayList<?>) {
return val;
}
}
}
Expand Down