All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • com.avaje.ebeaninternal.server.deploy.BeanSetHelp Maven / Gradle / Ivy

    There is a newer version: 2.8.1
    Show newest version
    package com.avaje.ebeaninternal.server.deploy;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.LinkedHashSet;
import java.util.Set;

import com.avaje.ebean.EbeanServer;
import com.avaje.ebean.InvalidValue;
import com.avaje.ebean.Query;
import com.avaje.ebean.Transaction;
import com.avaje.ebean.bean.BeanCollection;
import com.avaje.ebean.bean.BeanCollectionAdd;
import com.avaje.ebean.bean.BeanCollectionLoader;
import com.avaje.ebean.common.BeanSet;
import com.avaje.ebeaninternal.server.text.json.WriteJsonContext;

/**
 * Helper specifically for dealing with Sets.
 */
public final class BeanSetHelp implements BeanCollectionHelp {
	
	private final BeanPropertyAssocMany many;
	private final BeanDescriptor targetDescriptor;
	private BeanCollectionLoader loader;
	
	/**
	 * When attached to a specific many property.
	 */
	public BeanSetHelp(BeanPropertyAssocMany many){
		this.many = many;
		this.targetDescriptor = many.getTargetDescriptor();
	}
	
	/**
	 * For a query that returns a set.
	 */
	public BeanSetHelp(){
		this.many = null;
		this.targetDescriptor = null;
	}
	
	public void setLoader(BeanCollectionLoader loader){
		this.loader = loader;
	}
	
    public Iterator getIterator(Object collection) {
        return ((Set) collection).iterator();
    }
	
	public BeanCollectionAdd getBeanCollectionAdd(Object bc,String mapKey) {
	    if (bc instanceof BeanSet){
    		BeanSet beanSet = (BeanSet)bc;
    		if (beanSet.getActualSet() == null){
    			beanSet.setActualSet(new LinkedHashSet());
    		}
    		return beanSet;
	    } else if (bc instanceof Set) {
	        return new VanillaAdd((Set)bc);
	        
	    } else {
	        throw new RuntimeException("Unhandled type "+bc);
	    }
	}
	

    @SuppressWarnings("unchecked")
    static class VanillaAdd implements BeanCollectionAdd {

        @SuppressWarnings("rawtypes")
		private final Set set;

        private VanillaAdd(Set set) {
            this.set = set;
        }

        public void addBean(Object bean) {
            set.add(bean);
        }
    }
	
	public void add(BeanCollection collection, Object bean) {
		collection.internalAdd(bean);
	}

    @SuppressWarnings("unchecked")
    public Object copyCollection(Object source, CopyContext ctx, int maxDepth, Object parentBean) {
        if (source instanceof Set == false){
            return null;
        }
        Set s = ctx.isVanillaMode() ? new LinkedHashSet() : new BeanSet();

        if (source instanceof BeanSet == false){
            s.addAll((Set)source);
            return s;
            
        } 
        
        BeanSet bc = (BeanSet)source;
        if (!bc.isPopulated()) {
            if (ctx.isVanillaMode() || parentBean == null){
                return null;
            } else {
                return createReference(parentBean, many.getName());
            }
        }
        Set actual = bc.getActualSet();
        for (Object sourceDetail : actual) {
            Object destDetail = targetDescriptor.createCopy(sourceDetail, ctx, maxDepth-1);
            s.add((T)destDetail);
        }
        return s;
    }
    
    public Object createEmpty(boolean vanilla) {
	    return vanilla ? new LinkedHashSet() : new BeanSet();
	}

	public BeanCollection createReference(Object parentBean, String propertyName) {
		
		return new BeanSet(loader, parentBean, propertyName);
	}
	
	public ArrayList validate(Object manyValue) {
		
		ArrayList errs = null;
		
		Set set = (Set)manyValue;
		Iterator i = set.iterator();
		while (i.hasNext()) {
			Object detailBean = i.next();
			InvalidValue invalid = targetDescriptor.validate(true, detailBean);
			if (invalid != null){
				if (errs == null){
					errs = new ArrayList();
				}
				errs.add(invalid);
			}
		}
		return errs;
	}

	public void refresh(EbeanServer server, Query query, Transaction t, Object parentBean) {
		
		BeanSet newBeanSet = (BeanSet)server.findSet(query, t);
		refresh(newBeanSet, parentBean);
	}
	
	public void refresh(BeanCollection bc, Object parentBean) {
		
		BeanSet newBeanSet = (BeanSet)bc;
		
		Set current = (Set)many.getValueUnderlying(parentBean);
		
		newBeanSet.setModifyListening(many.getModifyListenMode());
		if (current == null){
			// the currentList is null?  Not really expecting this...
			many.setValue(parentBean,newBeanSet);
			
		} else if (current instanceof BeanSet) {
			// normally this case, replace just the underlying list
			BeanSet currentBeanSet = (BeanSet)current;
			currentBeanSet.setActualSet(newBeanSet.getActualSet());
			currentBeanSet.setModifyListening(many.getModifyListenMode());

		} else {
			// replace the entire set 
			many.setValue(parentBean, newBeanSet);
		}
	}
	
    public void jsonWrite(WriteJsonContext ctx, String name, Object collection, boolean explicitInclude) {
        
        Set set;
        if (collection instanceof BeanCollection){
            BeanSet bc = (BeanSet)collection;
            if (!bc.isPopulated()){
                if (explicitInclude){
                    // invoke lazy loading as collection 
                    // is explicitly included in the output
                    bc.size();
                } else {
                    return;
                }
            } 
            set = bc.getActualSet();
        } else {
            set = (Set)collection;
        }
        
        int count = 0;
        ctx.beginAssocMany(name);
        Iterator it = set.iterator();
        while (it.hasNext()) {
            Object detailBean = it.next();
            if (count++ > 0){
                ctx.appendComma();
            }
            targetDescriptor.jsonWrite(ctx, detailBean);
        }
        ctx.endAssocMany();        
    }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy