ebean/src/main/java/com/avaje/ebeaninternal/server/query/SqlTreeProperties.java

package com.avaje.ebeaninternal.server.query;

import java.util.ArrayList;
import java.util.LinkedHashSet;
import java.util.List;

import com.avaje.ebeaninternal.server.deploy.BeanProperty;
import com.avaje.ebeaninternal.server.deploy.TableJoin;

/**
 * The select properties for a node in the SqlTree.
 */
public class SqlTreeProperties {

  private static final TableJoin[] EMPTY_TABLE_JOINS = new TableJoin[0];

	/**
	 * True if this node of the tree should have read only entity beans.
	 */
	private boolean readOnly;

	/**
	 * set to false if the id field is not included.
	 */
	private boolean includeId = true;

	private TableJoin[] tableJoins = EMPTY_TABLE_JOINS;

	/**
	 * The bean properties in order.
	 */
	private List<BeanProperty> propsList = new ArrayList<BeanProperty>();

	/**
	 * Maintain a list of property names to detect embedded bean additions.
	 */
	private LinkedHashSet<String> propNames = new LinkedHashSet<String>();

  private boolean allProperties;
	
	public SqlTreeProperties() {
	}
	
  public boolean containsProperty(String propName){
		return propNames.contains(propName);
	}

	public void add(BeanProperty[] props) {
		for (BeanProperty beanProperty : props) {
			propsList.add(beanProperty);
		}
	}

	public void add(BeanProperty prop) {
    propsList.add(prop);
	  propNames.add(prop.getName());
	}
	
	public BeanProperty[] getProps() {
		return propsList.toArray(new BeanProperty[propsList.size()]);
	}

	public boolean isIncludeId() {
		return includeId;
	}

	public void setIncludeId(boolean includeId) {
		this.includeId = includeId;
	}

	public boolean isPartialObject() {
		return !allProperties;
	}

	public boolean isReadOnly() {
		return readOnly;
	}

	public void setReadOnly(boolean readOnly) {
		this.readOnly = readOnly;
	}

	public TableJoin[] getTableJoins() {
		return tableJoins;
	}

	public void setTableJoins(TableJoin[] tableJoins) {
		this.tableJoins = tableJoins;
	}

  public void setAllProperties(boolean allProperties) {
    this.allProperties = allProperties;
  }

  public boolean isAllProperties() {
    return allProperties;
  }
  
}