Metacat

package edu.ucsb.nceas.metacat.index;

import static org.junit.Assert.assertTrue;

import org.dataone.service.types.v1.Identifier;

import org.dataone.service.types.v1.SystemMetadata;

import org.junit.Test;

import com.hazelcast.core.IMap;

public class DistributedMapsFactoryIT {

    /**

     * Test to get the system metadata map

     */

    @Test

    public void testGetSystemMetadataMap() throws Exception {

        IMap<Identifier, SystemMetadata> map = DistributedMapsFactory.getSystemMetadataMap();

        System.out.println("the size of the map is "+map.size());

        assertTrue("The size of the system metadata map should equal or be greater than 0", map.size() >= 0);

    }

    /**

     * Test to get the system metadata map

     */

    @Test

    public void testGetObjectPathMap() throws Exception {

        IMap<Identifier, String> map = DistributedMapsFactory.getObjectPathMap();

        System.out.println("the size of the map is "+map.size());

        assertTrue("The size of the object path map should equal or be greater than 0", map.size() >= 0);

    }

}

/**

 *  '$RCSfile$'

 *    Purpose: A class that gets Accession Number, check for uniqueness

 *             and register it into db

 *  Copyright: 2000 Regents of the University of California and the

 *             National Center for Ecological Analysis and Synthesis

 *    Authors: Jivka Bojilova, Matt Jones

 *

 *   '$Author: leinfelder $'

 *     '$Date: 2011-11-02 20:40:12 -0700 (Wed, 02 Nov 2011) $'

 * '$Revision: 6595 $'

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 2 of the License, or

 * (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 */

package edu.ucsb.nceas.metacat.index;

import java.io.FileNotFoundException;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.dataone.configuration.Settings;

import org.dataone.service.exceptions.ServiceFailure;

import org.dataone.service.types.v1.Identifier;

import org.dataone.service.types.v1.SystemMetadata;

import com.hazelcast.client.ClientConfig;

import com.hazelcast.client.HazelcastClient;

import com.hazelcast.config.Config;

import com.hazelcast.config.FileSystemXmlConfig;

import com.hazelcast.core.IMap;

/**

 * A factory to get distributed maps from the haszel cast client.

 * @author tao

 *

 */

public class DistributedMapsFactory {

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

    private static HazelcastClient hzClient = null;

    private static String hzSystemMetadata = null;

    private static String hzObjectPath = null;

    /*

     * Start the hazel cast client

     */

    private static void startHazelCastClient() throws FileNotFoundException, ServiceFailure{

     // get config values

        hzSystemMetadata = Settings.getConfiguration().getString(

                "dataone.hazelcast.storageCluster.systemMetadataMap");

        hzObjectPath = Settings.getConfiguration().getString(

                "dataone.hazelcast.storageCluster.objectPathMap");

        String configFileName = Settings.getConfiguration().getString(

                "dataone.hazelcast.configFilePath");;

        Config hzConfig = null;

        try {

            hzConfig = new FileSystemXmlConfig(configFileName);

        } catch (FileNotFoundException e) {

            log.error("could not load hazelcast configuration file from: " + configFileName, e);

            throw e;

        }

        String hzGroupName = hzConfig.getGroupConfig().getName();

        String hzGroupPassword = hzConfig.getGroupConfig().getPassword();

        String hzAddress = hzConfig.getNetworkConfig().getInterfaces().getInterfaces().iterator().next() + ":" + hzConfig.getNetworkConfig().getPort();

        log.info("starting index entry listener...");

        log.info("System Metadata value: " + hzSystemMetadata);

        log.info("Object path value: " + hzObjectPath);

        log.info("Group Name: " + hzGroupName);

        log.info("Group Password: " + "*****"); // don't show value

        log.info("HZ Address: " + hzAddress);

        // connect to the HZ cluster

        ClientConfig cc = new ClientConfig();

        cc.getGroupConfig().setName(hzGroupName);

        cc.getGroupConfig().setPassword(hzGroupPassword);

        cc.addAddress(hzAddress);

        try {

            hzClient = HazelcastClient.newHazelcastClient(cc);

        } catch (Exception e) {

            log.error("Unable to create hazelcast client: ", e);

            throw new ServiceFailure("0000","DistributedMapsFactory.startHazelCastClient - can't start the hazel cast client since "+e.getMessage());

        }

    }

    /**

     * Get the system metadata map

     * @return

     * @throws FileNotFoundException

     * @throws ServiceFailure

     */

    public static IMap<Identifier, SystemMetadata> getSystemMetadataMap() throws FileNotFoundException, ServiceFailure {

        if(hzClient == null) {

            startHazelCastClient();

        }

        return hzClient.getMap(hzSystemMetadata);

    }

    /**

     * Get the distributed object path map from the haszel cast client.

     * @return

     * @throws FileNotFoundException

     * @throws ServiceFailure

     */

    public static IMap<Identifier, String> getObjectPathMap() throws FileNotFoundException, ServiceFailure {

        if(hzClient == null) {

            startHazelCastClient();

        }

        return hzClient.getMap(hzObjectPath);

    }

}