com.arangodb.impl.InternalImportDriverImpl 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,2013 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.util.Collection;
import java.util.Map;

import com.arangodb.ArangoConfigure;
import com.arangodb.ArangoException;
import com.arangodb.entity.EntityFactory;
import com.arangodb.entity.ImportResultEntity;
import com.arangodb.http.HttpManager;
import com.arangodb.http.HttpResponseEntity;
import com.arangodb.util.ImportOptions;
import com.arangodb.util.ImportOptionsJson;
import com.arangodb.util.ImportOptionsRaw;

/**
 * @author tamtam180 - kirscheless at gmail.com
 * @see 
 *      HttpBulkImports documentation
 */
public class InternalImportDriverImpl extends BaseArangoDriverImpl implements com.arangodb.InternalImportDriver {

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

	@Override
	public ImportResultEntity importDocuments(
		String database,
		String collection,
		Collection values,
		ImportOptionsJson importOptionsJson) throws ArangoException {

		Map map = importOptionsJson.toMap();
		map.put("type", "list");

		return importDocumentsInternal(database, collection, EntityFactory.toJsonString(values), map);
	}

	@Override
	public ImportResultEntity importDocumentsRaw(
		String database,
		String collection,
		String values,
		ImportOptionsRaw importOptionsRaw) throws ArangoException {

		return importDocumentsInternal(database, collection, values, importOptionsRaw.toMap());
	}

	@Override
	public ImportResultEntity importDocumentsByHeaderValues(
		String database,
		String collection,
		Collection> headerValues,
		ImportOptions importOptions) throws ArangoException {

		return importDocumentsByHeaderValuesInternal(database, collection,
			EntityFactory.toImportHeaderValues(headerValues), importOptions);
	}

	@Override
	public ImportResultEntity importDocumentsByHeaderValuesRaw(
		String database,
		String collection,
		String headerValues,
		ImportOptions importOptions) throws ArangoException {

		return importDocumentsByHeaderValuesInternal(database, collection, headerValues, importOptions);
	}

	private ImportResultEntity importDocumentsInternal(
		String database,
		String collection,
		String values,
		Map importOptions) throws ArangoException {

		importOptions.put(COLLECTION, collection);

		HttpResponseEntity res = httpManager.doPost(createEndpointUrl(database, "/_api/import"), importOptions, values);

		return createEntity(res, ImportResultEntity.class);
	}

	private ImportResultEntity importDocumentsByHeaderValuesInternal(
		String database,
		String collection,
		String headerValues,
		ImportOptions importOptions) throws ArangoException {

		Map map = importOptions.toMap();
		map.put(COLLECTION, collection);

		HttpResponseEntity res = httpManager.doPost(createEndpointUrl(database, "/_api/import"), map, headerValues);

		return createEntity(res, ImportResultEntity.class);

	}

}