Я пытаюсь написать движок на основе правил, используя 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