From 28c643cc4d3142503dfa5234777399586e8364e9 Mon Sep 17 00:00:00 2001
From: wuxw <928255095@qq.com>
Date: 星期六, 30 三月 2019 11:41:36 +0800
Subject: [PATCH] 添加员工服务端bug修复
---
java110-service/src/main/java/com/java110/service/aop/PageProcessAspect.java | 144 ++++++++++++++++++++++++++++++++++++++++++-----
1 files changed, 127 insertions(+), 17 deletions(-)
diff --git a/java110-service/src/main/java/com/java110/service/aop/PageProcessAspect.java b/java110-service/src/main/java/com/java110/service/aop/PageProcessAspect.java
index eac4f31..241eed0 100644
--- a/java110-service/src/main/java/com/java110/service/aop/PageProcessAspect.java
+++ b/java110-service/src/main/java/com/java110/service/aop/PageProcessAspect.java
@@ -1,19 +1,32 @@
package com.java110.service.aop;
-import com.java110.common.factory.PageDataFactory;
+import com.alibaba.fastjson.JSONObject;
+import com.java110.common.constant.CommonConstant;
+import com.java110.common.constant.ResponseConstant;
+import com.java110.common.exception.FilterException;
+import com.java110.core.context.IPageData;
+import com.java110.core.factory.PageDataFactory;
import com.java110.common.util.Assert;
-import com.java110.common.util.SequenceUtil;
-import com.java110.entity.service.PageData;
+import com.java110.core.factory.GenerateCodeFactory;
+import com.java110.common.util.StringUtil;
+import com.java110.core.context.PageData;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.*;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
+import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStream;
import java.io.InputStreamReader;
+import java.util.Map;
/**
* 鏁版嵁鍒濆鍖�
@@ -22,7 +35,10 @@
@Aspect
@Component
public class PageProcessAspect {
- @Pointcut("execution(public * com.java110..*.*Controller.*(..))")
+
+ private final static Logger logger = LoggerFactory.getLogger(PageProcessAspect.class);
+
+ @Pointcut("execution(public * com.java110..*.*Controller.*(..)) || execution(public * com.java110..*.*Rest.*(..))")
public void dataProcess(){}
/**
@@ -37,20 +53,70 @@
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
HttpServletRequest request = attributes.getRequest();
- if("POST".equals(request.getMethod())){
- BufferedReader reader = new BufferedReader(new InputStreamReader(request.getInputStream()));
+ IPageData pd = null;
+ String reqData = "";
+ String userId = "";
+ String sessionId = request.getSession().getId();
+ if("POST,PUT".contains(request.getMethod())){
+ InputStream in = request.getInputStream();
+ BufferedReader reader = new BufferedReader(new InputStreamReader(in));
//reader.
StringBuffer sb = new StringBuffer();
String str = "";
while ((str = reader.readLine()) != null) {
sb.append(str);
}
- str = sb.toString();
- if(Assert.isPageJsonObject(str)){
- PageData pd = PageDataFactory.newInstance().builder(str).setTransactionId(SequenceUtil.getPageTransactionId());
- request.setAttribute("pd",pd);
+ reqData = sb.toString();
+
+ }
+ //瀵� get鎯呭喌涓嬬殑鍙傛暟杩涜灏佽
+ else{
+ Map<String,String[]> params = request.getParameterMap();
+ if(params != null && !params.isEmpty()) {
+ JSONObject paramObj = new JSONObject();
+ for(String key : params.keySet()) {
+ if(params.get(key).length>0){
+ String value = "";
+ for(int paramIndex = 0 ; paramIndex < params.get(key).length;paramIndex++) {
+ value = params.get(key)[paramIndex] + ",";
+ }
+ value = value.endsWith(",")?value.substring(0,value.length()-1):value;
+ paramObj.put(key,value);
+ }
+ continue;
+ }
+ reqData = paramObj.toJSONString();
}
}
+ // 鑾峰彇 userId
+ if(request.getAttribute("claims") != null && request.getAttribute("claims") instanceof Map){
+ Map<String,String> userInfo = (Map<String,String>)request.getAttribute("claims");
+ if(userInfo.containsKey(CommonConstant.LOGIN_USER_ID)){
+ userId = userInfo.get(CommonConstant.LOGIN_USER_ID);
+ }
+ }
+
+ // 鑾峰彇缁勪欢鍚嶇О 鍜屾柟娉曞悕绉�
+ String url = request.getRequestURL()!=null?request.getRequestURL().toString():"";
+ String componentCode = "";
+ String componentMethod = "";
+ if(url.contains("callComponent")){ //缁勪欢澶勭悊
+ String []urls = url.split("/");
+
+ if(urls.length == 6){
+ componentCode = urls[4];
+ componentMethod = urls[5];
+ }
+ }else if(url.contains("flow")){ //娴佺▼澶勭悊
+ String []urls = url.split("/");
+
+ if(urls.length == 5){
+ componentCode = urls[4];
+ }
+ }
+
+ pd = PageData.newInstance().builder(userId,this.getToken(request),reqData,componentCode,componentMethod,url,sessionId);
+ request.setAttribute(CommonConstant.CONTEXT_PAGE_DATA,pd);
}
@@ -61,25 +127,30 @@
//鍚庣疆寮傚父閫氱煡
@AfterThrowing("dataProcess()")
- public void throwss(JoinPoint jp){
+ public void throwException(JoinPoint jp){
}
//鍚庣疆鏈�缁堥�氱煡,final澧炲己锛屼笉绠℃槸鎶涘嚭寮傚父鎴栬�呮甯搁��鍑洪兘浼氭墽琛�
@After("dataProcess()")
- public void after(JoinPoint jp){
+ public void after(JoinPoint jp) throws IOException {
// 鎺ユ敹鍒拌姹傦紝璁板綍璇锋眰鍐呭
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
- HttpServletRequest request = attributes.getRequest();
- PageData pd =request.getAttribute("pd") != null ?(PageData)request.getAttribute("pd"):null ;
- //淇濆瓨鏃ュ織澶勭悊
- if(pd != null){
+ HttpServletRequest request = attributes.getRequest();
+ PageData pd =request.getAttribute(CommonConstant.CONTEXT_PAGE_DATA) != null ?(PageData)request.getAttribute(CommonConstant.CONTEXT_PAGE_DATA):null ;
+ //淇濆瓨鏃ュ織澶勭悊
+ if(pd == null){
+ return ;
}
+
+ //鍐檆ookies淇℃伅
+ writeCookieInfo(pd,attributes);
+
}
//鐜粫閫氱煡,鐜粫澧炲己锛岀浉褰撲簬MethodInterceptor
@Around("dataProcess()")
- public Object arround(ProceedingJoinPoint pjp) {
+ public Object around(ProceedingJoinPoint pjp) {
try {
Object o = pjp.proceed();
return o;
@@ -88,4 +159,43 @@
return null;
}
}
+
+
+ /**
+ * 鑾峰彇TOKEN
+ * @param request
+ * @return
+ */
+ private String getToken(HttpServletRequest request) throws FilterException {
+ String token = "";
+ if(request.getCookies() == null || request.getCookies().length == 0){
+ return token;
+ }
+ for (Cookie cookie : request.getCookies()) {
+ if (CommonConstant.COOKIE_AUTH_TOKEN.equals(cookie.getName())) {
+ token = cookie.getValue();
+ }
+ }
+ return token;
+ }
+
+
+ /**
+ * 鍐檆ookie 淇℃伅
+ * @param pd 椤甸潰灏佽淇℃伅
+ * @param attributes
+ * @throws IOException
+ */
+ private void writeCookieInfo(IPageData pd,ServletRequestAttributes attributes) throws IOException {
+ // 杩欓噷鐩墠鍙啓鍒扮粍浠剁骇鍒紝濡傛灉闇�瑕� 鍐欐垚鏂规硶绾у埆
+ if(!StringUtil.isNullOrNone(pd.getToken()) && "login".equals(pd.getComponentCode())) {
+ HttpServletResponse response = attributes.getResponse();
+ Cookie cookie = new Cookie(CommonConstant.COOKIE_AUTH_TOKEN, pd.getToken());
+ cookie.setHttpOnly(true);
+ cookie.setPath("/");
+ response.addCookie(cookie);
+ response.flushBuffer();
+ }
+
+ }
}
--
Gitblit v1.8.0