package cn.autoform.web.interceptor.spring;
|
|
import cn.autoform.bean.system.UserInfo;
|
import cn.autoform.util.Des1;
|
import cn.autoform.util.thread.Keys;
|
import cn.autoform.util.thread.ThreadData;
|
import cn.autoform.util.tool.ObjectUtil;
|
import cn.autoform.web.interceptor.filter.CloneRequestFilter;
|
import cn.autoform.web.service.login.LoginService;
|
import com.codingapi.tx.aop.bean.TxCompensateLocal;
|
import com.codingapi.tx.aop.bean.TxTransactionLocal;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Service;
|
import org.springframework.web.servlet.HandlerInterceptor;
|
import org.springframework.web.servlet.ModelAndView;
|
|
import javax.annotation.Resource;
|
import javax.servlet.http.Cookie;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.net.URLDecoder;
|
import java.util.*;
|
|
/**
|
* 初始化线程参数
|
* @author WangYX
|
*
|
*/
|
@Service("initParamInterceptor")
|
@Slf4j
|
public class InitParamInterceptor implements HandlerInterceptor{
|
|
/**
|
* 初始化参数key
|
*/
|
private static final List<Keys> PARAM_KEY = new ArrayList<>();
|
|
/**
|
* 访问对外接口参数
|
*/
|
public static final Map<String,Keys> COOKIE_KEY = new HashMap<>();
|
|
@Resource
|
private LoginService loginService;
|
|
static{
|
// 表单id
|
PARAM_KEY.add(Keys.FORM_ID);
|
// 租户id
|
PARAM_KEY.add(Keys.TENANT_ID);
|
// 页码
|
PARAM_KEY.add(Keys.PAGE_NUM);
|
// 页条数
|
PARAM_KEY.add(Keys.PAGE_SIZE);
|
// 数据存储标识
|
PARAM_KEY.add(Keys.IO_FLAG);
|
// tenantId
|
COOKIE_KEY.put("tenantId",Keys.TENANT_ID);
|
// userName
|
COOKIE_KEY.put("userName",Keys.USER_NAME);
|
// openId
|
COOKIE_KEY.put("openId",Keys.OPEN_ID);
|
// comanyId
|
COOKIE_KEY.put("companyId",Keys.COMPANY_ID);
|
// serviceId
|
COOKIE_KEY.put("serviceId",Keys.SERVICE_ID);
|
// token
|
COOKIE_KEY.put("token",Keys.TOKEN);
|
// dataSourceId
|
COOKIE_KEY.put("dataSourceId",Keys.DATA_SOURCE_ID);
|
// 国际化
|
COOKIE_KEY.put("Language", Keys.USER_LANGUAGE);
|
|
}
|
|
@SuppressWarnings("unchecked")
|
@Override
|
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
|
final CloneRequestFilter.BodyReaderHttpServletRequestWrapper requestWrapper = new CloneRequestFilter.BodyReaderHttpServletRequestWrapper(request);
|
Map<String,String[]> paramMap = requestWrapper.getParameterMap();
|
PARAM_KEY.stream().forEach(key->{
|
String[] val = paramMap.get(key.getKey());
|
if(ObjectUtil.empty(val) || val.length == 0)
|
return;
|
ThreadData.set(key
|
, key.format(val[0]));
|
});
|
//取出cookie添加到线程中
|
Cookie[] cookie = request.getCookies();
|
if(cookie!=null){
|
for(Cookie cook : cookie){
|
if(!ObjectUtil.empty(COOKIE_KEY.get(cook.getName()))){
|
String str = URLDecoder.decode(cook.getValue(),"UTF-8").replaceAll(" ","+");
|
ThreadData.set(COOKIE_KEY.get(cook.getName()),Des1.desDecrypt(str,Des1.SECRETKEY));
|
if(Objects.equals("Language", cook.getName())){
|
com.iemsoft.framework.cloud.core.thread.ThreadData.set(
|
com.iemsoft.framework.cloud.core.thread.Keys.USER_LANGUAGE, Des1.desDecrypt(str,Des1.SECRETKEY)
|
);
|
}
|
}
|
}
|
}
|
loginService.getUserInfo(new UserInfo.LogInInfo(ThreadData.get(Keys.TOKEN),ThreadData.get(Keys.OPEN_ID)));
|
return true;
|
}
|
|
@Override
|
public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler,
|
ModelAndView modelAndView) throws Exception {
|
// 清空线程参数
|
TxCompensateLocal.clear();
|
TxTransactionLocal.clear();
|
ThreadData.clear();
|
com.iemsoft.framework.cloud.core.thread.ThreadData.clear();
|
}
|
|
@Override
|
public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex)
|
throws Exception {
|
// 清空线程参数
|
TxCompensateLocal.clear();
|
TxTransactionLocal.clear();
|
ThreadData.clear();
|
com.iemsoft.framework.cloud.core.thread.ThreadData.clear();
|
}
|
|
}
|
