org.cloudgraph.hbase.scan.PartialRowKeyScanAssembler Maven / Gradle / Ivy
/**
* CloudGraph Community Edition (CE) License
*
* This is a community release of CloudGraph, a dual-license suite of
* Service Data Object (SDO) 2.1 services designed for relational and
* big-table style "cloud" databases, such as HBase and others.
* This particular copy of the software is released under the
* version 2 of the GNU General Public License. CloudGraph was developed by
* TerraMeta Software, Inc.
*
* Copyright (c) 2013, TerraMeta Software, Inc. All rights reserved.
*
* General License information can be found below.
*
* This distribution may include materials developed by third
* parties. For license and attribution notices for these
* materials, please refer to the documentation that accompanies
* this distribution (see the "Licenses for Third-Party Components"
* appendix) or view the online documentation at
* resultFields = new ArrayList();
for (PreDefinedKeyFieldConfig field : this.graph.getPreDefinedRowKeyFields()) {
switch (field.getName()) {
case URI:
case TYPE:
resultFields.add(field);
break;
case UUID:
break; // not applicable
default:
}
}
int fieldCount = resultFields.size();
for (int i = 0; i < fieldCount; i++) {
PreDefinedKeyFieldConfig preDefinedField = resultFields.get(i);
if (startRowFieldCount > 0) {
this.startKey.put(graph.getRowKeyFieldDelimiterBytes());
}
if (stopRowFieldCount > 0) {
this.stopKey.put(graph.getRowKeyFieldDelimiterBytes());
}
byte[] paddedStartValue = getStartBytes(preDefinedField);
this.startKey.put(paddedStartValue);
startRowFieldCount++;
byte[] paddedStopValue = getStopBytes(preDefinedField, i >= (fieldCount -1));
this.stopKey.put(paddedStopValue);
stopRowFieldCount++;
}
}
private void assembleLiterals()
{
// first collect the set of field configs which have literals or
// predefined field config value(s), such that we can determine the
// last field value.
List resultFields = new ArrayList();
for (KeyFieldConfig fieldConfig : this.graph.getRowKeyFields()) {
if (fieldConfig instanceof PreDefinedKeyFieldConfig) {
PreDefinedKeyFieldConfig predefinedConfig = (PreDefinedKeyFieldConfig)fieldConfig;
switch (predefinedConfig.getName()) {
case UUID:
if (this.rootUUID != null)
resultFields.add(fieldConfig);
break;
default:
resultFields.add(fieldConfig);
break;
}
}
else {
UserDefinedRowKeyFieldConfig userFieldConfig = (UserDefinedRowKeyFieldConfig)fieldConfig;
List scanLiterals = this.scanLiterals.getLiterals(userFieldConfig);
if (scanLiterals != null)
resultFields.add(fieldConfig);
}
}
int fieldCount = resultFields.size();
for (int i = 0; i < fieldCount; i++) {
KeyFieldConfig fieldConfig = resultFields.get(i);
if (fieldConfig instanceof PreDefinedKeyFieldConfig) {
PreDefinedKeyFieldConfig predefinedConfig = (PreDefinedKeyFieldConfig)fieldConfig;
byte[] paddedStartValue = getStartBytes(predefinedConfig);
byte[] paddedStopValue = getStopBytes(predefinedConfig,
i >= (fieldCount -1));
if (startRowFieldCount > 0)
this.startKey.put(graph.getRowKeyFieldDelimiterBytes());
if (stopRowFieldCount > 0)
this.stopKey.put(graph.getRowKeyFieldDelimiterBytes());
this.startKey.put(paddedStartValue);
this.stopKey.put(paddedStopValue);
this.startRowFieldCount++;
this.stopRowFieldCount++;
}
else if (fieldConfig instanceof UserDefinedRowKeyFieldConfig) {
UserDefinedRowKeyFieldConfig userFieldConfig = (UserDefinedRowKeyFieldConfig)fieldConfig;
List scanLiterals = this.scanLiterals.getLiterals(userFieldConfig);
// We may have multiple literals but all may not have start/stop bytes, e.g.
// a String literal with a less-than-equal '<=' operator will not have a start bytes.
for (ScanLiteral scanLiteral : scanLiterals) {
byte[] startBytes = scanLiteral.getStartBytes();
if (startBytes.length > 0) {
if (this.startRowFieldCount > 0) {
this.startKey.put(graph.getRowKeyFieldDelimiterBytes());
}
this.startKey.put(startBytes);
this.startRowFieldCount++;
}
byte[] stopBytes = null;
// if not last field
if (i < (fieldCount -1)) {
stopBytes = scanLiteral.getStartBytes();
}
else {
// only use stop bytes is last field in (compound) key
stopBytes = scanLiteral.getStopBytes();
}
if (stopBytes.length > 0) {
if (this.stopRowFieldCount > 0) {
this.stopKey.put(graph.getRowKeyFieldDelimiterBytes());
}
this.stopKey.put(stopBytes);
this.stopRowFieldCount++;
}
}
}
}
}
private byte[] getStartBytes(PreDefinedKeyFieldConfig preDefinedField)
{
byte[] startValue = null;
switch (preDefinedField.getName()) {
case UUID:
if (this.rootUUID != null) {
startValue = this.rootUUID.getBytes(this.charset);
}
break;
default:
startValue = this.keySupport.getPredefinedFieldValueStartBytes(this.rootType,
hashing, preDefinedField);
break;
}
byte[] paddedStartValue = null;
if (preDefinedField.isHash()) {
paddedStartValue = this.padding.pad(startValue, preDefinedField.getMaxLength(),
DataFlavor.integral);
}
else {
paddedStartValue = this.padding.pad(startValue, preDefinedField.getMaxLength(),
preDefinedField.getDataFlavor());
}
return paddedStartValue;
}
private byte[] getStopBytes(PreDefinedKeyFieldConfig preDefinedField, boolean lastField)
{
byte[] stopValue = null;
switch (preDefinedField.getName()) {
case UUID:
if (this.rootUUID != null) {
stopValue = this.rootUUID.getBytes(this.charset);
}
break;
default:
if (!lastField)
stopValue = this.keySupport.getPredefinedFieldValueStartBytes(this.rootType,
hashing, preDefinedField);
else
stopValue = this.keySupport.getPredefinedFieldValueStopBytes(this.rootType,
hashing, preDefinedField);
break;
}
byte[] paddedStopValue = null;
if (preDefinedField.isHash()) {
paddedStopValue = this.padding.pad(stopValue, preDefinedField.getMaxLength(),
DataFlavor.integral);
}
else {
paddedStopValue = this.padding.pad(stopValue, preDefinedField.getMaxLength(),
preDefinedField.getDataFlavor());
}
return paddedStopValue;
}
/**
* Returns the start row key as a byte array.
* @return the start row key
* @throws IllegalStateException if row keys are not yet assembled
*/
@Override
public byte[] getStartKey() {
if (this.startKey == null)
throw new IllegalStateException("row keys not assembled - first call assemble(...)");
// ByteBuffer.array() returns unsized array so don't sent that back to clients
// to misuse.
// Use native arraycopy() method as it uses native memcopy to create result array
// and because
// ByteBuffer.get(byte[] dst,int offset, int length) is not native
byte [] result = new byte[this.startKey.position()];
System.arraycopy(this.startKey.array(), this.startKey.arrayOffset(), result, 0, this.startKey.position());
return result;
}
/**
* Returns the stop row key as a byte array.
* @return the stop row key
* @throws IllegalStateException if row keys are not yet assembled
*/
@Override
public byte[] getStopKey() {
if (this.stopKey == null)
throw new IllegalStateException("row keys not assembled - first call assemble(...)");
byte [] result = new byte[this.stopKey.position()];
System.arraycopy(this.stopKey.array(), this.stopKey.arrayOffset(), result, 0, this.stopKey.position());
return result;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy