xyz.erupt.core.proxy.EruptProxy Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of erupt-core Show documentation
core realize
There is a newer version: 1.12.17
package xyz.erupt.core.proxy;

import lombok.SneakyThrows;
import org.aopalliance.intercept.MethodInvocation;
import xyz.erupt.annotation.Erupt;
import xyz.erupt.annotation.sub_erupt.Drill;
import xyz.erupt.annotation.sub_erupt.Filter;
import xyz.erupt.annotation.sub_erupt.RowOperation;
import xyz.erupt.core.invoke.ExprInvoke;
import xyz.erupt.core.proxy.erupt.DrillProxy;
import xyz.erupt.core.proxy.erupt.FilterProxy;
import xyz.erupt.core.proxy.erupt.RowOperationProxy;

import java.util.ArrayList;
import java.util.List;

/**
 * @author YuePeng
 * date 2022/2/5 14:19
 */
public class EruptProxy extends AnnotationProxy {

    @Override
    @SneakyThrows
    protected Object invocation(MethodInvocation invocation) {
        switch (invocation.getMethod().getName()) {
            case "filter":
                Filter[] filters = this.rawAnnotation.filter();
                Filter[] proxyFilters = new Filter[filters.length];
                for (int i = 0; i < filters.length; i++) {
                    proxyFilters[i] = AnnotationProxyPool.getOrPut(filters[i], filter ->
                            new FilterProxy().newProxy(filter, this)
                    );
                }
                return proxyFilters;
            case "rowOperation":
                RowOperation[] rowOperations = this.rawAnnotation.rowOperation();
                List proxyOperations = new ArrayList<>();
                for (RowOperation rowOperation : rowOperations) {
                    if (ExprInvoke.getExpr(rowOperation.show())) {
                        proxyOperations.add(AnnotationProxyPool.getOrPut(rowOperation, it ->
                                new RowOperationProxy().newProxy(it, this)
                        ));
                    }
                }
                return proxyOperations.toArray(new RowOperation[0]);
            case "drills":
                Drill[] drills = this.rawAnnotation.drills();
                List proxyDrills = new ArrayList<>();
                for (Drill drill : drills) {
                    if (ExprInvoke.getExpr(drill.show())) {
                        proxyDrills.add(AnnotationProxyPool.getOrPut(drill, it ->
                                new DrillProxy().newProxy(it, this)
                        ));
                    }
                }
                return proxyDrills.toArray(new Drill[0]);
        }
        return this.invoke(invocation);
    }

}