com.blazebit.persistence.impl.BaseInsertCriteriaBuilderImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of blaze-persistence-core-impl Show documentation
There is a newer version: 1.2.0-Alpha1
Show newest version
/*
 * Copyright 2015 Blazebit.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.blazebit.persistence.impl;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

import javax.persistence.Query;

import com.blazebit.persistence.BaseInsertCriteriaBuilder;
import com.blazebit.persistence.SelectBuilder;
import com.blazebit.persistence.spi.DbmsStatementType;

/**
 *
 * @param  The query result type
 * @author Christian Beikov
 * @since 1.1.0
 */
public class BaseInsertCriteriaBuilderImpl, Y> extends AbstractModificationCriteriaBuilder implements BaseInsertCriteriaBuilder, SelectBuilder {

	private final Map bindingMap = new TreeMap();

	public BaseInsertCriteriaBuilderImpl(MainQuery mainQuery, boolean isMainQuery, Class clazz, String cteName, Class cteClass, Y result, CTEBuilderListener listener) {
		super(mainQuery, isMainQuery, DbmsStatementType.INSERT, clazz, null, cteName, cteClass, result, listener);
		
        if (!jpaProvider.supportsInsertStatement()) {
            throw new IllegalStateException("JPA provider does not support insert statements!");
        }
	}

    @Override
    @SuppressWarnings("unchecked")
	public X bind(String attributeName, Object value) {
        // NOTE: We are not resolving embedded properties, because hibernate does not support them
		// Just do that to assert the attribute exists
        if (entityType.getAttribute(attributeName) == null) {
            // Well, some implementations might not be fully spec compliant..
            throw new IllegalArgumentException("Attribute '" + attributeName + "' does not exist on '" + entityType.getName() + "'!");
        }
        Integer attributeBindIndex = bindingMap.get(attributeName);
        
        if (attributeBindIndex != null) {
            throw new IllegalArgumentException("The attribute [" + attributeName + "] has already been bound!");
        }
        
		bindingMap.put(attributeName, selectManager.getSelectInfos().size());
		selectManager.select(parameterManager.addParameterExpression(value), null);
		
		return (X) this;
	}

    @Override
	public SelectBuilder bind(String attributeName) {
        // NOTE: We are not resolving embedded properties, because hibernate does not support them
		// Just do that to assert the attribute exists
        if (entityType.getAttribute(attributeName) == null) {
            // Well, some implementations might not be fully spec compliant..
            throw new IllegalArgumentException("Attribute '" + attributeName + "' does not exist on '" + entityType.getName() + "'!");
        }
		Integer attributeBindIndex = bindingMap.get(attributeName);
		
		if (attributeBindIndex != null) {
			throw new IllegalArgumentException("The attribute [" + attributeName + "] has already been bound!");
		}
		
		bindingMap.put(attributeName, selectManager.getSelectInfos().size());
		return this;
	}
    
    @Override
    protected void prepareAndCheck() {
        List attributes = new ArrayList(bindingMap.size());
        List originalSelectInfos = new ArrayList(selectManager.getSelectInfos());
        List newSelectInfos = selectManager.getSelectInfos();
        newSelectInfos.clear();
        
        for (Map.Entry attributeEntry : bindingMap.entrySet()) {
            // Reorder select infos to fit the attribute order
            Integer newPosition = attributes.size();
            attributes.add(attributeEntry.getKey());
            
            SelectInfo selectInfo = originalSelectInfos.get(attributeEntry.getValue());
            newSelectInfos.add(selectInfo);
            attributeEntry.setValue(newPosition);
        }
        super.prepareAndCheck();
    }

	@Override
    protected boolean isJoinRequiredForSelect() {
	    // NOTE: since we aren't actually selecting properties but passing them through to the insert, we don't require joins
        return false;
    }

    @Override
	protected void getQueryString1(StringBuilder sbSelectFrom) {
		sbSelectFrom.append("INSERT INTO ");
		sbSelectFrom.append(entityType.getName()).append('(');
		
		boolean first = true;
		for (Map.Entry attributeEntry : bindingMap.entrySet()) {
			if (first) {
				first = false;
			} else {
				sbSelectFrom.append(", ");
			}
			
			sbSelectFrom.append(attributeEntry.getKey());
		}
		
		sbSelectFrom.append(")\n");
    	super.getQueryString1(sbSelectFrom);
	}

    @Override
    public Query getQuery() {
        if (jpaProvider.supportsInsertStatement()) {
            return super.getQuery();
        } else {
            // TODO: implement
            throw new UnsupportedOperationException("Not yet implemented!");
        }
    }

}