openjpa-jdbc/src/main/java/org/apache/openjpa/jdbc/kernel/SQLProjectionResultObjectProvider.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.    
 */
package org.apache.openjpa.jdbc.kernel;

import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;

import org.apache.openjpa.jdbc.meta.JavaSQLTypes;
import org.apache.openjpa.jdbc.sql.ResultSetResult;
import org.apache.openjpa.jdbc.sql.SQLExceptions;
import org.apache.openjpa.kernel.ResultPacker;
import org.apache.openjpa.lib.rop.ResultObjectProvider;
import org.apache.openjpa.util.StoreException;
import org.apache.openjpa.util.UnsupportedException;
import serp.util.Numbers;

/**
 * Provides all column data in a {@link ResultSet}.
 *
 * @author Abe White
 */
class SQLProjectionResultObjectProvider
    implements ResultObjectProvider {

    private final JDBCStore _store;
    private final JDBCFetchConfiguration _fetch;
    private final ResultSetResult _res;
    private final ResultPacker _packer;
    private final int _cols;

    /**
     * Constructor.
     *
     * @param res the result data
     * @param cls the result class; may be null for the default
     */
    public SQLProjectionResultObjectProvider(JDBCStore store,
        JDBCFetchConfiguration fetch, ResultSetResult res, Class cls)
        throws SQLException {
        _store = store;
        _fetch = fetch;

        ResultSetMetaData meta = res.getResultSet().getMetaData();
        _res = res;
        _cols = meta.getColumnCount();

        if (cls != null) {
            String[] aliases = new String[_cols];
            for (int i = 0; i < _cols; i++)
                aliases[i] = meta.getColumnLabel(i + 1);
            _packer = new ResultPacker(null, aliases, cls);
        } else
            _packer = null;
    }

    public boolean supportsRandomAccess() {
        try {
            return _res.supportsRandomAccess();
        } catch (Throwable t) {
            return false;
        }
    }

    public void open() {
    }

    public Object getResultObject()
        throws SQLException {
        if (_cols == 1) {
            Object val = _res.getObject(Numbers.valueOf(1),
                JavaSQLTypes.JDBC_DEFAULT, null);
            return (_packer == null) ? val : _packer.pack(val);
        }

        Object[] vals = new Object[_cols];
        for (int i = 0; i < vals.length; i++)
            vals[i] = _res.getObject(Numbers.valueOf(i + 1),
                JavaSQLTypes.JDBC_DEFAULT, null);
        return (_packer == null) ? vals : _packer.pack(vals);
    }

    public boolean next()
        throws SQLException {
        return _res.next();
    }

    public boolean absolute(int pos)
        throws SQLException {
        return _res.absolute(pos);
    }

    public int size()
        throws SQLException {
        if (_fetch.getLRSSize() == LRSSizes.SIZE_UNKNOWN
            || !supportsRandomAccess())
            return Integer.MAX_VALUE;
        return _res.size();
    }

    public void reset() {
        throw new UnsupportedException();
    }

    public void close() {
        _res.close();
    }

    public void handleCheckedException(Exception e) {
        if (e instanceof SQLException)
            throw SQLExceptions.getStore((SQLException) e,
                _store.getDBDictionary());
        throw new StoreException(e);
    }
}