com.arangodb.impl.InternalBatchDriverImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of arangodb-java-driver Show documentation
ArangoDB Java Driver
There is a newer version: 7.15.0
/*
 * Copyright (C) 2012 tamtam180
 *
 * 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 com.arangodb.impl;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import com.arangodb.ArangoConfigure;
import com.arangodb.ArangoException;
import com.arangodb.entity.BatchResponseEntity;
import com.arangodb.entity.BatchResponseListEntity;
import com.arangodb.entity.DefaultEntity;
import com.arangodb.http.BatchPart;
import com.arangodb.http.HttpManager;
import com.arangodb.http.HttpResponseEntity;
import com.arangodb.http.InvocationObject;

/**
 * @author Florian Bartels -- [email protected]
 *
 */
public class InternalBatchDriverImpl extends BaseArangoDriverImpl {

	InternalBatchDriverImpl(ArangoConfigure configure, HttpManager httpManager) {
		super(configure, httpManager);
	}

	public static String newline = System.getProperty("line.separator");

	public String delimiter = "dlmtrMLTPRT";

	private BatchResponseListEntity batchResponseListEntity;

	public DefaultEntity executeBatch(List callStack, String defaultDataBase) throws ArangoException {

		String body = "";

		Map resolver = new HashMap();

		for (BatchPart bp : callStack) {
			body += "--" + delimiter + newline;
			body += "Content-Type: application/x-arango-batchpart" + newline;
			body += "Content-Id: " + bp.getId() + newline + newline;
			body += bp.getMethod() + " " + bp.getUrl() + " " + "HTTP/1.1" + newline;
			body += "Host: " + this.configure.getHost() + newline + newline;
			body += bp.getBody() == null ? "" : bp.getBody() + newline + newline;
			resolver.put(bp.getId(), bp.getInvocationObject());
		}
		body += "--" + delimiter + "--";

		Map headers = new HashMap();
		headers.put("Content-Type", "multipart/form-data; boundary=" + delimiter);

		HttpResponseEntity res = httpManager.doPostWithHeaders(createEndpointUrl(defaultDataBase, "/_api/batch"), null,
			null, headers, body);

		String data = res.getText();
		res.setContentType("application/json");
		String currentId = null;
		Boolean fetchText = false;
		res.setText("");
		List batchResponseEntityList = new ArrayList();
		BatchResponseEntity batchResponseEntity = new BatchResponseEntity(null);
		String t = null;
		for (String line : data.split(newline)) {
			line.trim();
			line.replaceAll("\r", "");
			if (line.indexOf("Content-Id") != -1) {
				if (currentId != null) {
					batchResponseEntityList.add(batchResponseEntity);
				}
				currentId = line.split(" ")[1].trim();
				batchResponseEntity = new BatchResponseEntity(resolver.get(currentId));
				batchResponseEntity.setRequestId(currentId);
				continue;
			}
			if (line.indexOf("Content-Type:") != -1
					&& line.indexOf("Content-Type: application/x-arango-batchpart") == -1) {
				String ct = line.replaceAll("Content-Type: ", "");
				batchResponseEntity.httpResponseEntity.setContentType(ct);
				continue;
			}
			if (line.indexOf("Etag") != -1) {
				String etag = line.split(" ")[1].replaceAll("\"", "").trim();
				batchResponseEntity.httpResponseEntity.setEtag(Long.parseLong(etag));
				continue;
			}
			if (line.indexOf("HTTP/1.1") != -1) {
				batchResponseEntity.httpResponseEntity.setStatusCode(Integer.valueOf(line.split(" ")[1]));
				continue;
			}
			if (line.indexOf("Content-Length") != -1) {
				fetchText = true;
				t = "";
				continue;
			}
			if (line.indexOf("--" + delimiter) != -1 && resolver.get(currentId) != null) {
				fetchText = false;
				if (!batchResponseEntity.httpResponseEntity.isDumpResponse()) {
					batchResponseEntity.httpResponseEntity.setText(t);
				} else {
					InputStream is = new ByteArrayInputStream(t.getBytes());
					batchResponseEntity.httpResponseEntity.setStream(is);
				}
				continue;
			}
			if (fetchText == true && !line.equals(newline)) {
				t += line;
			}
		}
		if (batchResponseEntity.getHttpResponseEntity() != null) {
			batchResponseEntityList.add(batchResponseEntity);
		}
		BatchResponseListEntity batchResponseListEntity = new BatchResponseListEntity();
		batchResponseListEntity.setBatchResponseEntities(batchResponseEntityList);
		this.batchResponseListEntity = batchResponseListEntity;
		return createEntity(res, DefaultEntity.class, null, false);
	}

	public BatchResponseListEntity getBatchResponseListEntity() {
		return batchResponseListEntity;
	}
}