cn.imaq.autumn.aop.scanner.AspectBeanScanner Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of autumn-aop Show documentation

Aspect Oriented Programming support for Autumn

There is a newer version: 2020.10

package cn.imaq.autumn.aop.scanner;

import cn.imaq.autumn.aop.AopContext;
import cn.imaq.autumn.aop.HookModel;
import cn.imaq.autumn.aop.annotation.Aspect;
import cn.imaq.autumn.aop.annotation.Hook;
import cn.imaq.autumn.core.beans.BeanInfo;
import cn.imaq.autumn.core.beans.creator.NormalBeanCreator;
import cn.imaq.autumn.core.beans.scanner.BeanScanner;
import cn.imaq.autumn.core.context.AutumnContext;
import io.github.lukehutch.fastclasspathscanner.scanner.ScanSpec;
import lombok.extern.slf4j.Slf4j;

import java.lang.reflect.Method;

@Slf4j
public class AspectBeanScanner implements BeanScanner {
    @Override
    public void process(ScanSpec spec, AutumnContext context) {
        AopContext aopContext = AopContext.getFrom(context);
        spec.matchClassesWithAnnotation(Aspect.class, cls -> {
            Aspect anno = cls.getAnnotation(Aspect.class);
            String name = anno.value();
            if (name.isEmpty()) {
                name = cls.getSimpleName().toLowerCase();
            }
            context.addBeanInfo(BeanInfo.builder()
                    .name(name)
                    .type(cls)
                    .singleton(true)
                    .creator(new NormalBeanCreator(cls))
                    .build());

            for (Method method : cls.getDeclaredMethods()) {
                if (method.isAnnotationPresent(Hook.class)) {
                    method.setAccessible(true);
                    Hook hookAnno = method.getAnnotation(Hook.class);
                    log.info("Adding hook {}", hookAnno);
                    aopContext.addHook(new HookModel(hookAnno.value(), hookAnno.exclude(), method));
                }
            }
        });
    }
}