com.davfx.ninio.http.util.HttpServerHandlerToSimpleHttpServerHandler Maven / Gradle / Ivy

Go to download
package com.davfx.ninio.http.util;

import java.io.IOException;
import java.nio.ByteBuffer;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.davfx.ninio.common.Address;
import com.davfx.ninio.http.Http;
import com.davfx.ninio.http.HttpRequest;
import com.davfx.ninio.http.HttpResponse;
import com.davfx.ninio.http.HttpServerHandler;

public final class HttpServerHandlerToSimpleHttpServerHandler implements HttpServerHandler {
	private static final Logger LOGGER = LoggerFactory.getLogger(HttpServerHandlerToSimpleHttpServerHandler.class);
	
	private final SimpleHttpServerHandler handler;
	private final String contentType;
	private HttpRequest request = null;
	private InMemoryPost post = null;
	
	public HttpServerHandlerToSimpleHttpServerHandler(SimpleHttpServerHandler handler) {
		this(Http.ContentType.TEXT, handler);
	}
	public HttpServerHandlerToSimpleHttpServerHandler(String contentType, SimpleHttpServerHandler handler) {
		this.contentType = contentType;
		this.handler = handler;
	}
	
	@Override
	public void handle(HttpRequest request) {
		this.request = request;
	}

	@Override
	public void handle(Address address, ByteBuffer buffer) {
		if (post == null) {
			post = new InMemoryPost();
		}
		post.add(buffer);
	}
	
	@Override
	public void ready(Write write) {
		HttpQuery query = new HttpQuery(request.getPath());
		
		String path = query.getPath();
		Parameters parameters = query.getParameters();
		
		String response;
		switch (request.getMethod()) {
		case GET:
			response = handler.get(path, parameters);
			break;
		case DELETE:
			response = handler.delete(path, parameters);
			break;
		case HEAD:
			response = handler.head(path, parameters);
			break;
		case PUT:
			response = handler.put(path, parameters);
			break;
		case POST:
			response = handler.post(path, parameters, post);
			break;
		default:
			response = null;
			break;
		}
		HttpResponse r = new HttpResponse(Http.Status.OK, Http.Message.OK);
		if (contentType != null) {
			r.getHeaders().put(Http.CONTENT_TYPE, contentType);
		}
		ByteBuffer bb;
		if (response != null) {
			bb = ByteBuffer.wrap(response.getBytes(Http.UTF8_CHARSET));
			r.getHeaders().put(Http.CONTENT_LENGTH, String.valueOf(bb.remaining()));
		} else {
			bb = null;
		}
		write.write(r);
		if (bb != null) {
			write.handle(null, bb);
		}
		write.close();
		request = null;
		post = null;
	}
	
	@Override
	public void failed(IOException e) {
		LOGGER.debug("Client connection failed", e);
	}
	
	@Override
	public void close() {
		LOGGER.debug("Connection closed by peer");
	}
}