io.inverno.mod.http.server.internal.multipart.GenericPart Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of inverno-http-server Show documentation
Inverno HTTP 1.x/2 server module
There is a newer version: 1.11.0
/*
 * Copyright 2020 Jeremy KUHN
 *
 * 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
 *
 *    http://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.inverno.mod.http.server.internal.multipart;

import io.inverno.mod.base.Charsets;
import io.inverno.mod.base.converter.ObjectConverter;
import io.inverno.mod.http.base.InboundData;
import io.inverno.mod.http.base.header.Header;
import io.inverno.mod.http.server.Part;
import io.netty.buffer.ByteBuf;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import org.reactivestreams.Publisher;
import reactor.core.publisher.Flux;
import reactor.core.publisher.FluxSink;

/**
 * 
 * Generic {@link Part} implementation.
 * 
 * 
 * @author Jeremy Kuhn
 * @since 1.0
 */
class GenericPart implements Part {

	private String name;
	
	private String filename;

	private PartHeaders partHeaders;
	
	private Flux data;
	private FluxSink dataEmitter;
	
	private InboundData rawData;
	private InboundData stringData;
	
	/**
	 * 
	 * Creates a part.
	 * 
	 * 
	 * @param parameterConverter a string object converter
	 * @param name               the part's name
	 * @param headers            the part's headers
	 */
	public GenericPart(ObjectConverter parameterConverter, String name, Map> headers) {
		this(parameterConverter, name, null, headers);
	}
	
	/**
	 * 
	 * creates a file part.
	 * 
	 * 
	 * @param parameterConverter a string object converter
	 * @param name               the part's name
	 * @param filename           the part's file name
	 * @param headers            the part's headers
	 * @param contentType          the part's media type
	 * @param charset            the part's charset
	 * @param contentLength      the part's content length
	 */
	public GenericPart(ObjectConverter parameterConverter, String name, String filename, Map> headers) {
		this.name = name;
		this.filename = filename;
		this.partHeaders = new PartHeaders(headers, parameterConverter);
		
		this.data = Flux.create(emitter -> {
			this.dataEmitter = emitter;
		}).doOnDiscard(ByteBuf.class, ByteBuf::release);
	}
	
	/**
	 * 
	 * Returns the part's data sink.
	 * 
	 *
	 * @return an optional returning the payload data sink or an empty optional if the part has no body
	 */
	public Optional> getData() {
		return Optional.ofNullable(this.dataEmitter);
	}
	
	@Override
	public String getName() {
		return this.name;
	}
	
	@Override
	public Optional getFilename() {
		return Optional.ofNullable(this.filename);
	}
	
	@Override
	public PartHeaders headers() {
		return this.partHeaders;
	}
	
	// Emitted ByteBuf must be released 
	// But if nobody subscribe they are released
	@Override
	public InboundData raw() {
		if(this.rawData == null) {
			this.rawData = new RawInboundData();
		}
		return this.rawData;
	}

	@Override
	public InboundData string() {
		if(this.stringData == null) {
			this.stringData = new StringInboundData();
		}
		return this.stringData;
	}
	
	/**
	 * 
	 * Generic Part raw data.
	 * 
	 * 
	 * @author Jeremy Kuhn
	 * @since 1.0
	 */
	private class RawInboundData implements InboundData {

		@Override
		public Publisher stream() {
			return GenericPart.this.data;
		}
	}
	
	/**
	 * 
	 * Generic Part String data.
	 * 
	 * 
	 * @author Jeremy Kuhn
	 * @since 1.5
	 */
	private class StringInboundData implements InboundData {

		@Override
		public Publisher stream() {
			return GenericPart.this.data.map(buf -> {
				try {
					return buf.toString(Charsets.DEFAULT);
				}
				finally {
					buf.release();
				}
			});
		}
	}
}