Java中关于Aspectj面向切面与mongodb的应用(保存操作日志)

package com.aspectj;

import java.util.Date;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;

import org.apache.maven.surefire.shade.org.apache.maven.shared.utils.StringUtils;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import com.pojo.LogBean;
import com.pojo.UserBean;

//两个很重要的注解
@Aspect
@Component
public class LogAspectj {

	//注入
	@Autowired
	private MongoTemplate mongoTemplate;
	
	
	@Pointcut("execution(* com.service.impl..*.*(..))")
	public void logPointCut() {}
	
	
	@AfterReturning(value="logPointCut()",argNames="rtv",returning="rtv")
	public void saveLog(JoinPoint joinPoint,Object rtv) throws Exception {
		LogBean logBean = new LogBean();
		// 判断参数
		if (joinPoint.getArgs() == null) {// 没有参数
			return;
		}
		//获取类名
		String className = joinPoint.getTarget().getClass().getName();
		logBean.setClassName(className);
		logBean.setCreateTime(new Date());
		//获取方法名
		String methodName = joinPoint.getSignature().getName();
		logBean.setMethodName(methodName);
		//获取请求的参数
		String requestParams = "";
		Object[] args = joinPoint.getArgs();
		for (int i = 0; i < args.length; i++) {
			requestParams += "第"+(i+1)+"个参数="+args[i].toString();
		}
		logBean.setRequestParams(requestParams);
		//获取返回参数  关键点  rtv
		logBean.setResponseParams(rtv == null ? "" : rtv.toString());
		//获取request
		RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
		ServletRequestAttributes sra = (ServletRequestAttributes) requestAttributes;
		if (sra != null) {
			HttpServletRequest request = sra.getRequest();
			//判断request对象不为空时，才会取ip地址
			logBean.setIp(request != null ? getIp(request) : "");
			if (request != null) {
				HttpSession session = request.getSession();
				//判断session不为空时，获取用户的id，否则为-1
				if (session != null) {
					
					if (session.getAttribute(session.getId()) != null) {
						logBean.setUserId(((UserBean)session.getAttribute(session.getId())).getId());
					}
				}
			}
		}
		//调用mongoTemplate中的save方法，保存日志
		mongoTemplate.save(logBean);
	}
	
	
	//获取客户端ip
	public static String getIp(HttpServletRequest request) {
		String ip = request.getHeader("X-Forwarded-For");
		if (StringUtils.isNotEmpty(ip) && !"unKnown".equalsIgnoreCase(ip)) {
			//多次反向代理后会有多个ip值，第一个ip才是真实ip
			int index = ip.indexOf(",");
			if (index != -1) {
				return ip.substring(0,index);
			}else {
				return ip;
			}
		}
		
		ip = request.getHeader("X-Real-Ip");
		if (StringUtils.isNotEmpty(ip) && !"unKnown".equalsIgnoreCase(ip)) {
			return ip;
		}
		return request.getRemoteAddr();
	}
	
}