| | |
| | | |
| | | import com.alibaba.fastjson.JSON; |
| | | import com.baomidou.mybatisplus.mapper.EntityWrapper; |
| | | import com.core.annotations.AppAuth; |
| | | import com.core.annotations.ManagerAuth; |
| | | import com.core.common.BaseRes; |
| | | import com.core.common.Cools; |
| | |
| | | } |
| | | } |
| | | } |
| | | // 白名单 |
| | | // if (IpTools.gainRealIp(request).equals("127.0.0.1")) { |
| | | // request.setAttribute("userId", 9527); |
| | | // return true; |
| | | // } |
| | | // 跨域设置 |
| | | // response.setHeader("Access-Control-Allow-Origin", "*"); |
| | | HandlerMethod handlerMethod = (HandlerMethod) handler; |
| | | Method method = handlerMethod.getMethod(); |
| | | if (method.isAnnotationPresent(AppAuth.class)){ |
| | | AppAuth annotation = method.getAnnotation(AppAuth.class); |
| | | if (annotation.value().equals(AppAuth.Auth.CHECK)){ |
| | | request.setAttribute("appAuth", annotation.memo()); |
| | | } |
| | | } |
| | | if (method.isAnnotationPresent(ManagerAuth.class)){ |
| | | ManagerAuth annotation = method.getAnnotation(ManagerAuth.class); |
| | | if (annotation.value().equals(ManagerAuth.Auth.CHECK)){ |
| | |
| | | } |
| | | } |
| | | |
| | | @Override |
| | | public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, @Nullable Exception ex) throws Exception { |
| | | // Object r = request.getAttribute("cool-response"); |
| | | } |
| | | |
| | | private boolean check(HttpServletRequest request, HttpServletResponse response, String memo) { |
| | | try { |
| | | String token = request.getHeader("token"); |
| | | String token = new String(); |
| | | token = request.getHeader("token"); |
| | | if (Cools.isEmpty(token)){ |
| | | return true; |
| | | } |
| | | UserLogin userLogin = userLoginService.selectOne(new EntityWrapper<UserLogin>().eq("token", token)); |
| | | if (null == userLogin){ |
| | | Http.response(response, BaseRes.DENIED); |
| | |
| | | Http.response(response, BaseRes.NO_ACTIVATION); |
| | | return false; |
| | | } |
| | | // 记录操作日志 |
| | | OperateLog operateLog = new OperateLog(); |
| | | operateLog.setAction(Cools.isEmpty(memo)?request.getRequestURI():memo); |
| | | operateLog.setIp(request.getRemoteAddr()); |
| | | operateLog.setUserId(user.getId()); |
| | | operateLog.setRequest(JSON.toJSONString(request.getParameterMap())); |
| | | request.setAttribute("operateLog", operateLog); |
| | | // // 记录操作日志 |
| | | // OperateLog operateLog = new OperateLog(); |
| | | // operateLog.setAction(Cools.isEmpty(memo)?request.getRequestURI():memo); |
| | | // operateLog.setIp(request.getRemoteAddr()); |
| | | // operateLog.setUserId(user.getId()); |
| | | // operateLog.setRequest(JSON.toJSONString(request.getParameterMap())); |
| | | // request.setAttribute("operateLog", operateLog); |
| | | } |
| | | return true; |
| | | } catch (Exception e){ |