org.eclipse.emf.cdo.server.internal.mongodb.MongoDBStoreFactory.java Source code

Introduction

Here is the source code for org.eclipse.emf.cdo.server.internal.mongodb.MongoDBStoreFactory.java

Source

/*
 * Copyright (c) 2011, 2012, 2015 Eike Stepper (Berlin, Germany) and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    Eike Stepper - initial API and implementation
 */
package org.eclipse.emf.cdo.server.internal.mongodb;

import org.eclipse.emf.cdo.server.IStore;
import org.eclipse.emf.cdo.server.IStoreFactory;
import org.eclipse.emf.cdo.server.mongodb.CDOMongoDBUtil;
import org.eclipse.emf.cdo.spi.server.RepositoryConfigurator;

import org.eclipse.net4j.util.StringUtil;
import org.eclipse.net4j.util.WrappedException;

import com.mongodb.DB;
import com.mongodb.Mongo;
import com.mongodb.MongoURI;

import org.w3c.dom.Element;

import java.util.Map;

/**
 * @author Eike Stepper
 */
public class MongoDBStoreFactory implements IStoreFactory {
    public MongoDBStoreFactory() {
    }

    public String getStoreType() {
        return MongoDBStore.TYPE;
    }

    public IStore createStore(String repositoryName, Map<String, String> repositoryProperties,
            Element storeConfig) {
        Map<String, String> properties = RepositoryConfigurator.getProperties(storeConfig, 1);
        String uri = properties.get("uri");
        if (StringUtil.isEmpty(uri)) {
            throw new IllegalArgumentException("Property 'uri' missing");
        }

        MongoURI mongoURI = new MongoURI(uri);
        String dbName = properties.get("db");
        if (StringUtil.isEmpty(dbName)) {
            dbName = repositoryName;
        }

        String drop = properties.get("drop");
        if (Boolean.toString(true).equals(drop)) {
            dropDatabase(mongoURI, dbName);
        }

        return CDOMongoDBUtil.createStore(uri, dbName);
    }

    protected void dropDatabase(MongoURI mongoURI, String dbName) {
        Mongo mongo = null;

        try {
            mongo = new Mongo(mongoURI);
            DB db = mongo.getDB(dbName);
            db.dropDatabase();
        } catch (Exception ex) {
            throw WrappedException.wrap(ex);
        } finally {
            if (mongo != null) {
                mongo.close();
            }
        }
    }
}