xit-base
/
xit-foundation
17
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

PagingAid -> MapperSupport

Browse Source
master
mjkhan21 11 months ago
parent 936d9e68a3
commit 12fbd09de1
3 changed files with 110 additions and 217 deletions
Show Stats Download Patch File Download Diff File

110
src/main/java/cokr/xit/foundation/data/paging/MapperSupport.java
Unescape Escape View File

@ -0,0 +1,110 @@
package cokr.xit.foundation.data.paging;
import java.sql.SQLException;
import java.sql.Statement;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
import java.util.Map;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.executor.resultset.ResultSetHandler;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlCommandType;
import org.apache.ibatis.plugin.Intercepts;
import org.apache.ibatis.plugin.Signature;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;
import cokr.xit.foundation.AbstractEntity;
@Intercepts({
@Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class}),
@Signature(type = ResultSetHandler.class, method = "handleResultSets", args = {Statement.class}),
@Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class}),
})
public class MapperSupport extends MybatisPlugin {
private static final ThreadLocal<Pageable.Info> pagingInfo = new ThreadLocal<>();
@Override
protected Object query(Executor executor, MappedStatement mappedStatement, Object obj, RowBounds rowBounds, ResultHandler<?> resultHandler) throws SQLException {
Pageable.Info paging = Pageable.Info.create(obj);
if (paging != null && paging.isPaging())
pagingInfo.set(paging);
return super.query(executor, mappedStatement, obj, rowBounds, resultHandler);
}
@Override
protected Object handle(ResultSetHandler resultSetHandler, Statement statement) throws SQLException {
Object obj = super.handle(resultSetHandler, statement);
Pageable.Info paging = pagingInfo.get();
if (paging != null && (obj instanceof List)) {
pagingInfo.remove();
List<?> list = (List<?>)obj;
BoundedList<Object> boundedList = new BoundedList<>();
boundedList.setFetchSize(paging.fetchSize);
boundedList.addAll(list);
if (!boundedList.isEmpty()) {
boundedList.setStart((paging.pageNum - 1) * paging.fetchSize);
}
obj = boundedList;
}
return obj;
}
@Override
protected Object update(Executor executor, MappedStatement mappedStatement, Object obj) throws SQLException {
setEntityTimestamp(mappedStatement.getSqlCommandType(), obj, null);
return super.update(executor, mappedStatement, obj);
}
private static final SimpleDateFormat dateFormat = new SimpleDateFormat("yyyyMMddHHmmss");
private void setEntityTimestamp(SqlCommandType sqlType, Object arg, String now) {
if (isEmpty(arg)) return;
now = ifEmpty(now, () -> dateFormat.format(new Date()));
if (arg instanceof AbstractEntity) {
setTimestamp(sqlType, (AbstractEntity)arg, now);
} else if (arg instanceof Map<?, ?>) {
Map<?, ?> map = (Map<?, ?>)arg;
for (Object o: map.values())
setEntityTimestamp(sqlType, o, now);
} else if (arg instanceof Iterable<?>) {
Iterable<?> iterable = (Iterable<?>)arg;
for (Object o: iterable)
setEntityTimestamp(sqlType, o, now);
} else if (arg.getClass().isArray()) {
Object[] objs = (Object[])arg;
for (Object o: objs)
setEntityTimestamp(sqlType, o, now);
}
}
private void setTimestamp(SqlCommandType sqlType, AbstractEntity entity, String now) {
String userID = currentUser().getId();
switch (sqlType) {
case INSERT:
entity.setCreatedBy(userID);
entity.setCreatedAt(now);
entity.setModifiedBy(userID);
entity.setLastModified(now);
break;
case UPDATE:
entity.setModifiedBy(userID);
entity.setLastModified(now);
break;
case DELETE:
entity.setRemovedBy(userID);
entity.setRemovedAt(now);
entity.setUseYN("N");
break;
default: break;
}
}
}

51
src/main/java/cokr/xit/foundation/data/paging/PagingAid.java
Unescape Escape View File

@ -1,51 +0,0 @@
package cokr.xit.foundation.data.paging;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.List;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.executor.resultset.ResultSetHandler;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.plugin.Intercepts;
import org.apache.ibatis.plugin.Signature;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;
@Intercepts({
@Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class}),
@Signature(type = ResultSetHandler.class, method = "handleResultSets", args = {Statement.class})
})
public class PagingAid extends MybatisPlugin {
private static final ThreadLocal<Pageable.Info> pagingInfo = new ThreadLocal<>();
@Override
protected Object query(Executor executor, MappedStatement mappedStatement, Object obj, RowBounds rowBounds, ResultHandler<?> resultHandler) throws SQLException {
Pageable.Info paging = Pageable.Info.create(obj);
if (paging != null && paging.isPaging())
pagingInfo.set(paging);
return super.query(executor, mappedStatement, obj, rowBounds, resultHandler);
}
@Override
protected Object handle(ResultSetHandler resultSetHandler, Statement statement) throws SQLException {
Object obj = super.handle(resultSetHandler, statement);
Pageable.Info paging = pagingInfo.get();
if (paging != null && (obj instanceof List)) {
pagingInfo.remove();
List<?> list = (List<?>)obj;
BoundedList<Object> boundedList = new BoundedList<>();
boundedList.setFetchSize(paging.fetchSize);
boundedList.addAll(list);
if (!boundedList.isEmpty()) {
boundedList.setStart((paging.pageNum - 1) * paging.fetchSize);
}
obj = boundedList;
}
return obj;
}
}

