BinaryColumnTest.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.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.sql.ResultSet;
import java.sql.SQLException;

import org.postgresql.pljava.ResultSetProvider;

/**
 * Example using {@code ResultSetProvider} to return 100 rows of two
 * {@code bytea} columns each, which should be equal in each row, one being
 * set by {@link java.sql.ResultSet#updateBinaryStream updateBinaryStream}
 * and the other by {@link java.sql.ResultSet#updateBytes updateBytes}.
 */
public class BinaryColumnTest implements ResultSetProvider {
	public static ResultSetProvider getBinaryPairs() {
		return new BinaryColumnTest();
	}

	@Override
	public boolean assignRowValues(ResultSet rs, int rowCount)
			throws SQLException {
		try {
			if (rowCount >= 100)
				return false;

			int offset = rowCount * 100;
			ByteArrayOutputStream bld = new ByteArrayOutputStream();
			DataOutputStream da = new DataOutputStream(bld);
			for (int idx = 0; idx < 100; ++idx)
				da.writeInt(offset + idx);
			byte[] bytes = bld.toByteArray();
			ByteArrayInputStream input = new ByteArrayInputStream(bytes);
			rs.updateBinaryStream(1, input, bytes.length);
			rs.updateBytes(2, bytes);
			return true;
		} catch (IOException e) {
			throw new SQLException(e.getMessage());
		}
	}

	@Override
	public void close() throws SQLException {
	}
}