package com.tievd.jyz.plugin;
|
|
import com.tievd.jyz.util.MultiMinioUtil;
|
import org.apache.ibatis.executor.resultset.ResultSetHandler;
|
import org.apache.ibatis.plugin.*;
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import java.lang.reflect.Field;
|
import java.sql.Statement;
|
import java.util.List;
|
|
/**
|
* Author: wqy
|
*/
|
@Intercepts({@Signature(
|
type= ResultSetHandler.class,
|
method = "handleResultSets",
|
args = {Statement.class})})
|
public class EventInterceptPlugin implements Interceptor {
|
|
@Autowired
|
MultiMinioUtil s3Util;
|
|
/**
|
* 全局s3路径转换
|
* @param invocation
|
* @return
|
* @throws Throwable
|
*/
|
@Override
|
public Object intercept(Invocation invocation) throws Throwable {
|
Object returnValue = invocation.proceed();
|
if (returnValue != null && returnValue instanceof List) {
|
try {
|
List dataList = (List) returnValue;
|
if (dataList.size() == 0) {
|
return returnValue;
|
}
|
Class clazz = dataList.get(0).getClass();
|
if (clazz.getAnnotation(S3DataParse.class) == null) {
|
return returnValue;
|
}
|
Field[] fields = clazz.getDeclaredFields();
|
for (Field field : fields) {
|
if (field.getAnnotation(S3DataParse.class) == null) {
|
continue;
|
}
|
field.setAccessible(true);
|
String split = field.getAnnotation(S3DataParse.class).value();
|
for (Object o : dataList) {
|
Object val = field.get(o);
|
String path[] = String.valueOf(val).split(split);
|
if (path.length == 2) {
|
String url = s3Util.getObjectURL(path[0], path[1], 6000);
|
field.set(o, url);
|
}
|
}
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
}
|
return returnValue;
|
}
|
|
@Override
|
public Object plugin(Object target) {
|
if (target instanceof ResultSetHandler) {
|
return Plugin.wrap(target, this);
|
}
|
return target;
|
}
|
}
|
