基于服务器端保存用户的查询参数
最近公司项目有一个新的需求, 希望用户在PC查询的参数能够同步更新到APP端, 避免让用户在PC和APP端重复输入查询条件, 基于项目是前后端分离, APP和PC的请求参数都是一样的且都是POST请求, 因此考虑采用拦截器的方式拦截获取POST请求中的请求体, 同时前端请求后台时, 统一在Header中加入当前页面名称(pageName)的参数, 以用户Sessioin中的工号和pageName作为key, 将请求参数保存到Mongodb中, 基本的方案就是这样, 下面是具体实现
自定义线程池
/**
* 线程池配置
*
* @author Eric on 2019/5/4.
* @version 1.0
*/
@Configuration
public class ThreadPoolConfig {
private static final int PROCESSOR_NUM = Runtime.getRuntime().availableProcessors();
private static final int MIN_CORE_SIZE = PROCESSOR_NUM < 8 ? 16 : PROCESSOR_NUM * 2;
private static final String DEFAULT_THREAD = "default";
@Bean("default")
public ThreadPoolTaskExecutor executorService() {
ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
executor.setThreadNamePrefix(DEFAULT_THREAD);
executor.setCorePoolSize(MIN_CORE_SIZE);
executor.setMaxPoolSize(PROCESSOR_NUM * 16);
executor.setQueueCapacity(PROCESSOR_NUM * 2);
executor.setKeepAliveSeconds(30);
executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
executor.initialize();
return executor;
}
}
自定义拦截器
HelloService封装了保存到Mongodb或者更新到Mongodb的操作 用户信息是从session中获取, 具体用户信息的实体类可以自定义
/**
* 记录本次请求参数的拦截器
*
* @author wang.js on 2019/4/26.
* @version 1.0
*/
@Component
public class RequestParamInterceptor implements HandlerInterceptor {
@Resource
private HelloService helloService;
@Resource(name = "default")
private ThreadPoolTaskExecutor executor;
private static final Logger LOGGER = LoggerFactory.getLogger(RequestParamInterceptor.class);
@Override
public boolean preHandle(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Object o) {
HttpSession session = httpServletRequest.getSession();
if (session != null) {
String pageName = httpServletRequest.getHeader("pageName");
SystemUser user = (SystemUser) session.getAttribute("user");
String body = new RequestWrapper(httpServletRequest).getBody();
if (user != null && !StringUtils.isEmpty(pageName) && !StringUtils.isEmpty(body)) {
executor.submit(() -> {
helloService.saveOrUpdate(new MongoBean(pageName + user.getStaffCode(), body), "test");
LOGGER.info("保存成功, key:{}, value:{}", pageName + user.getStaffCode(), body);
});
}
}
return true;
}
@Override
public void postHandle(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Object o, ModelAndView modelAndView) {
}
@Override
public void afterCompletion(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Object o, Exception e) {
}
}
HellService
HellService只是封装了Mongodb的保存和更新操作, 仅供参考
/**
* @author wang.js
* @date 2018/10/29
* @copyright yougou.com
*/
@Service
public class HelloService {
@Autowired
private MongoBeanDao mongoBeanDao;
/**
* 保存
* @param mongoBean
*/
public void save(MongoBean mongoBean) {
mongoBeanDao.insert(mongoBean, "test");
}
/**
* 查找
* @param id
*/
public MongoBean find(String id, String collectionName) {
Query query = new Query(Criteria.where("id").is(id));
return mongoBeanDao.findByQuery(query, MongoBean.class, collectionName);
}
/**
* 更新
*
* @param mongoBean
* @param collectionName
* @return
*/
public boolean update(MongoBean mongoBean, String collectionName) {
Update update = new Update().set("value", mongoBean.getValue());
Query query = new Query(Criteria.where("id").is(mongoBean.getId()));
mongoBeanDao.updateByQuery(update, query, MongoBean.class, collectionName);
return true;
}
public void saveOrUpdate(MongoBean mongoBean, String collectionName) {
MongoBean mongoBean1 = find(mongoBean.getId(), collectionName);
if (mongoBean1 == null) {
save(mongoBean);
} else {
update(mongoBean, collectionName);
}
}
}
自定义HttpServletRequestWrapper
因为HttpServletRequest的输入流只能读取一次, 所以这里使用自定义HttpServletRequestWrapper将body中的信息保存起来, 方便后续多次使用 /** * @author wang.js on 2019/4/26. * @version 1.0 */ public class RequestWrapper extends HttpServletRequestWrapper {
private final String body;
public RequestWrapper(HttpServletRequest request) {
super(request);
StringBuilder stringBuilder = new StringBuilder();
BufferedReader bufferedReader = null;
InputStream inputStream = null;
try {
inputStream = request.getInputStream();
if (inputStream != null) {
bufferedReader = new BufferedReader(new InputStreamReader(inputStream));
char[] charBuffer = new char[128];
int bytesRead;
while ((bytesRead = bufferedReader.read(charBuffer)) > 0) {
stringBuilder.append(charBuffer, 0, bytesRead);
}
}
} catch (IOException ex) {
ex.printStackTrace();
} finally {
if (inputStream != null) {
try {
inputStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if (bufferedReader != null) {
try {
bufferedReader.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
body = stringBuilder.toString();
}
@Override
public ServletInputStream getInputStream() throws IOException {
final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(body.getBytes());
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();
}
};
}
@Override
public BufferedReader getReader() throws IOException {
return new BufferedReader(new InputStreamReader(this.getInputStream()));
}
String getBody() {
return this.body;
}
}
注册filter
/**
* 过滤器
*
* @author wang.js on 2019/4/26.
* @version 1.0
*/
@WebFilter(urlPatterns = "/*",filterName = "channelFilter")
public class ChannelFilter implements Filter {
@Override
public void init(FilterConfig filterConfig) throws ServletException {
}
@Override
public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
ServletRequest requestWrapper = null;
if(servletRequest instanceof HttpServletRequest) {
requestWrapper = new RequestWrapper((HttpServletRequest) servletRequest);
}
if(requestWrapper == null) {
filterChain.doFilter(servletRequest, servletResponse);
} else {
filterChain.doFilter(requestWrapper, servletResponse);
}
}
@Override
public void destroy() {
}
}
自定义webMvc的配置
/**
* 自定义webMvc的配置
*
* @author Eric on 2018/8/19.
* @version 1.0
*/
@Configuration
public class MyWebMvcConfig extends WebMvcConfigurerAdapter {
@Resource
private RequestParamInterceptor requestParamInterceptor;
/**
* 加入到Spring MVC的拦截器链中
*
* @param registry 拦截器注册
*/
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(requestParamInterceptor);
}
@Override
public void configurePathMatch(PathMatchConfigurer configurer) {
super.configurePathMatch(configurer);
}
}
以上操作就可以把在PC和APP端的请求参数保存到Mongodb中, 之后用户进入相应的页面时, 只需要根据当前页面的pageName来获取该用户上次的请求参数即可