tuon-framework/zen-orm/src/main/java/a/A.java

package a;

import com.gxzc.zen.orm.DynamicDataSource;
import com.gxzc.zen.orm.DynamicMultipleDataSource;
import kotlin.TypeCastException;
import kotlin.jvm.internal.Intrinsics;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.Signature;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.reflect.MethodSignature;
import org.jetbrains.annotations.NotNull;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;
import java.util.Objects;

@Aspect
@Order(-1)
@Component
public class A {

    @Before("execution(* com.gxzc.zen.api.sys.*.*Service.*(..))")
    public void dataSourceSys(@NotNull JoinPoint joinPoint) {
        Intrinsics.checkParameterIsNotNull(joinPoint, "joinPoint");
        this.setDataSource(joinPoint, "sysDataSource");
    }

    @Before("execution(* com.gxzc.zen.api.rec.*.*Service.*(..))")
    public void dataSourceRec(@NotNull JoinPoint joinPoint) {
        Intrinsics.checkParameterIsNotNull(joinPoint, "joinPoint");
        this.setDataSource(joinPoint, "recDataSource");
    }

    private void setDataSource(JoinPoint joinPoint, String defaultKey) {
        DynamicDataSource dynamicDataSource = this.currentMethod(joinPoint).getAnnotation(DynamicDataSource.class);
        if (Objects.isNull(dynamicDataSource)) {
            DynamicMultipleDataSource.Companion.setDataSource(defaultKey);
        } else {
            DynamicMultipleDataSource.Companion.setDataSource(dynamicDataSource.value());
        }
    }

    private Method currentMethod(JoinPoint joinPoint) {
        Signature var10000 = joinPoint.getSignature();
        if (var10000 == null) {
            throw new TypeCastException("null cannot be cast to non-null type org.aspectj.lang.reflect.MethodSignature");
        } else {
            Method var2 = ((MethodSignature) var10000).getMethod();
            Intrinsics.checkExpressionValueIsNotNull(var2, "(joinPoint.signature as MethodSignature).method");
            return var2;
        }
    }
}