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}