用切面来做日志记录器

最新推荐文章于 2024-06-14 09:58:39 发布

人月IT

最新推荐文章于 2024-06-14 09:58:39 发布

阅读量404

点赞数

文章标签： OS

本文链接：https://blog.csdn.net/dgutliangxuan/article/details/81078384

版权

@Aspect
@Component
@Slf4j
public class UserLoginLogAspect {

    @Autowired
    UserLoginLogMapper userLoginLogMapper;

    @Resource(name = "basicRedisTemplate")
    RedisTemplate redisTemplate;

    @Autowired
    SiteInformationMapper siteInformationMapper;

    @Pointcut("execution(* com.zemcho.cms.controller.site.SiteController.switchSite(..))")
    public void userPointCut() {

    }

    @After("(userPointCut()) && args(response,token,params,..)")
    public void writeLoginLog(HttpServletResponse response, String token, SwitchParams params) {

        System.out.println("token = " + token);
        System.out.println("params" + params);

        String siteId = params.getSiteId();
        SiteInformation siteInformation = siteInformationMapper.selectBySiteId(siteId);


        String redisToken = "token:" + token;
        LoginInformation loginInformation = (LoginInformation) redisTemplate.opsForValue().get(redisToken);

        if (siteInformation != null && loginInformation != null) {

            UserLoginLog entity = new UserLoginLog();

            entity.setUserId(loginInformation.getUserId());
            entity.setSiteId(siteId);
            entity.setNickname(loginInformation.getUserName());
            entity.setLoginSiteName(siteInformation.getSiteName());
            entity.setLoginTime(LocalDateTime.now());

            String ip = loginInformation.getIp();
            System.out.println("requestIp" + ip);
            entity.setLoginIp(ip);

            userLoginLogMapper.insertUserLoginLog(entity);
        }

        log.info("用户登陆，记录登陆日志");
    }

    public static String getRequestIp(HttpServletRequest request) {

        //如果XFF不为空，拿XFF的左边第一个
        String ip = request.getHeader("X-Forwarded-For");
        if (ip != null && !"unKnown".equalsIgnoreCase(ip)) {
            //多次反向代理后会有多个ip值，第一个ip才是真实ip
            int index = ip.indexOf(",");
            if (index != -1) {
                return ip.substring(0, index);
            } else {
                return ip;
            }
        }

        //如果XFF为空，拿XRI
        ip = request.getHeader("X-Real-IP");
        if (ip != null && !"unKnown".equalsIgnoreCase(ip)) {
            return ip;
        }

        //如果XRI为空，只能拿request.getRemoteAddr()，也就是只能拿到最直接发给他的机器ip了
        return request.getRemoteAddr();
    }

}

  @PostMapping("/switch")
    public Result switchSite(HttpServletResponse response, @RequestHeader("token") String token, @Validated @RequestBody SwitchParams params, BindingResult result){

        params.setToken(token);
        params.setResponse(response);
        return siteService.switchSite(params);
    }