Drools规则引擎中外部事实状态更新与条件重评估机制

rule "Activate Panic Mode for FMBTHQLA"
when
   $tradeEvent : TradeEvent(bookShortName == "FMBTHQLA")
   p : PanicButtonManager(panicModeEnabled == true) // 问题焦点：此条件未能实时更新
then
   modify ($tradeEvent){
      messageCode = "PM003",
      message = "HQLA: FMBTHQLA is restricted to HQLA mode. Panic status: " + p.isPanicModeEnabled(),
      tradeValidationStatus = STATUS.ERROR
   }
end

rule "Update PanicButtonManager State"
when
   // 假设有某种机制触发此规则，例如一个特定的事件或定时器
   // 或者，当PanicButtonManager的某个辅助字段被修改时
   $p : PanicButtonManager( /* 某些条件 */ )
then
   // 假设这里并没有直接修改 $p 的字段，但我们知道 $p 可能在外部被修改，
   // 或者我们希望强制其重新评估以响应其他规则的修改。
   // 显式调用 update(p) 来通知 Drools 重新评估所有依赖于 $p 的规则。
   update($p);
end

import org.kie.api.runtime.KieSession;
import org.kie.api.runtime.rule.FactHandle;

public class DroolsIntegration {

    private KieSession kieSession;
    private PanicButtonManager panicManager;
    private FactHandle panicManagerFactHandle;

    public DroolsIntegration(KieSession session) {
        this.kieSession = session;
        this.panicManager = new PanicButtonManager(); // 假设这是你的单例或共享对象
        // 将 panicManager 插入到 Drools 工作内存，并获取其 FactHandle
        this.panicManagerFactHandle = kieSession.insert(panicManager);
        kieSession.fireAllRules(); // 首次触发规则
    }

    public void togglePanicMode(boolean enable) {
        // 1. 在Java代码中修改对象状态
        panicManager.setPanicModeEnabled(enable);
        System.out.println("Panic mode set to: " + enable + " (Java side)");

        // 2. 显式通知Drools该事实已更新
        // 必须使用原始的 FactHandle 和更新后的对象
        kieSession.update(panicManagerFactHandle, panicManager);
        System.out.println("Drools engine notified of PanicButtonManager update.");

        // 3. 再次触发规则以响应状态变化
        kieSession.fireAllRules();
    }

    // 假设 PanicButtonManager 类定义
    public static class PanicButtonManager {
        private boolean panicModeEnabled;

        public boolean isPanicModeEnabled() {
            return panicModeEnabled;
        }

        public void setPanicModeEnabled(boolean panicModeEnabled) {
            this.panicModeEnabled = panicModeEnabled;
        }
    }

    // 假设 TradeEvent 类定义
    public static class TradeEvent {
        private String bookShortName;
        private String messageCode;
        private String message;
        private STATUS tradeValidationStatus;

        public TradeEvent(String bookShortName) {
            this.bookShortName = bookShortName;
        }

        public String getBookShortName() { return bookShortName; }
        public void setBookShortName(String bookShortName) { this.bookShortName = bookShortName; }
        public String getMessageCode() { return messageCode; }
        public void setMessageCode(String messageCode) { this.messageCode = messageCode; }
        public String getMessage() { return message; }
        public void setMessage(String message) { this.message = message; }
        public STATUS getTradeValidationStatus() { return tradeValidationStatus; }
        public void setTradeValidationStatus(STATUS tradeValidationStatus) { this.tradeValidationStatus = tradeValidationStatus; }
    }

    public enum STATUS { ERROR, OK }

    public static void main(String[] args) {
        // 假设 KieSession 已正确初始化
        KieSession kieSession = getInitializedKieSession(); // 实际项目中需要构建 KieBase 和 KieSession

        DroolsIntegration app = new DroolsIntegration(kieSession);

        // 插入一个 TradeEvent
        TradeEvent trade = new TradeEvent("FMBTHQLA");
        kieSession.insert(trade);
        kieSession.fireAllRules(); // 此时 panicModeEnabled 为 false，规则不激活

        System.out.println("\n--- Toggling panic mode ON ---");
        app.togglePanicMode(true); // panicModeEnabled 变为 true，规则应激活

        System.out.println("\n--- Toggling panic mode OFF ---");
        app.togglePanicMode(false); // panicModeEnabled 变为 false，规则应不再激活

        kieSession.dispose();
    }

    // 模拟获取 KieSession 的方法
    private static KieSession getInitializedKieSession() {
        // 实际项目中需要通过 KieServices, KieRepository, KieFileSystem 等构建
        // 这里仅为示例，简化处理
        return null; // 替换为实际的 KieSession 初始化代码
    }
}