TupleReturn.java

/*
 * Copyright (c) 2004-2013 Tada AB and other contributors, as listed below.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the The BSD 3-Clause License
 * which accompanies this distribution, and is available at
 * http://opensource.org/licenses/BSD-3-Clause
 *
 * Contributors:
 *   Tada AB
 */
package org.postgresql.pljava.example;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Timestamp;

import org.postgresql.pljava.ResultSetProvider;

/**
 * Illustrates various methods of returning composite values.
 * @author Thomas Hallgren
 */
public class TupleReturn implements ResultSetProvider {
	public static String makeString(ResultSet _testSetReturn)
			throws SQLException {
		int base = _testSetReturn.getInt(1);
		int incbase = _testSetReturn.getInt(2);
		Timestamp ctime = _testSetReturn.getTimestamp(3);
		return "Base = \"" + base + "\", incbase = \"" + incbase
				+ "\", ctime = \"" + ctime + "\"";
	}

	public static ResultSetProvider setReturn(int base, int increment)
			throws SQLException {
		return new TupleReturn(base, increment);
	}

	public static boolean tupleReturn(int base, int increment,
			ResultSet receiver) throws SQLException {
		receiver.updateInt(1, base);
		receiver.updateInt(2, base + increment);
		receiver.updateTimestamp(3, new Timestamp(System.currentTimeMillis()));
		return true;
	}

	public static boolean tupleReturn(Integer base, Integer increment,
			ResultSet receiver) throws SQLException {
		if (base == null) {
			receiver.updateNull(1);
			receiver.updateNull(2);
		} else {
			receiver.updateInt(1, base.intValue());
			if (increment == null)
				receiver.updateNull(2);
			else
				receiver.updateInt(2, base.intValue() + increment.intValue());
		}
		receiver.updateTimestamp(3, new Timestamp(System.currentTimeMillis()));
		return true;
	}

	private final int m_base;

	private final int m_increment;

	public TupleReturn(int base, int increment) {
		m_base = base;
		m_increment = increment;
	}

	@Override
	public boolean assignRowValues(ResultSet receiver, int currentRow)
			throws SQLException {
		// Stop when we reach 12 rows.
		//
		if (currentRow >= 12)
			return false;

		receiver.updateInt(1, m_base);
		receiver.updateInt(2, m_base + m_increment * currentRow);
		receiver.updateTimestamp(3, new Timestamp(System.currentTimeMillis()));
		return true;
	}

	@Override
	public void close() {
	}
}