org.eclipse.osee.framework.skynet.core/src/org/eclipse/osee/framework/skynet/core/revision/ArtifactNameDescriptorCache.java

/*******************************************************************************
 * Copyright (c) 2004, 2007 Boeing.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Boeing - initial API and implementation
 *******************************************************************************/
package org.eclipse.osee.framework.skynet.core.revision;

import java.util.HashMap;
import java.util.Map;
import org.eclipse.osee.framework.jdk.core.type.Pair;
import org.eclipse.osee.framework.skynet.core.artifact.ArtifactType;
import org.eclipse.osee.framework.skynet.core.artifact.Branch;

/**
 * Caches names and descriptors to an artifact id. This cache assumes that the user is already taking care of how this
 * data is appropriate for versions and branches.
 * 
 * @author Robert A. Fisher
 */
public class ArtifactNameDescriptorCache extends ArtifactNameDescriptorResolver {
   private Map<Integer, String> nameMap;
   private Map<Integer, ArtifactType> descriptorMap;
   private boolean backupAvailable;

   public ArtifactNameDescriptorCache() {
      this(null);
   }

   public ArtifactNameDescriptorCache(Branch branch) {
      super(branch);

      this.nameMap = new HashMap<Integer, String>();
      this.descriptorMap = new HashMap<Integer, ArtifactType>();
      //      this.backupAvailable = branch != null;
      this.backupAvailable = false;
   }

   public void cache(Integer artId, String name, ArtifactType descriptor) {
      if (artId == null) throw new IllegalArgumentException("artId must not be null");

      nameMap.put(artId, name);
      descriptorMap.put(artId, descriptor);
   }

   @Override
   public Pair<String, ArtifactType> get(Integer artId) {
      if (artId == null) throw new IllegalArgumentException("artId must not be null");

      if (nameMap.containsKey(artId) || !backupAvailable) {
         return new Pair<String, ArtifactType>(nameMap.get(artId), descriptorMap.get(artId));
      } else {
         return super.get(artId);
      }
   }
}