package com.zhd.exploit.api.config;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import javax.servlet.http.HttpServletResponseWrapper;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.mvc.method.annotation.ExtendedServletRequestDataBinder;
import com.alibaba.fastjson.JSONObject;
@Aspect
@Component
@Order(1)
public class ControllerLogInterceptor {
	private static final Logger log = LoggerFactory.getLogger(ControllerLogInterceptor.class);
    //創(chuàng)建Pointcut表示式，表示所有controller請求
	@Pointcut("execution(* com..*.controller..*(..))")
	private void controllerAspect() {
	}// 請求method前打印內(nèi)容
	@Around(value = "controllerAspect()")
	public void around(ProceedingJoinPoint pjp) throws Throwable {
	//通過uuid關(guān)聯(lián)請求參數(shù)和返回參數(shù)
		String uuid = UUID.randomUUID().toString().replaceAll("-", "");
		methodBefore(pjp, uuid);
		try {
			Object proceed = pjp.proceed();
			methodAfterReturing(proceed, uuid);
		} catch (Exception e) {
			log.error("[{}]Response異常內(nèi)容:{}", uuid, e);
			throw e;
		}
	}
	public void methodBefore(JoinPoint joinPoint, String uuid) {
		// 打印請求內(nèi)容
		try {
			// 下面兩個數(shù)組中，參數(shù)值和參數(shù)名的個數(shù)和位置是一一對應(yīng)的。
			Object[] objs = joinPoint.getArgs();
			String[] argNames = ((MethodSignature) joinPoint.getSignature()).getParameterNames(); // 參數(shù)名
			Map<String, Object> paramMap = new HashMap<String, Object>();
			for (int i = 0; i < objs.length; i++) {
				if (!(objs[i] instanceof ExtendedServletRequestDataBinder) && !(objs[i] instanceof HttpServletResponseWrapper)) {
					paramMap.put(argNames[i], objs[i]);
				}
			}
			if (paramMap.size() > 0) {
				log.info("\n[{}]方法:{}\n參數(shù):{}", uuid, joinPoint.getSignature(), JSONObject.toJSONString(paramMap));
			}
		} catch (Exception e) {
			log.error("[{}]AOP methodBefore:", uuid, e);
		}
	}
	public void methodAfterReturing(Object o, String uuid) {
		try {
			if (o != null)
				log.info("[{}]Response內(nèi)容:{}", uuid, JSONObject.toJSON(o));
		} catch (Exception e) {
			log.error("[{}]AOP methodAfterReturing:", uuid, e);
		}
	}
}

?? ?@RequestMapping(value = "/test0", method = RequestMethod.GET, produces = { MediaType.APPLICATION_JSON_VALUE })
?? ?public Result test0(@RequestParam String name, @RequestParam String password) {
?? ??? ?System.out.println("test0 OK");
?? ??? ?return new Result("1", "mock a Result");
?? ?}

?? ?@RequestMapping(value = "/test1", method = RequestMethod.GET, produces = { MediaType.APPLICATION_JSON_VALUE })
?? ?public Result test1(PayDTO payDTO) {
?? ??? ?System.out.println("test1 OK");
?? ??? ?return new Result("1", "mock a Result");
?? ?}

?? ?@RequestMapping(value = "/test2", method = RequestMethod.POST, produces = { MediaType.APPLICATION_JSON_VALUE })
?? ?public Result test2(@RequestBody PayDTO payDTO) {
?? ??? ?System.out.println("test2 OK");
?? ??? ?return new Result("2", "mock a Result");
?? ?}

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface Operate {
}

@Aspect
@Component
public class OperateInterceptor {?
? ? @Pointcut("@annotation(com.yili.web.entity.Operate)")
? ? public void interceptor() {
? ? }
?
? ? @Resource
? ? private SqlObjectUtil sqlObjectUtil;
?
? ? @Around(value = "interceptor()")
? ? public Object check(ProceedingJoinPoint pjp) throws Throwable {
? ? ? ? System.out.println("進(jìn)入interceptor");
? ? ? ? Signature signature = pjp.getSignature();
? ? ? ? if (!(signature instanceof MethodSignature)) {
? ? ? ? ? ? throw new IllegalArgumentException("該注解只適用于方法");
? ? ? ? }
? ? ? ? Object[] objects = pjp.getArgs();
? ? ? ? RequestAttributes attributes = RequestContextHolder.getRequestAttributes();
? ? ? ? ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) attributes;
? ? ? ? HttpServletRequest request = servletRequestAttributes.getRequest();
? ? ? ? String loginToken = getLoginToken(request); ? ? ? ?
? ? ? ? for (int i=0;i<objects.length;i++){
? ? ? ? ? ? if (SqlObject.class.isAssignableFrom(objects[i].getClass()))) {
? ? ? ? ? ? ? ? sqlObjectUtil.setOperatorInfo(loginToken,(SqlObject)objects[i]);
? ? ? ? ? ? }
? ? ? ? }
? ? ? ? return pjp.proceed(objects);
? ? }