com.olapdb.obase.data.index.Idx Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of olap-core Show documentation
The newest version!
package com.olapdb.obase.data.index;

import com.olapdb.obase.data.Bytez;
import com.olapdb.obase.data.ReferenceEntity;
import com.olapdb.obase.utils.Obase;
import org.apache.hadoop.hbase.Cell;
import org.apache.hadoop.hbase.CellUtil;
import org.apache.hadoop.hbase.client.*;
import org.apache.hadoop.hbase.util.Bytes;

import java.io.IOException;
import java.util.List;
import java.util.Vector;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class Idx extends ReferenceEntity implements Comparable{
	@SuppressWarnings("unused")
	private final static String tableName = "olapdb:idx";

	public Idx(byte[] row) {
		super(row);
	}

	public Idx(Result r) {
		super(r);
	}

	public Idx(Idc idc, Value value){
		this(Bytez.add(Bytez.from(idc.getId()), value.ranking()));
	}

	@Override
	public Table getBasisTable() {
		return Obase.getIdxTable();
	}

	@Override
	public int compareTo(Idx arg0) {
		if(this ==arg0)return 0;

		long v = this.getReference() - arg0.getReference();
		if(v > 0)return 1;
		if(v < 0)return -1;
		return 0;
	}

	public String getName(){
		return Bytez.toString(this.getRow());
	}

	public Stream stream(){
		byte[] irow = this.getRow();
		Scan scan = new Scan().withStartRow(irow).withStopRow(Bytez.next(irow));

		return Index.stream(scan).map(e->new IndexData(
                Bytes.copy(e.getRow(),irow.length,e.getRow().length-irow.length),
		        e.getData()));
	}

	public List getIndexRows(){
		byte[] irow = this.getRow();

		Scan scan = new Scan(irow, Bytez.next(irow));

		ResultScanner rs = null;
		try {
			rs = Obase.getIndexTable().getScanner(scan);
		} catch (IOException e) {
			e.printStackTrace();
			return new Vector();
		}
		List rows = new Vector();

		for(Result r: rs){
			Cell c = r.getColumnLatestCell(Bytez.from(Obase.FAMILY_ATTR), Bytez.from("data"));
			if(c != null){
				byte[] row = Bytes.copy(r.getRow(),irow.length,r.getRow().length-irow.length);
				byte[] data = Index.fromData(CellUtil.cloneValue(c));
				rows.add(new Find(this, row, data));
			}
		}

		return rows;
	}

	public List filterIndexRows(List rows){
		byte[] irow = this.getRow();

		//make get
		List gets = rows.stream()
				.map(t->new Get(Bytez.add(this.getRow(), t.getRow())))
				.collect(Collectors.toList());

		Result[] results = null;
		try {
			results = Obase.getIndexTable().get(gets);
		} catch (IOException e) {
			e.printStackTrace();
			return new Vector();
		}

		List rets = new Vector();
		for(Result r: results){
			Cell c = r.getColumnLatestCell(Bytez.from(Obase.FAMILY_ATTR), Bytez.from("data"));
			if(c != null){
				byte[] row = Bytez.copy(r.getRow(),irow.length);
				byte[] data = Index.fromData(CellUtil.cloneValue(c));
				rets.add(new Find(this, row, data));
			}
		}

		return rets;
	}
}