io.firebus.adapters.http.outbound.GeneralOutboundHandler 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.outbound;

import java.io.IOException;
import java.util.Iterator;

import javax.servlet.ServletException;

import org.apache.http.HttpEntity;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpUriRequest;
import org.apache.http.entity.ByteArrayEntity;
import org.apache.http.util.EntityUtils;

import io.firebus.Firebus;
import io.firebus.Payload;
import io.firebus.adapters.http.OutboundHandler;
import io.firebus.utils.DataException;
import io.firebus.utils.DataMap;

public class GeneralOutboundHandler extends OutboundHandler {

	public GeneralOutboundHandler(DataMap c, Firebus f) 
	{
		super(c, f);
	}

	@Override
	protected HttpUriRequest processRequest(Payload payload) throws ServletException, IOException, DataException {
		DataMap request = new DataMap(payload.getString());
		String url = (this.baseUrl != null && request.containsKey("path") ? baseUrl + "/" + request.getString("path") : request.getString("url"));
		String method = request.getString("method");
		HttpUriRequest httpRequest = null;
		if(method.equals("post")) 
		{
			httpRequest = new HttpPost(url);
			((HttpPost)httpRequest).setEntity(new ByteArrayEntity(request.get("body").toString().getBytes()));
		}
		else if(method.equals("get"))
		{
			httpRequest = new HttpGet(url);
		}
		if(request.containsKey("cookie")) {
			String cookie = "";
			Iterator it = request.getObject("cookie").keySet().iterator();
			while(it.hasNext()) {
				String key = it.next();
				cookie = cookie + key + "=" + request.getObject("cookie").getString(key) + (it.hasNext() ? ";" : "");
			}
			httpRequest.setHeader("Cookie", cookie);
		}
		if(request.containsKey("authorization")) {
			String authorization = request.getString("authorization");
			httpRequest.setHeader("Authorization", authorization);
		}
		return httpRequest;
	}

	@Override
	protected Payload processResponse(HttpEntity response) throws ServletException, IOException, DataException {
		String responseStr = EntityUtils.toString(response);
		EntityUtils.consume(response);
		Payload payload = new Payload(responseStr);
		return payload;
	}

}