Java CompositeType.build方法代码示例

import org.apache.cassandra.db.marshal.CompositeType; //导入方法依赖的package包/类
private ByteBuffer getPartitionKey(Map<String, ByteBuffer> keyColumns)
{
    ByteBuffer partitionKey;
    if (keyValidator instanceof CompositeType)
    {
        ByteBuffer[] keys = new ByteBuffer[partitionKeyColumns.length];
        for (int i = 0; i< keys.length; i++)
            keys[i] = keyColumns.get(partitionKeyColumns[i]);

        partitionKey = CompositeType.build(keys);
    }
    else
    {
        partitionKey = keyColumns.get(partitionKeyColumns[0]);
    }
    return partitionKey;
}
 

import org.apache.cassandra.db.marshal.CompositeType; //导入方法依赖的package包/类
/** check whether current row is at the end of range */
private boolean reachEndRange()
{
    // current row key
    ByteBuffer rowKey;
    if (keyValidator instanceof CompositeType)
    {
        ByteBuffer[] keys = new ByteBuffer[partitionBoundColumns.size()];
        for (int i = 0; i < partitionBoundColumns.size(); i++)
            keys[i] = partitionBoundColumns.get(i).value.duplicate();

        rowKey = CompositeType.build(keys);
    }
    else
    {
        rowKey = partitionBoundColumns.get(0).value;
    }

    String endToken = split.getEndToken();
    String currentToken = partitioner.getToken(rowKey).toString();
    logger.debug("End token: {}, current token: {}", endToken, currentToken);

    return endToken.equals(currentToken);
}
 

import org.apache.cassandra.db.marshal.CompositeType; //导入方法依赖的package包/类
private ByteBuffer getPartitionKey(Map<String, ByteBuffer> keyColumns)
{
    ByteBuffer partitionKey;
    if (partitionKeyColumns.size() > 1)
    {
        ByteBuffer[] keys = new ByteBuffer[partitionKeyColumns.size()];
        for (int i = 0; i< keys.length; i++)
            keys[i] = keyColumns.get(partitionKeyColumns.get(i).getName());

        partitionKey = CompositeType.build(keys);
    }
    else
    {
        partitionKey = keyColumns.get(partitionKeyColumns.get(0).getName());
    }
    return partitionKey;
}
 

import org.apache.cassandra.db.marshal.CompositeType; //导入方法依赖的package包/类
/**
 * check whether current row is at the end of range
 *
 * @return the boolean
 */
private boolean reachEndRange() {
    // current row key
    ByteBuffer rowKey;

    if (keyValidator instanceof CompositeType) {
        ByteBuffer[] keys = new ByteBuffer[partitionBoundColumns.size()];
        for (int i = 0; i < partitionBoundColumns.size(); i++) {
            keys[i] = partitionBoundColumns.get(i).value.duplicate();
        }

        rowKey = CompositeType.build(keys);
    } else {
        rowKey = partitionBoundColumns.get(0).value;
    }

    String endToken = String.valueOf(split.getEndToken());
    String currentToken = partitioner.getToken(rowKey).toString();

    return endToken.equals(currentToken);
}
 

import org.apache.cassandra.db.marshal.CompositeType; //导入方法依赖的package包/类
/**
 * Returns the partition key related to a given {@link Cells}.
 *
 * @param cells        {@link Cells} from Cassandra to extract the partition key.
 * @param keyValidator Cassandra key type.
 * @param numberOfKeys Number of keys.
 * @return Partition key.
 */
public static ByteBuffer getPartitionKey(Cells cells, AbstractType<?> keyValidator, int numberOfKeys) {
    ByteBuffer partitionKey;
    if (keyValidator instanceof CompositeType) {
        ByteBuffer[] keys = new ByteBuffer[numberOfKeys];

        for (int i = 0; i < cells.size(); i++) {
            Cell c = cells.getCellByIdx(i);

            if (c.isKey()) {
                keys[i] = DataType.serializeValue(c.getValue(), CassandraDeepJobConfig.PROTOCOL_VERSION);
            }
        }

        partitionKey = CompositeType.build(keys);
    } else {
        Cell cell = cells.getCellByIdx(0);
        partitionKey = DataType.serializeValue(cell.getValue(), CassandraDeepJobConfig.PROTOCOL_VERSION);
    }
    return partitionKey;
}
 

import org.apache.cassandra.db.marshal.CompositeType; //导入方法依赖的package包/类
/**
 * check whether current row is at the end of range
 */
private boolean reachEndRange() {
    // current row key
    ByteBuffer rowKey;
    if (keyValidator instanceof CompositeType) {
        ByteBuffer[] keys = new ByteBuffer[partitionBoundColumns.size()];
        for (int i = 0; i < partitionBoundColumns.size(); i++) {
            keys[i] = partitionBoundColumns.get(i).value.duplicate();
        }

        rowKey = CompositeType.build(keys);
    } else {
        rowKey = partitionBoundColumns.get(0).value;
    }

    String endToken = split.getEndToken();
    String currentToken = partitioner.getToken(rowKey).toString();
    logger.debug("End token: {}, current token: {}", endToken, currentToken);

    return endToken.equals(currentToken);
}
 

