openjpa-slice/src/main/java/org/apache/openjpa/slice/jdbc/DistributedTemplate.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.apache.openjpa.slice.jdbc;

import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.SQLWarning;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

/**
 * A template for multiple Statements being executed by multiple connections.
 * 
 * @author Pinaki Poddar 
 *
 */
class DistributedTemplate<T extends Statement> 
	implements Statement, Iterable<T> {
	protected List<T> stmts = new ArrayList<T>();
	protected final DistributedConnection con;
	protected T master;
	
	public DistributedTemplate(DistributedConnection c) {
		con = c;
	}
	
	public Iterator<T> iterator() {
		return stmts.iterator();
	}
	
	public void add(T s) {
		if (stmts.isEmpty())
			master = s;
		try {
			if (!con.contains(s.getConnection()))
				throw new IllegalArgumentException(s + " has different connection");
			stmts.add(s);
		} catch (SQLException e) {
			e.printStackTrace();
		}
	}
	
	public void addBatch(String sql) throws SQLException {
		for (T s:this)
			s.addBatch(sql);
	}

	public void cancel() throws SQLException {
		for (T s:this)
			s.cancel();
	}

	public void clearBatch() throws SQLException {
		for (T s:this)
			s.clearBatch();
	}

	public void clearWarnings() throws SQLException {
		for (T s:this)
			s.clearWarnings();
	}

	public void close() throws SQLException {
		for (T s:this)
			s.close();
	}

	public boolean execute(String arg0) throws SQLException {
		boolean ret = true;
		for (T s:this)
			ret = s.execute(arg0) & ret;
		return ret;
	}

	public boolean execute(String arg0, int arg1) throws SQLException {
		boolean ret = true;
		for (T s:this)
			ret = s.execute(arg0, arg1) & ret;
		return ret;
	}

	public boolean execute(String arg0, int[] arg1) throws SQLException {
		boolean ret = true;
		for (T s:this)
			ret = s.execute(arg0, arg1) & ret;
		return ret;
	}

	public boolean execute(String arg0, String[] arg1) throws SQLException {
		boolean ret = true;
		for (T s:this)
			ret = s.execute(arg0, arg1) & ret;
		return ret;
	}

	public int[] executeBatch() throws SQLException {
		int[] ret = new int[0];
		for (Statement s:this) {
			int[] tmp = s.executeBatch();
			ret = new int[ret.length + tmp.length];
			System.arraycopy(tmp, 0, ret, ret.length-tmp.length, tmp.length);
		}
		return ret;
	}

	public ResultSet executeQuery() throws SQLException {
		DistributedResultSet rs = new DistributedResultSet();
		for (T s:this)
			rs.add(s.executeQuery(null));
		return rs;
	}

	public ResultSet executeQuery(String arg0) throws SQLException {
		DistributedResultSet rs = new DistributedResultSet();
		for (T s:this)
			rs.add(s.executeQuery(arg0));
		return rs;
	}

	public int executeUpdate(String arg0) throws SQLException {
		int ret = 0;
		for (T s:this)
			ret += s.executeUpdate(arg0);
		return ret;
	}

	public int executeUpdate(String arg0, int arg1) throws SQLException {
		int ret = 0;
		for (T s:this)
			ret += s.executeUpdate(arg0, arg1);
		return ret;
	}

	public int executeUpdate(String arg0, int[] arg1) throws SQLException {
		int ret = 0;
		for (T s:this)
			ret += s.executeUpdate(arg0, arg1);
		return ret;
	}

	public int executeUpdate(String arg0, String[] arg1) throws SQLException {
		int ret = 0;
		for (T s:this)
			ret += s.executeUpdate(arg0, arg1);
		return ret;
	}

	public Connection getConnection() throws SQLException {
		return con;
	}

	public int getFetchDirection() throws SQLException {
		return master.getFetchDirection();
	}

	public int getFetchSize() throws SQLException {
		return master.getFetchSize();
	}

	public ResultSet getGeneratedKeys() throws SQLException {
		DistributedResultSet mrs = new DistributedResultSet();
		for (T s:this)
			mrs.add(s.getGeneratedKeys());
		return mrs;
	}

	public int getMaxFieldSize() throws SQLException {
		return master.getMaxFieldSize();
	}

	public int getMaxRows() throws SQLException {
		return master.getMaxRows();
	}

	public boolean getMoreResults() throws SQLException {
		for (T s:this)
			if (s.getMoreResults())
				return true;
		return false;
	}

	public boolean getMoreResults(int arg0) throws SQLException {
		for (T s:this)
			if (s.getMoreResults(arg0))
				return true;
		return false;
	}

	public int getQueryTimeout() throws SQLException {
		return master.getQueryTimeout();
	}

	public ResultSet getResultSet() throws SQLException {
		DistributedResultSet rs = new DistributedResultSet();
		for (T s:this)
			rs.add(s.getResultSet());
		return rs;
	}

	public int getResultSetConcurrency() throws SQLException {
		return master.getResultSetConcurrency();
	}

	public int getResultSetHoldability() throws SQLException {
		return master.getResultSetHoldability();
	}

	public int getResultSetType() throws SQLException {
		return master.getResultSetType();
	}

	public int getUpdateCount() throws SQLException {
		return master.getUpdateCount();
	}

	public SQLWarning getWarnings() throws SQLException {
		return master.getWarnings();
	}

	public void setCursorName(String name) throws SQLException {
		for (T s:this)
			s.setCursorName(name);
	}

	public void setEscapeProcessing(boolean flag) throws SQLException {
		for (T s:this)
			s.setEscapeProcessing(flag);
	}

	public void setFetchDirection(int dir) throws SQLException {
		for (T s:this)
			s.setFetchDirection(dir);
	}

	public void setFetchSize(int size) throws SQLException {
		for (T s:this)
			s.setFetchSize(size);
	}

	public void setMaxFieldSize(int size) throws SQLException {
		for (T s:this)
			s.setMaxFieldSize(size);
	}

	public void setMaxRows(int n) throws SQLException {
		for (T s:this)
			s.setMaxFieldSize(n);
	}
	
	public void setQueryTimeout(int n) throws SQLException {
		for (T s:this)
			s.setMaxFieldSize(n);
	}
}