Как написать Stateful и основанные на времени правила в Drools? - PullRequest
0 голосов
/ 24 января 2019

Я пытаюсь написать движок на основе правил, используя Drools. Правила идут так:

Правило «Тревога»: если статус «Тревога», немедленно отправьте уведомление.

Правило «Предупреждение»: если статус «Предупреждение», сохраните идентификатор устройства в памяти и подождите 5 минут. Если в течение 5 минут будет получено другое сообщение с таким же идентификатором устройства и статусом «Разрешено», отмените это правило. В противном случае измените статус на «Тревога» и активируйте правило «Тревога».

Правило «Разрешено»: если состояние «Разрешено» и идентификатор устройства уже находится в памяти, очистите «Предупреждение» для этого идентификатора устройства, т.е. удалите идентификатор устройства из памяти.

Правило Alarm простое и работает. Но правила Warning и Resolved явно не работают в моем коде. Как написать такие правила в Drools?

Вот код, который я написал до сих пор:

DroolsTest.java

package com.sample;

import org.drools.KnowledgeBase;
import org.drools.KnowledgeBaseFactory;
import org.drools.builder.KnowledgeBuilder;
import org.drools.builder.KnowledgeBuilderError;
import org.drools.builder.KnowledgeBuilderErrors;
import org.drools.builder.KnowledgeBuilderFactory;
import org.drools.builder.ResourceType;
import org.drools.io.ResourceFactory;
import org.drools.logger.KnowledgeRuntimeLogger;
import org.drools.logger.KnowledgeRuntimeLoggerFactory;
import org.drools.runtime.StatefulKnowledgeSession;

public class DroolsTest {

    public static final void main(String[] args) {
        try {
            // load up the knowledge base
            KnowledgeBase kbase = readKnowledgeBase();
            StatefulKnowledgeSession ksession = kbase.newStatefulKnowledgeSession();
            KnowledgeRuntimeLogger logger = KnowledgeRuntimeLoggerFactory.newFileLogger(ksession, "test");
            // go !
            MachineMessage[] messages = new MachineMessage[4]; 
            messages[0] = new MachineMessage("1", "Warning");
            messages[1] = new MachineMessage("2", "Alarm");
            messages[2] = new MachineMessage("3", "Alarm");
            messages[3] = new MachineMessage("1", "Resolved");

            for(int i = 0; i < messages.length; i++)
            {
                ksession.insert(messages[i]);
                ksession.fireAllRules();
                Thread.sleep(9000);
            }
            logger.close();
        } catch (Throwable t) {
            t.printStackTrace();
        }
    }

    private static KnowledgeBase readKnowledgeBase() throws Exception {
        KnowledgeBuilder kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
        kbuilder.add(ResourceFactory.newClassPathResource("Sample.drl"), ResourceType.DRL);
        KnowledgeBuilderErrors errors = kbuilder.getErrors();
        if (errors.size() > 0) {
            for (KnowledgeBuilderError error: errors) {
                System.err.println(error);
            }
            throw new IllegalArgumentException("Could not parse knowledge.");
        }
        KnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
        kbase.addKnowledgePackages(kbuilder.getKnowledgePackages());
        return kbase;
    }

    public static class MachineMessage {

        private String deviceID;
        private String status;

        public MachineMessage() { }
        public MachineMessage(String deviceID, String status) {
            this.deviceID = deviceID;
            this.status = status;           
        }

        public String getDeviceID() {
            return this.deviceID;
        }

        public void setDeviceID(String deviceID) {
            this.deviceID = deviceID;
        }

        public String getStatus() {
            return this.status;
        }

        public void setStatus(String status) {
            this.status = status;
        }
    }
}

Sample.drl

package com.sample

import com.sample.DroolsTest.MachineMessage;

rule "Alarm"
when
    m : MachineMessage( status == "Alarm", myDeviceID : deviceID )
then
    System.out.println("Send notification to technician, device in Alarm state. Device ID: " +  myDeviceID );
end

rule "Warning"
when
    m : MachineMessage( status == "Warning", myDeviceID : deviceID )
then
    System.out.println("Wait for 5 minutes, device in Warning state. Device ID: " + myDeviceID);        
    // if we get Machine Message with the same device ID and Resolved status in 5 minutes then 
    // we'll not trigger the following line
    m.setStatus("Alarm");
    update(m);
end

rule "Resolved"
when
    m : MachineMessage( status == "Resolved", myDeviceID : deviceID )
then
    System.out.println("Device has resolved its warning state. Device ID: " + myDeviceID);

end

Фактический вывод, который я получаю, таков:

    Wait for 5 minutes, device in Warning state. Device ID: 1
    Send notification to technician, device in Alarm state. Device ID: 1
    Send notification to technician, device in Alarm state. Device ID: 2
    Send notification to technician, device in Alarm state. Device ID: 3
    Device has resolved its warning state. Device ID: 1

Тогда как я хотел:

    Wait for 5 minutes, device in Warning state. Device ID: 1
    Send notification to technician, device in Alarm state. Device ID: 2
    Send notification to technician, device in Alarm state. Device ID: 3
    Device has resolved its warning state. Device ID: 1

1 Ответ

0 голосов
/ 25 января 2019

Согласно документации Drools по CEP События должны быть неизменными:

Рекомендуется, чтобы приложению разрешалось заполнять незаполненные атрибуты события (для обогащения события выведенными данными), но уже заполненные атрибуты никогда не следует изменять.

Очевидно, что правило «Предупреждение» уже не соответствует 1-му требованию CEP, как это предложено OP (изменяя поле «статус»).

Потенциальное предложение будет выглядеть следующим образом:

rule "Warning"
when
    m : MachineMessage( status == "Warning", myDeviceID : deviceID )
    not( MachineMessage( status == "Resolved", deviceID == m.deviceID, this after[0s, 30s] m) )
then
    System.out.println("I waited for 5 minutes, device in Warning state not followed by Resolved. Device ID: " + myDeviceID);        
    insert(new MachineMessage(m.deviceID, "Alarm"));
end

, который НЕ изменял бы никакие оцененные поля существующих Событий, и фактически делал бы вывод знаний CEP, пытаясь соответствовать бизнес-требованиям, описанным OP.

В целом пример CEP в руководстве Drools о спринклерной / пожарной сигнализации , похоже, соответствует бизнес-требованиям, описанным в OP.

...