Examples with MarshallerWriteContext org.drools.core.marshalling.impl.MarshallerWriteContext used on opensource projects

Example 1 with MarshallerWriteContext

use of org.drools.core.marshalling.impl.MarshallerWriteContext in project drools by kiegroup.

the class WorkItemInfo method transform.

// @PreUpdate
@Override
public void transform() {
    this.state = workItem.getState();
    ByteArrayOutputStream baos = new ByteArrayOutputStream();
    try {
        MarshallerWriteContext context = new MarshallerWriteContext(baos, null, null, null, null, this.env);
        ProtobufOutputMarshaller.writeWorkItem(context, workItem);
        context.close();
        this.workItemByteArray = baos.toByteArray();
    } catch (IOException e) {
        throw new IllegalArgumentException("IOException while storing workItem " + workItem.getId() + ": " + e.getMessage());
    }
}

Example 2 with MarshallerWriteContext

use of org.drools.core.marshalling.impl.MarshallerWriteContext in project drools by kiegroup.

the class FactHandleMarshallingTest method backwardsCompatibleEventFactHandleTest.

@Test
public void backwardsCompatibleEventFactHandleTest() throws IOException, ClassNotFoundException {
    InternalKnowledgeBase kBase = createKnowledgeBase();
    StatefulKnowledgeSessionImpl wm = createWorkingMemory(kBase);
    InternalFactHandle factHandle = createEventFactHandle(wm, kBase);
    // marshall/serialize workItem
    byte[] byteArray;
    {
        ObjectMarshallingStrategy[] strats = new ObjectMarshallingStrategy[] { MarshallerFactory.newSerializeMarshallingStrategy(), new MarshallerProviderImpl().newIdentityMarshallingStrategy() };
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        MarshallerWriteContext outContext = new MarshallerWriteContext(baos, null, null, null, new ObjectMarshallingStrategyStoreImpl(strats), true, true, null);
        OldOutputMarshallerMethods.writeFactHandle_v1(outContext, (ObjectOutputStream) outContext, outContext.objectMarshallingStrategyStore, 2, factHandle);
        outContext.close();
        byteArray = baos.toByteArray();
    }
    // unmarshall/deserialize workItem
    InternalFactHandle newFactHandle;
    {
        // Only put serialization strategy in
        ObjectMarshallingStrategy[] newStrats = new ObjectMarshallingStrategy[] { MarshallerFactory.newSerializeMarshallingStrategy() };
        ByteArrayInputStream bais = new ByteArrayInputStream(byteArray);
        MarshallerReaderContext inContext = new MarshallerReaderContext(bais, null, null, new ObjectMarshallingStrategyStoreImpl(newStrats), Collections.EMPTY_MAP, true, true, null);
        inContext.wm = wm;
        newFactHandle = InputMarshaller.readFactHandle(inContext);
        inContext.close();
    }
    assertTrue("Serialized FactHandle not the same as the original.", compareInstances(factHandle, newFactHandle));
}