org.codehaus.xfire.aegis.type.CustomTypeMapping Maven / Gradle / Ivy

package org.codehaus.xfire.aegis.type;

import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

import javax.xml.namespace.QName;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 * Contains type mappings for java/qname pairs.
 * 
 * @author Dan Diephouse
 * @since Feb 21, 2004
 */
public class CustomTypeMapping
    implements TypeMapping
{
    private static final Log LOG = LogFactory.getLog(CustomTypeMapping.class);
    
	private Map class2Type;

    private Map xml2Type;

    private Map class2xml;
    
    private TypeMapping defaultTM;
    
    private String encodingStyleURI;
    
    private TypeCreator typeCreator;
    
    public CustomTypeMapping( TypeMapping defaultTM )
    {
        this();

        this.defaultTM = defaultTM;
    }
    
    public CustomTypeMapping()
    {
        class2Type = Collections.synchronizedMap(new HashMap());
        class2xml = Collections.synchronizedMap(new HashMap());
        xml2Type = Collections.synchronizedMap(new HashMap());
    }

	public boolean isRegistered(Class javaType)
	{
        boolean registered = class2Type.containsKey(javaType);
        
        if ( !registered && defaultTM != null )
            registered = defaultTM.isRegistered(javaType);
        
        return registered;
	}

    public boolean isRegistered(QName xmlType)
    {
        boolean registered = xml2Type.containsKey(xmlType);
        
        if ( !registered && defaultTM != null )
            registered = defaultTM.isRegistered(xmlType);
        
        return registered;
    }

	public void register(Class javaType, QName xmlType, Type type)
    {
        type.setSchemaType(xmlType);
        type.setTypeClass(javaType);

        register(type);
    }

    public void register(Type type)
    {
        type.setTypeMapping(this);
        /*
         * -- [email protected]; changing this to only register the type for
         * actions that it supports, and it could be none.
         */
        if (type.getTypeClass() != null) {
            class2xml.put( type.getTypeClass(), type.getSchemaType() ); 
            class2Type.put( type.getTypeClass(), type );
        }
        if (type.getSchemaType() != null) {
            xml2Type.put( type.getSchemaType(), type );
        }
        if (type.getTypeClass() == null && type.getSchemaType() == null) {
            LOG.warn("The type " + type.getClass().getName()
                     + " supports neither serialization (non-null TypeClass) nor deserialization (non-null SchemaType).");
        }
    }

	public void removeType(Type type)
	{
        if (!xml2Type.containsKey(type.getSchemaType()))
        {
           defaultTM.removeType(type);
        }
        else
        {
            xml2Type.remove(type.getSchemaType());
            class2Type.remove(type.getTypeClass());
            class2xml.remove(type.getTypeClass());
        }
	}

    /**
     * @see org.codehaus.xfire.aegis.type.TypeMapping#getType(java.lang.Class)
     */
    public Type getType(Class javaType)
    {
        Type type = (Type) class2Type.get( javaType );

        if ( type == null && defaultTM != null )
        {
            type = defaultTM.getType( javaType );
        }
        
        return type;
    }

    /**
     * @see org.codehaus.xfire.aegis.type.TypeMapping#getType(javax.xml.namespace.QName)
     */
    public Type getType(QName xmlType)
    {
        Type type = (Type) xml2Type.get( xmlType );

        if ( type == null && defaultTM != null )
            type = defaultTM.getType( xmlType );
        
        return type;
    }

    /**
     * @see org.codehaus.xfire.aegis.type.TypeMapping#getTypeQName(java.lang.Class)
     */
    public QName getTypeQName(Class clazz)
    {
        QName qname = (QName) class2xml.get( clazz );

        if ( qname == null && defaultTM != null )
            qname = defaultTM.getTypeQName( clazz );
        
        return qname;
    }

    public String getEncodingStyleURI()
    {
        return encodingStyleURI;
    }
    
    public void setEncodingStyleURI( String encodingStyleURI )
    {
        this.encodingStyleURI = encodingStyleURI;
    }

    public TypeCreator getTypeCreator()
    {
        return typeCreator;
    }

    public void setTypeCreator(TypeCreator typeCreator)
    {
        this.typeCreator = typeCreator;
        
        typeCreator.setTypeMapping(this);
    }
    
    public TypeMapping getParent()
    {
        return defaultTM;
    }
}