001/* 002 * HA-JDBC: High-Availability JDBC 003 * Copyright (C) 2013 Paul Ferraro 004 * 005 * This program is free software: you can redistribute it and/or modify 006 * it under the terms of the GNU Lesser General Public License as published by 007 * the Free Software Foundation, either version 3 of the License, or 008 * (at your option) any later version. 009 * 010 * This program is distributed in the hope that it will be useful, 011 * but WITHOUT ANY WARRANTY; without even the implied warranty of 012 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 013 * GNU Lesser General Public License for more details. 014 * 015 * You should have received a copy of the GNU Lesser General Public License 016 * along with this program. If not, see <http://www.gnu.org/licenses/>. 017 */ 018package net.sf.hajdbc.io.simple; 019 020import java.io.ByteArrayInputStream; 021import java.io.ByteArrayOutputStream; 022import java.io.IOException; 023import java.io.InputStreamReader; 024import java.io.OutputStreamWriter; 025import java.io.Reader; 026 027import net.sf.hajdbc.io.InputSinkChannel; 028 029/** 030 * Reader channel for writing to, and reading from, an in-memory buffer sink. 031 * @author Paul Ferraro 032 */ 033public class SimpleReaderSinkChannel implements InputSinkChannel<Reader, byte[]> 034{ 035 @Override 036 public byte[] write(Reader reader) throws IOException 037 { 038 ByteArrayOutputStream output = new ByteArrayOutputStream(); 039 OutputStreamWriter writer = new OutputStreamWriter(output); 040 int c = reader.read(); 041 while (c >= 0) 042 { 043 writer.write(c); 044 c = reader.read(); 045 } 046 writer.flush(); 047 return output.toByteArray(); 048 } 049 050 @Override 051 public Reader read(byte[] sink) 052 { 053 return new InputStreamReader(new ByteArrayInputStream(sink)); 054 } 055}