D-Net

package eu.dnetlib.common.logging.dao;

import java.util.*;

import java.util.Map.Entry;

import com.google.common.collect.Lists;

import com.google.common.collect.Maps;

import com.mongodb.BasicDBObject;

import com.mongodb.BasicDBObjectBuilder;

import com.mongodb.DBObject;

import com.mongodb.client.model.Filters;

import eu.dnetlib.common.logging.LogMessage;

import org.apache.commons.lang.StringUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.bson.conversions.Bson;

import org.jongo.Jongo;

import org.jongo.MongoCollection;

import org.jongo.MongoCursor;

import org.springframework.beans.factory.annotation.Autowired;

public class DnetLoggerMongoDao implements DnetLoggerDao {

	private static final Log log = LogFactory.getLog(DnetLoggerMongoDao.class);

	@Autowired

	private Jongo db;

	@Override

	public void init(final String collection) {

		final Set<String> collections = db.getDatabase().getCollectionNames();

		if (!Lists.newArrayList(collections).contains(collection)) {

			log.info(String.format("creating collection %s", collection));

			db.getDatabase().createCollection(collection, new BasicDBObject());

		}

	}

	@Override

	public void configureIndex(final String collection, final Map<String, IndexConf> conf) {

		final MongoCollection coll = db.getCollection(collection);

		coll.dropIndexes();

		for (String key : conf.keySet()) {

			coll.ensureIndex("{'"+key+"':1}");

		}

	}

	@Override

	public void writeLog(final String collection, final Map<String, Object> map) {

		final MongoCollection coll = db.getCollection(collection);

		DBObject obj = BasicDBObjectBuilder.start(replaceKeyNames(map)).get();

		coll.insert(obj);

	}

	private Map<String, Object> replaceKeyNames(final Map<String, Object> inputMap) {

		final Map<String, Object> outMap = Maps.newHashMap();

		if (inputMap != null) {

			for (Entry<String, Object> e : inputMap.entrySet()) {

				final String k = e.getKey();

				if (!StringUtils.isBlank(k)) {

					Object v = e.getValue();

					outMap.put(k.replaceAll("\\.", "_"), v != null ? v : "null");

				}

			}

		}

		return outMap;

	}

	@Override

	public Map<String, String> findOne(final String collection, final String key, final String value) {

		final MongoCollection coll = db.getCollection(collection);

		//final DBObject obj = coll.find(Filters.eq(key, value)).first();

		final DBObject obj = coll.findOne(getBasicQuery(key, value)).as(BasicDBObject.class);

		return dbObject2Map(obj);

	}

	@Override

	public Iterator<Map<String, String>> obtainLogIterator(final String collection) {

		final MongoCollection coll = db.getCollection(collection);

		return iter(collection, coll.find().as(BasicDBObject.class));

	}

	@Override

	public Iterator<Map<String, String>> find(final String collection, final String key, final String value) {

		final MongoCollection coll = db.getCollection(collection);

		return iter(collection, coll.find(getBasicQuery(key, value)).as(BasicDBObject.class));

	}

	@Override

	public Iterator<Map<String, String>> find(final String collection, final Map<String, Object> criteria) {

		final MongoCollection coll = db.getCollection(collection);

		final String query = getCustomQuery(criteria);

		return iter(collection, coll.find(query).as(BasicDBObject.class));

	}

	@Override

	public Iterator<Map<String, String>> findByDateRange(final String collection, final Date startDate, final Date endDate) {

		final MongoCollection coll = db.getCollection(collection);

		//Bson dateQuery = dateRangeQuery(startDate, endDate);

		//Filters.and(Filters.gte(LogMessage.LOG_DATE_FIELD, startDate.getTime()), Filters.lt(LogMessage.LOG_DATE_FIELD, endDate.getTime()));

		final String query = getDateQuery(startDate, endDate);

		// System.out.println(mongoQueryObject);

		return iter(collection, coll.find(query).as(BasicDBObject.class));

	}

	@Override

