Search in sources :

Example 1 with UriMatchTemplate

use of io.micronaut.http.uri.UriMatchTemplate in project micronaut-core by micronaut-projects.

the class ClientTypesRule method validate.

@Override
public RouteValidationResult validate(List<UriMatchTemplate> templates, ParameterElement[] parameters, MethodElement method) {
    String[] errors = new String[] {};
    if (method.hasAnnotation(Client.class)) {
        final Stream.Builder<ClassElement> builder = Stream.<ClassElement>builder().add(method.getReturnType());
        for (ParameterElement param : method.getParameters()) {
            builder.add(param.getType());
        }
        errors = builder.build().filter(type -> {
            for (Class<?> clazz : SERVER_TYPES) {
                if (type.isAssignable(clazz)) {
                    return true;
                }
            }
            return false;
        }).map(type -> "The type [" + type + "] must not be used in declarative client methods. The type is specific to server based usages.").toArray(String[]::new);
    }
    return new RouteValidationResult(errors);
}
Also used : List(java.util.List) RouteValidationResult(io.micronaut.validation.routes.RouteValidationResult) Stream(java.util.stream.Stream) MultipartBody(io.micronaut.http.server.multipart.MultipartBody) StreamedFile(io.micronaut.http.server.types.files.StreamedFile) MethodElement(io.micronaut.inject.ast.MethodElement) UriMatchTemplate(io.micronaut.http.uri.UriMatchTemplate) ClassElement(io.micronaut.inject.ast.ClassElement) Client(io.micronaut.http.client.annotation.Client) SystemFile(io.micronaut.http.server.types.files.SystemFile) ParameterElement(io.micronaut.inject.ast.ParameterElement) Stream(java.util.stream.Stream) ClassElement(io.micronaut.inject.ast.ClassElement) ParameterElement(io.micronaut.inject.ast.ParameterElement) RouteValidationResult(io.micronaut.validation.routes.RouteValidationResult)

Example 2 with UriMatchTemplate

use of io.micronaut.http.uri.UriMatchTemplate in project micronaut-core by micronaut-projects.

the class DefaultRouter method findAllClosest.

@NonNull
@Override
public <T, R> List<UriRouteMatch<T, R>> findAllClosest(@NonNull HttpRequest<?> request) {
    final HttpMethod httpMethod = request.getMethod();
    final MediaType contentType = request.getContentType().orElse(null);
    boolean permitsBody = HttpMethod.permitsRequestBody(httpMethod);
    final Collection<MediaType> acceptedProducedTypes = request.accept();
    List<UriRouteMatch<T, R>> uriRoutes = this.find(request.getMethodName(), request.getPath(), routeMatch -> routeMatch.test(request) && (!permitsBody || routeMatch.doesConsume(contentType)) && routeMatch.doesProduce(acceptedProducedTypes));
    int routeCount = uriRoutes.size();
    if (routeCount <= 1) {
        return uriRoutes;
    }
    if (CollectionUtils.isNotEmpty(acceptedProducedTypes)) {
        // take the highest priority accepted type
        final MediaType mediaType = acceptedProducedTypes.iterator().next();
        List<UriRouteMatch<T, R>> mostSpecific = new ArrayList<>(uriRoutes.size());
        for (UriRouteMatch<T, R> routeMatch : uriRoutes) {
            if (routeMatch.explicitlyProduces(mediaType)) {
                mostSpecific.add(routeMatch);
            }
        }
        if (!mostSpecific.isEmpty() || !acceptedProducedTypes.contains(MediaType.ALL_TYPE)) {
            uriRoutes = mostSpecific;
        }
    }
    routeCount = uriRoutes.size();
    if (routeCount > 1 && permitsBody) {
        List<UriRouteMatch<T, R>> explicitlyConsumedRoutes = new ArrayList<>(routeCount);
        List<UriRouteMatch<T, R>> consumesRoutes = new ArrayList<>(routeCount);
        for (UriRouteMatch<T, R> match : uriRoutes) {
            if (match.explicitlyConsumes(contentType != null ? contentType : MediaType.ALL_TYPE)) {
                explicitlyConsumedRoutes.add(match);
            }
            if (explicitlyConsumedRoutes.isEmpty() && match.doesConsume(contentType)) {
                consumesRoutes.add(match);
            }
        }
        uriRoutes = explicitlyConsumedRoutes.isEmpty() ? consumesRoutes : explicitlyConsumedRoutes;
    }
    /*
         * Any changes to the logic below may also need changes to {@link io.micronaut.http.uri.UriTemplate#compareTo(UriTemplate)}
         */
    routeCount = uriRoutes.size();
    if (routeCount > 1) {
        long variableCount = 0;
        long rawLength = 0;
        List<UriRouteMatch<T, R>> closestMatches = new ArrayList<>(routeCount);
        for (int i = 0; i < routeCount; i++) {
            UriRouteMatch<T, R> match = uriRoutes.get(i);
            UriMatchTemplate template = match.getRoute().getUriMatchTemplate();
            long variable = template.getPathVariableSegmentCount();
            long raw = template.getRawSegmentLength();
            if (i == 0) {
                variableCount = variable;
                rawLength = raw;
            }
            if (variable > variableCount || raw < rawLength) {
                break;
            }
            closestMatches.add(match);
        }
        uriRoutes = closestMatches;
    }
    return uriRoutes;
}
Also used : UriMatchTemplate(io.micronaut.http.uri.UriMatchTemplate) NonNull(io.micronaut.core.annotation.NonNull)

