import org.apache.commons.io.IOUtils;

import org.aspectj.lang.ProceedingJoinPoint;

import org.aspectj.lang.annotation.Aspect;

import org.aspectj.lang.annotation.Pointcut;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.cloud.netflix.feign.ribbon.LoadBalancerFeignClient;

import org.springframework.stereotype.Component;

import feign.Request;

import feign.Response;

import java.io.StringWriter;

@Aspect

@Component

public class FeignClientAspect {

    private static final Logger LOGGER = LoggerFactory.getLogger(FeignClientAspect.class);

    @Pointcut("execution(* org.springframework.cloud.netflix.feign.ribbon.LoadBalancerFeignClient.*(..))")

    public void feignClientPointcut() {

    }

    @org.aspectj.lang.annotation.Around("feignClientPointcut()")

    public Object doAround(ProceedingJoinPoint joinPoint) throws Throwable {

        LOGGER.info("feignclient begin");

        long start = System.currentTimeMillis();

        Object object = joinPoint.getTarget();

        if (!(object instanceof LoadBalancerFeignClient)) {

            LOGGER.info("feignclient not LoadBalancerFeignClient");

            return joinPoint.proceed();

        }

        Object[] args = joinPoint.getArgs();

        for (Object obj : args) {

            if (obj instanceof Request) {

                Request request = (Request) obj;

                LOGGER.info("feignclient request url:{}", request.url());

            }

        }

        Object result = joinPoint.proceed();

        if (result instanceof Response) {

            Response response = (Response) result;

            StringWriter writer = new StringWriter();

            IOUtils.copy(response.body().asInputStream(), writer, "UTF-8");

            String theString = writer.toString();

            LOGGER.info("feignclient response body:{}", theString);

        }

        LOGGER.info("feignclient end ,耗时:{}", (System.currentTimeMillis() - start));

        return result;

    }

}