/*

* 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() {

}

}