import com.fasterxml.jackson.databind.ObjectMapper;
import com.myjo.luim.common.utils.AddressUtils;
import com.myjo.luim.common.utils.IpUtils;
import com.myjo.luim.common.utils.ServletUtils;
import lombok.extern.slf4j.Slf4j;
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.springframework.data.domain.PageRequest;
import org.springframework.stereotype.Component;

import org.springframework.data.domain.Pageable;

import java.text.Format;

@Aspect
@Component
@Slf4j
public class PageSizeAspect {

@Pointcut("execution(public * com..*.*Controller.*(..))")
public void doOperation() {
log.info("分页size参数检验AOP");
}

@Around("doOperation()")
public Object doBefore(ProceedingJoinPoint joinPoint) throws Throwable {
Object[] args = joinPoint.getArgs();
for (int i = 0; i < args.length-1; i++) {
if (args[i] instanceof Pageable) {
if (((Pageable) args[i]).getPageSize() > 200){
Pageable pageable = new PageRequest(((Pageable) args[i]).getPageNumber(), 200,((Pageable) args[i]).getSort());
log.info(String.format("拦截到过大size:%s-IP地址:%s",((Pageable) args[i]).getPageSize(),IpUtils.getIpAddr(ServletUtils.getRequest())));
args[i] = pageable;
break;
}
}
}
return joinPoint.proceed(args);
}

}