166
src/main/java/cokr/xit/foundation/data/paging/PagingSupport.java
Unescape Escape View File

@ -1,166 +0,0 @@
package cokr.xit.foundation.data.paging;
import java.sql.Statement;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
import java.util.Map;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.executor.resultset.ResultSetHandler;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlCommandType;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.plugin.Intercepts;
import org.apache.ibatis.plugin.Invocation;
import org.apache.ibatis.plugin.Signature;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;
import cokr.xit.foundation.AbstractComponent;
import cokr.xit.foundation.AbstractEntity;
import cokr.xit.foundation.component.QueryRequest;
/**조회 결과의 페이징 처리를 지원하는 MyBatis Interceptor
* @author mjkhan
*/
@Intercepts({
@Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class}),
@Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class}),
@Signature(type = ResultSetHandler.class, method = "handleResultSets", args = {Statement.class})
})
public class PagingSupport extends AbstractComponent implements Interceptor {
private boolean paging;
private int pageNum;
private int fetchSize;
@Override
public Object intercept(Invocation invocation) throws Throwable {
Object[] args = invocation.getArgs();
Object arg0 = args[0];
if (arg0 instanceof MappedStatement) {
MappedStatement stmt = (MappedStatement)arg0;
SqlCommandType sqlType = stmt.getSqlCommandType();
switch (sqlType) {
case SELECT: return setPaging(invocation);
case INSERT:
case UPDATE:
case DELETE: return setEntityProperties(invocation, sqlType);
default: return invocation.proceed();
}
} else
switch (getStep(invocation)) {
case "ResultSetHandler.handleResultSets": return setPagingInfo(invocation);
default: return invocation.proceed();
}
}
private static final SimpleDateFormat dateFormat = new SimpleDateFormat("yyyyMMddHHmmss");
private Object setEntityProperties(Invocation invocation, SqlCommandType sqlType) throws Exception {
Object[] args = invocation.getArgs();
Object arg = args[1];
setEntityProperties(sqlType, arg, null);
return invocation.proceed();
}
private void setEntityProperties(SqlCommandType sqlType, Object obj, String now) {
if (isEmpty(obj)) return;
now = ifEmpty(now, () -> dateFormat.format(new Date()));
if (obj instanceof AbstractEntity) {
setProperties(sqlType, (AbstractEntity)obj, now);
} else if (obj instanceof Map<?, ?>) {
Map<?, ?> map = (Map<?, ?>)obj;
for (Object o: map.values())
setEntityProperties(sqlType, o, now);
} else if (obj instanceof Iterable<?>) {
Iterable<?> iterable = (Iterable<?>)obj;
for (Object o: iterable)
setEntityProperties(sqlType, o, now);
} else if (obj.getClass().isArray()) {
Object[] objs = (Object[])obj;
for (Object o: objs)
setEntityProperties(sqlType, o, now);
}
}
private void setProperties(SqlCommandType sqlType, AbstractEntity entity, String now) {
String userID = currentUser().getId();
switch (sqlType) {
case INSERT:
entity.setCreatedBy(userID);
entity.setCreatedAt(now);
entity.setModifiedBy(userID);
entity.setLastModified(now);
break;
case UPDATE:
entity.setModifiedBy(userID);
entity.setLastModified(now);
break;
case DELETE:
entity.setRemovedBy(userID);
entity.setRemovedAt(now);
entity.setUseYN("N");
break;
default: break;
}
}
private static final Class<?>[] handlers = {Executor.class, ResultSetHandler.class};
private String getStep(Invocation invocation) {
Object obj = invocation.getTarget();
String result = "";
for (Class<?> klass: handlers) {
if (klass.isInstance(obj)) {
result = klass.getSimpleName();
break;
}
}
return result += "." + invocation.getMethod().getName();
}
private Object setPaging(Invocation invocation) throws Exception {
Object[] args = invocation.getArgs();
Object arg = args[1];
if (arg instanceof Map) {
@SuppressWarnings("unchecked")
Map<String, ?> params = (Map<String, ?>)arg;
fetchSize = Pageable.getFetchSize(params);
pageNum = Pageable.getPageNum(params);
} else if (arg instanceof QueryRequest) {
QueryRequest req = (QueryRequest)arg;
fetchSize = req.getFetchSize();
pageNum = req.getPageNum();
} else if (arg instanceof Pageable) {
Pageable pageable = (Pageable)arg;
fetchSize = pageable.getFetchSize();
pageNum = pageable.getPageNum();
}
paging = pageNum > 0 && fetchSize > 0;
return invocation.proceed();
}
private Object setPagingInfo(Invocation invocation) throws Exception {
Object obj = invocation.proceed();
if (paging && (obj instanceof List)) {
List<?> list = (List<?>)obj;
BoundedList<Object> boundedList = new BoundedList<>();
boundedList.setFetchSize(fetchSize);
boundedList.addAll(list);
if (!boundedList.isEmpty()) {
boundedList.setStart((pageNum - 1) * fetchSize);
}
obj = boundedList;
}
paging = false;
pageNum = fetchSize = 0;
return obj;
}
}
Write Preview
Loading…
Cancel
Save