org.butor.json.BinStreamHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of butor-json Show documentation

This module enables fast and easy creation of sync., async., req./resp., req./resp. stream HTTP/json services.

There is a newer version: 1.0.29

/*******************************************************************************
 * Copyright 2013 butor.com
 * 
 * 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 org.butor.json;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.ByteBuffer;
import java.util.HashMap;
import java.util.Map;

import org.butor.json.service.BinResponseHandler;
import org.butor.json.service.ResponseHandler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.base.Strings;

public class BinStreamHandler implements StreamHandler {
	protected Logger _logger = LoggerFactory.getLogger(getClass());

	@Override
	public void parse(InputStream is,
			ResponseHandler handler, String loReqInfo) throws IOException {
		//_reqLogInfo = loReqInfo_;
		// if content-type and headers are set, the stream starts
		// with "___content_type___\n"
		String boundary = "___content_type___\n";
		BinResponseHandler brh = (BinResponseHandler)handler;
		OutputStream os = brh.getOutputStream();

		int n = 0;
		int bl = boundary.length();
		byte[] buffer = new byte[1024];
		while (n < bl) {
			byte b = (byte)is.read();
			if (b == -1) {
				break;
			}
			buffer[n++] = b;
		}
		if (new String(buffer, 0, bl).equalsIgnoreCase(boundary)) {
			// then retrieve content-type and others headers till reaching \n\n
			ByteBuffer bb = ByteBuffer.allocate(1024); // will grows if more space required
			bb.clear();
			byte prevB = 0, b=0;
			while (true) {
				prevB = b;
				b = (byte) is.read();
				if (b == 10 && prevB == 10) { // end of headers
					break;
				}
				bb.put(b);
			}
			bb.flip();
			String[] lines = new String(bb.array(), 0, bb.limit()).split("\n");
			bb.clear();
			String contentType = null;
			Map headers = new HashMap();
			for (String h : lines) {
				if (Strings.isNullOrEmpty(h)) {
					continue;
				}
				int pos = h.indexOf(":");
				String hn = h.substring(0, pos++);
				String hv = h.substring(pos);
				if (hn .equalsIgnoreCase("Content-Type")) {
					contentType = hv;
				} else {
					headers.put(hn, hv);
				}
			}
			brh.setContentType(contentType, headers);

		} else {
			os.write(buffer, 0, bl);
		}

		while ((n = is.read(buffer)) > -1) {
			os.write(buffer, 0, n);
		}
		brh.end();
	}

}