我需要一个Java函数，以InputStream参数的形式返回SQL SELECT查询的结果，以便另一个系统将结果通过网络发送。
然而，InputStream必须是一个带有自定义分隔符的String（通常是CSV格式，但不总是）。
虽然我可以轻松创建一个函数来检索结果、创建分隔符的String，最后将该String转换为InputStream，但SQL结果通常太大而无法在内存中处理。另外，在返回结果之前处理整个结果集将导致不必要的等待时间。
如何返回一个InputStream以遍历SQL结果，并在从数据库返回已处理（分隔）数据时发送？
以下是（未经测试的）代码片段，它应该能给你一个基本的思路：
/**
* Implementors of this interface should only convert current row to byte array and return it.
*
* @author yura
*/
public interface RowToByteArrayConverter {
byte[] rowToByteArray(ResultSet resultSet);
}
public class ResultSetAsInputStream extends InputStream {
private final RowToByteArrayConverter converter;
private final PreparedStatement statement;
private final ResultSet resultSet;
private byte[] buffer;
private int position;
public ResultSetAsInputStream(final RowToByteArrayConverter converter, final Connection connection, final String sql, final Object... parameters) throws SQLException {
this.converter = converter;
statement = createStatement(connection, sql, parameters);
resultSet = statement.executeQuery();
}
private static PreparedStatement createStatement(final Connection connection, final String sql, final Object[] parameters) {
// PreparedStatement should be created here from passed connection, sql and parameters
return null;
}
@Override
public int read() throws IOException {
try {
if(buffer == null) {
// first call of read method
if(!resultSet.next()) {
return -1; // no rows - empty input stream
} else {
buffer = converter.rowToByteArray(resultSet);
position = 0;
return buffer[position++] & (0xff);
}
} else {
// not first call of read method
if(position < buffer.length) {
// buffer already has some data in, which hasn't been read yet - returning it
return buffer[position++] & (0xff);
} else {
// all data from buffer was read - checking whether there is next row and re-filling buffer
if(!resultSet.next()) {
return -1; // the buffer was read to the end and there is no rows - end of input stream
} else {
// there is next row - converting it to byte array and re-filling buffer
buffer = converter.rowToByteArray(resultSet);
position = 0;
return buffer[position++] & (0xff);
}
}
}
} catch(final SQLException ex) {
throw new IOException(ex);
}
}
@Override
public void close() throws IOException {
try {
statement.close();
} catch(final SQLException ex) {
throw new IOException(ex);
}
}
}
这是一个非常简单的实现，可以通过以下方式进行改进：
new byte[]是昂贵的操作）。然后应该更改RowToByteArrayConverter.rowToByteArray方法的签名为int fillByteArrayFromRow(ResultSet rs, byte[] array)，它应该返回填充的字节数并填充传递的字节数组。因为字节数组包含有符号字节，所以它可以包含-1（实际上是无符号字节的255），从而指示流的结束位置不正确，因此使用& (0xff)将有符号字节转换为无符号字节作为整数值。有关详细信息，请参阅Java如何将int转换为byte？。
还请注意，如果网络传输速度较慢，这可能会使结果集保持打开状态很长时间，从而给数据库带来问题。
希望这可以帮助...
public abstract class RowToByteArrayConverter {
public byte[] rowToByteArray(ResultSet resultSet) {
parseResultSet(dataOutputStream, resultSet);
return byteArrayOutputSteam.toByteArray();
}
public RowToByteArrayConverter() {
dataOutputStream = new DataOutputStream(byteArrayOutputStream);
}
protected DataOutputStream dataOutputStream;
protected ByteArrayOutputStream byteArrayOutputStream;
protected abstract void parseResultSet(DataOutputStream dataOutputStresm, ResultSet rs);
}
InputStream中发送。你能展示一下可以实现这个功能的简短例子/方法吗？ - Cameron S