com.avaje.ebeaninternal.server.deploy.id.IdBinderMultiple Maven / Gradle / Ivy

Go to download
package com.avaje.ebeaninternal.server.deploy.id;

import com.avaje.ebeaninternal.api.SpiExpressionRequest;
import com.avaje.ebeaninternal.server.core.DefaultSqlUpdate;
import com.avaje.ebeaninternal.server.core.InternString;
import com.avaje.ebeaninternal.server.deploy.BeanProperty;
import com.avaje.ebeaninternal.server.deploy.DbReadContext;
import com.avaje.ebeaninternal.server.deploy.DbSqlContext;
import com.avaje.ebeaninternal.server.lib.util.MapFromString;
import com.avaje.ebeaninternal.server.type.DataBind;

import javax.naming.InvalidNameException;
import javax.naming.ldap.LdapName;
import javax.naming.ldap.Rdn;
import javax.persistence.PersistenceException;
import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;
import java.sql.SQLException;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

/**
 * Bind an Id that is made up of multiple separate properties.
 * 
 * The id passed in for binding is expected to be a map with the key being the
 * String name of the property and the value being that properties bind value.
 * 
 */
public final class IdBinderMultiple implements IdBinder {

	private final BeanProperty[] props;

	private final String idProperties;
	
	private final String idInValueSql;
	
	public IdBinderMultiple(BeanProperty[] idProps) {
		this.props = idProps;
		
		StringBuilder sb = new StringBuilder();
		for (int i = 0; i < idProps.length; i++) {
			if (i > 0){
				sb.append(",");
			}
			sb.append(idProps[i].getName());
		}
		idProperties = InternString.intern(sb.toString());
		
		sb = new StringBuilder();
		sb.append("(");
		for (int i = 0; i < props.length; i++) {
			if (i > 0){
				sb.append(",");
			}
			sb.append("?");
		}
		sb.append(")");
		
		idInValueSql = sb.toString();
	}
	
	public void initialise(){
		// do nothing
	}
	
