/*
* (non-Javadoc)
*
* @see org.kie.common.AgendaI#focusStackSize()
*/
public int focusStackSize() {
int size = 0;
for (final AgendaGroup group : this.focusStack) {
size += group.size();
}
return size;
}
/*
* (non-Javadoc)
*
* @see org.kie.common.AgendaI#getActivations()
*/
public Activation[] getActivations() {
final List<Activation> list = new ArrayList<Activation>();
for (final Iterator<InternalAgendaGroup> it = this.agendaGroups.values().iterator();
it.hasNext(); ) {
final AgendaGroup group = it.next();
for (Match activation : group.getActivations()) {
list.add((Activation) activation);
}
}
return list.toArray(new Activation[list.size()]);
}
public void setFocus(final PropagationContext ctx, final String name) {
AgendaGroup agendaGroup = getAgendaGroup(name);
agendaGroup.setAutoFocusActivator(ctx);
setFocus(agendaGroup);
}
public void addAgendaGroup(final AgendaGroup agendaGroup) {
this.agendaGroups.put(agendaGroup.getName(), (InternalAgendaGroup) agendaGroup);
}
public int compare(AgendaGroup group1, AgendaGroup group2) {
return group1.getName().compareTo(group2.getName());
}
private static void writeAgenda(
MarshallerWriteContext context, ProtobufMessages.RuleData.Builder _ksb) throws IOException {
InternalWorkingMemory wm = context.wm;
InternalAgenda agenda = (InternalAgenda) wm.getAgenda();
org.drools.core.marshalling.impl.ProtobufMessages.Agenda.Builder _ab =
ProtobufMessages.Agenda.newBuilder();
AgendaGroup[] agendaGroups =
(AgendaGroup[])
agenda
.getAgendaGroupsMap()
.values()
.toArray(new AgendaGroup[agenda.getAgendaGroupsMap().size()]);
Arrays.sort(agendaGroups, AgendaGroupSorter.instance);
for (AgendaGroup ag : agendaGroups) {
AgendaGroupQueueImpl group = (AgendaGroupQueueImpl) ag;
org.drools.core.marshalling.impl.ProtobufMessages.Agenda.AgendaGroup.Builder _agb =
ProtobufMessages.Agenda.AgendaGroup.newBuilder();
_agb.setName(group.getName())
.setIsActive(group.isActive())
.setIsAutoDeactivate(group.isAutoDeactivate())
.setClearedForRecency(group.getClearedForRecency())
.setHasRuleFlowLister(group.isRuleFlowListener())
.setActivatedForRecency(group.getActivatedForRecency());
Map<Long, String> nodeInstances = group.getNodeInstances();
for (Map.Entry<Long, String> entry : nodeInstances.entrySet()) {
org.drools.core.marshalling.impl.ProtobufMessages.Agenda.AgendaGroup.NodeInstance.Builder
_nib = ProtobufMessages.Agenda.AgendaGroup.NodeInstance.newBuilder();
_nib.setProcessInstanceId(entry.getKey());
_nib.setNodeInstanceId(entry.getValue());
_agb.addNodeInstance(_nib.build());
}
_ab.addAgendaGroup(_agb.build());
}
org.drools.core.marshalling.impl.ProtobufMessages.Agenda.FocusStack.Builder _fsb =
ProtobufMessages.Agenda.FocusStack.newBuilder();
LinkedList<AgendaGroup> focusStack = agenda.getStackList();
for (Iterator<AgendaGroup> it = focusStack.iterator(); it.hasNext(); ) {
AgendaGroup group = it.next();
_fsb.addGroupName(group.getName());
}
_ab.setFocusStack(_fsb.build());
// serialize all dormant activations
org.drools.core.util.Iterator it = ActivationIterator.iterator(wm);
List<org.drools.core.spi.Activation> dormant = new ArrayList<org.drools.core.spi.Activation>();
for (org.drools.core.spi.Activation item = (org.drools.core.spi.Activation) it.next();
item != null;
item = (org.drools.core.spi.Activation) it.next()) {
if (!item.isQueued()) {
dormant.add(item);
}
}
Collections.sort(dormant, ActivationsSorter.INSTANCE);
for (org.drools.core.spi.Activation activation : dormant) {
_ab.addMatch(writeActivation(context, (AgendaItem) activation));
}
// serialize all network evaluator activations
for (Activation activation : agenda.getActivations()) {
if (activation.isRuleAgendaItem()) {
// serialize it
_ab.addRuleActivation(writeActivation(context, (AgendaItem) activation));
}
}
_ksb.setAgenda(_ab.build());
}
