use of org.infinispan.protostream.descriptors.Type in project protostream by infinispan.
the class DescriptorsTest method testAnnotationParser.
@Test
public void testAnnotationParser() throws Exception {
Configuration config = Configuration.builder().annotationsConfig().annotation("Indexed", AnnotationElement.AnnotationTarget.MESSAGE).attribute(AnnotationElement.Annotation.VALUE_DEFAULT_ATTRIBUTE).type(AnnotationElement.AttributeType.BOOLEAN).defaultValue(true).metadataCreator((descriptor, annotation) -> annotation.getDefaultAttributeValue().getValue()).build();
FileDescriptorSource fileDescriptorSource = FileDescriptorSource.fromResources("sample_bank_account/bank.proto");
Map<String, FileDescriptor> descriptors = parseAndResolve(fileDescriptorSource, config);
FileDescriptor fileDescriptor = descriptors.get("sample_bank_account/bank.proto");
List<Descriptor> messageTypes = fileDescriptor.getMessageTypes();
Descriptor userMessageType = messageTypes.get(0);
assertEquals("sample_bank_account.User", userMessageType.getFullName());
assertEquals(Boolean.TRUE, userMessageType.getProcessedAnnotation("Indexed"));
Descriptor accountMessageType = messageTypes.get(1);
assertEquals("sample_bank_account.Account", accountMessageType.getFullName());
assertEquals(Boolean.TRUE, accountMessageType.getProcessedAnnotation("Indexed"));
}
use of org.infinispan.protostream.descriptors.Type in project protostream by infinispan.
the class DescriptorsTest method testDocAnnotations.
@Test
public void testDocAnnotations() {
String file1 = "package test1;\n" + "/** \n" + " * @Foo(fooValue) \n" + " * some more doc text \n" + " **/\n\n" + "message X {\n" + " /**\n" + " * @Bar(barValue) \n\n" + " */\n" + " optional int32 field1 = 1;\n" + "}\n";
Configuration config = Configuration.builder().annotationsConfig().annotation("Foo", AnnotationElement.AnnotationTarget.MESSAGE).attribute(AnnotationElement.Annotation.VALUE_DEFAULT_ATTRIBUTE).type(AnnotationElement.AttributeType.IDENTIFIER).metadataCreator((descriptor, annotation) -> annotation.getDefaultAttributeValue().getValue()).annotation("Bar", AnnotationElement.AnnotationTarget.FIELD).attribute(AnnotationElement.Annotation.VALUE_DEFAULT_ATTRIBUTE).type(AnnotationElement.AttributeType.IDENTIFIER).metadataCreator((fieldDescriptor, annotation) -> annotation.getDefaultAttributeValue().getValue()).build();
FileDescriptorSource fileDescriptorSource = new FileDescriptorSource();
fileDescriptorSource.addProtoFile("file1.proto", file1);
Map<String, FileDescriptor> descriptors = parseAndResolve(fileDescriptorSource, config);
assertEquals(1, descriptors.size());
assertTrue(descriptors.containsKey("file1.proto"));
Map<String, GenericDescriptor> types = descriptors.get("file1.proto").getTypes();
Descriptor typeX = (Descriptor) types.get("test1.X");
assertNotNull(typeX);
assertEquals(1, typeX.getFields().size());
FieldDescriptor field1 = typeX.getFields().get(0);
assertEquals("@Foo(fooValue) \n some more doc text", typeX.getDocumentation());
Map<String, AnnotationElement.Annotation> typeAnnotations = typeX.getAnnotations();
assertEquals("fooValue", typeAnnotations.get("Foo").getDefaultAttributeValue().getValue());
assertEquals("fooValue", typeX.getProcessedAnnotation("Foo"));
assertEquals("@Bar(barValue)", field1.getDocumentation());
Map<String, AnnotationElement.Annotation> fieldAnnotations = field1.getAnnotations();
assertEquals("barValue", fieldAnnotations.get("Bar").getDefaultAttributeValue().getValue());
assertEquals("barValue", field1.getProcessedAnnotation("Bar"));
}
Aggregations