net.sf.mongodb_jdbc_driver.zql.ZInsert Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mongojdbcdriver Show documentation
JDBC driver for MongoDB
The newest version!
/*
 * This file is part of Zql.
 *
 * Zql is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Zql is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Zql.  If not, see .
 */

package net.sf.mongodb_jdbc_driver.zql;

import java.util.Vector;

/**
 * ZInsert: an SQL INSERT statement
 */
public class ZInsert implements ZStatement
{

	private static final long serialVersionUID = -2577795635628419391L;

	String table_;

	Vector columns_ = null;

	ZExp valueSpec_ = null;

	/**
	 * Create an INSERT statement on a given table
	 */
	public ZInsert(String tab)
	{
		table_ = new String(tab);
	}

	/**
	 * Get the name of the table involved in the INSERT statement.
	 * 
	 * @return A String equal to the table name
	 */
	public String getTable()
	{
		return table_;
	}

	/**
	 * Get the columns involved in the INSERT statement.
	 * 
	 * @return A Vector of Strings equal to the column names
	 */
	public Vector getColumns()
	{
		return columns_;
	}

	/**
	 * Specify which columns to insert
	 * 
	 * @param c
	 *           A vector of column names (Strings)
	 */
	public void addColumns(Vector c)
	{
		columns_ = c;
	}

	/**
	 * Specify the VALUES part or SQL sub-query of the INSERT statement
	 * 
	 * @param e
	 *           An SQL expression or a SELECT statement. If it is a list of SQL expressions, e should be
	 *           represented by ONE SQL expression with operator = "," and operands = the expressions in the
	 *           list. If it is a SELECT statement, e should be a ZQuery object.
	 */
	public void addValueSpec(ZExp e)
	{
		valueSpec_ = e;
	}

	/**
	 * Get the VALUES part of the INSERT statement
	 * 
	 * @return A vector of SQL Expressions (ZExp objects); If there's no VALUES but a subquery, returns null
	 *         (use getQuery() method).
	 */
	public Vector getValues()
	{
		if (!(valueSpec_ instanceof ZExpression)) return null;
		return ((ZExpression) valueSpec_).getOperands();
	}

	/**
	 * Get the sub-query (ex. in INSERT INTO table1 SELECT * FROM table2;, the sub-query is SELECT * FROM
	 * table2;)
	 * 
	 * @return A ZQuery object (A SELECT statement), or null if there's no sub-query (in that case, use the
	 *         getValues() method to get the VALUES part).
	 */
	public ZQuery getQuery()
	{
		if (!(valueSpec_ instanceof ZQuery)) return null;
		return (ZQuery) valueSpec_;
	}

	public String toString()
	{
		StringBuffer buf = new StringBuffer("insert into " + table_);
		if (columns_ != null && columns_.size() > 0)
		{
			// buf.append(" " + columns_.toString());
			buf.append("(" + columns_.elementAt(0));
			for (int i = 1; i < columns_.size(); i++)
			{
				buf.append("," + columns_.elementAt(i));
			}
			buf.append(")");
		}

		String vlist = valueSpec_.toString();
		buf.append(" ");
		if (getValues() != null) buf.append("values ");
		if (vlist.startsWith("(")) buf.append(vlist);
		else buf.append(" (" + vlist + ")");

		return buf.toString();
	}
};