org.onetwo.common.utils.map.TableMap Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of onetwo-common Show documentation

There is a newer version: 4.7.2

package org.onetwo.common.utils.map;

import java.util.HashMap;
import java.util.Map;

public class TableMap {

	private Map> rowMap = new HashMap<>();
	private Map> columnMap = new HashMap<>();

	public int rowSize(){
		return rowMap.size();
	}
	public int size(){
		int size = 0;
		for(Map row: rowMap.values()){
			size += row.size();
		}
		return size;
	}

	public boolean containsRow(RK rowKey){
		return rowMap.containsKey(rowKey);
	}
	public boolean containsColumn(CK columnKey){
		return columnMap.containsKey(columnKey);
	}
	public boolean contains(RK rowKey, CK columnKey){
		return rowMap.containsKey(rowKey) && columnMap.containsKey(columnKey);
	}
	public V get(RK rowKey, CK columKey){
		return get(rowKey, columKey, null);
	}
	public V getByRC(RK rowKey, CK columKey){
		return get(rowKey, columKey, null);
	}
	public V getByCR(CK columKey, RK rowKey){
		return get(rowKey, columKey, null);
	}
	
	public V get(RK rowKey, CK columKey, V defaultValue){
		Map row = getRow(rowKey);
		if(row==null)
			return defaultValue;
		return row.containsKey(columKey)?row.get(columKey):defaultValue;
	}
	public Map getRow(RK rowKey){
		return rowMap.get(rowKey);
	}
	public Map getColumn(RK columnKey){
		return columnMap.get(columnKey);
	}

	public V put(RK rowKey, CK columKey, V value){
		putIntoRow(rowKey, columKey, value);
		V v = putIntoColumn(rowKey, columKey, value);
		return v;
	}
	
	public V putIntoRow(RK rowKey, CK columKey, V value){
		Map row = rowMap.get(rowKey);
		if(row==null){
			row = new HashMap<>();
			this.rowMap.put(rowKey, row);
		}
		return row.put(columKey, value);
	}
	
	public V putIntoColumn(RK rowKey, CK columKey, V value){
		Map columns = columnMap.get(columKey);
		if(columns==null){
			columns = new HashMap<>();
			this.columnMap.put(columKey, columns);
		}
		return columns.put(rowKey, value);
	}

}