com.bizmda.bizsip.integrator.checkrule.FieldCheckRuleHelper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of integrator-spring-boot-starter Show documentation

There is a newer version: 1.0.0.Beta9

Show newest version

package com.bizmda.bizsip.integrator.checkrule;

import cn.hutool.core.thread.ThreadUtil;
import cn.hutool.json.JSONObject;
import com.bizmda.bizsip.common.BizException;
import com.bizmda.bizsip.common.BizResultEnum;
import lombok.extern.slf4j.Slf4j;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.*;

@Slf4j
public class FieldCheckRuleHelper {
    private FieldCheckRuleHelper() {

    }

    private static ExecutorService fieldAssertExecutorService = ThreadUtil.newExecutor(Runtime.getRuntime().availableProcessors());

    public static List checkFieldRule(JSONObject jsonObject, List fieldCheckRuleList, CheckMode checkMode) throws BizException {
        CompletionService service = new ExecutorCompletionService<>(fieldAssertExecutorService);
        List> futureList = new ArrayList<>();
        for (FieldCheckRule fieldCheckRule : fieldCheckRuleList) {
            Future future = service.submit(new FieldCheckRuleThread(jsonObject, fieldCheckRule));
            futureList.add(future);
        }
        List fieldChcekRuleResultList = new ArrayList<>();
        for (int i = 0; i < fieldCheckRuleList.size(); i++) {
            if (takeFieldCheckRuleResult(checkMode, service, futureList, fieldChcekRuleResultList))
                return fieldChcekRuleResultList;
        }
        return fieldChcekRuleResultList;
    }

    private static boolean takeFieldCheckRuleResult(CheckMode checkMode, CompletionService service, List> futureList, List fieldChcekRuleResultList) throws BizException {
        Future take = null;
        try {
            take = service.take();
        } catch (InterruptedException e) {
            log.error("域校验计算被中断",e);
            Thread.currentThread().interrupt();
            throw new BizException(BizResultEnum.CHECKRULE_FIELD_CHECK_THREAD_ERROR, e);
        }
        FieldChcekRuleResult fieldChcekRuleResult = null;
        try {
            fieldChcekRuleResult = take.get();
        }
        catch (InterruptedException e) {
            log.error("域校验计算被中断",e);
            Thread.currentThread().interrupt();
            throw new BizException(BizResultEnum.CHECKRULE_FIELD_CHECK_THREAD_ERROR, e);
        }
        catch (ExecutionException e) {
            throw new BizException(BizResultEnum.CHECKRULE_FIELD_CHECK_THREAD_ERROR, e);
        }
        if (fieldChcekRuleResult != null) {
            if (checkMode == CheckMode.ONE) {
                for (Future future : futureList) {
                    future.cancel(true);
                }
                fieldChcekRuleResultList.add(fieldChcekRuleResult);
                return true;
            }
            else {
                fieldChcekRuleResultList.add(fieldChcekRuleResult);
            }
        }
        return false;
    }
}