All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • io.micronaut.http.server.netty.converters.NettyConverters Maven / Gradle / Ivy

    /*
 * Copyright 2017-2020 original authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.micronaut.http.server.netty.converters;

import io.micronaut.buffer.netty.NettyByteBufferFactory;
import io.micronaut.context.BeanProvider;
import io.micronaut.context.annotation.Prototype;
import io.micronaut.core.annotation.Internal;
import io.micronaut.core.convert.ArgumentConversionContext;
import io.micronaut.core.convert.ConversionService;
import io.micronaut.core.convert.MutableConversionService;
import io.micronaut.core.convert.TypeConverterRegistrar;
import io.micronaut.core.naming.NameUtils;
import io.micronaut.core.type.Argument;
import io.micronaut.core.util.StringUtils;
import io.micronaut.http.MediaType;
import io.micronaut.http.body.MessageBodyHandlerRegistry;
import io.micronaut.http.body.MessageBodyReader;
import io.micronaut.http.codec.MediaTypeCodec;
import io.micronaut.http.codec.MediaTypeCodecRegistry;
import io.micronaut.http.netty.channel.converters.ChannelOptionFactory;
import io.micronaut.http.server.netty.multipart.NettyPartData;
import io.micronaut.http.simple.SimpleHttpHeaders;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.ByteBufInputStream;
import io.netty.channel.ChannelOption;
import io.netty.handler.codec.http.multipart.Attribute;
import io.netty.handler.codec.http.multipart.FileUpload;
import io.netty.util.ReferenceCounted;

import java.io.IOException;
import java.io.InputStream;
import java.nio.ByteBuffer;
import java.util.Locale;
import java.util.Optional;

/**
 * Factory for bytebuf related converters.
 *
 * @author graemerocher
 * @since 1.0
 */
@Prototype
@Internal
public final class NettyConverters implements TypeConverterRegistrar {

    private final ConversionService conversionService;
    private final BeanProvider decoderRegistryProvider;
    private final BeanProvider messageBodyHandlerRegistries;
    private final BeanProvider channelOptionFactory;

    /**
     * Default constructor.
     *
     * @param conversionService            The conversion service
     * @param decoderRegistryProvider      The decoder registry provider
     * @param messageBodyHandlerRegistries The message body handlers
     * @param channelOptionFactory         The decoder channel option factory
     */
    public NettyConverters(ConversionService conversionService,
                           //Prevent early initialization of the codecs
                           BeanProvider decoderRegistryProvider,
                           BeanProvider messageBodyHandlerRegistries,
                           BeanProvider channelOptionFactory) {
        this.conversionService = conversionService;
        this.decoderRegistryProvider = decoderRegistryProvider;
        this.messageBodyHandlerRegistries = messageBodyHandlerRegistries;
        this.channelOptionFactory = channelOptionFactory;
    }

    @Override
    public void register(MutableConversionService conversionService) {
        conversionService.addConverter(
                CharSequence.class,
                ChannelOption.class,
                (object, targetType, context) -> {
                    String str = object.toString();
                    String name = NameUtils.underscoreSeparate(str).toUpperCase(Locale.ENGLISH);
                    return Optional.of(channelOptionFactory.get().channelOption(name));
                }
        );

        conversionService.addConverter(
                ByteBuf.class,
                Object.class,
                (object, targetType, context) -> conversionService.convert(object.toString(context.getCharset()), targetType, context)
        );

        conversionService.addConverter(
                FileUpload.class,
                Object.class,
                (object, targetType, context) -> {
                    try {
                        if (!object.isCompleted()) {
                            return Optional.empty();
                        }
                        Argument argument = context instanceof ArgumentConversionContext argumentConversionContext ? (Argument) argumentConversionContext.getArgument() : Argument.of(targetType);
                        String contentType = object.getContentType();
                        ByteBuf byteBuf = object.getByteBuf();
                        MediaType mediaType = StringUtils.isEmpty(contentType) ? null : MediaType.of(contentType);
                        MediaTypeCodec codec = decoderRegistryProvider.get().findCodec(mediaType).orElse(null);
                        if (codec != null) {
                            return Optional.of(
                                codec.decode(argument, new ByteBufInputStream(byteBuf))
                            );
                        }
                        MessageBodyReader messageBodyReader = messageBodyHandlerRegistries.get().findReader(argument, mediaType).orElse(null);
                        if (messageBodyReader != null) {
                            return Optional.of(
                                messageBodyReader.read(argument, mediaType, new SimpleHttpHeaders(), NettyByteBufferFactory.DEFAULT.wrap(byteBuf))
                            );
                        }
                        return conversionService.convert(byteBuf, targetType, context);
                    } catch (Exception e) {
                        context.reject(e);
                        return Optional.empty();
                    }
                }
        );

        conversionService.addConverter(
                NettyPartData.class,
                Object.class,
                (object, targetType, context) -> {
                    try {
                        if (targetType.isAssignableFrom(ByteBuffer.class)) {
                            return Optional.of(object.getByteBuffer());
                        } else if (targetType.isAssignableFrom(InputStream.class)) {
                            return Optional.of(object.getInputStream());
                        } else {
                            ByteBuf byteBuf = object.getByteBuf();
                            try {
                                return this.conversionService.convert(byteBuf, targetType, context);
                            } finally {
                                byteBuf.release();
                            }
                        }
                    } catch (IOException e) {
                        context.reject(e);
                        return Optional.empty();
                    }
                }
        );

        conversionService.addConverter(
                Attribute.class,
                Object.class,
                (object, targetType, context) -> {
                    try {
                        final String value = object.getValue();
                        if (targetType.isInstance(value)) {
                            return Optional.of(value);
                        } else {
                            return this.conversionService.convert(value, targetType, context);
                        }
                    } catch (IOException e) {
                        context.reject(e);
                        return Optional.empty();
                    }
                }
        );

        conversionService.addConverter(
                String.class,
                ChannelOption.class,
                s -> channelOptionFactory.get().channelOption(NameUtils.environmentName(s))
        );
    }

    /**
     * This method converts a
     * {@link io.netty.util.ReferenceCounted netty reference counted object} and transfers release
     * ownership to the new object.
     *
     * @param service The conversion service
     * @param context The context to convert to
     * @param input   The object to convert
     * @param      Target type
     * @return The converted object
     */
    public static  Optional refCountAwareConvert(ConversionService service, ReferenceCounted input, ArgumentConversionContext context) {
        Optional converted = service.convert(input, context);
        postProcess(input, converted);
        return converted;
    }

    @SuppressWarnings("OptionalUsedAsFieldOrParameterType")
    public static  void postProcess(ReferenceCounted input, Optional converted) {
        if (converted.isPresent()) {
            input.touch();
            T item = converted.get();
            // this is not great, but what can we do?
            boolean targetRefCounted = item instanceof ReferenceCounted || item instanceof io.micronaut.core.io.buffer.ReferenceCounted;
            if (!targetRefCounted) {
                input.release();
            }
        } else {
            input.release();
        }
    }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy