org.datacleaner.data.MockInputRow Maven / Gradle / Ivy

Go to download
/**
 * DataCleaner (community edition)
 * Copyright (C) 2014 Neopost - Customer Information Management
 *
 * This copyrighted material is made available to anyone wishing to use, modify,
 * copy, or redistribute it subject to the terms and conditions of the GNU
 * Lesser General Public License, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
 * for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this distribution; if not, write to:
 * Free Software Foundation, Inc.
 * 51 Franklin Street, Fifth Floor
 * Boston, MA  02110-1301  USA
 */
package org.datacleaner.data;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;

import org.datacleaner.api.InputColumn;

/**
 * A mock implementation of the InputRow interface. Allows for adhoc generation
 * of a row using the put(...) method.
 */
public class MockInputRow extends AbstractInputRow {

    private static final long serialVersionUID = 1L;

    private static final AtomicInteger _idGenerator = new AtomicInteger(Integer.MIN_VALUE);

    private final Map, Object> _values;
    private final int _id;

    public MockInputRow() {
        this(_idGenerator.getAndIncrement());
    }

    public MockInputRow(int id, Map, Object> values) {
        _values = values;
        _id = id;
    }

    public MockInputRow(Map, Object> values) {
        this(_idGenerator.getAndIncrement(), values);
    }

    public MockInputRow(int id) {
        this(id, new LinkedHashMap, Object>());
    }

    public MockInputRow(InputColumn[] columns, Object[] values) {
        this(_idGenerator.getAndIncrement(), columns, values);
    }

    public MockInputRow(int id, InputColumn[] columns, Object[] values) {
        this(id);
        for (int i = 0; i < values.length; i++) {
            put(columns[i], values[i]);
        }
    }

    @Override
    public int getId() {
        return _id;
    }

    @Override
    public List> getInputColumns() {
        return new ArrayList>(_values.keySet());
    }

    @Override
    public boolean containsInputColumn(InputColumn inputColumn) {
        return _values.containsKey(inputColumn);
    }

    @SuppressWarnings("unchecked")
    @Override
    public  E getValueInternal(InputColumn column) {
        return (E) _values.get(column);
    }

    /**
     * Puts/adds a new value to the row.
     * 
     * @param column
     * @param value
     * @return
     */
    public MockInputRow put(InputColumn column, Object value) {
        _values.put(column, value);
        return this;
    }

    /**
     * Puts/adds new values to the row.
     * 
     * @param values
     * @return
     */
    public MockInputRow putAll(Map, Object> values) {
        _values.putAll(values);
        return this;
    }

    @Override
    public int hashCode() {
        final int prime = 31;
        int result = 1;
        result = prime * result + _id;
        result = prime * result + ((_values == null) ? 0 : _values.hashCode());
        return result;
    }

    @Override
    public boolean equals(Object obj) {
        if (this == obj)
            return true;
        if (obj == null)
            return false;
        if (getClass() != obj.getClass())
            return false;
        MockInputRow other = (MockInputRow) obj;
        if (_id != other._id)
            return false;
        if (_values == null) {
            if (other._values != null)
                return false;
        } else if (!_values.equals(other._values))
            return false;
        return true;
    }

    @Override
    public String toString() {
        return "MockInputRow[id=" + _id + "]";
    }
}