    public String getOrderBy(String pathPrefix, boolean ascending){
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < props.length; i++) {
            if (i > 0) {
                sb.append(" ");
            }
            if (pathPrefix != null){
                sb.append(pathPrefix).append(".");
            }
            sb.append(props[i].getName());
            if (!ascending){
                sb.append(" desc");
            }
        }
        return sb.toString();
    }
    
    public void createLdapNameById(LdapName name, Object id) throws InvalidNameException {
        
        if (id instanceof Map == false){
            throw new RuntimeException("Expecting a Map for concatinated key");
        }
        
        Map mapId = (Map)id;
        for (int i = 0; i < props.length; i++) {
            
            Object v = mapId.get(props[i].getName());
            if (v == null){
                throw new RuntimeException("No value in Map for key "+props[i].getName());
            }
            
            Rdn rdn = new Rdn(props[i].getDbColumn(), v);
            name.add(rdn);            
        }
    }
    
    public void buildSelectExpressionChain(String prefix, List selectChain) {

        for (int i = 0; i < props.length; i++) {            
            props[i].buildSelectExpressionChain(prefix, selectChain);
        }
    }

	public void createLdapNameByBean(LdapName name, Object bean) throws InvalidNameException {

        for (int i = 0; i < props.length; i++) {
            
            Object v = props[i].getValue(bean);            
            Rdn rdn = new Rdn(props[i].getDbColumn(), v);
            name.add(rdn);            
        }
    }

    public int getPropertyCount() {
		return props.length;
	}

	public String getIdProperty() {
		return idProperties;
	}

	
	public BeanProperty findBeanProperty(String dbColumnName) {
		
		for (int i = 0; i < props.length; i++) {
			if (dbColumnName.equalsIgnoreCase(props[i].getDbColumn())){
				return props[i];
			}
		}

		return null;
	}
	
	public boolean isComplexId(){
		return true;
	}
	
	public String getDefaultOrderBy() {
		
		StringBuilder sb = new StringBuilder();
		for (int i = 0; i < props.length; i++) {
			if (i > 0){
				sb.append(",");
			}
			
			sb.append(props[i].getName());
		}
		
		return sb.toString();
	}
	
	public BeanProperty[] getProperties() {
		return props;
	}
	
	public void addIdInBindValue(SpiExpressionRequest request, Object value) {
		for (int i = 0; i < props.length; i++) {
			request.addBindValue(props[i].getValue(value));
		}
	}

    public String getIdInValueExprDelete(int size) {
        return getIdInValueExpr(size);
    }

    public String getIdInValueExpr(int size) {
        StringBuilder sb = new StringBuilder();
        sb.append(" in");
        sb.append(" (");
        sb.append(idInValueSql);
        for (int i = 1; i < size; i++) {            
            sb.append(",").append(idInValueSql);
        }
        sb.append(") ");
        return sb.toString();        
    }
	   
	public String getBindIdInSql(String baseTableAlias) {
		StringBuilder sb = new StringBuilder();
		sb.append("(");
		for (int i = 0; i < props.length; i++) {
			if (i > 0) {
				sb.append(",");
			}
			if (baseTableAlias != null){
    			sb.append(baseTableAlias);
    			sb.append(".");
			}
			sb.append(props[i].getDbColumn());
		}
		sb.append(")");
		return sb.toString();
	}

	public Object[] getIdValues(Object bean){
		Object[] bindvalues = new Object[props.length];
		for (int i = 0; i < props.length; i++) {
			bindvalues[i] = props[i].getValue(bean);
		}
		return bindvalues;
	}
	
	@SuppressWarnings("unchecked")
	public Object[] getBindValues(Object idValue){
		
		Object[] bindvalues = new Object[props.length];
		// concatenated id as a Map
		try {
			Map uidMap = (Map) idValue;

			for (int i = 0; i < props.length; i++) {
				Object value = uidMap.get(props[i].getName());
				bindvalues[i] = value;
			}

			return bindvalues;
			
		} catch (ClassCastException e) {
			String msg = "Expecting concatinated idValue to be a Map";
			throw new PersistenceException(msg, e);
		}
	}
	
    
    public Object readTerm(String idTermValue) {
        
        String[] split = idTermValue.split("|");
        if (split.length != props.length){
            String msg = "Failed to split ["+idTermValue+"] using | for id.";
            throw new PersistenceException(msg);
        }
        Map uidMap = new LinkedHashMap();
        for (int i = 0; i < props.length; i++) {
            Object v = props[i].getScalarType().parse(split[i]);
            uidMap.put(props[i].getName(), v);
        }
        return uidMap;
    }

    @SuppressWarnings("unchecked")
    public String writeTerm(Object idValue) {

        Map uidMap = (Map) idValue;
        
        StringBuilder sb = new StringBuilder();
        
        for (int i = 0; i < props.length; i++) {
            Object v = uidMap.get(props[i].getName());
            String formatValue = props[i].getScalarType().format(v);
            if (i > 0){
                sb.append("|");
            }
            sb.append(formatValue);
        }    
        return sb.toString();
    }
	
    public Object readData(DataInput dataInput) throws IOException {
        
        LinkedHashMap map = new LinkedHashMap();
        
        boolean notNull = true;

        for (int i = 0; i < props.length; i++) {
            Object value = props[i].readData(dataInput);
            map.put(props[i].getName(), value);
            if (value == null) {
                notNull = false;
            }
        }

        if (notNull) {
            return map;
        } else {
            return null;
        }
    }

    @SuppressWarnings("unchecked")
    public void writeData(DataOutput dataOutput, Object idValue) throws IOException {
        
        Map map = (Map)idValue;
        for (int i = 0; i < props.length; i++) {
            Object embFieldValue = map.get(props[i].getName());
            //Object embFieldValue = props[i].getValue(idValue);
            props[i].writeData(dataOutput, embFieldValue);
        }
    }
	
	public void loadIgnore(DbReadContext ctx) {
        for (int i = 0; i < props.length; i++) {
            props[i].loadIgnore(ctx);
        }
    }

    public Object readSet(DbReadContext ctx, Object bean) throws SQLException {
		
		LinkedHashMap map = new LinkedHashMap();
		boolean notNull = false;
		for (int i = 0; i < props.length; i++) {
			Object value = props[i].readSet(ctx, bean, null);
			if (value != null){
				map.put(props[i].getName(), value);
				notNull = true;
			}
		}
		if (notNull){
			return map;
		} else {
			return null;
		}
	}
	
	public Object read(DbReadContext ctx) throws SQLException {
		
		LinkedHashMap map = new LinkedHashMap();
		boolean notNull = false;
		for (int i = 0; i < props.length; i++) {
			Object value = props[i].read(ctx);
			if (value != null){
				map.put(props[i].getName(), value);
				notNull = true;
			}
		}
		if (notNull){
			return map;
		} else {
			return null;
		}
	}	
	
    @SuppressWarnings("unchecked")
	public void bindId(DefaultSqlUpdate sqlUpdate, Object idValue) {
        // concatenated id as a Map
        try {
            Map uidMap = (Map) idValue;

            for (int i = 0; i < props.length; i++) {
                Object value = uidMap.get(props[i].getName());
                sqlUpdate.addParameter(value);
            }
            
        } catch (ClassCastException e) {
            String msg = "Expecting concatinated idValue to be a Map";
            throw new PersistenceException(msg, e);
        }
    }

    @SuppressWarnings("unchecked")
	public void bindId(DataBind bind, Object idValue) throws SQLException {

		// concatenated id as a Map
		try {
			Map uidMap = (Map) idValue;

			for (int i = 0; i < props.length; i++) {
				Object value = uidMap.get(props[i].getName());
				props[i].bind(bind, value);
			}
			
		} catch (ClassCastException e) {
			String msg = "Expecting concatinated idValue to be a Map";
			throw new PersistenceException(msg, e);
		}
	}
	
	public void appendSelect(DbSqlContext ctx, boolean subQuery) {
		for (int i = 0; i < props.length; i++) {
    		props[i].appendSelect(ctx, subQuery);
		}
	}
	        
    public String getAssocIdInExpr(String prefix) {

        StringBuilder sb = new StringBuilder();
        sb.append("(");
        for (int i = 0; i < props.length; i++) {
            if (i > 0) {
                sb.append(",");
            }
            if (prefix != null) {
                sb.append(prefix);
                sb.append(".");
            }
            sb.append(props[i].getName());
        }
        sb.append(")");
        return sb.toString();
    }
	
	public String getAssocOneIdExpr(String prefix, String operator){

		StringBuilder sb = new StringBuilder();
		for (int i = 0; i < props.length; i++) {
			if (i > 0) {
				sb.append(" and ");
			}
			if (prefix != null){
				sb.append(prefix);
				sb.append(".");				
			}
			sb.append(props[i].getName());
			sb.append(operator);
		}
		return sb.toString();		
	}
	
	public String getBindIdSql(String baseTableAlias) {
		
		StringBuilder sb = new StringBuilder();
		for (int i = 0; i < props.length; i++) {
			if (i > 0) {
				sb.append(" and ");
			}
			if (baseTableAlias != null){
    			sb.append(baseTableAlias);
    			sb.append(".");
			}
			sb.append(props[i].getDbColumn());
			sb.append(" = ? ");
		}
		return sb.toString();
	}
	
	public Object convertSetId(Object idValue, Object bean) {

		// allow Map or String for concatenated id
		Map mapVal = null;
		if (idValue instanceof Map) {
			mapVal = (Map) idValue;
		} else {
			mapVal = MapFromString.parse(idValue.toString());
		}

		// Use a new LinkedHashMap to control the order
		LinkedHashMap newMap = new LinkedHashMap();

		for (int i = 0; i < props.length; i++) {
			BeanProperty prop = props[i];

			Object value = mapVal.get(prop.getName());

			// Convert the property type if required
			value = props[i].getScalarType().toBeanType(value);
			newMap.put(prop.getName(), value);
			if (bean != null) {
				// support PropertyChangeSupport
				prop.setValueIntercept(bean, value);
			}
		}

		return newMap;
	}
}