Metacat

package edu.ucsb.nceas.metacat.dataone.hazelcast;

import java.sql.SQLException;

import java.util.Collection;

import java.util.HashMap;

import java.util.Map;

import java.util.Set;

import org.apache.log4j.Logger;

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

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

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

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

import com.hazelcast.core.Hazelcast;

import com.hazelcast.core.MapLoader;

import com.hazelcast.core.MapStore;

import edu.ucsb.nceas.metacat.IdentifierManager;

import edu.ucsb.nceas.metacat.McdbDocNotFoundException;

import edu.ucsb.nceas.metacat.properties.PropertyService;

import edu.ucsb.nceas.utilities.PropertyNotFoundException;

/**

 * Storage implementation for Hazelcast System Metadata

 * @author leinfelder

 *

 */

public class SystemMetadataMap 

    implements MapStore<Identifier, SystemMetadata>, MapLoader<Identifier, SystemMetadata> {

  private Logger logMetacat = null;

	@Override

	public void delete(Identifier arg0) {

		// we do not delete system metadata

	}

	@Override

	public void deleteAll(Collection<Identifier> arg0) {

		// we do not delete system metadata	

	}

	@Override

	public void store(Identifier pid, SystemMetadata sm) {

		try {

			//System.out.println("Storing System Metadata to store: " + pid.getValue());

			if (!IdentifierManager.getInstance().systemMetadataExists(pid.getValue())) {

				IdentifierManager.getInstance().createSystemMetadata(sm);

			} else {

				IdentifierManager.getInstance().updateSystemMetadata(sm);

			}

		} catch (McdbDocNotFoundException e) {

			throw new RuntimeException(e.getMessage(), e);

		} catch (SQLException e) {

	      throw new RuntimeException(e.getMessage(), e);

    }

	}

	@Override

	public void storeAll(Map<Identifier, SystemMetadata> map) {

		for (Identifier key: map.keySet()) {

			store(key, map.get(key));

		}

	}

	@Override

	public SystemMetadata load(Identifier pid) {

		SystemMetadata sm = null;

    logMetacat = Logger.getLogger(SystemMetadataMap.class);

    try {

			logMetacat.debug("loading from store: " + pid.getValue());

			sm = IdentifierManager.getInstance().getSystemMetadata(pid.getValue());

		} catch (McdbDocNotFoundException e) {

			//throw new RuntimeException(e.getMessage(), e);

			// not found => null

			return null;

		}

		catch (Exception e) {

			throw new RuntimeException(e.getMessage(), e);

		}

		return sm;

	}

	@Override

	public Map<Identifier, SystemMetadata> loadAll(Collection<Identifier> keys) {

		Map<Identifier, SystemMetadata> map = new HashMap<Identifier, SystemMetadata>();

		for (Identifier key: keys) {

			SystemMetadata value = load(key);

			map.put(key, value);

		}

		return map;

	}

	@Override

	public Set<Identifier> loadAllKeys() {

	    //Set<Identifier> pids = new HashSet<Identifier>();

	  String identifiersSet;

    try {

        identifiersSet = 

            PropertyService.getProperty("dataone.hazelcast.storageCluster.identifiersSet");

    } catch (PropertyNotFoundException e1) {

        identifiersSet = "hzIdentifiers";

    }

    Set<Identifier> pids = Hazelcast.getSet(identifiersSet);

		try {

			ObjectList ol = IdentifierManager.getInstance().querySystemMetadata(

					null, //startTime, 

					null, //endTime, 

					null, //objectFormatId, 

					false, //replicaStatus, 

					0, //start, 

					-1 //count

					);

			for (ObjectInfo o: ol.getObjectInfoList()) {

				Identifier pid = o.getIdentifier();

				if ( !pids.contains(pid) ) {

		        pids.add(pid);

				}				

			}

		} catch (Exception e) {

			throw new RuntimeException(e.getMessage(), e);

		}

		return pids;

	}

}