at.ac.tuwien.dsg.smartcom.manager.messaging.logging.dao.MongoDBLoggingDAO.java Source code

Introduction

Here is the source code for at.ac.tuwien.dsg.smartcom.manager.messaging.logging.dao.MongoDBLoggingDAO.java

Source

/**
 * Copyright (c) 2014 Technische Universitat Wien (TUW), Distributed Systems Group E184 (http://dsg.tuwien.ac.at)
 *
 * This work was partially supported by the EU FP7 FET SmartSociety (http://www.smart-society-project.eu/).
 *
 *    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 at.ac.tuwien.dsg.smartcom.manager.messaging.logging.dao;

import at.ac.tuwien.dsg.smartcom.model.Identifier;
import at.ac.tuwien.dsg.smartcom.model.Message;
import com.mongodb.BasicDBObject;
import com.mongodb.DBCollection;
import com.mongodb.MongoClient;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.net.UnknownHostException;
import java.util.Date;

/**
 * @author Philipp Zeppezauer (philipp.zeppezauer@gmail.com)
 * @version 1.0
 */
public class MongoDBLoggingDAO implements LoggingDAO {
    private static final Logger log = LoggerFactory.getLogger(MongoDBLoggingDAO.class);
    private static final String LOGGING_COLLECTION = "MESSAGE_LOGGING_COLLECTION";

    private final DBCollection coll;

    public MongoDBLoggingDAO(String host, int port, String database) throws UnknownHostException {
        this(new MongoClient(host, port), database, LOGGING_COLLECTION);
    }

    public MongoDBLoggingDAO(MongoClient client, String database, String collection) {
        coll = client.getDB(database).getCollection(collection);
    }

    @Override
    public void persist(Message message) {
        log.trace("Persisting message to log: {}", message);
        BasicDBObject dbObject = serializeMessage(message);
        log.trace("Created document for message: {}", dbObject);
        coll.insert(dbObject);
    }

    private BasicDBObject serializeMessage(Message message) {
        return new BasicDBObject().append("_id", message.getId().getId()).append("type", message.getType())
                .append("subtype", message.getSubtype())
                .append("sender", serializeIdentifier(message.getSenderId()))
                .append("receiver", serializeIdentifier(message.getReceiverId()))
                .append("content", message.getContent()).append("conversationId", message.getConversationId())
                .append("ttl", message.getTtl()).append("language", message.getLanguage())
                .append("securityToken", message.getSecurityToken()) //TODO should we log the security token too?
                .append("created", new Date());
    }

    private BasicDBObject serializeIdentifier(Identifier identifier) {
        try {
            return new BasicDBObject().append("type", identifier.getType().toString()).append("id",
                    identifier.getId());
        } catch (Exception e) {
            return null;
        }
    }
}