com.avaje.ebeaninternal.server.ddl.VisitorUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ebean Show documentation
Ebean
There is a newer version: 2.8.1
package com.avaje.ebeaninternal.server.ddl;

import java.util.List;

import com.avaje.ebeaninternal.api.SpiEbeanServer;
import com.avaje.ebeaninternal.server.deploy.BeanDescriptor;
import com.avaje.ebeaninternal.server.deploy.BeanProperty;
import com.avaje.ebeaninternal.server.deploy.BeanPropertyAssocMany;
import com.avaje.ebeaninternal.server.deploy.BeanPropertyAssocOne;
import com.avaje.ebeaninternal.server.deploy.BeanPropertyCompound;

/**
 * Utility object to use BeanVisitor.
 */
public class VisitorUtil {

	/**
	 * Visit all the descriptors for a given server.
	 */
	public static void visit(SpiEbeanServer server, BeanVisitor visitor){

		visit(server.getBeanDescriptors(), visitor);
	}


	/**
	 * Visit all the descriptors in the list.
	 */
	public static void visit(List> descriptors, BeanVisitor visitor){

		visitor.visitBegin();

		for (BeanDescriptor desc : descriptors) {

			if (desc.getBaseTable() != null){
				visitBean(desc, visitor);
			} 
		}

		visitor.visitEnd();
	}

	/**
	 * Visit the bean using a visitor.
	 */
	public static void visitBean(BeanDescriptor desc, BeanVisitor visitor) {

		if (visitor.visitBean(desc)) {

		    BeanProperty[] propertiesId = desc.propertiesId();
		    for (int i = 0; i < propertiesId.length; i++) {
		        visit(visitor, propertiesId[i]);
            }
		    BeanPropertyAssocOne unidirectional = desc.getUnidirectional();
		    if (unidirectional != null){
                visit(visitor, unidirectional);
		    }
		    BeanProperty[] propertiesNonTransient = desc.propertiesNonTransient();
		    for (int i = 0; i < propertiesNonTransient.length; i++) {
		        BeanProperty p = propertiesNonTransient[i];
		        if (!p.isFormula() && !p.isSecondaryTable()){
    		        visit(visitor, p);
    		    }
            }
	
			visitor.visitBeanEnd(desc);
		}
	}
	
	private static void visit(BeanVisitor visitor, BeanProperty p) {
	    PropertyVisitor pv = visitor.visitProperty(p);
        if (pv != null){
            visit(p, pv);
        }
	}

	/**
	 * Visit all the properties.
	 */
	public static void visit(BeanProperty[] p, PropertyVisitor pv){

		for (int i = 0; i < p.length; i++) {
			visit(p[i], pv);
		}
	}

	/**
	 * Visit the property.
	 */
	public static void visit(BeanProperty p, PropertyVisitor pv) {

		if (p instanceof BeanPropertyAssocMany){
			pv.visitMany((BeanPropertyAssocMany)p);

		} else if (p instanceof BeanPropertyAssocOne){
			BeanPropertyAssocOne assocOne = (BeanPropertyAssocOne)p;
			if (assocOne.isEmbedded()){
				pv.visitEmbedded(assocOne);
				BeanProperty[] embProps = assocOne.getProperties();
				for (int i = 0; i < embProps.length; i++) {
					pv.visitEmbeddedScalar(embProps[i], assocOne);
				}


			} else if (assocOne.isOneToOneExported()){
				pv.visitOneExported(assocOne);

			} else {
				pv.visitOneImported(assocOne);

			}
        } else if (p instanceof BeanPropertyCompound){
            BeanPropertyCompound compound = (BeanPropertyCompound)p;
            pv.visitCompound(compound);
            
            BeanProperty[] properties = compound.getScalarProperties();
            for (int i = 0; i < properties.length; i++) {
                pv.visitCompoundScalar(compound, properties[i]);
            }
            
		} else {
			pv.visitScalar(p);
		}
	}
}