基于服务器端保存用户的查询参数

最近公司项目有一个新的需求, 希望用户在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 {@Resourceprivate HelloService helloService;@Resource(name = "default")private ThreadPoolTaskExecutor executor;private static final Logger LOGGER = LoggerFactory.getLogger(RequestParamInterceptor.class);@Overridepublic 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;}@Overridepublic void postHandle(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Object o, ModelAndView modelAndView) {}@Overridepublic 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 {@Autowiredprivate 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();}@Overridepublic ServletInputStream getInputStream() throws IOException {final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(body.getBytes());return new ServletInputStream() {@Overridepublic boolean isFinished() {return false;}@Overridepublic boolean isReady() {return false;}@Overridepublic void setReadListener(ReadListener readListener) {}@Overridepublic int read() throws IOException {return byteArrayInputStream.read();}};}@Overridepublic 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 {@Overridepublic void init(FilterConfig filterConfig) throws ServletException {}@Overridepublic 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);}}@Overridepublic void destroy() {}
}

自定义webMvc的配置

/*** 自定义webMvc的配置** @author Eric on 2018/8/19.* @version 1.0*/
@Configuration
public class MyWebMvcConfig extends WebMvcConfigurerAdapter {@Resourceprivate RequestParamInterceptor requestParamInterceptor;/*** 加入到Spring MVC的拦截器链中** @param registry 拦截器注册*/@Overridepublic void addInterceptors(InterceptorRegistry registry) {registry.addInterceptor(requestParamInterceptor);}@Overridepublic void configurePathMatch(PathMatchConfigurer configurer) {super.configurePathMatch(configurer);}
}

以上操作就可以把在PC和APP端的请求参数保存到Mongodb中, 之后用户进入相应的页面时, 只需要根据当前页面的pageName来获取该用户上次的请求参数即可