xmlbeansschemasupport.java

Xfire文件 用于开发web service 的一个开源工具 很好用的

package org.codehaus.xfire.gen.xmlbeans;

import javax.xml.namespace.QName;

import org.apache.xmlbeans.SchemaType;
import org.apache.xmlbeans.SchemaTypeLoader;
import org.apache.xmlbeans.XmlBeans;
import org.apache.xmlbeans.XmlObject;
import org.codehaus.xfire.aegis.AegisBindingProvider;
import org.codehaus.xfire.aegis.type.DefaultTypeMappingRegistry;
import org.codehaus.xfire.aegis.type.Type;
import org.codehaus.xfire.aegis.type.TypeMapping;
import org.codehaus.xfire.gen.GenerationContext;
import org.codehaus.xfire.gen.GenerationException;
import org.codehaus.xfire.gen.SchemaSupport;
import org.codehaus.xfire.service.MessagePartInfo;
import org.codehaus.xfire.service.binding.BindingProvider;
import org.codehaus.xfire.xmlbeans.XmlBeansTypeRegistry;
import org.jdom.Element;

import com.sun.codemodel.JClass;
import com.sun.codemodel.JExpr;
import com.sun.codemodel.JExpression;
import com.sun.codemodel.JInvocation;
import com.sun.codemodel.JType;

public class XmlBeansSchemaSupport
    implements SchemaSupport
{
    SchemaTypeLoader loader = 
        XmlBeans.typeLoaderForClassLoader(XmlObject.class.getClassLoader());

    private DefaultTypeMappingRegistry reg = new XmlBeansTypeRegistry();
    private AegisBindingProvider provider = new AegisBindingProvider(reg);
    private TypeMapping tm;
    
    public XmlBeansSchemaSupport()
    {
        reg.createDefaultMappings();
        tm = reg.getDefaultTypeMapping();
    }

    public void initialize(GenerationContext context)
    {
     	
    }
    
	public JType getType(GenerationContext context, MessagePartInfo part)
			throws GenerationException {

		QName concreteType = part.getName();
		QName schemaType = part.getSchemaType().getSchemaType();
		
		return getType(context, concreteType, schemaType);
    }

    public JType getType(GenerationContext context, QName concreteType, QName schemaType) 
        throws GenerationException
    {
   
        SchemaType xst = loader.findDocumentType(concreteType);
        
        if (xst == null)
        {    
            xst = loader.findDocumentType(schemaType);
        }
        
        if (xst != null)
        {    
            return context.getCodeModel().ref(xst.getJavaClass());
        }
        else 
        {
            Class cls;
            Type type = tm.getType(schemaType);
            cls = (type != null) ? cls = type.getTypeClass() : XmlObject.class;
            
            return context.getCodeModel().ref(cls);
        }
    }
    
    public BindingProvider getBindingProvider()
    {
        return provider;
    }

    public JExpression getBindingProviderExpr(GenerationContext context)
    {
        JType aegis = context.getCodeModel()._ref(AegisBindingProvider.class);
        JInvocation expr = JExpr._new(aegis);
        
        JType xbean = context.getCodeModel()._ref(XmlBeansTypeRegistry.class);
        expr.arg(JExpr._new(xbean));
        
        return expr;//JExpr._new()
    }
    
    public String getServiceFactory()
    {
        return "#xmlbeansServiceFactory";
    }
    
    public Element getServiceFactoryBean()
    {
        return null;
    }
}