org.teiid.query.sql.lang.BatchedUpdateCommand Maven / Gradle / Ivy

Go to download
/*
 * Copyright Red Hat, Inc. and/or its affiliates
 * and other contributors as indicated by the @author tags and
 * the COPYRIGHT.txt file distributed with this work.
 *
 * 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 org.teiid.query.sql.lang;

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

import org.teiid.core.util.EquivalenceUtil;
import org.teiid.query.sql.LanguageObject;
import org.teiid.query.sql.LanguageVisitor;
import org.teiid.query.sql.util.VariableContext;



/** 
 * Represents a batch of INSERT, UPDATE, DELETE, and SELECT INTO commands
 * @since 4.2
 */
public class BatchedUpdateCommand extends Command {
    
	protected List commands;
    private List variableContexts; //processing state
    private boolean singleResult;
    
    /**
     *  
     * @param updateCommands
     * @since 4.2
     */
    public BatchedUpdateCommand(List updateCommands) {
        this(updateCommands, false);
    }
    
    public BatchedUpdateCommand(List updateCommands, boolean singleResult) {
        this.commands = new ArrayList(updateCommands);
        this.singleResult = singleResult;
    }
    
    /**
     * Gets the List of updates contained in this batch
     * @return
     * @since 4.2
     */
    public List getUpdateCommands() {
        return commands;
    }

    /** 
     * @see org.teiid.query.sql.lang.Command#getType()
     * @since 4.2
     */
    public int getType() {
        return Command.TYPE_BATCHED_UPDATE;
    }

    /** 
     * @see org.teiid.query.sql.lang.Command#getProjectedSymbols()
     * @since 4.2
     */
    public List getProjectedSymbols() {
        return Command.getUpdateCommandSymbol();
    }
    
    /** 
     * @since 4.2
     */
    public void acceptVisitor(LanguageVisitor visitor) {
        visitor.visit(this);
    }

    /** 
     * @see java.lang.Object#clone()
     * @since 4.2
     */
    public Object clone() {
        List clonedCommands = LanguageObject.Util.deepClone(this.commands, Command.class);
        BatchedUpdateCommand copy = new BatchedUpdateCommand(clonedCommands);
        copy.singleResult = this.singleResult;
        copyMetadataState(copy);
        return copy;
    }
    
	/**
	 * @see org.teiid.query.sql.lang.Command#areResultsCachable()
	 */
	public boolean areResultsCachable() {
		return false;
	}

    public String toString() {
       return getStringForm(false);
    }
    
	public void setVariableContexts(List variableContexts) {
		this.variableContexts = variableContexts;
	}

	public List getVariableContexts() {
		return variableContexts;
	}
	
    @Override
	public int hashCode() {
    	return commands.hashCode();
	}

	@Override
	public boolean equals(Object obj) {
		if (this == obj) {
			return true;
		}
		if (!(obj instanceof BatchedUpdateCommand)) {
			return false;
		}
		BatchedUpdateCommand other = (BatchedUpdateCommand)obj;
		return EquivalenceUtil.areEqual(commands, other.commands) && this.singleResult == other.singleResult;
	}
	
	public void setSingleResult(boolean singleResult) {
		this.singleResult = singleResult;
	}
	
	public boolean isSingleResult() {
		return singleResult;
	}

	public String getStringForm(boolean full) {
		StringBuffer val = new StringBuffer(); 
		if (!full) {
			val.append("BatchedUpdate{"); //$NON-NLS-1$
		}
        if (commands != null && commands.size() > 0) {
            for (int i = 0; i < commands.size(); i++) {
            	if (i > 0) { 
            		if (full) {
            			val.append(";\n"); //$NON-NLS-1$
            		} else {
            			val.append(","); //$NON-NLS-1$
            		}
            	}
            	if (full) {
            		val.append(commands.get(i));
            	} else {
            		val.append(getCommandToken(commands.get(i).getType()));	
            	}
            }
        }
        if (!full) {
        	val.append('}');
        }
	    return val.toString();
	}
	
}