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.InputStream;
024
025import net.sf.hajdbc.io.InputSinkChannel;
026
027/**
028 * Input stream channel for writing to, and reading from, an in-memory buffer sink.
029 * @author Paul Ferraro
030 */
031public class SimpleInputStreamSinkChannel implements InputSinkChannel<InputStream, byte[]>
032{
033        @Override
034        public byte[] write(InputStream input) throws IOException
035        {
036                ByteArrayOutputStream output = new ByteArrayOutputStream();
037                int b = input.read();
038                while (b >= 0)
039                {
040                        output.write(b);
041                        b = input.read();
042                }
043                return output.toByteArray();
044        }
045
046        @Override
047        public InputStream read(byte[] sink)
048        {
049                return new ByteArrayInputStream(sink);
050        }
051}