package eu.dnetlib.data.transform;

import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import com.google.common.collect.Maps;

public class Row implements Iterable<Column<String, byte[]>> {

	private String columnFamily;

	private String key;

	private Map<String, Column<String, byte[]>> columns;

	public Row(final Row row) {
		this.columnFamily = row.getColumnFamily();
		this.key = row.getKey();
		this.columns = row.columns;
	}

	public Row(final String columnFamily, final String key) {
		this.columnFamily = columnFamily;
		this.key = key;
		this.columns = Maps.newHashMap();
	}

	public Row(final String columnFamily, final String key, final List<Column<String, byte[]>> columns) {
		this(columnFamily, key);
		this.setColumns(columns);
	}

	public void addColumn(final Column<String, byte[]> column) {
		getColumns().add(column);
	}

	public String getKey() {
		return key;
	}

	public Collection<Column<String, byte[]>> getColumns() {
		return columns.values();
	}

	public Column<String, byte[]> getColumn(final String name) {
		return columns.get(name);
	}

	public void setColumn(final String name, final Column<String, byte[]> newColumn) {
		columns.put(name, newColumn);
	}

	public void setColumns(final List<Column<String, byte[]>> columns) {
		this.columns.clear();
		for (final Column<String, byte[]> col : columns) {
			this.columns.put(col.getName(), col);
		}
	}

	public String getColumnFamily() {
		return columnFamily;
	}

	@Override
	public Iterator<Column<String, byte[]>> iterator() {
		return getColumns().iterator();
	}

	@Override
	public String toString() {
		return "\nRow [ KEY: " + key + ",\n\tCF: " + columnFamily + "\n\tCOLS: {" + columns + "}]";
	}

}