com.github.dreamroute.args.printer.spring.boot.starter.ArgsPrinterConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of args-printer-spring-boot-starter Show documentation
print args.
The newest version!
package com.github.dreamroute.args.printer.spring.boot.starter;

import cn.hutool.core.annotation.AnnotationUtil;
import lombok.extern.slf4j.Slf4j;
import org.aopalliance.intercept.MethodInterceptor;
import org.springframework.aop.Advisor;
import org.springframework.aop.aspectj.AspectJExpressionPointcut;
import org.springframework.aop.support.DefaultPointcutAdvisor;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.beans.factory.support.DefaultListableBeanFactory;
import org.springframework.context.annotation.ImportBeanDefinitionRegistrar;
import org.springframework.core.type.AnnotationMetadata;
import org.springframework.core.type.StandardAnnotationMetadata;
import org.springframework.lang.NonNull;
import org.springframework.util.ObjectUtils;
import org.springframework.util.StopWatch;

import java.io.Serializable;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

import static com.alibaba.fastjson.JSON.toJSONString;

/**
 * @author w.dehai.2021/9/9.14:58
 */
@Slf4j
public class ArgsPrinterConfig implements ImportBeanDefinitionRegistrar {

private static final String PREFIX = "\r\n--------------------------------------------------------";

    @Override
    public void registerBeanDefinitions(@NonNull AnnotationMetadata importingClassMetadata, @NonNull BeanDefinitionRegistry registry) {
        ImportBeanDefinitionRegistrar.super.registerBeanDefinitions(importingClassMetadata, registry);
        StandardAnnotationMetadata icm = (StandardAnnotationMetadata) importingClassMetadata;
        Class ic = icm.getIntrospectedClass();

        // 之所以这里解析所有包，然后拼接execution表达式，是因为有时候使用execution表达式拦截包以及子包时候子包不生效，必须要明确拼接在execution中才生效，所以这里就解析出来具体的子包然后拼接，原因百思不得其解
        String[] pkgs = AnnotationUtil.getAnnotationValue(ic, EnableArgsPrinter.class);
        String[] excludePkgs = AnnotationUtil.getAnnotationValue(ic, EnableArgsPrinter.class, "exclude");
        Set all = new HashSet<>();
        Set exclude = new HashSet<>();

        // 获取所有包
        for (String pkg : pkgs) {
            String[] ps = ClassPathUtil.resolvePackage(pkg);
            if (!ObjectUtils.isEmpty(ps)) {
                all.addAll(Arrays.asList(ps));
            }
        }

        // 移除需要过滤的包
        for (String pkg : excludePkgs) {
            String[] ps = ClassPathUtil.resolvePackage(pkg);
            exclude.addAll(Arrays.asList(ps));
        }

        // 最终的包: all与exclude求差集
        all.removeAll(exclude);

        if (registry instanceof DefaultListableBeanFactory) {
            DefaultListableBeanFactory factory = (DefaultListableBeanFactory) registry;
            if (!ObjectUtils.isEmpty(all)) {
                Advisor advisor = createAdvisor(all.toArray(new String[0]));
                factory.registerSingleton("argsPrinter", advisor);
            }
        }
    }

    private Advisor createAdvisor(String[] pkg) {
        AspectJExpressionPointcut pointcut = new AspectJExpressionPointcut();
        String execution = Arrays.stream(pkg).map(e -> "execution(* " + e.trim() + ".*.*(..))").collect(Collectors.joining(" || "));
        pointcut.setExpression(execution);

        MethodInterceptor interceptor = invocation -> {
            Object[] args = invocation.getArguments();
            String methodName = invocation.getMethod().getDeclaringClass().getSimpleName() + "." + invocation.getMethod().getName();
            String input = null;
            if (args != null && args.length > 0) {
                List