use of com.palantir.conjure.spec.ErrorNamespace in project conjure-java by palantir.
the class ErrorGenerator method generateErrorTypesForNamespace.
private static JavaFile generateErrorTypesForNamespace(TypeMapper typeMapper, String conjurePackage, ErrorNamespace namespace, List<ErrorDefinition> errorTypeDefinitions) {
ClassName className = errorTypesClassName(conjurePackage, namespace);
// Generate ErrorType definitions
List<FieldSpec> fieldSpecs = errorTypeDefinitions.stream().map(errorDef -> {
CodeBlock initializer = CodeBlock.of("ErrorType.create(ErrorType.Code.$L, \"$L:$L\")", errorDef.getCode().get(), namespace.get(), errorDef.getErrorName().getName());
FieldSpec.Builder fieldSpecBuilder = FieldSpec.builder(ClassName.get(ErrorType.class), CaseFormat.UPPER_CAMEL.to(CaseFormat.UPPER_UNDERSCORE, errorDef.getErrorName().getName()), Modifier.PUBLIC, Modifier.STATIC, Modifier.FINAL).initializer(initializer);
errorDef.getDocs().ifPresent(docs -> fieldSpecBuilder.addJavadoc(docs.get()));
return fieldSpecBuilder.build();
}).collect(Collectors.toList());
// Generate ServiceException factory methods
List<MethodSpec> methodSpecs = errorTypeDefinitions.stream().flatMap(entry -> {
MethodSpec withoutCause = generateExceptionFactory(typeMapper, entry, false);
MethodSpec withCause = generateExceptionFactory(typeMapper, entry, true);
return Stream.of(withoutCause, withCause);
}).collect(Collectors.toList());
// Generate ServiceException factory check methods
List<MethodSpec> checkMethodSpecs = errorTypeDefinitions.stream().map(entry -> {
String exceptionMethodName = CaseFormat.UPPER_CAMEL.to(CaseFormat.LOWER_CAMEL, entry.getErrorName().getName());
String methodName = "throwIf" + entry.getErrorName().getName();
String shouldThrowVar = "shouldThrow";
MethodSpec.Builder methodBuilder = MethodSpec.methodBuilder(methodName).addModifiers(Modifier.PUBLIC, Modifier.STATIC).addParameter(TypeName.BOOLEAN, shouldThrowVar);
methodBuilder.addJavadoc("Throws a {@link ServiceException} of type $L when {@code $L} is true.\n", entry.getErrorName().getName(), shouldThrowVar);
methodBuilder.addJavadoc("@param $L $L\n", shouldThrowVar, "Cause the method to throw when true");
Streams.concat(entry.getSafeArgs().stream(), entry.getUnsafeArgs().stream()).forEach(arg -> {
methodBuilder.addParameter(typeMapper.getClassName(arg.getType()), arg.getFieldName().get());
methodBuilder.addJavadoc("@param $L $L", arg.getFieldName().get(), StringUtils.appendIfMissing(arg.getDocs().map(Javadoc::render).orElse(""), "\n"));
});
methodBuilder.addCode("if ($L) {", shouldThrowVar);
methodBuilder.addCode("throw $L;", Expressions.localMethodCall(exceptionMethodName, Streams.concat(entry.getSafeArgs().stream(), entry.getUnsafeArgs().stream()).map(arg -> arg.getFieldName().get()).collect(Collectors.toList())));
methodBuilder.addCode("}");
return methodBuilder.build();
}).collect(Collectors.toList());
List<MethodSpec> isRemoteExceptionDefinitions = errorTypeDefinitions.stream().map(entry -> {
String typeName = CaseFormat.UPPER_CAMEL.to(CaseFormat.UPPER_UNDERSCORE, entry.getErrorName().getName());
String methodName = "is" + entry.getErrorName().getName();
return MethodSpec.methodBuilder(methodName).addModifiers(Modifier.PUBLIC, Modifier.STATIC).addParameter(RemoteException.class, REMOTE_EXCEPTION_VAR).returns(TypeName.BOOLEAN).addStatement(Expressions.requireNonNull(REMOTE_EXCEPTION_VAR, "remote exception must not be null")).addStatement("return $N.name().equals($N.getError().errorName())", typeName, REMOTE_EXCEPTION_VAR).addJavadoc("Returns true if the {@link $T} is named $L:$L", RemoteException.class, entry.getNamespace(), entry.getErrorName().getName()).build();
}).collect(Collectors.toList());
TypeSpec.Builder typeBuilder = TypeSpec.classBuilder(className).addMethod(privateConstructor()).addModifiers(Modifier.PUBLIC, Modifier.FINAL).addFields(fieldSpecs).addMethods(methodSpecs).addMethods(checkMethodSpecs).addMethods(isRemoteExceptionDefinitions).addAnnotation(ConjureAnnotations.getConjureGeneratedAnnotation(ErrorGenerator.class));
return JavaFile.builder(conjurePackage, typeBuilder.build()).skipJavaLangImports(true).indent(" ").build();
}
use of com.palantir.conjure.spec.ErrorNamespace in project conjure-java by palantir.
the class ErrorGenerator method splitErrorDefsByNamespace.
private static Map<String, Map<ErrorNamespace, List<ErrorDefinition>>> splitErrorDefsByNamespace(List<ErrorDefinition> errorTypeNameToDef) {
Map<String, Map<ErrorNamespace, List<ErrorDefinition>>> pkgToNamespacedErrorDefs = new HashMap<>();
errorTypeNameToDef.forEach(errorDef -> {
String errorPkg = errorDef.getErrorName().getPackage();
pkgToNamespacedErrorDefs.computeIfAbsent(errorPkg, key -> new HashMap<>());
Map<ErrorNamespace, List<ErrorDefinition>> namespacedErrorDefs = pkgToNamespacedErrorDefs.get(errorPkg);
ErrorNamespace namespace = errorDef.getNamespace();
// TODO(rfink): Use Multimap?
namespacedErrorDefs.computeIfAbsent(namespace, key -> new ArrayList<>());
namespacedErrorDefs.get(namespace).add(errorDef);
});
return pkgToNamespacedErrorDefs;
}
Aggregations