@Service
@Slf4j
public class KieServiceImpl implements KieService {

    private StatelessKieSession kieSession;

    private KnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();

    @Resource
    private RiskConfigService riskConfigService;
    //商品规则处理
    @Resource
    private ProductRiskService productRiskService;

    @Override
    public void loadRules(String ruleCode) {
        RiskConfig riskConfig = riskConfigService.queryRiskConfig(ruleCode);
        if(riskConfig!=null){
            loadRule(ruleCode,riskConfig.getRiskContent());
        }
    }

    
    @Override
    public void removeRule(String ruleName) {
        if (kbase.getRule("rules", ruleName) != null) {
            kbase.removeRule("rules", ruleName);
            log.info("remove rule: rules." + ruleName);
            kieSession = kbase.newStatelessKieSession();
            printRules();
        } else {
            log.error("no rule: rules." + ruleName);
        }
    }

    
    @Override
    public void execute(Object object) {
        this.kieSession.execute(object);
    }

    
    @PostConstruct
    public void init() {
        loadRulesAll();
    }

    
    private void loadRulesAll() {
        RiskConfig config = new RiskConfig();
        List<RiskConfig> riskConfigList = riskConfigService.queryRiskConfigs(config);
        if(riskConfigList!=null){
            for(RiskConfig riskConfig : riskConfigList){
                loadRule(riskConfig.getRiskCode(),riskConfig.getRiskContent());
            }
        }
    }

    
    private void loadRule(String ruleCode,String ruleContent){
        try {
            KnowledgeBuilder kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
            kbuilder.add(ResourceFactory.newByteArrayResource(ruleContent.getBytes()), ResourceType.DRL);
            if(kbuilder.hasErrors()){
                log.error("加载规则模板引擎异常{},{}",ruleCode,kbuilder.getErrors());
                return;
            }
            kbase.addKnowledgePackages(kbuilder.getKnowledgePackages());
            kieSession = kbase.newStatelessKieSession();
            printRules();
            setGlobal();
        }catch (Exception e){
            log.error("加载规则模板引擎异常",e);
        }
    }

    
    private void setGlobal(){
        kieSession.setGlobal("productRiskService", productRiskService);
    }

    
    private void printRules() {
        log.info("print rule start: ---------");
       kbase.getKnowledgePackages().forEach(knowledgePackage ->
                knowledgePackage.getRules().forEach(rule ->
                        log.info("print rule: " + knowledgePackage.getName() + "." + rule.getName())));
        log.info("print rule end: ---------");
    }
}