com.univocity.app.data.Data.java Source code

Introduction

Here is the source code for com.univocity.app.data.Data.java

Source

/*******************************************************************************
 * Copyright (c) 2014 uniVocity Software Pty Ltd. All rights reserved.
 * This file is subject to the terms and conditions defined in file
 * 'LICENSE.txt', which is part of this source code package.
 ******************************************************************************/
package com.univocity.app.data;

import java.util.*;

import org.springframework.jdbc.support.rowset.*;

import com.univocity.app.utils.*;

public class Data {
    protected final Database database;
    private final String selectScript;

    private final List<Object[]> data;
    private String[] columnNames;

    public Data(Database database, String selectScript) {
        this.database = database;
        this.selectScript = selectScript;

        this.data = new ArrayList<Object[]>();
        this.columnNames = new String[0];
    }

    public void reloadData() {
        data.clear();

        SqlRowSet queryResult = database.getJdbcTemplate().queryForRowSet(selectScript);
        SqlRowSetMetaData metaData = queryResult.getMetaData();
        columnNames = metaData.getColumnNames();

        for (int i = 0; i < columnNames.length; i++) {
            String label = metaData.getColumnLabel(i + 1);
            if (label != null) {
                columnNames[i] = label;
            }
        }

        while (queryResult.next()) {
            Object[] row = new Object[columnNames.length];
            for (int i = 0; i < row.length; i++) {
                row[i] = queryResult.getObject(i + 1);
            }
            data.add(row);
        }

    }

    public List<Object[]> getRows() {
        return data;
    }

    public String[] getColumnNames() {
        return columnNames;
    }
}