MongoMapEntityStoreAssembler.java

/*
 * Copyright 2011 Paul Merlin.
 *
 * 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 org.qi4j.entitystore.mongodb;

import com.mongodb.ServerAddress;
import java.net.UnknownHostException;
import java.util.ArrayList;
import java.util.List;
import org.qi4j.api.common.Visibility;
import org.qi4j.bootstrap.Assembler;
import org.qi4j.bootstrap.AssemblyException;
import org.qi4j.bootstrap.ModuleAssembly;
import org.qi4j.entitystore.mongodb.MongoEntityStoreConfiguration.WriteConcern;
import org.qi4j.spi.uuid.UuidIdentityGeneratorService;

public class MongoMapEntityStoreAssembler
    implements Assembler
{

    private Visibility visibility = Visibility.application;
    private ModuleAssembly configModule;
    private Visibility configVisibility = Visibility.layer;
    private String hostname = "127.0.0.1";
    private Integer port = 27017;
    private String database;
    private String collection;
    private WriteConcern writeConcern;
    private List<ServerAddress> serverAddresses;

    public MongoMapEntityStoreAssembler withVisibility( Visibility visibility )
    {
        this.visibility = visibility;
        return this;
    }

    public MongoMapEntityStoreAssembler withConfigModule( ModuleAssembly configModule )
    {
        this.configModule = configModule;
        return this;
    }

    public MongoMapEntityStoreAssembler withConfigVisibility( Visibility configVisibility )
    {
        this.configVisibility = configVisibility;
        return this;
    }

    /**
     * Add a MongoDB node's hostname and port.
     *
     * Calling this method once disable the default behavior that use the MongoDB defaults: 127.0.0.1 27017
     */
    public MongoMapEntityStoreAssembler addHostnameAndPort( String hostname, Integer port )
        throws UnknownHostException
    {
        this.hostname = null;
        this.port = null;
        if( serverAddresses == null )
        {
            serverAddresses = new ArrayList<ServerAddress>();
        }
        serverAddresses.add( new ServerAddress( hostname, port ) );
        return this;
    }

    public MongoMapEntityStoreAssembler withDatabase( String database )
    {
        this.database = database;
        return this;
    }

    public MongoMapEntityStoreAssembler withCollection( String collection )
    {
        this.collection = collection;
        return this;
    }

    public MongoMapEntityStoreAssembler withWriteConcern( WriteConcern writeConcern )
    {
        this.writeConcern = writeConcern;
        return this;
    }

    @Override
    public void assemble( ModuleAssembly module )
        throws AssemblyException
    {
        if( configModule == null )
        {
            configModule = module;
        }
        onAssemble( module, visibility, configModule, configVisibility );
    }

    private void onAssemble( ModuleAssembly module, Visibility visibility, ModuleAssembly configModule, Visibility configVisibility )
    {
        module.services( MongoMapEntityStoreService.class ).visibleIn( visibility );
        module.services( UuidIdentityGeneratorService.class ).visibleIn( visibility );

        configModule.entities( MongoEntityStoreConfiguration.class ).visibleIn( configVisibility );
        MongoEntityStoreConfiguration mongoConfig = configModule.forMixin( MongoEntityStoreConfiguration.class ).declareDefaults();
        if( hostname != null )
        {
            mongoConfig.hostname().set( hostname );
        }
        if( port != null )
        {
            mongoConfig.port().set( port );
        }
        if( database != null )
        {
            mongoConfig.database().set( database );
        }
        if( collection != null )
        {
            mongoConfig.collection().set( collection );
        }
        if( writeConcern != null )
        {
            mongoConfig.writeConcern().set( writeConcern );
        }
        if( serverAddresses != null )
        {
            mongoConfig.nodes().set( serverAddresses );
        }
    }
}