SpringAOP笔记

AOP记录, 它可以做很多的事情. 留着查看

1.日志处理

2.权限处理

3.限制,异常控制

package com.pahaoche.portal.qiantai.weizhang.aop;

import java.lang.reflect.Method;

import javax.servlet.http.HttpServletRequest;

import org.apache.commons.lang.StringUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import com.pahaoche.portal.base.memcached.MemcachedService;

@Aspect
@Component("systemLogAspect")
public class SystemLogAspect {
	
	private Log logger = LogFactory.getLog(SystemLogAspect.class);
	
	@Autowired
	@Qualifier("memcachedService")
	private MemcachedService memcachedService;
	
	/**
	 * 频繁请求控制
	 */
	@Pointcut("@annotation(com.pahaoche.portal.qiantai.weizhang.aop.FrequentRequestAsp)")  
	public void pointCutFrequentLog(){}
	
	@Pointcut("@annotation(com.pahaoche.portal.qiantai.weizhang.aop.SystemControllerLog)")
    public void pointCutMethod() {}
	/**
	 * controller 执行消耗时间ms
	 */
	@Pointcut("execution(public * com.pahaoche.portal.qiantai.weizhang.controller.*Controller.*(..))")
	public void pointCutUseTime(){}
	/**
	 * service 方法执行消耗时间ms
	 */
	@Pointcut("execution(public * com.pahaoche.portal.base.weizhang.service..*.*(..))")
	public void pointCutServiceUseTime(){}
	
	
	@Before("pointCutMethod()")
	public void doBefore() {
		
	}
	@Around("pointCutUseTime()")
	public Object doAroundUseTime(ProceedingJoinPoint pjp) throws Throwable{
		Long startTime = System.currentTimeMillis();
        String ip = getRemoteAddr();
		String className = pjp.getSignature().getDeclaringType().getName();
		String methodName = pjp.getSignature().getName();
		Object o = pjp.proceed();	
		Long endTime = System.currentTimeMillis();
		logger.info("controller:"+methodName+"(),useTime:"+(endTime-startTime)+"ms."+"path:"+className+"."+methodName+"()"+";IP:"+ip);
		return o;  
	}
	
	
	@Around("pointCutServiceUseTime()")
	public Object doAroundServiceUseTime(ProceedingJoinPoint pjp) throws Throwable{
		Long startTime = System.currentTimeMillis();
		String className = pjp.getSignature().getDeclaringType().getName();
		String methodName = pjp.getSignature().getName();
		Object o = pjp.proceed();	
		Long endTime = System.currentTimeMillis();
        logger.info("service:"+methodName+"(),useTime:"+(endTime-startTime)+"ms."+"path:"+className+"."+methodName+"()");
		return o;  
	}
	
	
	@Around("pointCutFrequentLog()")
	public Object doAround(ProceedingJoinPoint pjp) throws Throwable {  
        String ip = getRemoteAddr();
        String methodName = pjp.getSignature().getName();
        Object o = null;
        //IP不为null时做判断
        if(StringUtils.isNotEmpty(ip) && checkIpMemcached(ip, methodName)){
        	logger.info("频繁请求不处理. IP:"+ip+";methodName:"+methodName);
        }else{
        	o = pjp.proceed();	
        }
        return o;  
    }  
	
	private String getRemoteAddr(){
		String ip = "";
		if(null == RequestContextHolder.getRequestAttributes() 
				|| null == ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest()){
			return  ip;
		}
		HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        ip = request.getHeader("x-forwarded-for"); 
        if(StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) { 
            ip = request.getHeader("Proxy-Client-IP"); 
            if(StringUtils.isNotEmpty(ip) && !"unknown".equalsIgnoreCase(ip)){
            	ip +="#Proxy-Client-IP";	
            }
        } 
        if(StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) { 
            ip = request.getHeader("WL-Proxy-Client-IP");
            if(StringUtils.isNotEmpty(ip) && !"unknown".equalsIgnoreCase(ip)){
            	ip +="#WL-Proxy-Client-IP";
            }
        } 
        if (StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) {
            ip = request.getHeader("HTTP_CLIENT_IP");
            if(StringUtils.isNotEmpty(ip) && !"unknown".equalsIgnoreCase(ip)){
            	ip +="#HTTP_CLIENT_IP";
            }
        }
        if (StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) {
            ip = request.getHeader("HTTP_X_FORWARDED_FOR");
            if(StringUtils.isNotEmpty(ip) && !"unknown".equalsIgnoreCase(ip)){
            	ip +="#HTTP_X_FORWARDED_FOR";
            }
        }
        if (StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) {
            ip = request.getRemoteAddr();
            if(StringUtils.isNotEmpty(ip) && !"unknown".equalsIgnoreCase(ip)){
            	ip +="#RemoteAddr";
            }
        }
        return ip;
        /*
         * 
         * 	可是，如果通过了多级反向代理的话，X-Forwarded-For的值并不止一个，而是一串IP值，究竟哪个才是真正的用户端的真实IP呢？
			答案是取X-Forwarded-For中第一个非unknown的有效IP字符串。
			如：
			X-Forwarded-For：192.168.1.110， 192.168.1.120， 192.168.1.130， 192.168.1.100 
			用户真实IP为： 192.168.1.110
         * 
         */
	}
	
	/**
	 * 如果6秒内同一个IP对同一个方法有请求,则不作处理
	 * @param ip
	 * @param methodName
	 * @return
	 */
	private boolean checkIpMemcached(String ip,String methodName){
		String key =(ip+":"+methodName).replace(",", "%").replace(".", "$");
		//wwwtest.pahaoche.com:202.69.18.197, 172.30.50.250:resultViolation
		if(key.length() > 40){
			logger.warn("Key too long. key="+key);
			return false;
		}
		Long currentTime = System.currentTimeMillis();
		int exp = 20 ; //过期时间, 单位秒
		if(null == memcachedService.get(key)){
			memcachedService.set(key,exp,currentTime);	
		}else{
			Long oldTime = (Long)memcachedService.get(key);
			if(currentTime - oldTime   < (6*1000)){
				return true;
			}else{
				memcachedService.set(key,exp,currentTime);
				logger.info("更新IP:methodName缓存. key="+key);
			}
		}
		return false;
	}

	
	/**
	 * 获取注解中对方法的描述信息 用于Controller层注解
	 * 
	 * @param joinPoint
	 *            切点
	 * @return 方法描述
	 * @throws Exception
	 */
	public static String getControllerMethodDescription(ProceedingJoinPoint joinPoint) throws Exception {
		String targetName = joinPoint.getTarget().getClass().getName();
		String methodName = joinPoint.getSignature().getName();
		Object[] arguments = joinPoint.getArgs();
		Class targetClass = Class.forName(targetName);
		Method[] methods = targetClass.getMethods();
		String description = "";
		for (Method method : methods) {
			if (method.getName().equals(methodName)) {
				Class[] clazzs = method.getParameterTypes();
				if (clazzs.length == arguments.length) {
					description = method.getAnnotation(SystemControllerLog.class).description();
					break;
				}
			}
		}
		return description;
	}
	
}

applicationContext.xml中加入aop注解扫描，如果controller需要用到aop,将spring-mvc.xml配置即可。

<aop:aspectj-autoproxy proxy-target-class="true"/>