use of org.infinispan.protostream.annotations.ProtoUnknownFieldSet in project protostream by infinispan.
the class ProtoMessageTypeMetadata method discoverFields.
private void discoverFields(XClass clazz, Set<XClass> examinedClasses) {
if (!examinedClasses.add(clazz)) {
// avoid re-examining classes due to multiple interface inheritance
return;
}
if (clazz.getSuperclass() != null) {
discoverFields(clazz.getSuperclass(), examinedClasses);
}
for (XClass i : clazz.getInterfaces()) {
discoverFields(i, examinedClasses);
}
for (XField field : clazz.getDeclaredFields()) {
if (field.getAnnotation(ProtoUnknownFieldSet.class) != null) {
if (isAdapter) {
throw new ProtoSchemaBuilderException("No ProtoStream annotations should be present on fields when @ProtoAdapter is present on a class : " + clazz.getCanonicalName() + '.' + field);
}
if (unknownFieldSetField != null || unknownFieldSetGetter != null || unknownFieldSetSetter != null) {
throw new ProtoSchemaBuilderException("The @ProtoUnknownFieldSet annotation should not occur more than once in a class and its superclasses and superinterfaces : " + clazz.getCanonicalName() + '.' + field);
}
unknownFieldSetField = field;
} else {
ProtoField annotation = field.getAnnotation(ProtoField.class);
if (annotation != null) {
if (isAdapter) {
throw new ProtoSchemaBuilderException("No ProtoStream annotations should be present on fields when @ProtoAdapter is present on a class : " + clazz.getCanonicalName() + '.' + field);
}
if (field.isStatic()) {
throw new ProtoSchemaBuilderException("Static fields cannot be @ProtoField annotated: " + clazz.getCanonicalName() + '.' + field);
}
if (factory == null && field.isFinal()) {
// todo [anistor] maybe allow this
throw new ProtoSchemaBuilderException("Final fields cannot be @ProtoField annotated: " + clazz.getCanonicalName() + '.' + field);
}
if (field.isPrivate()) {
throw new ProtoSchemaBuilderException("Private fields cannot be @ProtoField annotated: " + clazz.getCanonicalName() + '.' + field);
}
int number = getNumber(annotation, field);
String fieldName = annotation.name();
if (fieldName.isEmpty()) {
fieldName = field.getName();
}
Type protobufType = annotation.type();
if (field.getType() == typeFactory.fromClass(byte[].class) && protobufType == Type.MESSAGE) {
// MESSAGE is the default and stands for 'undefined', we can override it with a better default
protobufType = Type.BYTES;
}
boolean isArray = isArray(field.getType(), protobufType);
boolean isRepeated = isRepeated(field.getType(), protobufType);
boolean isRequired = annotation.required();
if (isRepeated && isRequired) {
throw new ProtoSchemaBuilderException("Repeated field '" + fieldName + "' of " + clazz.getCanonicalName() + " cannot be marked required.");
}
XClass javaType = getJavaTypeFromAnnotation(annotation);
if (javaType == typeFactory.fromClass(void.class)) {
javaType = isRepeated ? field.determineRepeatedElementType() : field.getType();
}
if (javaType == typeFactory.fromClass(byte[].class) && protobufType == Type.MESSAGE) {
// MESSAGE is the default and stands for 'undefined', we can override it with a better default
protobufType = Type.BYTES;
}
if (!javaType.isArray() && !javaType.isPrimitive() && javaType.isAbstract() && !javaType.isEnum()) {
throw new ProtoSchemaBuilderException("The type " + javaType.getCanonicalName() + " of field '" + fieldName + "' of " + clazz.getCanonicalName() + " should not be abstract.");
}
protobufType = getProtobufType(javaType, protobufType);
Object defaultValue = getDefaultValue(clazz, fieldName, javaType, protobufType, annotation.defaultValue());
if (!isRequired && !isRepeated && javaType.isPrimitive() && defaultValue == null) {
throw new ProtoSchemaBuilderException("Primitive field '" + fieldName + "' of " + clazz.getCanonicalName() + " is not nullable so it should be either marked required or should have a default value.");
}
XClass collectionImplementation = getCollectionImplementation(clazz, field.getType(), getCollectionImplementationFromAnnotation(annotation), fieldName, isRepeated);
if (isArray) {
collectionImplementation = typeFactory.fromClass(ArrayList.class);
}
ProtoTypeMetadata protoTypeMetadata = null;
if (protobufType.getJavaType() == JavaType.ENUM || protobufType.getJavaType() == JavaType.MESSAGE) {
protoTypeMetadata = protoSchemaGenerator.scanAnnotations(javaType);
}
ProtoFieldMetadata fieldMetadata = new ProtoFieldMetadata(number, fieldName, javaType, collectionImplementation, protobufType, protoTypeMetadata, isRequired, isRepeated, isArray, defaultValue, field);
ProtoFieldMetadata existing = fieldsByNumber.get(number);
if (existing != null) {
throw new ProtoSchemaBuilderException("Duplicate field number definition. Found two field definitions with number " + number + ": in " + fieldMetadata.getLocation() + " and in " + existing.getLocation());
}
existing = fieldsByName.get(fieldMetadata.getName());
if (existing != null) {
throw new ProtoSchemaBuilderException("Duplicate field name definition. Found two field definitions with name '" + fieldMetadata.getName() + "': in " + fieldMetadata.getLocation() + " and in " + existing.getLocation());
}
checkReserved(fieldMetadata);
fieldsByNumber.put(fieldMetadata.getNumber(), fieldMetadata);
fieldsByName.put(fieldName, fieldMetadata);
}
}
}
for (XMethod method : clazz.getDeclaredMethods()) {
if (method.getAnnotation(ProtoUnknownFieldSet.class) != null) {
if (unknownFieldSetField != null || unknownFieldSetGetter != null || unknownFieldSetSetter != null) {
throw new ProtoSchemaBuilderException("The @ProtoUnknownFieldSet annotation should not occur more than once in a class and its superclasses and superinterfaces : " + method);
}
String propertyName;
if (method.getReturnType() == typeFactory.fromClass(void.class)) {
// this method is expected to be a setter
if (method.getName().startsWith("set") && method.getName().length() > 3) {
propertyName = Character.toLowerCase(method.getName().charAt(3)) + method.getName().substring(4);
} else {
throw new ProtoSchemaBuilderException("Illegal setter method signature: " + method);
}
if (isAdapter && method.getParameterTypes().length != 2 || !isAdapter && method.getParameterTypes().length != 1) {
throw new ProtoSchemaBuilderException("Illegal setter method signature: " + method);
}
// TODO [anistor] also check setter args
unknownFieldSetSetter = method;
unknownFieldSetGetter = findGetter(propertyName, method.getParameterTypes()[0]);
} else {
// this method is expected to be a getter
if (method.getName().startsWith("get") && method.getName().length() > 3) {
propertyName = Character.toLowerCase(method.getName().charAt(3)) + method.getName().substring(4);
} else if (method.getName().startsWith("is") && method.getName().length() > 2) {
propertyName = Character.toLowerCase(method.getName().charAt(2)) + method.getName().substring(3);
} else {
throw new ProtoSchemaBuilderException("Illegal getter method signature: " + method);
}
if (isAdapter && method.getParameterTypes().length != 1 || !isAdapter && method.getParameterTypes().length != 0) {
throw new ProtoSchemaBuilderException("Illegal getter method signature: " + method);
}
// TODO [anistor] also check getter args
unknownFieldSetGetter = method;
unknownFieldSetSetter = findSetter(propertyName, unknownFieldSetGetter.getReturnType());
}
} else {
ProtoField annotation = method.getAnnotation(ProtoField.class);
if (annotation != null) {
if (method.isPrivate()) {
throw new ProtoSchemaBuilderException("Private methods cannot be @ProtoField annotated: " + method);
}
if (!isAdapter && method.isStatic()) {
throw new ProtoSchemaBuilderException("Static methods cannot be @ProtoField annotated: " + method);
}
String propertyName;
XMethod getter;
XMethod setter;
XClass getterReturnType;
// we can have the annotation present on either getter or setter but not both
if (method.getReturnType() == typeFactory.fromClass(void.class)) {
// this method is expected to be a setter
if (method.getName().startsWith("set") && method.getName().length() >= 4) {
propertyName = Character.toLowerCase(method.getName().charAt(3)) + method.getName().substring(4);
} else {
// not a standard java-beans setter, use the whole name as property name
propertyName = method.getName();
}
if (isAdapter && method.getParameterTypes().length != 2 || !isAdapter && method.getParameterTypes().length != 1) {
throw new ProtoSchemaBuilderException("Illegal setter method signature: " + method);
}
// TODO [anistor] also check setter args
setter = method;
getter = findGetter(propertyName, method.getParameterTypes()[0]);
getterReturnType = getter.getReturnType();
if (getterReturnType == typeFactory.fromClass(Optional.class)) {
getterReturnType = getter.determineOptionalReturnType();
}
} else {
// this method is expected to be a getter
if (method.getName().startsWith("get") && method.getName().length() >= 4) {
propertyName = Character.toLowerCase(method.getName().charAt(3)) + method.getName().substring(4);
} else if (method.getName().startsWith("is") && method.getName().length() >= 3) {
propertyName = Character.toLowerCase(method.getName().charAt(2)) + method.getName().substring(3);
} else {
// not a standard java-beans getter
propertyName = method.getName();
}
if (isAdapter && method.getParameterTypes().length != 1 || !isAdapter && method.getParameterTypes().length != 0) {
throw new ProtoSchemaBuilderException("Illegal setter method signature: " + method);
}
// TODO [anistor] also check getter args
getter = method;
getterReturnType = getter.getReturnType();
if (getterReturnType == typeFactory.fromClass(Optional.class)) {
getterReturnType = getter.determineOptionalReturnType();
}
setter = factory == null ? findSetter(propertyName, getterReturnType) : null;
}
int number = getNumber(annotation, method);
String fieldName = annotation.name();
if (fieldName.isEmpty()) {
fieldName = propertyName;
}
Type protobufType = annotation.type();
if (getterReturnType == typeFactory.fromClass(byte[].class) && protobufType == Type.MESSAGE) {
// MESSAGE is the default and stands for 'undefined', we can override it with a better default
protobufType = Type.BYTES;
}
boolean isArray = isArray(getterReturnType, protobufType);
boolean isRepeated = isRepeated(getterReturnType, protobufType);
boolean isRequired = annotation.required();
if (isRepeated && isRequired) {
throw new ProtoSchemaBuilderException("Repeated field '" + fieldName + "' of " + clazz.getCanonicalName() + " cannot be marked required.");
}
XClass javaType = getJavaTypeFromAnnotation(annotation);
if (javaType == typeFactory.fromClass(void.class)) {
javaType = isRepeated ? getter.determineRepeatedElementType() : getterReturnType;
}
if (javaType == typeFactory.fromClass(byte[].class) && protobufType == Type.MESSAGE) {
// MESSAGE is the default and stands for 'undefined', we can override it with a better default
protobufType = Type.BYTES;
}
if (!javaType.isArray() && !javaType.isPrimitive() && javaType.isAbstract() && !javaType.isEnum()) {
throw new ProtoSchemaBuilderException("The type " + javaType.getCanonicalName() + " of field '" + fieldName + "' of " + clazz.getCanonicalName() + " should not be abstract.");
}
protobufType = getProtobufType(javaType, protobufType);
Object defaultValue = getDefaultValue(clazz, fieldName, javaType, protobufType, annotation.defaultValue());
if (!isRequired && !isRepeated && javaType.isPrimitive() && defaultValue == null) {
throw new ProtoSchemaBuilderException("Primitive field '" + fieldName + "' of " + clazz.getCanonicalName() + " is not nullable so it should be either marked required or should have a default value.");
}
XClass collectionImplementation = getCollectionImplementation(clazz, getterReturnType, getCollectionImplementationFromAnnotation(annotation), fieldName, isRepeated);
if (isArray) {
collectionImplementation = typeFactory.fromClass(ArrayList.class);
}
ProtoTypeMetadata protoTypeMetadata = null;
if (protobufType.getJavaType() == JavaType.ENUM || protobufType.getJavaType() == JavaType.MESSAGE) {
protoTypeMetadata = protoSchemaGenerator.scanAnnotations(javaType);
}
ProtoFieldMetadata fieldMetadata = new ProtoFieldMetadata(number, fieldName, javaType, collectionImplementation, protobufType, protoTypeMetadata, isRequired, isRepeated, isArray, defaultValue, propertyName, method, getter, setter);
ProtoFieldMetadata existing = fieldsByNumber.get(number);
if (existing != null) {
throw new ProtoSchemaBuilderException("Duplicate field definition. Found two field definitions with number " + number + ": in " + fieldMetadata.getLocation() + " and in " + existing.getLocation());
}
existing = fieldsByName.get(fieldMetadata.getName());
if (existing != null) {
throw new ProtoSchemaBuilderException("Duplicate field definition. Found two field definitions with name '" + fieldMetadata.getName() + "': in " + fieldMetadata.getLocation() + " and in " + existing.getLocation());
}
checkReserved(fieldMetadata);
fieldsByNumber.put(number, fieldMetadata);
fieldsByName.put(fieldName, fieldMetadata);
}
}
}
}
Aggregations