ch.bfh.uniboard.persistence.mongodb.ConnectionManagerTestImpl.java Source code

Introduction

Here is the source code for ch.bfh.uniboard.persistence.mongodb.ConnectionManagerTestImpl.java

Source

/*
 * Copyright (c) 2014 Berner Fachhochschule, Switzerland.
 * Bern University of Applied Sciences, Engineering and Information Technology,
 * Research Institute for Security in the Information Society, E-Voting Group,
 * Biel, Switzerland.
 *
 * Project UniBoard.
 *
 * Distributable under GPL license.
 * See terms of license at gnu.org.
 */
package ch.bfh.uniboard.persistence.mongodb;

import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.MongoClient;
import de.flapdoodle.embed.mongo.MongodExecutable;
import de.flapdoodle.embed.mongo.MongodProcess;
import de.flapdoodle.embed.mongo.MongodStarter;
import de.flapdoodle.embed.mongo.config.MongodConfigBuilder;
import de.flapdoodle.embed.mongo.config.Net;
import de.flapdoodle.embed.mongo.distribution.Version;
import de.flapdoodle.embed.process.runtime.Network;
import java.io.IOException;
import java.net.UnknownHostException;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
import javax.ejb.LocalBean;
import javax.ejb.Singleton;
import javax.ejb.Startup;

/**
 *
 * @author Severin Hauser <severin.hauser@bfh.ch>
 */
@Singleton
@Startup
@LocalBean
public class ConnectionManagerTestImpl implements ConnectionManager {

    private static final Logger logger = Logger.getLogger(ConnectionManagerImpl.class.getName());

    //TODO put this in a config file
    private static final String host = "localhost";
    private static final String dbName = "testDB";
    private static final String collectionName = "test";
    protected static final int port = 27017;
    private static final String username = "test";
    private static final String password = "test";
    //must be false in Unit testing config
    private static final boolean authentication = false;

    private DBCollection collection;
    private MongoClient mongoClient;
    private boolean connected = false;

    private static final MongodStarter starter = MongodStarter.getDefaultInstance();
    private static MongodExecutable mongodExe;
    private static MongodProcess mongod;

    @PostConstruct
    private void init() {

        try {
            mongodExe = starter.prepare(new MongodConfigBuilder().version(Version.Main.PRODUCTION)
                    .net(new Net(27017, Network.localhostIsIPv6())).build());
            mongod = mongodExe.start();
            mongoClient = new MongoClient(host, port);
            this.connected = true;
        } catch (UnknownHostException ex) {
            logger.log(Level.SEVERE, "DB creation error", ex);
            return;
        } catch (IOException ex) {
            logger.log(Level.SEVERE, "DB creation error", ex);
            return;
        }

        //Create or load the database
        DB db = mongoClient.getDB(dbName);

        //create the collection if it does not exist
        if (!db.collectionExists(collectionName)) {
            collection = db.createCollection(collectionName, null);
        }
        //load the collection
        collection = db.getCollection(collectionName);
    }

    @PreDestroy
    private void preDestroy() {
        mongoClient.close();
    }

    @Override
    public DBCollection getCollection() {
        return this.collection;
    }

    @Override
    public boolean isConnected() {
        return connected;
    }

}