* {@inheritDoc}
*/
@Override
public void handleOperation(Exchange exchange, KnowledgeOperation operation) throws HandlerException {
Integer sessionId = null;
Message inputMessage = exchange.getMessage();
ExchangePattern exchangePattern = exchange.getContract().getProviderOperation().getExchangePattern();
Map<String, Object> expressionVariables = new HashMap<String, Object>();
RulesOperationType operationType = (RulesOperationType)operation.getType();
switch (operationType) {
case EXECUTE: {
KnowledgeSession session = newStatelessSession();
sessionId = session.getId();
setGlobals(inputMessage, operation, session);
if (ExchangePattern.IN_ONLY.equals(exchangePattern)) {
List<Object> facts = getInputList(inputMessage, operation, session);
session.getStateless().execute(facts);
} else if (ExchangePattern.IN_OUT.equals(exchangePattern)) {
KieCommands cmds = KieServices.Factory.get().getCommands();
List<Command<?>> batch = new ArrayList<Command<?>>();
Map<String, Object> inouts = getInputOutputMap(inputMessage, operation, session);
for (Entry<String, Object> inout : inouts.entrySet()) {
batch.add(cmds.newInsert(inout.getValue(), inout.getKey()));
}
List<Object> facts = getInputOnlyList(inputMessage, operation, session);
batch.add(cmds.newInsertElements(facts));
batch.add(cmds.newFireAllRules());
BatchExecutionCommand exec = cmds.newBatchExecution(batch, KnowledgeConstants.RESULT);
ExecutionResults results = session.getStateless().execute(exec);
for (String id : inouts.keySet()) {
expressionVariables.put(id, results.getValue(id));
}
expressionVariables.putAll(getGlobalVariables(session));
}
break;
}
case INSERT:
case FIRE_ALL_RULES: {
/*
if (!isContinue(exchange)) {
disposeStatefulSession();
}
*/
KnowledgeSession session = getStatefulSession();
sessionId = session.getId();
setGlobals(inputMessage, operation, session);
List<Object> facts = getInputList(inputMessage, operation, session);
for (Object fact : facts) {
session.getStateful().insert(fact);
}
if (RulesOperationType.FIRE_ALL_RULES.equals(operationType)) {
session.getStateful().fireAllRules();
}
if (ExchangePattern.IN_OUT.equals(exchangePattern)) {
expressionVariables.putAll(getGlobalVariables(session));
}
if (isDispose(exchange, inputMessage)) {
disposeStatefulSession();
}
break;
}
case FIRE_UNTIL_HALT: {
/*
if (!isContinue(exchange, inputMessage)) {
disposeStatefulSession();
}
*/
KnowledgeSession session = getStatefulSession();
sessionId = session.getId();
setGlobals(inputMessage, operation, session);
if (_fireUntilHaltThread == null) {
ClassLoader fireUntilHaltLoader = Classes.getTCCL();
if (fireUntilHaltLoader == null) {
fireUntilHaltLoader = getLoader();
}
FireUntilHalt fireUntilHalt = new FireUntilHalt(this, session, fireUntilHaltLoader);
session.addDisposals(fireUntilHalt);
_fireUntilHaltThread = fireUntilHalt.startThread();
}
final String undefinedVariable = toVariable(exchange);
Map<String, List<Object>> inputMap = getListMap(inputMessage, operation.getInputExpressionMappings(), true, undefinedVariable);
if (inputMap.size() > 0) {
for (Entry<String, List<Object>> inputEntry : inputMap.entrySet()) {
String key = inputEntry.getKey();
if (undefinedVariable.equals(key)) {
String eventId = Strings.trimToNull(operation.getEventId());
if (eventId != null) {
key = eventId;
}
}
List<Object> facts = inputEntry.getValue();
if (undefinedVariable.equals(key)) {
for (Object fact : facts) {
session.getStateful().insert(fact);
}
} else {
EntryPoint entryPoint = session.getStateful().getEntryPoint(key);
if (entryPoint != null) {
for (Object fact : facts) {
entryPoint.insert(fact);
}
} else {
throw RulesMessages.MESSAGES.unknownEntryPoint(key);
}
}
}
} else {
List<Object> facts = getInputList(inputMessage, operation, session);
for (Object fact : facts) {
session.getStateful().insert(fact);
}
}
if (ExchangePattern.IN_OUT.equals(exchangePattern)) {
expressionVariables.putAll(getGlobalVariables(session));
}
if (isDispose(exchange, inputMessage)) {
disposeStatefulSession();
}
break;
}
default: {
throw RulesMessages.MESSAGES.unsupportedOperationType(operationType.toString());
}
}
if (ExchangePattern.IN_OUT.equals(exchangePattern)) {
Message outputMessage = exchange.createMessage();
Context outputContext = exchange.getContext(outputMessage);
if (sessionId != null && sessionId.intValue() > 0) {
outputContext.setProperty(RulesConstants.SESSION_ID_PROPERTY, sessionId);
}
setFaults(outputMessage, operation, expressionVariables);
if (outputMessage.getContent() != null) {
exchange.sendFault(outputMessage);
} else {
setOutputs(outputMessage, operation, expressionVariables);
exchange.send(outputMessage);
}