	public Iterator<Map<String, String>> findByDateRange(final String collection, final Date startDate, final Date endDate, final String key, final String value) {

		final MongoCollection coll = db.getCollection(collection);

		final String dateQuery = getDateQuery(startDate, endDate);

		final String customQuery = getBasicQuery(key, value);

		final StringBuilder sb = new StringBuilder("{'"+key+"':'"+value+"',");

		sb.append(StringUtils.removeEnd(sb.toString(), ","))

				.append("'" + LogMessage.LOG_DATE_FIELD + "' : {$gte:"+startDate.getTime()+"},")

				.append("'" + LogMessage.LOG_DATE_FIELD + "' : {$lt:"+endDate.getTime()+"}}");

		return iter(collection, coll.find(sb.toString()).as(BasicDBObject.class));

	}

	@Override

	public Iterator<Map<String, String>> findByDateRange(final String collection, final Date startDate, final Date endDate, final Map<String, Object> criteria) {

		final MongoCollection coll = db.getCollection(collection);

		//final Bson dateQuery = dateRangeQuery(startDate, endDate);

		//final BasicDBObject customQuery = new BasicDBObject(criteria);

		final StringBuilder sb = new StringBuilder("{");

		for(Entry<String, Object> e : criteria.entrySet()) {

			sb.append("'" + e.getKey() + "':'" + e.getValue() + "', ");

		}

		sb.append(StringUtils.removeEnd(sb.toString(), ","))

			.append("'" + LogMessage.LOG_DATE_FIELD + "' : {$gte:"+startDate.getTime()+"},")

			.append("'" + LogMessage.LOG_DATE_FIELD + "' : {$lt:"+endDate.getTime()+"}}");

		return iter(collection, coll.find(sb.toString()).as(BasicDBObject.class));

	}

	@Override

	public Iterator<Map<String, String>> find(final String collection, final String mongoquery, final List<String> fields) {

		final MongoCollection coll = db.getCollection(collection);

		return iter(collection, coll.find(mongoquery).as(BasicDBObject.class));

	}

	private String getBasicQuery(final String key, final String value) {

		return "{'"+key+"':'"+value+"'}";

	}

	private String getDateQuery(final Date startDate, final Date endDate) {

		return "{'" + LogMessage.LOG_DATE_FIELD + "' : {$gte:"+startDate.getTime()+"}, '" + LogMessage.LOG_DATE_FIELD + "' : {$lt:"+endDate.getTime()+"} }";

	}

	private String getCustomQuery(final Map<String, Object> criteria) {

		final StringBuilder sb = new StringBuilder("{");

		for(Entry<String, Object> e : criteria.entrySet()) {

			sb.append("'" + e.getKey() + "':'" + e.getValue() + "', ");

		}

		return StringUtils.removeEnd(sb.toString(), ",") + "}";

	}

	private Bson dateRangeQuery(final Date startDate, final Date endDate) {

		Bson dateFilter = Filters.and(Filters.gte(LogMessage.LOG_DATE_FIELD, startDate.getTime()), Filters.lt(LogMessage.LOG_DATE_FIELD, endDate.getTime()));

		log.debug("Date filter created: " + dateFilter);

		//return new BasicDBObject(LogMessage.LOG_DATE_FIELD, BasicDBObjectBuilder.start("$gte", startDate.getTime()).append("$lt", endDate.getTime()).get());

		return dateFilter;

	}

	private Iterator<Map<String, String>> iter(final String collection, final MongoCursor<BasicDBObject> cursor) {

		return new Iterator<Map<String, String>>() {

			@Override

			public boolean hasNext() {

				return cursor.hasNext();

			}

			@Override

			public Map<String, String> next() {

				return dbObject2Map(cursor.next());

			}

			@Override

			public void remove() {

				throw new RuntimeException("NOT IMPLEMENTED");

			}

		};

	}

	private Map<String, String> dbObject2Map(final DBObject obj) {

		final Map<String, String> res = Maps.newHashMap();

		if (obj != null) {

			for (String k : obj.keySet()) {

				res.put(k, "" + obj.get(k));

			}

		}

		return res;

	}

}