DataSetMetadataImpl.java

/*
 * Copyright 2014 Red Hat, Inc. and/or its affiliates.
 *
 * Licensed 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.dashbuilder.dataset.impl;

import java.util.List;
import java.util.ArrayList;

import org.dashbuilder.dataset.ColumnType;
import org.dashbuilder.dataset.DataColumn;
import org.dashbuilder.dataset.DataSetMetadata;
import org.dashbuilder.dataset.def.DataSetDef;

public class DataSetMetadataImpl implements DataSetMetadata {

    protected DataSetDef definition;
    protected String uuid;
    protected int numberOfRows;
    protected int numberOfColumns;
    protected List<String> columnIds = new ArrayList<String>();
    protected List<ColumnType> columnTypes = new ArrayList<ColumnType>();
    protected int estimatedSize;

    public DataSetMetadataImpl() {
    }

    public DataSetMetadataImpl(DataSetDef definition, String uuid, int numberOfRows, int numberOfColumns,
            List<String> columnIds, List<ColumnType> columnTypes, int estimatedSize) {
        this.definition = definition;
        this.uuid = uuid;
        this.numberOfRows = numberOfRows;
        this.numberOfColumns = numberOfColumns;
        this.columnIds = columnIds;
        this.columnTypes = columnTypes;
        this.estimatedSize = estimatedSize;
    }

    public DataSetMetadataImpl(DataSetImpl dataSet) {
        this.definition = dataSet.getDefinition();
        this.uuid = dataSet.uuid;
        this.numberOfRows = dataSet.getRowCount();
        this.estimatedSize = (int) dataSet.getEstimatedSize();
        this.numberOfColumns = dataSet.getColumns().size();
        for (DataColumn column : dataSet.getColumns()) {
            columnIds.add(column.getId());
            columnTypes.add(column.getColumnType());
        }
    }

    public String getUUID() {
        return uuid;
    }

    public int getNumberOfRows() {
        return numberOfRows;
    }

    public int getNumberOfColumns() {
        return numberOfColumns;
    }

    public int getEstimatedSize() {
        return estimatedSize;
    }

    public String getColumnId(int columnIndex) {
        return columnIds.get(columnIndex);
    }

    public ColumnType getColumnType(int columnIndex) {
        return columnTypes.get(columnIndex);
    }

    public ColumnType getColumnType(String columnId) {
        if (columnId == null) {
            return null;
        }
        for (int i=0; i<columnIds.size(); i++) {
            if (columnId.equalsIgnoreCase(columnIds.get(i))) {
                return columnTypes.get(i);
            }
        }
        return null;
    }

    public List<String> getColumnIds() {
        return columnIds;
    }

    public List<ColumnType> getColumnTypes() {
        return columnTypes;
    }

    public DataSetDef getDefinition() {
        return definition;
    }

    public void setDefinition(DataSetDef definition) {
        this.definition = definition;
    }

    public void setEstimatedSize(int estimatedSize) {
        this.estimatedSize = estimatedSize;
    }

    public void setNumberOfRows(int numberOfRows) {
        this.numberOfRows = numberOfRows;
    }

    public boolean equals(Object obj) {
        try {
            if (obj == null) return false;
            if (obj == this) return true;
            DataSetMetadataImpl other = (DataSetMetadataImpl) obj;

            if (numberOfColumns != other.numberOfColumns) {
                return false;
            }
            if (numberOfRows != other.numberOfRows) {
                return false;
            }
            if (estimatedSize != other.estimatedSize) {
                return false;
            }
            for (int i=0; i<columnIds.size(); i++) {
                if (!columnIds.get(i).equalsIgnoreCase(other.columnIds.get(i))) {
                    return false;
                }
            }
            for (int i=0; i<columnTypes.size(); i++) {
                if (!columnTypes.get(i).equals(other.columnTypes.get(i))) {
                    return false;
                }
            }
            return true;
        }
        catch (ClassCastException e) {
            return false;
        }
    }
}