/*

  * Copyright 2002-2004 The Apache Software Foundation

  *

  * Licensed under the Apache License, Version 2.0 (the "License");

  * you may not use this file except in compliance with the License.

  * You may obtain a copy of the License at

  *

  *     http://www.apache.org/licenses/LICENSE-2.0

  *

  * Unless required by applicable law or agreed to in writing, software

  * distributed under the License is distributed on an "AS IS" BASIS,

  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  * See the License for the specific language governing permissions and

  * limitations under the License.

  */

 package org.apache.commons.clazz;

 import java.util.HashMap;

 import org.apache.commons.clazz.common.GroupClazzLoader;

 /**

  * An implementation of <code>ClazzLoader</code> that functions as the top-

  * level clazz loader for a model and caches all clazzes by name.

  *

  * @author <a href="mailto:dmitri@apache.org">Dmitri Plotnikov</a>

  * @version $Id: ModelClazzLoader.java 155436 2005-02-26 13:17:48Z dirkv $

  */

 public class ModelClazzLoader extends GroupClazzLoader {

     private String model;

     private HashMap clazzMap = new HashMap();

     public ModelClazzLoader(String model) {

         super(null);

         this.model = model;

     }

     public String getModel() {

         return model;

     }

     /**

      * @see ClazzLoader#getClazzForName(String)

      */

     public Clazz getClazzForName(String name) {

         Clazz clazz = (Clazz) clazzMap.get(name);

         if (clazz == null) {

             clazz = super.getClazzForName(name);

             if (clazz != null) {

                 clazzMap.put(name, clazz);

             }

         }

         return clazz;

     }

     /**

      * @see ClazzLoader#defineClazz(String, Class, Class)

      */

     public Clazz defineClazz(

             String name,

             Class clazzType,

             Class instanceClass) 

     {

         Clazz clazz = super.defineClazz(name, clazzType, instanceClass);

         if (clazz != null) {

             clazzMap.put(name, clazz);

         }

         return clazz;

     }

 }