import org.apache.cassandra.db.marshal.CompositeType; //导入方法依赖的package包/类
public static ByteBuffer serializePartitionKey(ClusteringPrefix keyAsClustering)
{
    // TODO: we should stop using Clustering for partition keys. Maybe we can add
    // a few methods to DecoratedKey so we don't have to (note that while using a Clustering
    // allows to use buildBound(), it's actually used for partition keys only when every restriction
    // is an equal, so we could easily create a specific method for keys for that.
    if (keyAsClustering.size() == 1)
        return keyAsClustering.get(0);

    ByteBuffer[] values = new ByteBuffer[keyAsClustering.size()];
    for (int i = 0; i < keyAsClustering.size(); i++)
        values[i] = keyAsClustering.get(i);
    return CompositeType.build(values);
}
 

import org.apache.cassandra.db.marshal.CompositeType; //导入方法依赖的package包/类
public static void addMutation(RowMutation rm, String columnFamilyName, String superColumnName, long columnName, String value, long timestamp)
{
    ByteBuffer cname = superColumnName == null
                     ? getBytes(columnName)
                     : CompositeType.build(ByteBufferUtil.bytes(superColumnName), getBytes(columnName));
    rm.add(columnFamilyName, cname, ByteBufferUtil.bytes(value), timestamp);
}
 

import org.apache.cassandra.db.marshal.CompositeType; //导入方法依赖的package包/类
private DecoratedKey makeCurrentPartitionKey()
{
    ByteBuffer rawKey = viewMetadata.partitionKeyColumns().size() == 1
                      ? currentViewEntryPartitionKey[0]
                      : CompositeType.build(currentViewEntryPartitionKey);

    return viewMetadata.decorateKey(rawKey);
}
 

import org.apache.cassandra.db.marshal.CompositeType; //导入方法依赖的package包/类
private static ByteBuffer makeCellName(Clustering clustering, ColumnDefinition c, CellPath path)
{
    int cs = clustering.size();
    ByteBuffer[] values = new ByteBuffer[cs + 1 + (path == null ? 0 : path.size())];
    for (int i = 0; i < cs; i++)
        values[i] = clustering.get(i);
    values[cs] = c.name.bytes;
    if (path != null)
        for (int i = 0; i < path.size(); i++)
            values[cs + 1 + i] = path.get(i);
    return CompositeType.build(values);
}
 

import org.apache.cassandra.db.marshal.CompositeType; //导入方法依赖的package包/类
private ByteBuffer getPartitionKey(Map<String, ByteBuffer> keyColumns) {
    ByteBuffer partitionKey;
    if (keyValidator instanceof CompositeType) {
        ByteBuffer[] keys = new ByteBuffer[partitionKeyColumns.length];
        for (int i = 0; i < keys.length; i++) {
            keys[i] = keyColumns.get(partitionKeyColumns[i]);
        }

        partitionKey = CompositeType.build(keys);
    } else {
        partitionKey = keyColumns.get(partitionKeyColumns[0]);
    }
    return partitionKey;
}
 

import org.apache.cassandra.db.marshal.CompositeType; //导入方法依赖的package包/类
public static ByteBuffer toByteBuffer(final Object value) {
  if (value == null) {
    return ByteBufferUtil.EMPTY_BYTE_BUFFER;
  } else if (value instanceof CharSequence) {
    return ByteBufferUtil.bytes(value.toString());
  } else if (value instanceof Double) {
    return ByteBufferUtil.bytes((Double) value);
  } else if (value instanceof Float) {
    return ByteBufferUtil.bytes((Float) value);
  } else if (value instanceof Integer) {
    return ByteBufferUtil.bytes((Integer) value);
  } else if (value instanceof Long) {
    return ByteBufferUtil.bytes((Long) value);
  } else if (value instanceof ByteBuffer) {
    return ByteBufferUtil.clone((ByteBuffer) value);
  } else if (value instanceof GenericData.Array) {
    return serializeList((GenericData.Array)value);
  } else if (value instanceof SpecificRecord) {
    List<ByteBuffer> buffers = Lists.newArrayList();
    SpecificRecord record = (SpecificRecord) value;
    for (Schema.Field field : record.getSchema().getFields()) {
      buffers.add(toByteBuffer(record.get(field.pos())));
    }
    return CompositeType.build(buffers.toArray(new ByteBuffer[0]));
  } else if (value instanceof Map) {
    return serializeMap((Map<?, ?>) value);
  } else if (value instanceof Set) {
    return serializeSet((Set<?>) value);
  } else if (value instanceof List) {
    return serializeList((List<?>) value);
  } else if (value instanceof UUID) {
    return ByteBufferUtil.bytes((UUID) value);
  }


  throw new CrunchRuntimeException("Can not transform field (class: " + value.getClass() + ") to ByteBuffer");
}
 

import org.apache.cassandra.db.marshal.CompositeType; //导入方法依赖的package包/类
public ByteBuffer getIndexedValue(ByteBuffer partitionKey,
                                  Clustering clustering,
                                  CellPath path, ByteBuffer cellValue)
{
    return CompositeType.build(path.get(0), cellValue);
}
 

import org.apache.cassandra.db.marshal.CompositeType; //导入方法依赖的package包/类
@Override
public ByteBuffer bindAndGet(List<ByteBuffer> variables) throws InvalidRequestException
{
    return CompositeType.build(bindInternal(variables));
}