Metacat

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

import java.util.Collection;

import java.util.HashMap;

import java.util.HashSet;

import java.util.Map;

import java.util.Set;

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.MapLoader;

import com.hazelcast.core.MapStore;

import edu.ucsb.nceas.metacat.IdentifierManager;

import edu.ucsb.nceas.metacat.McdbDocNotFoundException;

/**

 * Storage implementation for Hazelcast System Metadata

 * @author leinfelder

 *

 */

public class SystemMetadataMap implements MapStore<Identifier, SystemMetadata>, MapLoader<Identifier, SystemMetadata> {

	@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);

		}

	}

	@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;

		try {

			System.out.println("loading from store: " + pid.getValue());

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

		} catch (McdbDocNotFoundException 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>();

		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();

				pids.add(pid);

			}

		} catch (Exception e) {

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

		}

		return pids;

	}

}