001    /**
002     * Licensed to the Apache Software Foundation (ASF) under one or more
003     * contributor license agreements.  See the NOTICE file distributed with
004     * this work for additional information regarding copyright ownership.
005     * The ASF licenses this file to You under the Apache License, Version 2.0
006     * (the "License"); you may not use this file except in compliance with
007     * the License.  You may obtain a copy of the License at
008     *
009     *      http://www.apache.org/licenses/LICENSE-2.0
010     *
011     * Unless required by applicable law or agreed to in writing, software
012     * distributed under the License is distributed on an "AS IS" BASIS,
013     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014     * See the License for the specific language governing permissions and
015     * limitations under the License.
016     */
017    package org.apache.activemq.store.jdbc.adapter;
018    
019    import java.io.ByteArrayOutputStream;
020    import java.io.IOException;
021    import java.io.InputStream;
022    import java.sql.PreparedStatement;
023    import java.sql.ResultSet;
024    import java.sql.SQLException;
025    
026    import org.apache.activemq.util.ByteArrayInputStream;
027    
028    /**
029     * This JDBCAdapter inserts and extracts BLOB data using the
030     * setBinaryStream()/getBinaryStream() operations.
031     * 
032     * The databases/JDBC drivers that use this adapter are:
033     * <ul>
034     * <li>Axion</li>
035     * </ul>
036     * 
037     * @org.apache.xbean.XBean element="streamJDBCAdapter"
038     * 
039     * 
040     */
041    public class StreamJDBCAdapter extends DefaultJDBCAdapter {
042    
043        /**
044         * @see org.apache.activemq.store.jdbc.adapter.DefaultJDBCAdapter#getBinaryData(java.sql.ResultSet,
045         *      int)
046         */
047        @Override
048        protected byte[] getBinaryData(ResultSet rs, int index) throws SQLException {
049    
050            try {
051                InputStream is = rs.getBinaryStream(index);
052                ByteArrayOutputStream os = new ByteArrayOutputStream(1024 * 4);
053    
054                int ch;
055                while ((ch = is.read()) >= 0) {
056                    os.write(ch);
057                }
058                is.close();
059                os.close();
060    
061                return os.toByteArray();
062            } catch (IOException e) {
063                throw (SQLException)new SQLException("Error reading binary parameter: " + index).initCause(e);
064            }
065        }
066    
067        /**
068         * @see org.apache.activemq.store.jdbc.adapter.DefaultJDBCAdapter#setBinaryData(java.sql.PreparedStatement,
069         *      int, byte[])
070         */
071        @Override
072        protected void setBinaryData(PreparedStatement s, int index, byte[] data) throws SQLException {
073            s.setBinaryStream(index, new ByteArrayInputStream(data), data.length);
074        }
075    
076    }