背景:每一次请求都用记录请求的用户和IP等业务信息,在rest方法中写代码及其冗余,特使用动态处理的方式。
楼主赋值的是:ip和user
处理:
使用Filter和HttpServletRequestWrapper的方法进行动态赋值
1.创建参数过滤器
package com.bxm.sentinel.web.resolver;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.annotation.Order;
import org.springframework.web.filter.OncePerRequestFilter;
import javax.servlet.FilterChain;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebFilter;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
/**
* 过滤器拦截所有请求
* 括号中的数字越大,在多个过滤器的执行顺序越靠前
*
* @Author kk.xie
* @Date 2019/9/19 19:35
* @Version 1.0
**/
@WebFilter(filterName = "paramFilter", urlPatterns = "/*")
@Order(100)
public class ParamFilter extends OncePerRequestFilter {
@Autowired
private ParameterRequestWrapperFactory factory;
@Override
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException {
//定义一个新的request(名称是wrapper)
HttpServletRequest wrapper = factory.build(request);
//将修改过的request(wrapper)返回
filterChain.doFilter(wrapper, response);
}
}
2.创建请求参数工厂类,有些参数需要调用service获取
package com.bxm.sentinel.web.resolver;
import com.alibaba.fastjson.JSONObject;
import com.bxm.acl.facade.AclUserManagerService;
import com.bxm.acl.facade.Utils;
import com.bxm.acl.facade.model.ResultModel;
import com.bxm.acl.facade.model.UserVo;
import com.bxm.warcar.utils.IpHelper;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.*;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Enumeration;
/**
* 请求参数动态赋值工厂类
*
* @Author kk.xie
* @Date 2019/9/20 10:50
* @Version 1.0
**/
@Component
public class ParameterRequestWrapperFactory {
private Logger logger = LoggerFactory.getLogger(this.getClass());
private final String IP = "ip";
private final String USER = "user";
@Autowired
private AclUserManagerService aclUserManagerService;
public ParameterRequestWrapper build(HttpServletRequest request) throws IOException {
return new ParameterRequestWrapper(request);
}
class ParameterRequestWrapper extends HttpServletRequestWrapper {
private Logger logger = LoggerFactory.getLogger(this.getClass());
private byte[] bytes;
private HttpServletRequest request;
/**
* POST请求的RequestBody添加 ip和user
*
* @param request
* @return
* @throws
* @author kk.xie
* @date 2019/9/20 11:04
*/
public ParameterRequestWrapper(HttpServletRequest request) throws IOException {
super(request);
this.request = request;
try (BufferedInputStream bis = new BufferedInputStream(request.getInputStream());
ByteArrayOutputStream baos = new ByteArrayOutputStream()) {
byte[] buffer = new byte[1024];
int len;
while ((len = bis.read(buffer)) > 0) {
baos.write(buffer, 0, len);
}
bytes = baos.toByteArray();
String body = new String(bytes);
if(StringUtils.isNotBlank(body)){
JSONObject json = JSONObject.parseObject(body);
// 获取IP
String ip = IpHelper.getIpFromHeader(request);
json.put(IP, ip);
//获取user
String user = getUser(request);
json.put(USER, user);
body = json.toJSONString();
bytes = body.getBytes();
}
logger.info("body: {}", body);
} catch (IOException ex) {
throw ex;
}
}
/**
* POST请求的RequestBody添加 ip和user
*
* @param
* @return javax.servlet.ServletInputStream
* @throws
* @author kk.xie
* @date 2019/9/20 11:04
*/
@Override
public ServletInputStream getInputStream() throws IOException {
final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(bytes);
return new ServletInputStream() {
@Override
public boolean isFinished() {
return false;
}
@Override
public boolean isReady() {
return false;
}
@Override
public void setReadListener(ReadListener readListener) {
}
@Override
public int read() throws IOException {
return byteArrayInputStream.read();
}
};
}
/**
* POST请求的RequestBody添加 ip和user
*
* @param
* @return java.io.BufferedReader
* @throws
* @author kk.xie
* @date 2019/9/20 11:04
*/
@Override
public BufferedReader getReader() throws IOException {
return new BufferedReader(new InputStreamReader(this.getInputStream()));
}
/**
* GET 请求添加 ip和user
*
* @param
* @return java.util.Enumeration<java.lang.String>
* @throws
* @author kk.xie
* @date 2019/9/20 11:07
*/
@Override
public Enumeration<String> getParameterNames() {
Enumeration<String> enumeration = super.getParameterNames();
ArrayList<String> list = Collections.list(enumeration);
list.add(IP);
list.add(USER);
return Collections.enumeration(list);
}
/**
* GET 请求添加 ip和user
*
* @param name
* @return java.lang.String
* @throws
* @author kk.xie
* @date 2019/9/20 11:08
*/
@Override
public String getParameter(String name) {
if(IP.equals(name)){
return IpHelper.getIpFromHeader(request);
}else if(USER.equals(name)){
return getUser(request);
}
return super.getParameter(name);
}
/**
* GET 请求添加 ip和user
*
* @param name
* @return java.lang.String[]
* @throws
* @author kk.xie
* @date 2019/9/20 11:39
*/
@Override
public String[] getParameterValues(String name) {
if(IP.equals(name)){
return new String[]{IpHelper.getIpFromHeader(request)};
}else if(USER.equals(name)){
return new String[]{getUser(request)};
}
return super.getParameterValues(name);
}
}
private String getUser(HttpServletRequest request) {
String username = null;
try {
// 获取token
String token = Utils.getToken(request);
ResultModel<UserVo> result = aclUserManagerService.auth(token, 7, request.getRequestURI());
UserVo userVo = result.getReturnValue();
username = userVo.getUsername();
} catch (Exception e) {
logger.error("get user error!", e);
}
return username;
}
}
3.Application修改,添加@ServletComponentScan 注解
@ServletComponentScan
@SpringBootApplication
public class SentinelApiApplication {
public static void main(String[] args) {
SpringApplication application = new SpringApplication(SentinelApiApplication.class);
application.run(args);
}
}
示例rest接口
import com.bxm.sentinel.model.base.PrimaryKeyDto;
import com.bxm.sentinel.model.dto.MonitorStrategyDto;
import com.bxm.sentinel.model.dto.MonitorStrategyPageDto;
import com.bxm.sentinel.model.vo.MonitorStrategyVo;
import com.bxm.sentinel.service.monitor.MonitorStrategyService;
import com.bxm.warcar.utils.response.ResultModel;
import com.bxm.warcar.utils.response.ResultModelFactory;
import com.github.pagehelper.PageInfo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RestController;
import javax.servlet.http.HttpServletRequest;
/**
* 监控策略控制器
*
* @Author kk.xie
* @Date 2019/9/18 16:18
* @Version 1.0
**/
@RestController
@RequestMapping("/monitorStrategy")
public class MonitorStrategyController {
@Autowired
private MonitorStrategyService monitorStrategyService;
/**
* 新增
*
* @param monitorStrategyDto
* @return com.bxm.warcar.utils.response.ResultModel<java.lang.Boolean>
* @throws
* @author kk.xie
* @date 2019/9/18 16:49
*/
@RequestMapping(value = "/add", method = RequestMethod.POST)
public ResultModel<Boolean> add(@RequestBody MonitorStrategyDto monitorStrategyDto){
return ResultModelFactory.SUCCESS(monitorStrategyService.add(monitorStrategyDto));
}
/**
* 查询
*
* @param monitorStrategyPageDto
* @return com.bxm.warcar.utils.response.ResultModel<com.github.pagehelper.PageInfo<com.bxm.sentinel.model.vo.MonitorStrategyVo>>
* @throws
* @author kk.xie
* @date 2019/9/18 16:50
*/
@RequestMapping(value = "/selectAll", method = RequestMethod.GET)
public ResultModel<PageInfo<MonitorStrategyVo>> selectAll(MonitorStrategyPageDto monitorStrategyPageDto){
return ResultModelFactory.SUCCESS(monitorStrategyService.selectAll(monitorStrategyPageDto));
}
}
MonitorStrategyDto和MonitorStrategyPageDto均继承CommonBaseDto
package com.bxm.sentinel.model.base;
import lombok.Data;
/**
* 通用请求对象
*
* @Author kk.xie
* @Date 2019/9/19 15:17
* @Version 1.0
**/
@Data
public class CommonBaseDto {
/**
* 请求IP
*/
private String ip;
/**
* 请求的用户账号
*/
private String user;
}