use of org.teiid.translator.marshallers.G2 in project teiid by teiid.
the class TestTeiidTableMarsheller method buildG2.
private G2 buildG2() {
G3 g3 = new G3();
g3.setE1(1);
g3.setE2("bar");
G4 g41 = new G4();
g41.setE1(1);
g41.setE2("hello");
G4 g42 = new G4();
g42.setE1(2);
g42.setE2("world");
G2 g2 = new G2();
g2.setE1(1);
g2.setE2("foo");
g2.setG3(g3);
g2.setG4(Arrays.asList(g41, g42));
g2.setE5("Hello Infinispan".getBytes());
g2.setE6(new Timestamp(1489835322801L));
return g2;
}
use of org.teiid.translator.marshallers.G2 in project teiid by teiid.
the class TestTeiidTableMarsheller method testReadComplex.
@Test
public void testReadComplex() throws Exception {
IckleConversionVisitor visitor = helpExecute("select * from G2");
TeiidTableMarsheller readMarshaller = new TeiidTableMarsheller(ProtobufMetadataProcessor.getMessageName(visitor.getParentTable()), MarshallerBuilder.getWireMap(visitor.getParentTable(), visitor.getMetadata()));
SerializationContext ctx = ProtobufUtil.newSerializationContext(Configuration.builder().build());
ctx.registerProtoFiles(FileDescriptorSource.fromString("tables.proto", ObjectConverterUtil.convertFileToString(UnitTestUtil.getTestDataFile("tables.proto"))));
ctx.registerMarshaller(new G3Marshaller());
ctx.registerMarshaller(new G4Marshaller());
G2Marshaller writeMarshaller = new G2Marshaller() {
@Override
public G2 readFrom(ProtoStreamReader reader) throws IOException {
throw new RuntimeException("Use Teiid marshaller for reading for this test..");
}
};
G2 g2 = buildG2();
// this is used for writing, if reading is being attempted then fail
ctx.registerMarshaller(writeMarshaller);
ByteArrayOutputStream baos = new ByteArrayOutputStream();
RawProtoStreamWriter out = RawProtoStreamWriterImpl.newInstance(baos);
WrappedMessage.writeMessage(ctx, out, g2);
out.flush();
baos.flush();
ctx.unregisterMarshaller(writeMarshaller);
ctx.registerMarshaller(readMarshaller);
RawProtoStreamReader in = RawProtoStreamReaderImpl.newInstance(baos.toByteArray());
InfinispanDocument result = WrappedMessage.readMessage(ctx, in);
// System.out.println(result.flatten());
assertG2(result);
ctx.unregisterMarshaller(readMarshaller);
}
use of org.teiid.translator.marshallers.G2 in project teiid by teiid.
the class TestTeiidTableMarsheller method testWriteComplex.
@Test
public void testWriteComplex() throws Exception {
IckleConversionVisitor visitor = helpExecute("select * from G2");
TeiidTableMarsheller writeMarshaller = new TeiidTableMarsheller(ProtobufMetadataProcessor.getMessageName(visitor.getParentTable()), MarshallerBuilder.getWireMap(visitor.getParentTable(), visitor.getMetadata()));
SerializationContext ctx = ProtobufUtil.newSerializationContext(Configuration.builder().build());
ctx.registerProtoFiles(FileDescriptorSource.fromString("tables.proto", ObjectConverterUtil.convertFileToString(UnitTestUtil.getTestDataFile("tables.proto"))));
ctx.registerMarshaller(new G3Marshaller());
ctx.registerMarshaller(new G4Marshaller());
G2Marshaller readMarshaller = new G2Marshaller() {
@Override
public void writeTo(ProtoStreamWriter writer, G2 g2) throws IOException {
throw new RuntimeException("Use Teiid marshaller for writing for this test..");
}
};
InfinispanDocument g2 = buildG2(visitor);
ctx.registerMarshaller(writeMarshaller);
ByteArrayOutputStream baos = new ByteArrayOutputStream();
RawProtoStreamWriter out = RawProtoStreamWriterImpl.newInstance(baos);
WrappedMessage.writeMessage(ctx, out, g2);
out.flush();
baos.flush();
ctx.unregisterMarshaller(writeMarshaller);
// this is used for writing, if reading is being attempted then fail
ctx.registerMarshaller(readMarshaller);
RawProtoStreamReader in = RawProtoStreamReaderImpl.newInstance(baos.toByteArray());
G2 result = WrappedMessage.readMessage(ctx, in);
ctx.unregisterMarshaller(readMarshaller);
assertEquals(buildG2(), result);
}
use of org.teiid.translator.marshallers.G2 in project teiid by teiid.
the class TestTeiidTableMarsheller method testMarshallWithComplexNative.
// this is for sanity debugging while writing the protocol decoder.
@Test
public void testMarshallWithComplexNative() throws Exception {
SerializationContext ctx = ProtobufUtil.newSerializationContext(Configuration.builder().build());
ctx.registerProtoFiles(FileDescriptorSource.fromString("tables.proto", ObjectConverterUtil.convertFileToString(UnitTestUtil.getTestDataFile("tables.proto"))));
ctx.registerMarshaller(new G3Marshaller());
ctx.registerMarshaller(new G4Marshaller());
ctx.registerMarshaller(new G2Marshaller());
G2 g2 = buildG2();
ByteArrayOutputStream baos = new ByteArrayOutputStream();
RawProtoStreamWriter out = RawProtoStreamWriterImpl.newInstance(baos);
WrappedMessage.writeMessage(ctx, out, g2);
out.flush();
baos.flush();
RawProtoStreamReader in = RawProtoStreamReaderImpl.newInstance(baos.toByteArray());
G2 result = WrappedMessage.readMessage(ctx, in);
// System.out.println(result);
assertNotNull(result);
assertEquals(g2, result);
}
Aggregations