D-Net

{"type_source": "SVN", "goal": "package -U -T 4C source:jar", "url": "http://svn-public.driver.research-infrastructures.eu/driver/dnet40/modules/dnet-mongo-logging/trunk/", "deploy_repository": "dnet4-snapshots", "version": "4", "mail": "sandro.labruzzo@isti.cnr.it,michele.artini@isti.cnr.it, claudio.atzori@isti.cnr.it, alessia.bardi@isti.cnr.it", "deploy_repository_url": "http://maven.research-infrastructures.eu/nexus/content/repositories/dnet4-snapshots", "name": "dnet-mongo-logging"}

package eu.dnetlib.common.logging.dao;

import java.util.Date;

import java.util.Iterator;

import java.util.Map;

import java.util.Map.Entry;

import org.apache.commons.lang.StringUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

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

import com.google.common.collect.Maps;

import com.mongodb.BasicDBObject;

import com.mongodb.BasicDBObjectBuilder;

import com.mongodb.DB;

import com.mongodb.DBCollection;

import com.mongodb.DBCursor;

import com.mongodb.DBObject;

import com.mongodb.QueryBuilder;

import eu.dnetlib.common.logging.LogMessage;

public class DnetLoggerMongoDao implements DnetLoggerDao {

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

	private DB db;

	@Override

	public void init(final String collection) {

		if (!db.collectionExists(collection)) {

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

			db.createCollection(collection, null);

		}

	}

	@Override

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

		final DBCollection coll = db.getCollection(collection);

		coll.dropIndexes();

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

			coll.createIndex(new BasicDBObject(key, 1));

		}

	}

	@Override

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

		final DBCollection 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 DBCollection coll = db.getCollection(collection);

		final DBObject obj = coll.findOne(new BasicDBObject(key, value));

		return dbObject2Map(obj);

	}

	@Override

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

		final DBCollection coll = db.getCollection(collection);

		return iter(collection, coll.find());

	}

	@Override

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

		final DBCollection coll = db.getCollection(collection);

		return iter(collection, coll.find(new BasicDBObject(key, value)));

	}

	@Override

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

		final DBCollection coll = db.getCollection(collection);

		return iter(collection, coll.find(new BasicDBObject(criteria)));

	}

	@Override

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

		final DBCollection coll = db.getCollection(collection);

		DBObject mongoQueryObject = dateRangeQuery(startDate, endDate);

		// System.out.println(mongoQueryObject);

		return iter(collection, coll.find(mongoQueryObject));

	}

	@Override

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

		final DBCollection coll = db.getCollection(collection);

		final DBObject dateQuery = dateRangeQuery(startDate, endDate);

		final BasicDBObject customQuery = new BasicDBObject(key, value);

		return iter(collection, coll.find(QueryBuilder.start().and(dateQuery, customQuery).get()));

	}

	@Override

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

		final DBCollection coll = db.getCollection(collection);

		final DBObject dateQuery = dateRangeQuery(startDate, endDate);

		final BasicDBObject customQuery = new BasicDBObject(criteria);

		return iter(collection, coll.find(QueryBuilder.start().and(dateQuery, customQuery).get()));

	}

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

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

	}

	private Iterator<Map<String, String>> iter(final String collection, final DBCursor 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");

			}

		};

	}

	public DB getDb() {

		return db;

	}

	@Required

	public void setDb(final DB db) {

		this.db = db;

	}

	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;

	}

}

<?xml version="1.0" encoding="UTF-8"?>

<beans xmlns="http://www.springframework.org/schema/beans"

	xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:jaxws="http://cxf.apache.org/jaxws"

	xmlns:sec="http://cxf.apache.org/configuration/security" xmlns:wsa="http://cxf.apache.org/ws/addressing"

	xmlns:p="http://www.springframework.org/schema/p" xmlns:http="http://cxf.apache.org/transports/http/configuration"

	xmlns:t="http://dnetlib.eu/springbeans/t" xmlns:template="http://dnetlib.eu/springbeans/template"

	xmlns:util="http://www.springframework.org/schema/util"

	xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd

                                    http://cxf.apache.org/ws/addressing http://cxf.apache.org/schemas/ws-addr-conf.xsd

                                    http://cxf.apache.org/configuration/security http://cxf.apache.org/schemas/configuration/security.xsd

                                    http://cxf.apache.org/transports/http/configuration http://cxf.apache.org/schemas/configuration/http-conf.xsd

                            http://cxf.apache.org/jaxws http://cxf.apache.org/schemas/jaxws.xsd

                            http://www.springframework.org/schema/util http://www.springframework.org/schema/util/spring-util-2.0.xsd

                            http://dnetlib.eu/springbeans/template http://dnetlib.eu/springbeans/template.xsd">

	<bean id="dnetLoggerMongoDao" class="eu.dnetlib.common.logging.dao.DnetLoggerMongoDao"

		p:db-ref="loggingMongoDB" />

	<bean id="loggingMongoServer" class="com.mongodb.Mongo">

		<constructor-arg index="0" type="com.mongodb.ServerAddress">

			<bean class="com.mongodb.ServerAddress">

				<constructor-arg index="0"

					value="${dnet.logger.mongo.host}" />

				<constructor-arg index="1"

					value="${dnet.logger.mongo.port}" />

			</bean>

		</constructor-arg>

	</bean>

	<bean id="loggingMongoDB" factory-bean="loggingMongoServer"

		factory-method="getDB">

		<constructor-arg index="0" value="${dnet.logger.mongo.db}" />

	</bean>

</beans>

dnet.logger.mongo.host = localhost

dnet.logger.mongo.port = 27017

dnet.logger.mongo.db   = dnet_logs

<?xml version="1.0" encoding="UTF-8"?>

<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/maven-v4_0_0.xsd">

	<parent>

		<groupId>eu.dnetlib</groupId>

		<artifactId>dnet-parent</artifactId>

		<version>1.0.0</version>

		<relativePath />

	</parent>

	<modelVersion>4.0.0</modelVersion>

	<groupId>eu.dnetlib</groupId>

	<artifactId>dnet-mongo-logging</artifactId>

	<packaging>jar</packaging>

	<version>1.1.2</version>

	<scm>

	  <developerConnection>scm:svn:https://svn.driver.research-infrastructures.eu/driver/dnet40/modules/dnet-mongo-logging/tags/dnet-mongo-logging-1.1.2</developerConnection>

	</scm>

	<dependencies>

		<dependency>

			<groupId>org.mongodb</groupId>

			<artifactId>mongo-java-driver</artifactId>

			<version>${mongodb.driver.version}</version>

		</dependency>

		<dependency>

			<groupId>eu.dnetlib</groupId>

			<artifactId>dnet-modular-logging</artifactId>

			<version>[1.0.0,2.0.0)</version>

		</dependency>

	</dependencies>

</project>