BerkeleyDbFormatModuleAccessor.java

/**
 * JHOVE2 - Next-generation architecture for format-aware characterization
 *
 * Copyright (c) 2009 by The Regents of the University of California,
 * Ithaka Harbors, Inc., and The Board of Trustees of the Leland Stanford
 * Junior University.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * o Redistributions of source code must retain the above copyright notice,
 *   this list of conditions and the following disclaimer.
 *
 * o Redistributions in binary form must reproduce the above copyright notice,
 *   this list of conditions and the following disclaimer in the documentation
 *   and/or other materials provided with the distribution.
 *
 * o Neither the name of the University of California/California Digital
 *   Library, Ithaka Harbors/Portico, or Stanford University, nor the names of
 *   its contributors may be used to endorse or promote products derived from
 *   this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */
package org.jhove2.persist.berkeleydpl;


import java.util.ArrayList;
import java.util.List;

import org.jhove2.core.JHOVE2Exception;
import org.jhove2.module.AbstractModule;
import org.jhove2.module.format.AbstractFormatProfile;
import org.jhove2.module.format.FormatModule;
import org.jhove2.module.format.FormatProfile;
import org.jhove2.persist.FormatModuleAccessor;

import com.sleepycat.je.DatabaseException;
import com.sleepycat.persist.EntityCursor;
import com.sleepycat.persist.EntityIndex;
import com.sleepycat.persist.model.Persistent;

/**
 * @author smorrissey
 *
 */
@Persistent
public class BerkeleyDbFormatModuleAccessor extends
		BerkeleyDbBaseModuleAccessor implements FormatModuleAccessor {

	public BerkeleyDbFormatModuleAccessor(){
		super();
	}
	/* (non-Javadoc)
	 * @see org.jhove2.persist.FormatModuleAccessor#getProfiles(org.jhove2.module.format.BaseFormatModule)
	 */
	@Override
	public List<FormatProfile> getProfiles(FormatModule module)
			throws JHOVE2Exception {
		List<FormatProfile> profiles = new ArrayList<FormatProfile>();
		if (module != null){
			EntityIndex<Long, AbstractFormatProfile> subIndex = null;
			EntityCursor<AbstractFormatProfile> cursor = null;
			try{
				if (module.getModuleId()== null){ // new module, never persisted
					this.persistModule(module);
				}
				subIndex = 
					this.getBerkeleyDbPersistenceManager().
						getFormatProfileByParentFormatModule().subIndex(module.getModuleId());
				cursor = subIndex.entities();
				for (AbstractModule formatProfile:cursor){
					profiles.add((FormatProfile)formatProfile);
				}
			}
			catch (DatabaseException e){
				throw new JHOVE2Exception ("Unable to get FormatProfiles " , e);
			}
			finally{
				if (cursor != null){
					try{
						cursor.close();
					}
					catch (DatabaseException e){
						throw new JHOVE2Exception ("Unable to close cursor", e);
					}
				}
			}
		}
		return profiles;
	}

	/* (non-Javadoc)
	 * @see org.jhove2.persist.FormatModuleAccessor#setProfiles(org.jhove2.module.format.BaseFormatModule, java.util.List)
	 */
	@Override
	public List<FormatProfile> setProfiles(FormatModule module,
			List<FormatProfile> profiles) throws JHOVE2Exception {
		List<FormatProfile> childProfiles = new ArrayList<FormatProfile>();
		if (module != null){
			try{
				if (module.getModuleId()== null){ // new module, never persisted
					this.persistModule(module);
				}
				// un-link old profiles
				List<FormatProfile> oldProfiles = this.getProfiles(module);
				if (oldProfiles != null){
					for (FormatProfile profile:oldProfiles){
						this.deleteProfile(module, profile);
					}
				}
				//now link new ones
				if (profiles != null){
					for (FormatProfile profile:profiles){
						this.addProfile(module, profile);
						childProfiles.add(profile);
					}
				}
			}
			catch (DatabaseException e){
				throw new JHOVE2Exception ("Unable to set FormatProfiles " , e);
			}
		}
		return childProfiles;
	}
	@Override
	public FormatProfile addProfile(FormatModule module, FormatProfile profile)
			throws JHOVE2Exception {		
		if (module != null && profile != null){
			if (module.getModuleId()==null){
				this.persistModule(module);
			}
			profile.setFormatModuleId(module.getModuleId());
		}		
		return profile;
	}
	@Override
	public FormatProfile deleteProfile(FormatModule module,
			FormatProfile profile) throws JHOVE2Exception {
		if (module != null && profile != null){
			if (module.getModuleId()==null){
				this.persistModule(module);
			}
			Long parentKey = profile.getFormatModuleId();
			if (parentKey !=null && parentKey.equals(module.getModuleId())){
				profile.setFormatModuleId(null);
			}
		}		
		return profile;
	}



}