ro.nextreports.engine.queryexec.QueryResult Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nextreports-engine Show documentation
NextReports Engine is a lightweight Java platform development library which can be used to run NextReports inside your applications.
There is a newer version: 9.2
Show newest version
/*
 * 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 ro.nextreports.engine.queryexec;

import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;

/**
 * @author Decebal Suiu
 */
public class QueryResult {

    private ResultSet resultSet;
    private int numberOfRows;
    private int numberOfColumns;
    private long executeTime;

    // cache
    private Map columnNames;
    private Map columnIndexes;
    private Map columnClassNames;
    private Map columnTypes;
    
    public QueryResult(ResultSet rs, int count, long executeTime) throws QueryException {
        this.resultSet = rs;
        this.executeTime = executeTime;
        this.numberOfRows = count;

        initCache();
    }

    private void initCache() throws QueryException {
        columnNames = new HashMap();
        columnIndexes = new HashMap();
        columnClassNames = new HashMap();
        columnTypes = new HashMap();
        
        try {
            ResultSetMetaData metadata = resultSet.getMetaData();
            numberOfColumns = metadata.getColumnCount();
            for (int i = 0; i < numberOfColumns; i++) {
                columnNames.put(i, metadata.getColumnLabel(i + 1));
                columnIndexes.put(metadata.getColumnLabel(i + 1), i);
                columnClassNames.put(i, metadata.getColumnClassName(i + 1));
                columnTypes.put(i, metadata.getColumnType(i + 1));
            }
        } catch (SQLException e) {
            throw new QueryException(e);
        }
    }
    
    public boolean hasNext() throws QueryException {
        if (resultSet == null) {
            return false;
        }
        
        try {
            return resultSet.next();
        } catch (SQLException e) {
            throw new QueryException(e);
        }
    }
    
    public Object nextValue(String columnName) throws QueryException {
        if (resultSet == null) {
            return null;
        }
        
        try {
            return resultSet.getObject(columnName);
        } catch (SQLException e) {
            throw new QueryException(e);
        }
    }

    public Object nextValue(int columnIndex) throws QueryException {
        if (resultSet == null) {
            return null;
        }
        
        try {
            return resultSet.getObject(columnIndex + 1);
        } catch (SQLException e) {
            throw new QueryException(e);
        }
    }
    
    public ResultSet getResultSet() {
        return resultSet;
    }

    public int getColumnCount() {
        return numberOfColumns;
    }
    
    public int getRowCount() {
        return numberOfRows;
    }
    
    public String getColumnName(int columnIndex) {
        return columnNames.get(columnIndex);
    }

    public int getColumnIndex(String columnName) {
       return columnIndexes.get(columnName);  
    }

    public String getColumnClassName(int columnIndex) {
        return columnClassNames.get(columnIndex);
    }
    
    public int getColumnType(int columnIndex) {
    	return columnTypes.get(columnIndex);
    }
    
    public int getColumnType(String columnName) {    	
    	return getColumnType(getColumnIndex(columnName));
    }
    
    public Object getValueAt(int rowIndex, int columnIndex) throws QueryException {
        if (resultSet == null) {
            return null;
        }
        
        try {
            int row = resultSet.getRow();
            resultSet.absolute(rowIndex + 1);
            Object object = resultSet.getObject(columnIndex + 1);
            if (row == 0) {
                resultSet.beforeFirst();
            } else {
                resultSet.absolute(row);
            }
            return object;            
        } catch (SQLException e) {
            throw new QueryException(e);
        }
    }

    public Object getValueAt(int rowIndex, String columnName) throws QueryException {
        if (resultSet == null) {
            return null;
        }

        try {
            int row = resultSet.getRow();
            resultSet.absolute(rowIndex + 1);
            Object object = resultSet.getObject(columnName);
            if (row == 0) {
                resultSet.beforeFirst();
            } else {
                resultSet.absolute(row);
            }
            return object;
        } catch (SQLException e) {
            throw new QueryException(e);
        }
    }
    
    /**
     * Get the execution time in milliseconds.
     */
    public long getExecuteTime() {
        return executeTime;
    }

    public void close() {
        if (resultSet != null) {
            try {
                resultSet.close();
            } catch (SQLException e) {
                e.printStackTrace();  
            }
        }
    }
    
}