package com.moral.common.aop;

import java.lang.reflect.Method;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.HashMap;
import java.util.Map;

import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.Signature;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;

import com.moral.common.bean.ResultBean;
import com.moral.common.exception.BusinessException;
import com.moral.common.exception.ValidateException;

import lombok.extern.log4j.Log4j;

@Log4j
@Aspect
@Component
public class ControllerAOP {

	@Around("execution(* com.moral.controller.*Controller.*(..))")
	public Object handlerControllerMethod(ProceedingJoinPoint pjp) throws Exception {

		long startTime = System.currentTimeMillis();
		Object result;

		try {
			result = pjp.proceed();
			log.info(pjp.getSignature() + "use time:" + (System.currentTimeMillis() - startTime));
		} catch (Throwable e) {
			result = handlerException(pjp, e);
		}

		return result;
	}

	@SuppressWarnings("rawtypes")
	private Object handlerException(ProceedingJoinPoint pjp, Throwable e) throws Exception {
		Signature sig = pjp.getSignature();
		MethodSignature msig = null;
		if (!(sig instanceof MethodSignature)) {
			throw new IllegalArgumentException("该注解只能用于方法");
		}
		msig = (MethodSignature) sig;
		Object target = pjp.getTarget();
		Method currentMethod = target.getClass().getMethod(msig.getName(), msig.getParameterTypes());
		Type type = currentMethod.getGenericReturnType();
		String message = "";
		if (e instanceof BusinessException) {
			message = e.getLocalizedMessage();
		} else if (e instanceof ValidateException) {
			message = e.getLocalizedMessage();
		} else {
			log.error(pjp.getSignature() + " error ", e);
			message = e.toString();
		}

		if (type instanceof ParameterizedType) {
			Type rawType = ((ParameterizedType) type).getRawType();
			if (rawType == ResultBean.class) {
				ResultBean<?> resultBean = new ResultBean(ResultBean.FAIL);
				resultBean.setMessage(message);
				return resultBean;
			} else if (rawType == Map.class) {
				Map<String, Object> result = new HashMap<String, Object>();
				result.put("msg",message);
				return result;
			}
		}
		return null;

	}
}