package com.cloud.attendance.filter;
|
|
import com.alibaba.fastjson.JSONObject;
|
import com.cloud.attendance.utils.EnumStr;
|
import com.cloud.model.common.Result;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.io.IOUtils;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.http.*;
|
import org.springframework.stereotype.Component;
|
import org.springframework.util.LinkedMultiValueMap;
|
import org.springframework.util.MultiValueMap;
|
import org.springframework.web.client.HttpClientErrorException;
|
import org.springframework.web.client.HttpServerErrorException;
|
import org.springframework.web.client.RestTemplate;
|
|
import javax.servlet.*;
|
import javax.servlet.annotation.WebFilter;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.IOException;
|
import java.io.PrintWriter;
|
import java.util.*;
|
import java.util.regex.Matcher;
|
import java.util.regex.Pattern;
|
|
@Slf4j
|
@Component
|
@WebFilter(urlPatterns = "/*", filterName = "loginFilter")
|
public class LoginDataFilter implements Filter {
|
|
@Autowired
|
private RestTemplate restTemplate;
|
@Autowired
|
private EnumStr enumStr;
|
|
private static final Pattern QUERY_PARAM_PATTREN = Pattern.compile("([^&=]+)(=?)([^&]+)?");
|
|
@Override
|
public void init(FilterConfig filterConfig) throws ServletException {
|
|
}
|
|
@Override
|
public void doFilter(ServletRequest request, ServletResponse response, FilterChain fc) throws IOException, ServletException {
|
HttpServletRequest req = (HttpServletRequest) request;
|
HttpServletResponse resp = (HttpServletResponse) response;
|
String url = req.getRequestURI();
|
|
if (url.startsWith("/loginData")) {
|
Map<String, String[]> parMap = req.getParameterMap();
|
|
Enumeration<String> names1 = req.getParameterNames();
|
|
Set<String> set = parMap.keySet();
|
MultiValueMap<String, String> mulMap = new LinkedMultiValueMap<>();
|
for (String key : parMap.keySet()) {
|
String[] names = parMap.get(key);
|
for (int i = 0; i < names.length; i++) {
|
mulMap.add(key, names[i]);
|
}
|
}
|
log.info(mulMap + "mulMap");
|
String method = req.getMethod();
|
log.info(method + "请求方式");
|
HttpHeaders headers = new HttpHeaders();
|
String body = null;
|
String auth = req.getHeader("Authorization");
|
headers.add("Accept", MediaType.APPLICATION_JSON.toString());
|
// headers.add("content-type", req.getHeader("content-type"));
|
if (auth != null) {
|
mulMap.add("access_token", auth.substring(7));
|
headers.add("Authorization", auth);
|
}
|
// url = "http://"+ req.getServerName() +":"+enumStr.getUserHTTP_PORT()+"/data"+ url.substring(10) ;
|
url = "http://" + enumStr.getUserHOSTNAME() + ":" + enumStr.getUserHTTP_PORT() + "/data" + url.substring(10);
|
log.info(url + "查询url");
|
HttpEntity<MultiValueMap> reqEntity = new HttpEntity<MultiValueMap>(mulMap, headers);
|
if ("get".equalsIgnoreCase(method)) {
|
body = restTemplate.getForObject(expandURL(url, mulMap), String.class, mulMap);
|
// body =restTemplate.exchange(url, HttpMethod.GET, reqEntity,String.class).getBody();
|
} else if ("post".equalsIgnoreCase(method)) {
|
headers.setContentType(MediaType.APPLICATION_FORM_URLENCODED);
|
body = restTemplate.postForEntity(url, reqEntity, String.class).getBody();
|
} else if ("put".equalsIgnoreCase(method)) {
|
headers.setContentType(MediaType.APPLICATION_FORM_URLENCODED);
|
ServletInputStream inputStream = req.getInputStream();
|
String str = IOUtils.toString(inputStream);
|
log.info("put inputstream获取发送数据："+str);
|
StringBuilder sb = new StringBuilder();
|
if (StringUtils.isBlank(str)){
|
for (String key : mulMap.keySet()) {
|
List<String> values = mulMap.get(key);
|
for (String value : values){
|
sb.append("&"+key+"="+value);
|
}
|
}
|
if (sb.length() > 0)str = sb.substring(1);
|
}
|
log.info("put request getParamer获取发送数据："+str);
|
HttpEntity<String> entity = new HttpEntity<>(str, headers);
|
ResponseEntity<String> resultEntity = null;
|
try {
|
resultEntity = restTemplate.exchange(url, HttpMethod.PUT, entity, String.class);
|
body = resultEntity.getBody();
|
} catch (HttpClientErrorException e) {
|
String responseBodyAsString = e.getResponseBodyAsString();
|
body = responseBodyAsString;
|
// e.printStackTrace();
|
}catch (HttpServerErrorException e) {
|
String responseBodyAsString = e.getResponseBodyAsString();
|
body = responseBodyAsString;
|
// e.printStackTrace();
|
}catch (Exception e){
|
body = JSONObject.toJSONString(Result.error(e));
|
e.printStackTrace();
|
}
|
// log.info("put返回数据："+body);
|
}
|
log.info(body + "返回内容");
|
if (body != null) {
|
resp.setContentType("application/json;charset=utf-8");
|
resp.setStatus(200);
|
PrintWriter pw = resp.getWriter();
|
pw.write(body);
|
pw.close();
|
}
|
} else if (url.startsWith("/httpImage")) {
|
log.info(url + "图片 跳转");
|
log.info(enumStr.getHttpImgUrl() + url.substring(11) + "图片 装换 跳转");
|
resp.sendRedirect(enumStr.getHttpImgUrl() + url.substring(11));
|
resp.setHeader("Access-Control-Allow-Origin", "*");
|
return;
|
} else {
|
log.info(url + "其他url 跳转");
|
fc.doFilter(request, response);
|
}
|
}
|
|
// 添加url
|
private static String expandURL(String url, MultiValueMap<String, String> map) {
|
Matcher mc = QUERY_PARAM_PATTREN.matcher(url);
|
StringBuilder sb = new StringBuilder(url);
|
if (!mc.find()) {
|
sb.append("&");
|
} else {
|
sb.append("?");
|
}
|
Set<String> keys = map.keySet();
|
for (String key : keys) {
|
List<String> value = map.get(key);
|
for (String val : value) {
|
sb.append(key).append("=").append(val).append("&"); // .append("}")
|
}
|
}
|
return sb.deleteCharAt(sb.length() - 1).toString();
|
}
|
|
@Override
|
public void destroy() {
|
|
}
|
}
|
