io.firebus.adapters.http.MasterHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of firebus-http Show documentation

Firebus http gateway adapter

There is a newer version: 1.3.3

package io.firebus.adapters.http;


import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.List;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;


public class MasterHandler extends HttpServlet 
{
	private static final long serialVersionUID = 1L;
	
	protected List handlerMap; 
	protected HttpHandler defaultHandler;
	protected String rootForward;

	public MasterHandler()
	{
		handlerMap = new ArrayList();
		rootForward = null;
	}
	
	public void addHttpHandler(String path, String method, HttpHandler handler)
	{
		handlerMap.add(new HttpHandlerEntry(path, method, handler));
	}
	
	public void setDefaultHander(HttpHandler dh)
	{
		defaultHandler = dh;
	}
	
	public void setRootForward(String path)
	{
		rootForward = path;
	}
	

	protected void service(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException
	{
		String path = req.getRequestURI();
		String method = req.getMethod();
		
		if(path.equals("/") && rootForward != null)
			path = rootForward;

		HttpHandler selected = null;
		for(int i = 0; i < handlerMap.size() && selected == null; i++) 
		{
			HttpHandlerEntry entry = handlerMap.get(i);
			boolean match = true;
			if(entry.method == null || (entry.method != null && entry.method.equalsIgnoreCase(method)))
			{
				if(entry.path.endsWith("/*")) 
				{
					String shortEntryPath = entry.path.substring(0, entry.path.length() - 2);
					if(path.startsWith(shortEntryPath + "/") || path.equals(shortEntryPath))
						match = true;
					else
						match = false;
				}
				else
				{
					if(entry.path.equals(path))
						match = true;
					else
						match = false;
				}
			}
			else
			{
				match = false;
			}
			if(match)
				selected = entry.handler;
		}
		if(selected != null)
		{
			selected.service(req, resp);
		}
		else if(defaultHandler != null)
		{
			defaultHandler.service(req, resp);
		}
		else
		{
			resp.setStatus(HttpServletResponse.SC_NOT_FOUND);
	        PrintWriter writer = resp.getWriter();
			writer.println("No mapping, fool.");
		}			
	}	
}