wuxw
2019-03-30 28c643cc4d3142503dfa5234777399586e8364e9
java110-service/src/main/java/com/java110/service/aop/PageProcessAspect.java
@@ -56,6 +56,7 @@
        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));
@@ -99,16 +100,22 @@
        String url = request.getRequestURL()!=null?request.getRequestURL().toString():"";
        String componentCode = "";
        String componentMethod = "";
        if(url.contains("callComponent")){
        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);
         pd = PageData.newInstance().builder(userId,this.getToken(request),reqData,componentCode,componentMethod,url,sessionId);
        request.setAttribute(CommonConstant.CONTEXT_PAGE_DATA,pd);
    }
@@ -136,14 +143,8 @@
            return ;
        }
        if(!StringUtil.isNullOrNone(pd.getToken())) {
            HttpServletResponse response = attributes.getResponse();
            Cookie cookie = new Cookie(CommonConstant.COOKIE_AUTH_TOKEN, pd.getToken());
            cookie.setHttpOnly(true);
            cookie.setPath("/");
            response.addCookie(cookie);
            response.flushBuffer();
        }
        //写cookies信息
        writeCookieInfo(pd,attributes);
    }
@@ -177,4 +178,24 @@
        }
        return token;
    }
    /**
     * 写cookie 信息
     * @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();
        }
    }
}