com.lordofthejars.nosqlunit.couchdb.DataLoader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of nosqlunit-couchdb Show documentation

There is a newer version: 1.0.0

package com.lordofthejars.nosqlunit.couchdb;

import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.Map;

import org.codehaus.jackson.JsonParseException;
import org.codehaus.jackson.JsonProcessingException;
import org.codehaus.jackson.map.JsonMappingException;
import org.codehaus.jackson.map.ObjectMapper;
import org.ektorp.CouchDbConnector;

public class DataLoader {

	private static final ObjectMapper MAPPER = new ObjectMapper();

	public static final String ROOT_ELEMENT = "data";

	private CouchDbConnector connector;

	public DataLoader(CouchDbConnector connector) {
		this.connector = connector;
	}

	public void load(InputStream dataScript) {

		try {
			List> documentsIterator = getDocuments(dataScript);
			insertDocuments(documentsIterator);

		} catch (JsonParseException e) {
			throw new IllegalArgumentException(e);
		} catch (JsonMappingException e) {
			throw new IllegalArgumentException(e);
		} catch (IOException e) {
			throw new IllegalArgumentException(e);
		}

	}

	private void insertDocuments(List> documentsIterator) {
		for (Map map : documentsIterator) {
			this.connector.create(map);
		}
	}

	public static List> getDocuments(InputStream dataScript) throws IOException, JsonProcessingException {
		Map rootNode = MAPPER.readValue(dataScript, Map.class);

		Object dataElements = rootNode.get(ROOT_ELEMENT);

		if(dataElements instanceof List) {
			return (List>)dataElements;
		} else {
			throw new IllegalArgumentException("Array of documents are required.");
		}
		
	}

}