private static void writeBeliefSet(
MarshallerWriteContext context,
BeliefSet beliefSet,
org.drools.core.marshalling.impl.ProtobufMessages.EqualityKey.Builder _key)
throws IOException {
ProtobufMessages.BeliefSet.Builder _beliefSet = ProtobufMessages.BeliefSet.newBuilder();
_beliefSet.setHandleId(beliefSet.getFactHandle().getId());
ObjectMarshallingStrategyStore objectMarshallingStrategyStore =
context.objectMarshallingStrategyStore;
// for ( LinkedListEntry node = (LinkedListEntry) beliefSet.getFirst(); node != null; node =
// (LinkedListEntry) node.getNext() ) {
FastIterator it = beliefSet.iterator();
for (LinkedListEntry node = (LinkedListEntry) beliefSet.getFirst();
node != null;
node = (LinkedListEntry) it.next(node)) {
LogicalDependency belief = (LogicalDependency) node.getObject();
ProtobufMessages.LogicalDependency.Builder _logicalDependency =
ProtobufMessages.LogicalDependency.newBuilder();
// _belief.setActivation( value )
LogicalDependency dependency = (LogicalDependency) node.getObject();
org.drools.core.spi.Activation activation = dependency.getJustifier();
ProtobufMessages.Activation _activation =
ProtobufMessages.Activation.newBuilder()
.setPackageName(activation.getRule().getPackage())
.setRuleName(activation.getRule().getName())
.setTuple(PersisterHelper.createTuple(activation.getTuple()))
.build();
_logicalDependency.setActivation(_activation);
if (belief.getObject() != null) {
ObjectMarshallingStrategy strategy =
objectMarshallingStrategyStore.getStrategyObject(belief.getObject());
Integer index = context.getStrategyIndex(strategy);
_logicalDependency.setObjectStrategyIndex(index.intValue());
_logicalDependency.setObject(
ByteString.copyFrom(
strategy.marshal(
context.strategyContext.get(strategy), context, belief.getObject())));
}
if (belief.getValue() != null) {
ObjectMarshallingStrategy strategy =
objectMarshallingStrategyStore.getStrategyObject(belief.getValue());
Integer index = context.getStrategyIndex(strategy);
_logicalDependency.setValueStrategyIndex(index.intValue());
_logicalDependency.setValue(
ByteString.copyFrom(
strategy.marshal(
context.strategyContext.get(strategy), context, belief.getValue())));
}
_beliefSet.addLogicalDependency(_logicalDependency.build());
}
_key.setBeliefSet(_beliefSet);
}
private static void readBeliefSet(
MarshallerReaderContext context,
TruthMaintenanceSystem tms,
EqualityKey key,
ProtobufMessages.BeliefSet _beliefSet)
throws IOException, ClassNotFoundException {
InternalFactHandle handle = (InternalFactHandle) context.handles.get(_beliefSet.getHandleId());
for (ProtobufMessages.LogicalDependency _logicalDependency :
_beliefSet.getLogicalDependencyList()) {
ProtobufMessages.Activation _activation = _logicalDependency.getActivation();
Activation activation =
(Activation)
context
.filter
.getTuplesCache()
.get(
PersisterHelper.createActivationKey(
_activation.getPackageName(),
_activation.getRuleName(),
_activation.getTuple()))
.getObject();
Object object = null;
ObjectMarshallingStrategy strategy = null;
if (_logicalDependency.hasObjectStrategyIndex()) {
strategy = context.usedStrategies.get(_logicalDependency.getObjectStrategyIndex());
object =
strategy.unmarshal(
context.strategyContexts.get(strategy),
context,
_logicalDependency.getObject().toByteArray(),
(context.ruleBase == null) ? null : context.ruleBase.getRootClassLoader());
}
Object value = null;
if (_logicalDependency.hasValueStrategyIndex()) {
strategy = context.usedStrategies.get(_logicalDependency.getValueStrategyIndex());
value =
strategy.unmarshal(
context.strategyContexts.get(strategy),
context,
_logicalDependency.getValue().toByteArray(),
(context.ruleBase == null) ? null : context.ruleBase.getRootClassLoader());
}
ObjectTypeConf typeConf =
context
.wm
.getObjectTypeConfigurationRegistry()
.getObjectTypeConf(
((NamedEntryPoint) handle.getEntryPoint()).getEntryPoint(), handle.getObject());
tms.readLogicalDependency(
handle,
object,
value,
activation,
activation.getPropagationContext(),
activation.getRule(),
typeConf);
}
}