Example 3 with UriMatchTemplate

use of io.micronaut.http.uri.UriMatchTemplate in project micronaut-core by micronaut-projects.

the class UriTemplateTest method testUriTemplate.

@Test
public void testUriTemplate() {
    // tag::match[]
    UriMatchTemplate template = UriMatchTemplate.of("/hello/{name}");
    // <1>
    assertTrue(template.match("/hello/John").isPresent());
    assertEquals("/hello/John", // <2>
    template.expand(Collections.singletonMap("name", "John")));
// end::match[]
}
Also used : UriMatchTemplate(io.micronaut.http.uri.UriMatchTemplate) Test(org.junit.Test)

Example 4 with UriMatchTemplate

use of io.micronaut.http.uri.UriMatchTemplate in project micronaut-core by micronaut-projects.

the class HttpClientIntroductionAdvice method intercept.

/**
 * Interceptor to apply headers, cookies, parameter and body arguements.
 *
 * @param context The context
 * @return httpClient or future
 */
@Nullable
@Override
public Object intercept(MethodInvocationContext<Object, Object> context) {
    if (!context.hasStereotype(Client.class)) {
        throw new IllegalStateException("Client advice called from type that is not annotated with @Client: " + context);
    }
    final AnnotationMetadata annotationMetadata = context.getAnnotationMetadata();
    Class<?> declaringType = context.getDeclaringType();
    if (Closeable.class == declaringType || AutoCloseable.class == declaringType) {
        clientFactory.disposeClient(annotationMetadata);
        return null;
    }
    Optional<Class<? extends Annotation>> httpMethodMapping = context.getAnnotationTypeByStereotype(HttpMethodMapping.class);
    HttpClient httpClient = clientFactory.getClient(annotationMetadata);
    if (context.hasStereotype(HttpMethodMapping.class) && httpClient != null) {
        AnnotationValue<HttpMethodMapping> mapping = context.getAnnotation(HttpMethodMapping.class);
        String uri = mapping.getRequiredValue(String.class);
        if (StringUtils.isEmpty(uri)) {
            uri = "/" + context.getMethodName();
        }
        Class<? extends Annotation> annotationType = httpMethodMapping.get();
        HttpMethod httpMethod = HttpMethod.parse(annotationType.getSimpleName().toUpperCase(Locale.ENGLISH));
        String httpMethodName = context.stringValue(CustomHttpMethod.class, "method").orElse(httpMethod.name());
        MutableHttpRequest<?> request = HttpRequest.create(httpMethod, "", httpMethodName);
        UriMatchTemplate uriTemplate = UriMatchTemplate.of("");
        if (!(uri.length() == 1 && uri.charAt(0) == '/')) {
            uriTemplate = uriTemplate.nest(uri);
        }
        Map<String, Object> pathParams = new HashMap<>();
        Map<String, List<String>> queryParams = new LinkedHashMap<>();
        ClientRequestUriContext uriContext = new ClientRequestUriContext(uriTemplate, pathParams, queryParams);
        List<Argument> bodyArguments = new ArrayList<>();
        List<String> uriVariables = uriTemplate.getVariableNames();
        Map<String, MutableArgumentValue<?>> parameters = context.getParameters();
        ClientArgumentRequestBinder<Object> defaultBinder = (ctx, uriCtx, value, req) -> {
            Argument<?> argument = ctx.getArgument();
            if (uriCtx.getUriTemplate().getVariableNames().contains(argument.getName())) {
                String name = argument.getAnnotationMetadata().stringValue(Bindable.class).orElse(argument.getName());
                // Convert and put as path param
                if (argument.getAnnotationMetadata().hasStereotype(Format.class)) {
                    ConversionService.SHARED.convert(value, ConversionContext.STRING.with(argument.getAnnotationMetadata())).ifPresent(v -> pathParams.put(name, v));
                } else {
                    pathParams.put(name, value);
                }
            } else {
                bodyArguments.add(ctx.getArgument());
            }
        };
        // Apply all the method binders
        List<Class<? extends Annotation>> methodBinderTypes = context.getAnnotationTypesByStereotype(Bindable.class);
        // @Version is not a bindable, so it needs to looked for separately
        methodBinderTypes.addAll(context.getAnnotationTypesByStereotype(Version.class));
        if (!CollectionUtils.isEmpty(methodBinderTypes)) {
            for (Class<? extends Annotation> binderType : methodBinderTypes) {
                binderRegistry.findAnnotatedBinder(binderType).ifPresent(b -> b.bind(context, uriContext, request));
            }
        }
        InterceptedMethod interceptedMethod = InterceptedMethod.of(context);
        // Apply all the argument binders
        Argument[] arguments = context.getArguments();
        if (arguments.length > 0) {
            for (Argument argument : arguments) {
                Object definedValue = getValue(argument, context, parameters);
                if (definedValue != null) {
                    final ClientArgumentRequestBinder<Object> binder = (ClientArgumentRequestBinder<Object>) binderRegistry.findArgumentBinder((Argument<Object>) argument).orElse(defaultBinder);
                    ArgumentConversionContext conversionContext = ConversionContext.of(argument);
                    binder.bind(conversionContext, uriContext, definedValue, request);
                    if (conversionContext.hasErrors()) {
                        return interceptedMethod.handleException(new ConversionErrorException(argument, conversionContext.getLastError().get()));
                    }
                }
            }
        }
        Object body = request.getBody().orElse(null);
        if (body == null && !bodyArguments.isEmpty()) {
            Map<String, Object> bodyMap = new LinkedHashMap<>();
            for (Argument bodyArgument : bodyArguments) {
                String argumentName = bodyArgument.getName();
                MutableArgumentValue<?> value = parameters.get(argumentName);
                bodyMap.put(argumentName, value.getValue());
            }
            body = bodyMap;
            request.body(body);
        }
        boolean variableSatisfied = uriVariables.isEmpty() || pathParams.keySet().containsAll(uriVariables);
        if (body != null && !variableSatisfied) {
            if (body instanceof Map) {
                for (Map.Entry<Object, Object> entry : ((Map<Object, Object>) body).entrySet()) {
                    String k = entry.getKey().toString();
                    Object v = entry.getValue();
                    if (v != null) {
                        pathParams.putIfAbsent(k, v);
                    }
                }
            } else {
                BeanMap<Object> beanMap = BeanMap.of(body);
                for (Map.Entry<String, Object> entry : beanMap.entrySet()) {
                    String k = entry.getKey();
                    Object v = entry.getValue();
                    if (v != null) {
                        pathParams.putIfAbsent(k, v);
                    }
                }
            }
        }
        if (!HttpMethod.permitsRequestBody(httpMethod)) {
            // If a binder set the body and the method does not permit it, reset to null
            request.body(null);
            body = null;
        }
        uri = uriTemplate.expand(pathParams);
        // Remove all the pathParams that have already been used.
        // Other path parameters are added to query
        uriVariables.forEach(pathParams::remove);
        addParametersToQuery(pathParams, uriContext);
        // The original query can be added by getting it from the request.getUri() and appending
        request.uri(URI.create(appendQuery(uri, uriContext.getQueryParameters())));
        if (body != null && !request.getContentType().isPresent()) {
            MediaType[] contentTypes = MediaType.of(context.stringValues(Produces.class));
            if (ArrayUtils.isEmpty(contentTypes)) {
                contentTypes = DEFAULT_ACCEPT_TYPES;
            }
            if (ArrayUtils.isNotEmpty(contentTypes)) {
                request.contentType(contentTypes[0]);
            }
        }
        request.setAttribute(HttpAttributes.INVOCATION_CONTEXT, context);
        // Set the URI template used to make the request for tracing purposes
        request.setAttribute(HttpAttributes.URI_TEMPLATE, resolveTemplate(annotationMetadata, uriTemplate.toString()));
        String serviceId = getClientId(annotationMetadata);
        Argument<?> errorType = annotationMetadata.classValue(Client.class, "errorType").map((Function<Class, Argument>) Argument::of).orElse(HttpClient.DEFAULT_ERROR_TYPE);
        request.setAttribute(HttpAttributes.SERVICE_ID, serviceId);
        final MediaType[] acceptTypes;
        Collection<MediaType> accept = request.accept();
        if (accept.isEmpty()) {
            String[] consumesMediaType = context.stringValues(Consumes.class);
            if (ArrayUtils.isEmpty(consumesMediaType)) {
                acceptTypes = DEFAULT_ACCEPT_TYPES;
            } else {
                acceptTypes = MediaType.of(consumesMediaType);
            }
            request.accept(acceptTypes);
        } else {
            acceptTypes = accept.toArray(MediaType.EMPTY_ARRAY);
        }
        ReturnType<?> returnType = context.getReturnType();
        try {
            Argument<?> valueType = interceptedMethod.returnTypeValue();
            Class<?> reactiveValueType = valueType.getType();
            switch(interceptedMethod.resultType()) {
                case PUBLISHER:
                    boolean isSingle = returnType.isSingleResult() || returnType.isCompletable() || HttpResponse.class.isAssignableFrom(reactiveValueType) || HttpStatus.class == reactiveValueType;
                    Publisher<?> publisher;
                    if (!isSingle && httpClient instanceof StreamingHttpClient) {
                        publisher = httpClientResponseStreamingPublisher((StreamingHttpClient) httpClient, acceptTypes, request, errorType, valueType);
                    } else {
                        publisher = httpClientResponsePublisher(httpClient, request, returnType, errorType, valueType);
                    }
                    Object finalPublisher = interceptedMethod.handleResult(publisher);
                    for (ReactiveClientResultTransformer transformer : transformers) {
                        finalPublisher = transformer.transform(finalPublisher);
                    }
                    return finalPublisher;
                case COMPLETION_STAGE:
                    Publisher<?> csPublisher = httpClientResponsePublisher(httpClient, request, returnType, errorType, valueType);
                    CompletableFuture<Object> future = new CompletableFuture<>();
                    csPublisher.subscribe(new CompletionAwareSubscriber<Object>() {

                        AtomicReference<Object> reference = new AtomicReference<>();

                        @Override
                        protected void doOnSubscribe(Subscription subscription) {
                            subscription.request(1);
                        }

                        @Override
                        protected void doOnNext(Object message) {
                            if (Void.class != reactiveValueType) {
                                reference.set(message);
                            }
                        }

                        @Override
                        protected void doOnError(Throwable t) {
                            if (t instanceof HttpClientResponseException) {
                                HttpClientResponseException e = (HttpClientResponseException) t;
                                if (e.getStatus() == HttpStatus.NOT_FOUND) {
                                    if (reactiveValueType == Optional.class) {
                                        future.complete(Optional.empty());
                                    } else if (HttpResponse.class.isAssignableFrom(reactiveValueType)) {
                                        future.complete(e.getResponse());
                                    } else {
                                        future.complete(null);
                                    }
                                    return;
                                }
                            }
                            if (LOG.isErrorEnabled()) {
                                LOG.error("Client [" + declaringType.getName() + "] received HTTP error response: " + t.getMessage(), t);
                            }
                            future.completeExceptionally(t);
                        }

                        @Override
                        protected void doOnComplete() {
                            future.complete(reference.get());
                        }
                    });
                    return interceptedMethod.handleResult(future);
                case SYNCHRONOUS:
                    Class<?> javaReturnType = returnType.getType();
                    BlockingHttpClient blockingHttpClient = httpClient.toBlocking();
                    if (void.class == javaReturnType || httpMethod == HttpMethod.HEAD) {
                        request.getHeaders().remove(HttpHeaders.ACCEPT);
                    }
                    if (HttpResponse.class.isAssignableFrom(javaReturnType)) {
                        return handleBlockingCall(javaReturnType, () -> blockingHttpClient.exchange(request, returnType.asArgument().getFirstTypeVariable().orElse(Argument.OBJECT_ARGUMENT), errorType));
                    } else if (void.class == javaReturnType) {
                        return handleBlockingCall(javaReturnType, () -> blockingHttpClient.exchange(request, null, errorType));
                    } else {
                        return handleBlockingCall(javaReturnType, () -> blockingHttpClient.retrieve(request, returnType.asArgument(), errorType));
                    }
                default:
                    return interceptedMethod.unsupported();
            }
        } catch (Exception e) {
            return interceptedMethod.handleException(e);
        }
    }
    // try other introduction advice
    return context.proceed();
}
Also used : Publishers(io.micronaut.core.async.publisher.Publishers) Bindable(io.micronaut.core.bind.annotation.Bindable) Arrays(java.util.Arrays) Version(io.micronaut.core.version.annotation.Version) CustomHttpMethod(io.micronaut.http.annotation.CustomHttpMethod) Produces(io.micronaut.http.annotation.Produces) ArrayUtils(io.micronaut.core.util.ArrayUtils) LoggerFactory(org.slf4j.LoggerFactory) HttpHeaders(io.micronaut.http.HttpHeaders) SseClient(io.micronaut.http.client.sse.SseClient) Internal(io.micronaut.core.annotation.Internal) Event(io.micronaut.http.sse.Event) CompletionAwareSubscriber(io.micronaut.core.async.subscriber.CompletionAwareSubscriber) HttpStatus(io.micronaut.http.HttpStatus) MediaType(io.micronaut.http.MediaType) Locale(java.util.Locale) Map(java.util.Map) ClientArgumentRequestBinder(io.micronaut.http.client.bind.ClientArgumentRequestBinder) HttpResponse(io.micronaut.http.HttpResponse) HttpClientResponseException(io.micronaut.http.client.exceptions.HttpClientResponseException) URI(java.net.URI) HttpClient(io.micronaut.http.client.HttpClient) JsonMediaTypeCodec(io.micronaut.json.codec.JsonMediaTypeCodec) MutableArgumentValue(io.micronaut.core.type.MutableArgumentValue) Collection(java.util.Collection) Singleton(jakarta.inject.Singleton) HttpClientBinderRegistry(io.micronaut.http.client.bind.HttpClientBinderRegistry) StringUtils(io.micronaut.core.util.StringUtils) Format(io.micronaut.core.convert.format.Format) List(java.util.List) MethodInvocationContext(io.micronaut.aop.MethodInvocationContext) AnnotationValue(io.micronaut.core.annotation.AnnotationValue) Annotation(java.lang.annotation.Annotation) Optional(java.util.Optional) HttpAttributes(io.micronaut.http.HttpAttributes) StreamingHttpClient(io.micronaut.http.client.StreamingHttpClient) MutableHttpRequest(io.micronaut.http.MutableHttpRequest) HttpClientRegistry(io.micronaut.http.client.HttpClientRegistry) UriBuilder(io.micronaut.http.uri.UriBuilder) ArgumentConversionContext(io.micronaut.core.convert.ArgumentConversionContext) HttpMethodMapping(io.micronaut.http.annotation.HttpMethodMapping) ReactiveClientResultTransformer(io.micronaut.http.client.ReactiveClientResultTransformer) ConversionErrorException(io.micronaut.core.convert.exceptions.ConversionErrorException) UriMatchTemplate(io.micronaut.http.uri.UriMatchTemplate) HashMap(java.util.HashMap) CompletableFuture(java.util.concurrent.CompletableFuture) Client(io.micronaut.http.client.annotation.Client) InterceptedMethod(io.micronaut.aop.InterceptedMethod) AtomicReference(java.util.concurrent.atomic.AtomicReference) Function(java.util.function.Function) Supplier(java.util.function.Supplier) BeanMap(io.micronaut.core.beans.BeanMap) ArrayList(java.util.ArrayList) LinkedHashMap(java.util.LinkedHashMap) BlockingHttpClient(io.micronaut.http.client.BlockingHttpClient) MethodInterceptor(io.micronaut.aop.MethodInterceptor) BootstrapContextCompatible(io.micronaut.context.annotation.BootstrapContextCompatible) Nullable(io.micronaut.core.annotation.Nullable) ByteBuffer(io.micronaut.core.io.buffer.ByteBuffer) ReturnType(io.micronaut.core.type.ReturnType) Argument(io.micronaut.core.type.Argument) HttpRequest(io.micronaut.http.HttpRequest) ClientRequestUriContext(io.micronaut.http.client.bind.ClientRequestUriContext) ConversionService(io.micronaut.core.convert.ConversionService) ConfigurationException(io.micronaut.context.exceptions.ConfigurationException) HttpMethod(io.micronaut.http.HttpMethod) Logger(org.slf4j.Logger) Publisher(org.reactivestreams.Publisher) ConversionContext(io.micronaut.core.convert.ConversionContext) CollectionUtils(io.micronaut.core.util.CollectionUtils) Consumes(io.micronaut.http.annotation.Consumes) Subscription(org.reactivestreams.Subscription) Closeable(java.io.Closeable) AnnotationMetadata(io.micronaut.core.annotation.AnnotationMetadata) Collections(java.util.Collections) HashMap(java.util.HashMap) LinkedHashMap(java.util.LinkedHashMap) Closeable(java.io.Closeable) ArrayList(java.util.ArrayList) CustomHttpMethod(io.micronaut.http.annotation.CustomHttpMethod) AnnotationMetadata(io.micronaut.core.annotation.AnnotationMetadata) LinkedHashMap(java.util.LinkedHashMap) Version(io.micronaut.core.version.annotation.Version) MediaType(io.micronaut.http.MediaType) List(java.util.List) ArrayList(java.util.ArrayList) ClientArgumentRequestBinder(io.micronaut.http.client.bind.ClientArgumentRequestBinder) SseClient(io.micronaut.http.client.sse.SseClient) HttpClient(io.micronaut.http.client.HttpClient) StreamingHttpClient(io.micronaut.http.client.StreamingHttpClient) Client(io.micronaut.http.client.annotation.Client) BlockingHttpClient(io.micronaut.http.client.BlockingHttpClient) Subscription(org.reactivestreams.Subscription) Optional(java.util.Optional) HttpStatus(io.micronaut.http.HttpStatus) BlockingHttpClient(io.micronaut.http.client.BlockingHttpClient) ClientRequestUriContext(io.micronaut.http.client.bind.ClientRequestUriContext) Produces(io.micronaut.http.annotation.Produces) Map(java.util.Map) HashMap(java.util.HashMap) BeanMap(io.micronaut.core.beans.BeanMap) LinkedHashMap(java.util.LinkedHashMap) CustomHttpMethod(io.micronaut.http.annotation.CustomHttpMethod) HttpMethod(io.micronaut.http.HttpMethod) StreamingHttpClient(io.micronaut.http.client.StreamingHttpClient) Argument(io.micronaut.core.type.Argument) ConversionErrorException(io.micronaut.core.convert.exceptions.ConversionErrorException) UriMatchTemplate(io.micronaut.http.uri.UriMatchTemplate) Function(java.util.function.Function) CompletableFuture(java.util.concurrent.CompletableFuture) Format(io.micronaut.core.convert.format.Format) ArgumentConversionContext(io.micronaut.core.convert.ArgumentConversionContext) ReactiveClientResultTransformer(io.micronaut.http.client.ReactiveClientResultTransformer) InterceptedMethod(io.micronaut.aop.InterceptedMethod) HttpMethodMapping(io.micronaut.http.annotation.HttpMethodMapping) MutableArgumentValue(io.micronaut.core.type.MutableArgumentValue) AtomicReference(java.util.concurrent.atomic.AtomicReference) Annotation(java.lang.annotation.Annotation) HttpClientResponseException(io.micronaut.http.client.exceptions.HttpClientResponseException) ConversionErrorException(io.micronaut.core.convert.exceptions.ConversionErrorException) ConfigurationException(io.micronaut.context.exceptions.ConfigurationException) HttpClientResponseException(io.micronaut.http.client.exceptions.HttpClientResponseException) HttpClient(io.micronaut.http.client.HttpClient) StreamingHttpClient(io.micronaut.http.client.StreamingHttpClient) BlockingHttpClient(io.micronaut.http.client.BlockingHttpClient) Nullable(io.micronaut.core.annotation.Nullable)

