ca.carleton.gcrc.couch.date.impl.DateSourceCouchWithCluster Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nunaliit2-couch-date Show documentation
There is a newer version: 2.2.7
package ca.carleton.gcrc.couch.date.impl;

import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.List;

import org.json.JSONArray;
import org.json.JSONObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import ca.carleton.gcrc.couch.client.CouchDesignDocument;
import ca.carleton.gcrc.couch.client.CouchQuery;
import ca.carleton.gcrc.couch.client.CouchQueryResults;
import ca.carleton.gcrc.couch.date.cluster.Tree;
import ca.carleton.gcrc.couch.date.cluster.CouchIntervalClusterTreeFactory;

public class DateSourceCouchWithCluster implements DateSource, SerializableToDot, SerializableToInfo {

	final protected Logger logger = LoggerFactory.getLogger(this.getClass());

	private CouchDesignDocument atlasDesignDocument;	
	private Tree clusterTree = null;
	
	public DateSourceCouchWithCluster(CouchDesignDocument atlasDesignDocument) throws Exception {
		this.atlasDesignDocument = atlasDesignDocument;

		try {
			CouchIntervalClusterTreeFactory factory = new CouchIntervalClusterTreeFactory();
			clusterTree = factory.createClusterTree(atlasDesignDocument);
		} catch(Exception e) {
			logger.error("Unable to create a date cluster tree",e);
			throw new Exception("Unable to create a date cluster tree",e);
		}
	}
	
	public Tree getClusterTree(){
		return clusterTree;
	}
	
	@Override
	public SearchResults getAllDateIntervals() throws Exception {
		SearchResults results = new SearchResults();
		
		CouchQuery query = new CouchQuery();
		query.setViewName("date-index");
		query.setIncludeDocs(false);
		query.setReduce(false);
		CouchQueryResults queryResults = atlasDesignDocument.performQuery(query);
		
		results.documentWithIntervals = new ArrayList(queryResults.getRows().size());
		for(JSONObject row : queryResults.getRows()){
			String docId = row.optString("id");
			JSONObject jsonInterval = row.optJSONObject("value");
			if( null != docId && null != jsonInterval ){
				TimeInterval interval = TimeInterval.fromJson(jsonInterval);
				DocumentWithInterval docWithInt = new DocumentWithInterval(docId, interval);
				results.documentWithIntervals.add(docWithInt);
			}
		}
		
		return results;
	}

	@Override
	public SearchResults getDateIntervalsIntersectingWith(TimeInterval interval, NowReference now) throws Exception {
		SearchResults results = new SearchResults();
		
		List clusterIds = clusterTree.clusterIdsFromInterval(interval, now);
		results.clusterCount = clusterIds.size();
		JSONArray keys = new JSONArray();
		keys.put(JSONObject.NULL); // always include un-indexed intervals
		for(Integer clusterId : clusterIds){
			keys.put(clusterId.intValue());
		}
		
		CouchQuery query = new CouchQuery();
		query.setViewName("date-index");
		query.setIncludeDocs(false);
		query.setReduce(false);
		query.setKeys(keys);
		CouchQueryResults queryResults = atlasDesignDocument.performQuery(query);
		
		results.documentWithIntervals = new ArrayList(queryResults.getRows().size());
		for(JSONObject row : queryResults.getRows()){
			String docId = row.optString("id");
			JSONObject jsonInterval = row.optJSONObject("value");
			results.intervalCount++;
			if( null != docId && null != jsonInterval ){
				TimeInterval docInterval = TimeInterval.fromJson(jsonInterval);
				if( docInterval.intersectsWith(interval, now) ){
					DocumentWithInterval docWithInt = new DocumentWithInterval(docId, docInterval);
					results.documentWithIntervals.add(docWithInt);
					results.intervalMatched++;
				}
			}
		}
		
		return results;
	}

	@Override
	public void printDot(PrintWriter pw) throws Exception {
		Tree.treeToDot(clusterTree, pw);
	}

	@Override
	public void printInfo(PrintWriter pw) throws Exception {
		Tree.treeToInfo(clusterTree, pw);
	}
}