package com.irdstudio.basic.framework.core.annotation.checkAnnotation;

import java.lang.annotation.*;

/**
 * 需求編號：1=1校驗
 * 問題編號：條件查詢檢查可選條件是否全部為空
 * 開發人員：【xieziwei】
 * 創建日期：【2021/3/4 17:05】
 * 功能描述：
 */
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
public @interface QueryParamsNullCheck {

	/**
	 * 校驗對象
	 *
	 * @return
	 */
	String objName() default "";

	/**
	 * 需要校驗的字段名字符串數組
	 *
	 * @return
	 */
	String[] queryParamNames() default {};
}

package com.irdstudio.basic.framework.core.annotation.checkAnnotation;

import com.irdstudio.basic.framework.core.exception.ValidateException;
import com.irdstudio.basic.framework.core.util.StringUtil;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.reflect.MethodSignature;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

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

/**
 * @author xieziwei
 */
@Aspect
@Component
public class QueryParamNullCheckAspect {

    private Logger logger = LoggerFactory.getLogger(QueryParamNullCheckAspect.class);

    @Around("@annotation(com.irdstudio.basic.framework.core.annotation.checkAnnotation.QueryParamsNullCheck)")
    public Object around(ProceedingJoinPoint joinPoint) throws Throwable {
        logger.info("查詢參數是否空切面校驗開始>>>>>>>>>>>>>>>>>>>>>>>>");
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();
        Method method = signature.getMethod();
        //獲取函數的參數名
        String[] parameterNames = signature.getParameterNames();
        //獲取redislock注解對象
        QueryParamsNullCheck annotation = method.getAnnotation(QueryParamsNullCheck.class);
        //獲取校驗對象的參數名
        String objName = annotation.objName();
        //獲取校驗字段的字段名
        String[] ParamNames = annotation.queryParamNames();
        //獲取校驗對象的參數值(這里只有一個，故取數組的第一個)
        Object obj = joinPoint.getArgs()[0];
        try {
            // 取對象的所有方法；
            Method[] methods = obj.getClass().getDeclaredMethods();
            Map<String,String> attrMap = new HashMap<>();
            if (methods.length > 0){
                // 遍歷數組，保存 屬性名-屬性值 到map中
                for (Method item : methods){
                    if (item.getName().startsWith("get")){
                        Object o = item.invoke(obj);
                        String attrName = item.getName().substring(3);
                        String attrValue = "";
                        if (Objects.nonNull(o) && StringUtil.isNotEmpty(o.toString())){
                            attrValue = o.toString();
                        }
                        attrMap.put(attrName,attrValue);
                    }
                }
                // 遍歷map 比較不等參數后的屬性是否為空
                if (attrMap.size() > 0){
                    for (String col : ParamNames){
                        for (Map.Entry<String,String> entry : attrMap.entrySet()){
                            if (entry.getKey().equalsIgnoreCase(col)){
                                if (StringUtil.isNotEmpty(entry.getValue())){
                                    logger.info("查詢參數是否空切面校驗通過>>>>>>>>>>>>>>>>>>>>>>>>");
                                    try {
                                        return joinPoint.proceed();
                                    } catch (Throwable throwable) {
                                        throwable.printStackTrace();
                                    }
                                }
                            }
                        }
                    }
                    logger.info("查詢參數是否全空切面校驗不通過>>>>>>>>>>>>>>>>>>>>>>>>");
                    throw new Throwable();
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        logger.info("查詢參數是否全空切面校驗不通過>>>>>>>>>>>>>>>>>>>>>>>>");
        throw new Throwable();
    }
}