Example 5 with UriMatchTemplate

use of io.micronaut.http.uri.UriMatchTemplate in project micronaut-core by micronaut-projects.

the class NullableParameterRule method validate.

@Override
public RouteValidationResult validate(List<UriMatchTemplate> templates, ParameterElement[] parameters, MethodElement method) {
    List<String> errorMessages = new ArrayList<>();
    boolean isClient = method.hasAnnotation("io.micronaut.http.client.annotation.Client");
    // Optional variables can be required in clients
    if (!isClient) {
        Map<String, UriMatchVariable> variables = new HashMap<>();
        Set<UriMatchVariable> required = new HashSet<>();
        for (UriMatchTemplate template : templates) {
            for (UriMatchVariable variable : template.getVariables()) {
                if (!variable.isOptional() || variable.isExploded()) {
                    required.add(variable);
                }
                variables.compute(variable.getName(), (key, var) -> {
                    if (var == null) {
                        if (variable.isOptional() && !variable.isExploded()) {
                            return variable;
                        } else {
                            return null;
                        }
                    } else {
                        if (!var.isOptional() || var.isExploded()) {
                            if (variable.isOptional() && !variable.isExploded()) {
                                return variable;
                            } else {
                                return var;
                            }
                        } else {
                            return var;
                        }
                    }
                });
            }
        }
        for (UriMatchVariable variable : required) {
            if (templates.stream().anyMatch(t -> !t.getVariableNames().contains(variable.getName()))) {
                variables.putIfAbsent(variable.getName(), variable);
            }
        }
        for (UriMatchVariable variable : variables.values()) {
            Arrays.stream(parameters).flatMap(p -> getTypedElements(p).stream()).filter(p -> p.getName().equals(variable.getName())).forEach(p -> {
                ClassElement type = p.getType();
                boolean hasDefaultValue = p.findAnnotation(Bindable.class).flatMap(av -> av.stringValue("defaultValue")).isPresent();
                if (!isNullable(p) && type != null && !type.isAssignable(Optional.class) && !hasDefaultValue) {
                    errorMessages.add(String.format("The uri variable [%s] is optional, but the corresponding method argument [%s %s] is not defined as an Optional or annotated with a Nullable annotation.", variable.getName(), p.getType().toString(), p.getName()));
                }
            });
        }
    }
    return new RouteValidationResult(errorMessages.toArray(new String[0]));
}
Also used : Bindable(io.micronaut.core.bind.annotation.Bindable) Arrays(java.util.Arrays) RouteValidationResult(io.micronaut.validation.routes.RouteValidationResult) UriMatchVariable(io.micronaut.http.uri.UriMatchVariable) Set(java.util.Set) UriMatchTemplate(io.micronaut.http.uri.UriMatchTemplate) ClassElement(io.micronaut.inject.ast.ClassElement) HashMap(java.util.HashMap) Collectors(java.util.stream.Collectors) ParameterElement(io.micronaut.inject.ast.ParameterElement) ArrayList(java.util.ArrayList) HashSet(java.util.HashSet) List(java.util.List) TypedElement(io.micronaut.inject.ast.TypedElement) Map(java.util.Map) MethodElement(io.micronaut.inject.ast.MethodElement) Optional(java.util.Optional) NameUtils(io.micronaut.core.naming.NameUtils) Collections(java.util.Collections) Optional(java.util.Optional) HashMap(java.util.HashMap) ArrayList(java.util.ArrayList) ClassElement(io.micronaut.inject.ast.ClassElement) UriMatchTemplate(io.micronaut.http.uri.UriMatchTemplate) UriMatchVariable(io.micronaut.http.uri.UriMatchVariable) RouteValidationResult(io.micronaut.validation.routes.RouteValidationResult) HashSet(java.util.HashSet)

Aggregations

UriMatchTemplate (io.micronaut.http.uri.UriMatchTemplate)6 List (java.util.List)4 ParameterElement (io.micronaut.inject.ast.ParameterElement)3 ArrayList (java.util.ArrayList)3 Bindable (io.micronaut.core.bind.annotation.Bindable)2 Arrays (java.util.Arrays)2 Collections (java.util.Collections)2 HashMap (java.util.HashMap)2 Map (java.util.Map)2 Optional (java.util.Optional)2 InterceptedMethod (io.micronaut.aop.InterceptedMethod)1 MethodInterceptor (io.micronaut.aop.MethodInterceptor)1 MethodInvocationContext (io.micronaut.aop.MethodInvocationContext)1 BootstrapContextCompatible (io.micronaut.context.annotation.BootstrapContextCompatible)1 ConfigurationException (io.micronaut.context.exceptions.ConfigurationException)1 AnnotationMetadata (io.micronaut.core.annotation.AnnotationMetadata)1 AnnotationValue (io.micronaut.core.annotation.AnnotationValue)1 Internal (io.micronaut.core.annotation.Internal)1 NonNull (io.micronaut.core.annotation.NonNull)1 Nullable (io.micronaut.core.annotation.Nullable)1